Rust integration

.github/workflows/ci.yml

@@ -22,9 +22,20 @@ jobs:
               with:
                   node-version-file: '.nvmrc'
                   cache: 'yarn'
+
+            - name: Setup Rust
+              uses: dtolnay/rust-toolchain@stable
+
+            - uses: Swatinem/rust-cache@v2
+              with:
+                workspaces: './src/wasm-lib'
 
             - run: yarn install
 
+            - run: yarn install:wasm
+
+            - run: yarn build:wasm
+
             - run: yarn build
 
             - run: yarn test

.gitignore

@@ -35,3 +35,11 @@ yarn-error.log*
 /tests/extension.spec.ts-snapshots/*darwin*
 
 .env*
+
+# rust
+src/wasm-lib/target
+src/wasm-lib/bindings
+src/wasm-lib/kcl/bindings
+public/wasm_lib_bg.wasm
+src/wasm-lib/lcov.info
+src/wasm-lib/grackle/test_json_output

manifest.json

@@ -19,7 +19,8 @@
         "https://github.com/",
         "https://api.github.com/",
         "https://media.githubusercontent.com/",
-        "https://api.kittycad.io/"
+        "https://api.kittycad.io/",
+        "https://api.zoo.dev/"
     ],
     "content_scripts": [
         {
@@ -36,5 +37,9 @@
     "background": {
         "service_worker": "src/chrome/background.ts",
         "type": "module"
+    },
+    "content_security_policy": {
+        "extension_pages": "script-src 'self' 'wasm-unsafe-eval'; object-src 'self';",
+        "sandbox": "sandbox allow-scripts allow-forms allow-popups allow-modals; script-src 'self' 'unsafe-inline' 'unsafe-eval'; child-src 'self';"
     }
 }

package.json

@@ -36,6 +36,10 @@
     },
     "scripts": {
         "start": "vite",
+        "install:rust": "curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh",
+        "install:wasm": "curl https://rustwasm.github.io/wasm-pack/installer/init.sh -sSf | sh",
+        "build:wasm:dev": "(cd src/wasm-lib && wasm-pack build --dev --target web --out-dir pkg) && cp src/wasm-lib/pkg/wasm_lib_bg.wasm public",
+        "build:wasm": "(cd src/wasm-lib && wasm-pack build --target web --out-dir pkg) && cp src/wasm-lib/pkg/wasm_lib_bg.wasm public",
         "build": "vite build",
         "test": "vitest",
         "e2e": "yarn build && yarn playwright test",

src/chrome/background.ts

@@ -18,6 +18,7 @@ import {
     setStorageKittycadToken,
 } from './storage'
 import { getFileBlob, getFileDiff } from './diff'
+import init, { greet } from '../wasm-lib/pkg/wasm_lib'
 
 let github: Octokit | undefined
 let kittycad: Client | undefined
@@ -35,7 +36,9 @@ async function initGithubApi() {
 
 async function initKittycadApi() {
     try {
-        kittycad = new Client(await getStorageKittycadToken())
+        const token = await getStorageKittycadToken()
+        console.log('greet(token)', await greet(token))
+        kittycad = new Client(token)
         const response = await users.get_user_self({ client: kittycad })
         if ('error_code' in response) throw response
         const { email } = response
@@ -59,9 +62,24 @@ async function saveKittycadTokenAndReload(token: string): Promise<void> {
     await initKittycadApi()
 }
 
-;(async () => {
+async function initialiseWasm() {
+    try {
+        const fullUrl = '/wasm_lib_bg.wasm'
+        const input = await fetch(fullUrl)
+        const buffer = await input.arrayBuffer()
+        const output = await init(buffer)
+        console.log('Wasm loaded: ', output)
+        return output
+    } catch (e) {
+        console.log('Error initialising WASM', e)
+        throw e
+    }
+}
+
+; (async () => {
     // Delay to allow for external storage sets before auth, like in e2e
     await new Promise(resolve => setTimeout(resolve, 1000))
+    await initialiseWasm()
     await initKittycadApi()
     await initGithubApi()
 })()