1 files changed, 99 insertions, 0 deletions
diff --git a/ext/wasm/speedtest1-worker.js b/ext/wasm/speedtest1-worker.js
new file mode 100644
index 000000000..8512bdbbf
--- /dev/null
+++ b/ext/wasm/speedtest1-worker.js
@@ -0,0 +1,99 @@
+'use strict';
+(function(){
+  importScripts('common/whwasmutil.js','speedtest1.js');
+  /**
+     If this environment contains OPFS, this function initializes it and
+     returns the name of the dir on which OPFS is mounted, else it returns
+     an empty string.
+  */
+  const opfsDir = function f(wasmUtil){
+    if(undefined !== f._) return f._;
+    const pdir = '/persistent';
+    if( !self.FileSystemHandle
+        || !self.FileSystemDirectoryHandle
+        || !self.FileSystemFileHandle){
+      return f._ = "";
+    }
+    try{
+      if(0===wasmUtil.xCallWrapped(
+        'sqlite3_wasm_init_opfs', 'i32', ['string'], pdir
+      )){
+        return f._ = pdir;
+      }else{
+        return f._ = "";
+      }
+    }catch(e){
+      // sqlite3_wasm_init_opfs() is not available
+      return f._ = "";
+    }
+  };
+  opfsDir._ = undefined;
+
+  const mPost = function(msgType,payload){
+    postMessage({type: msgType, data: payload});
+  };
+
+  const App = Object.create(null);
+  App.logBuffer = [];
+  const logMsg = (type,msgArgs)=>{
+    const msg = msgArgs.join(' ');
+    App.logBuffer.push(msg);
+    mPost(type,msg);
+  };
+  const log = (...args)=>logMsg('stdout',args);
+  const logErr = (...args)=>logMsg('stderr',args);
+  
+  const runSpeedtest = function(cliFlagsArray){
+    const scope = App.wasm.scopedAllocPush();
+    const dbFile = 0 ? "" : App.pDir+"/speedtest1.db";
+    try{
+      const argv = [
+        "speedtest1.wasm", ...cliFlagsArray, dbFile
+      ];
+      App.logBuffer.length = 0;
+      mPost('run-start', [...argv]);
+      App.wasm.xCall('__main_argc_argv', argv.length,
+                     App.wasm.scopedAllocMainArgv(argv));
+    }catch(e){
+      mPost('error',e.message);
+    }finally{
+      App.wasm.scopedAllocPop(scope);
+      App.unlink(dbFile);
+      mPost('run-end', App.logBuffer.join('\n'));
+      App.logBuffer.length = 0;
+    }
+  };
+
+  self.onmessage = function(msg){
+    msg = msg.data;
+    switch(msg.type){
+        case 'run': runSpeedtest(msg.data || []); break;
+        default:
+          logErr("Unhandled worker message type:",msg.type);
+          break;
+    }
+  };
+
+  const EmscriptenModule = {
+    print: log,
+    printErr: logErr,
+    setStatus: (text)=>mPost('load-status',text)
+  };
+  self.sqlite3Speedtest1InitModule(EmscriptenModule).then(function(EmscriptenModule){
+    log("Module inited.");
+    App.wasm = {
+      exports: EmscriptenModule.asm,
+      alloc: (n)=>EmscriptenModule._malloc(n),
+      dealloc: (m)=>EmscriptenModule._free(m),
+      memory: EmscriptenModule.asm.memory || EmscriptenModule.wasmMemory
+    };
+    //console.debug('wasm =',wasm);
+    self.WhWasmUtilInstaller(App.wasm);
+    App.unlink = App.wasm.xWrap("sqlite3_wasm_vfs_unlink", "int", ["string"]);
+    App.pDir = opfsDir(App.wasm);
+    if(App.pDir){
+      log("Persistent storage:",pDir);
+    }
+    mPost('ready',true);
+  });
+})();