@cloudglides/veil 0.1.0 → 0.1.1

package/tsup.config.ts

@@ -10,26 +10,32 @@ export default defineConfig({
   clean: true,
   shims: true,
   outDir: "dist",
-  external: ["./veil_core.js"],
+  external: ["./veil_core.js", "../veil-core/pkg/veil_core.js"],
+  noExternal: [],
+  splitting: true,
   onSuccess: async () => {
     const srcWasmDir = resolve("veil-core/pkg");
     const distDir = resolve("dist");
+    const exampleDir = resolve("example");
 
     await mkdir(distDir, { recursive: true });
+    await mkdir(exampleDir, { recursive: true });
 
-    await cp(`${srcWasmDir}/veil_core_bg.wasm`, `${distDir}/veil_core_bg.wasm`);
-    await cp(`${srcWasmDir}/veil_core.js`, `${distDir}/veil_core.js`);
-    await cp(`${srcWasmDir}/veil_core.d.ts`, `${distDir}/veil_core.d.ts`);
-    await cp(
-      `${srcWasmDir}/veil_core_bg.wasm.d.ts`,
-      `${distDir}/veil_core_bg.wasm.d.ts`,
-    );
+    for (const dir of [distDir, exampleDir]) {
+      await cp(`${srcWasmDir}/veil_core_bg.wasm`, `${dir}/veil_core_bg.wasm`);
+      await cp(`${srcWasmDir}/veil_core.js`, `${dir}/veil_core.js`);
+      await cp(`${srcWasmDir}/veil_core.d.ts`, `${dir}/veil_core.d.ts`);
+      await cp(
+        `${srcWasmDir}/veil_core_bg.wasm.d.ts`,
+        `${dir}/veil_core_bg.wasm.d.ts`,
+      );
 
-    let wasmContent = readFileSync(`${distDir}/veil_core.js`, "utf-8");
-    wasmContent = wasmContent.replace(
-      /module_or_path = new URL\('veil_core_bg\.wasm', import\.meta\.url\);/g,
-      "module_or_path = new URL('veil_core_bg.wasm', import.meta.url).href;",
-    );
-    writeFileSync(`${distDir}/veil_core.js`, wasmContent);
+      let wasmContent = readFileSync(`${dir}/veil_core.js`, "utf-8");
+      wasmContent = wasmContent.replace(
+        /module_or_path = new URL\('veil_core_bg\.wasm', import\.meta\.url\);/g,
+        "module_or_path = new URL('veil_core_bg.wasm', import.meta.url).href;",
+      );
+      writeFileSync(`${dir}/veil_core.js`, wasmContent);
+    }
   },
 });

package/veil-core/src/lib.rs

@@ -3,6 +3,7 @@ use std::usize;
 use wasm_bindgen::prelude::*;
 
 mod entropy;
+mod similarity;
 
 const MURMUR_SEED: u32 = 0x811c9dc5;
 const FNV_PRIME: u32 = 16777619;
@@ -88,3 +89,5 @@ pub fn approx_entropy(samples: &[f64], m: usize) -> f64 {
 pub fn sample_ent(samples: &[f64], m: usize, r: f64) -> f64 {
     entropy::sample_entropy(samples, m, r)
 }
+
+pub use similarity::{levenshtein_distance, similarity_score, cosine_similarity};

package/veil-core/src/similarity.rs

@@ -0,0 +1,67 @@
+pub use wasm_bindgen::prelude::*;
+
+#[wasm_bindgen]
+pub fn levenshtein_distance(s1: &str, s2: &str) -> u32 {
+    let len1 = s1.len();
+    let len2 = s2.len();
+    let mut matrix = vec![vec![0u32; len2 + 1]; len1 + 1];
+
+    for i in 0..=len1 {
+        matrix[i][0] = i as u32;
+    }
+    for j in 0..=len2 {
+        matrix[0][j] = j as u32;
+    }
+
+    let bytes1 = s1.as_bytes();
+    let bytes2 = s2.as_bytes();
+
+    for i in 1..=len1 {
+        for j in 1..=len2 {
+            let cost = if bytes1[i - 1] == bytes2[j - 1] { 0 } else { 1 };
+            matrix[i][j] = std::cmp::min(
+                std::cmp::min(matrix[i - 1][j] + 1, matrix[i][j - 1] + 1),
+                matrix[i - 1][j - 1] + cost,
+            );
+        }
+    }
+
+    matrix[len1][len2]
+}
+
+#[wasm_bindgen]
+pub fn similarity_score(s1: &str, s2: &str) -> f64 {
+    let distance = levenshtein_distance(s1, s2) as f64;
+    let max_len = std::cmp::max(s1.len(), s2.len()) as f64;
+    if max_len == 0.0 {
+        1.0
+    } else {
+        1.0 - (distance / max_len)
+    }
+}
+
+#[wasm_bindgen]
+pub fn cosine_similarity(v1: &[f64], v2: &[f64]) -> f64 {
+    if v1.len() != v2.len() || v1.is_empty() {
+        return 0.0;
+    }
+
+    let mut dot = 0.0;
+    let mut mag1 = 0.0;
+    let mut mag2 = 0.0;
+
+    for i in 0..v1.len() {
+        dot += v1[i] * v2[i];
+        mag1 += v1[i] * v1[i];
+        mag2 += v2[i] * v2[i];
+    }
+
+    mag1 = mag1.sqrt();
+    mag2 = mag2.sqrt();
+
+    if mag1 == 0.0 || mag2 == 0.0 {
+        0.0
+    } else {
+        dot / (mag1 * mag2)
+    }
+}