atoa & btoa global functions

Author: richarddavison

src/encoding/mod.rs

@@ -15,7 +15,9 @@ use crate::{
     },
 };
 
-use self::encoder::{bytes_from_hex, bytes_to_hex_string, Encoder};
+use self::encoder::{
+    bytes_from_b64, bytes_from_hex, bytes_to_b64_string, bytes_to_hex_string, Encoder,
+};
 
 pub struct HexModule;
 
@@ -139,9 +141,21 @@ impl StringBuilder {
     }
 }
 
+pub fn atob<'js>(ctx: Ctx<'js>, encoded_value: String) -> Result<String> {
+    let vec = bytes_from_b64(encoded_value.as_bytes()).or_throw(&ctx)?;
+    Ok(unsafe { String::from_utf8_unchecked(vec) })
+}
+
+pub fn btoa(value: String) -> String {
+    bytes_to_b64_string(value.as_bytes())
+}
+
 pub fn init(ctx: &Ctx<'_>) -> Result<()> {
     let globals = ctx.globals();
 
+    globals.set("atob", Func::from(atob))?;
+    globals.set("btoa", Func::from(btoa))?;
+
     Class::<TextEncoder>::define(&globals)?;
     Class::<TextDecoder>::define(&globals)?;
     Class::<StringBuilder>::define(&globals)?;

tests/encoding.test.ts

@@ -1,16 +1,28 @@
 import hex from "hex";
 
-it("should encode/decode text", () => {
-  let hello = "hello";
-  const encoded = new TextEncoder().encode(hello);
-  const decoded = new TextDecoder().decode(encoded);
+describe("hex", () => {
+  it("should encode/decode text", () => {
+    let hello = "hello";
+    const encoded = new TextEncoder().encode(hello);
+    const decoded = new TextDecoder().decode(encoded);
 
-  assert.equal(decoded, hello);
-});
+    assert.equal(decoded, hello);
+  });
+
+  it("should encode/decode hex", () => {
+    const byteArray = new TextEncoder().encode("hello");
+    const encoded = hex.encode(byteArray);
 
-it("should encode/decode hex", () => {
-  const byteArray = new TextEncoder().encode("hello");
-  const encoded = hex.encode(byteArray);
+    assert.equal(encoded, "68656c6c6f");
+  });
+});
 
-  assert.equal(encoded, "68656c6c6f");
+describe("atoa & btoa", () => {
+  it("btoa/atob", () => {
+    const text = "Hello, world!";
+    const encodedData = btoa(text);
+    assert.equal(encodedData, "SGVsbG8sIHdvcmxkIQ==");
+    const decodedData = atob(encodedData);
+    assert.equal(decodedData, text);
+  });
 });