Feature ASCII encoding and decoding (#27)

jtenner · web-flow · commit 6e0743c4958a · 2020-03-13T00:13:31.000-04:00
* start implementation

* fix operator grouping, now to optimize

* remove lookup table, switch to simple bitwise

* mask with shorter literal, remove explicit cast

* add decorator
diff --git a/assembly/buffer/index.ts b/assembly/buffer/index.ts
@@ -287,6 +287,34 @@ export class Buffer extends Uint8Array {
 }
 
 export namespace Buffer {
+  export namespace ASCII {
+    export function encode(str: string): ArrayBuffer {
+      let length = str.length;
+      let output = __alloc(length, idof<ArrayBuffer>());
+      for (let i = 0; i < length; i++) {
+        let char = load<u16>(changetype<usize>(str) + <usize>(i << 1));
+        store<u8>(output + <usize>i, char & 0x7F);
+      }
+      return changetype<ArrayBuffer>(output);
+    }
+
+    export function decode(buffer: ArrayBuffer): String {
+      return decodeUnsafe(changetype<usize>(buffer), buffer.byteLength);
+    }
+
+    // @ts-ignore: decorator
+    @unsafe export function decodeUnsafe(pointer: usize, length: i32): String {
+      let result = __alloc(<usize>length << 1, idof<string>());
+
+      for (let i = 0; i < length; i++) {
+        let byte = load<u8>(pointer + <usize>i);
+        store<u16>(result + <usize>(i << 1), byte & 0x7F);
+      }
+
+      return changetype<String>(result);
+    }
+  }
+
   export namespace HEX {
     /** Calculates the byte length of the specified string when encoded as HEX. */
     export function byteLength(str: string): i32 {
@@ -358,6 +386,7 @@ export namespace Buffer {
     }
 
     /** Decodes a chunk of memory to a utf16le encoded string in hex format. */
+    // @ts-ignore: decorator
     @unsafe export function decodeUnsafe(ptr: usize, length: i32): string {
       let stringByteLength = length << 2; // length * (2 bytes per char) * (2 chars per input byte)
       let result = __alloc(stringByteLength, idof<String>());
@@ -374,6 +403,7 @@ export namespace Buffer {
     }
 
     /** Calculates the two char combination from the byte. */
+    // @ts-ignore: decorator
     @inline function charsFromByte(byte: u32): u32 {
       let hi = (byte >>> 4) & 0xF;
       let lo = byte & 0xF;
diff --git a/assembly/node.d.ts b/assembly/node.d.ts
@@ -86,6 +86,7 @@ declare class Buffer extends Uint8Array {
 }
 
 declare namespace Buffer {
+  /** The HEX encoding and decoding namespace. */
   export namespace HEX {
     /** Creates an ArrayBuffer from a given string that is encoded in the hex format. */
     export function encode(str: string): ArrayBuffer;
@@ -94,4 +95,13 @@ declare namespace Buffer {
     /** Decodes a chunk of memory to a utf16le encoded string in hex format. */
     export function decodeUnsafe(ptr: usize, byteLength: i32): string;
   }
+  /** The ASCII encoding and decoding namespace. */
+  export namespace ASCII {
+    /** Creates an ArrayBuffer from a given string that is encoded in the ASCII format. */
+    export function encode(str: string): ArrayBuffer;
+    /** Creates a string from a given ArrayBuffer that is decoded into ASCII format. */
+    export function decode(buffer: ArrayBuffer): string;
+    /** Decodes a chunk of memory to a utf16le encoded string in ASCII format. */
+    export function decodeUnsafe(ptr: usize, byteLength: i32): string;
+  }
 }
diff --git a/tests/buffer.spec.ts b/tests/buffer.spec.ts
@@ -550,4 +550,18 @@ describe("buffer", () => {
     let decoded = Buffer.HEX.decode(exampleBuffer.buffer);
     expect(decoded).toStrictEqual(expected);
   });
+
+  test("#ASCII.encode", () => {
+    let actual = "D34dB3eF";
+    let exampleBuffer = create<Buffer>([0x44, 0x33, 0x34, 0x64, 0x42, 0x33, 0x65, 0x46]);
+    let encoded = Buffer.ASCII.encode(actual);
+    expect(encoded).toStrictEqual(exampleBuffer.buffer);
+  });
+
+  test("#ASCII.decode", () => {
+    let expected = create<Buffer>([0x44, 0x33, 0x34, 0x64, 0x42, 0x33, 0x65, 0x46]);
+    let example = "D34dB3eF";
+    let encoded = Buffer.ASCII.decode(expected.buffer);
+    expect(encoded).toStrictEqual(example);
+  });
 });
