From 860977355284db0cb56ccf82537f9c4e144d6858 Mon Sep 17 00:00:00 2001
From: zer0x64 <dugre.philippe@hotmail.com>
Date: Sat, 25 Jan 2020 22:52:54 -0500
Subject: [PATCH] WASM bindings draft

---
 Cargo.toml  |  9 +++++++++
 src/lib.rs  |  3 +++
 src/wasm.rs | 22 ++++++++++++++++++++++
 3 files changed, 34 insertions(+)
 create mode 100644 src/wasm.rs

diff --git a/Cargo.toml b/Cargo.toml
index abe5a4b..7941fa6 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -17,9 +17,18 @@ codecov = { repository = "https://github.com/c0dearm/sharks" }
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
+[lib]
+crate-type = ["cdylib", "rlib"]
+
 [dependencies]
 rand = "0.7"
 
+[target.'cfg(target_arch="wasm32")'.dependencies]
+serde = "1.0.104"
+serde_derive = "1.0.104"
+rand = { version = "0.7", features = ["wasm-bindgen"] }
+wasm-bindgen = { version = "0.2.58", features = ["serde-serialize"] }
+
 [dev-dependencies]
 criterion = "0.3"
 
diff --git a/src/lib.rs b/src/lib.rs
index 0580124..f261a2a 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -19,6 +19,9 @@ mod field;
 mod math;
 mod share;
 
+#[cfg(target_arch = "wasm32")]
+pub mod wasm;
+
 use std::collections::HashSet;
 
 use field::GF256;
diff --git a/src/wasm.rs b/src/wasm.rs
new file mode 100644
index 0000000..62a815e
--- /dev/null
+++ b/src/wasm.rs
@@ -0,0 +1,22 @@
+use wasm_bindgen::prelude::*;
+
+use crate::{ Sharks, Share };
+
+#[wasm_bindgen]
+pub fn generate_shares(n_shares: u8, threshold: u8, secret: &[u8]) -> JsValue {
+    let sharks = Sharks(threshold);
+    let dealer = sharks.dealer(secret);
+    let shares: Vec<Vec<u8>> = dealer.take(n_shares as usize).map(|s| (&s).into()).collect();
+
+    JsValue::from_serde(&shares).expect("A Vec<Vec<u8>> should always be JSON serializable.")
+}
+
+#[wasm_bindgen]
+pub fn recover(threshold: u8, shares: JsValue) -> Vec<u8> {
+    let sharks = Sharks(threshold);
+
+    let shares: Vec<Vec<u8>> = shares.into_serde().expect("will implement proper error handling later");
+
+    let shares: Vec<Share> = shares.iter().map(|s| s.as_slice().into()).collect();
+    sharks.recover(&shares).expect("will implement proper error handling later").into()
+}
\ No newline at end of file