jsshaker 0.0.0 → 0.1.1

package/README.md

@@ -0,0 +1,49 @@
+# JsShaker
+
+🪚 Code size optimizer for JavaScript.
+
+https://github.com/kermanx/jsshaker
+
+## WASM & N-API
+
+```ts
+import { shakeSingleModule, shakeMultiModule } from "jsshaker";
+```
+
+## CLI
+
+```sh
+pnpx jsshaker ./entry.js
+
+# --preset(-p, optional): "safest" | "recommended" | "smallest" | "disabled"
+# --minify(-m, optional): boolean
+# --outdir(-o, optional): string
+```
+
+## Vite Plugin
+
+```ts
+import { defineConfig } from "vite";
+import { shakeMultiModule } from "jsshaker";
+import * as fs from "fs";
+
+export default defineConfig({
+  plugins: [
+    {
+      name: "jsshaker",
+      enforce: "post",
+      buildEnd() {
+        const { output, diagnostics } = shakeMultiModule("./dist/index.js", {
+          minify: true,
+        })
+        for (const diag of diagnostics) {
+          console.error(diag);
+        }
+        for (const file in output) {
+          fs.writeFileSync(file, output[file]);
+        }
+      },
+    },
+  ],
+});
+```

package/cli.js

@@ -1,7 +1,7 @@
 #!/usr/bin/env node
 
 const { parseArgs } = require("node:util");
-const { shakeSingleModule } = require("./index.js");
+const { shakeMultiModule } = require("./index.js");
 const { writeFile, mkdir } = require("node:fs/promises");
 const { join, dirname } = require("node:path");
 
@@ -28,7 +28,7 @@ if (positionals.length !== 1) {
   throw new Error("Must provide exactly one entry js file path.");
 }
 
-const result = shakeSingleModule(
+const result = shakeMultiModule(
   positionals[0],
   {
     preset: values.preset,

package/jsshaker.wasi.cjs

@@ -0,0 +1,112 @@
+/* eslint-disable */
+/* prettier-ignore */
+
+/* auto-generated by NAPI-RS */
+
+const __nodeFs = require('node:fs')
+const __nodePath = require('node:path')
+const { WASI: __nodeWASI } = require('node:wasi')
+const { Worker } = require('node:worker_threads')
+
+const {
+  createOnMessage: __wasmCreateOnMessageForFsProxy,
+  getDefaultContext: __emnapiGetDefaultContext,
+  instantiateNapiModuleSync: __emnapiInstantiateNapiModuleSync,
+} = require('@napi-rs/wasm-runtime')
+
+const __rootDir = __nodePath.parse(process.cwd()).root
+
+const __wasi = new __nodeWASI({
+  version: 'preview1',
+  env: process.env,
+  preopens: {
+    [__rootDir]: __rootDir,
+  }
+})
+
+const __emnapiContext = __emnapiGetDefaultContext()
+
+const __sharedMemory = new WebAssembly.Memory({
+  initial: 4000,
+  maximum: 65536,
+  shared: true,
+})
+
+let __wasmFilePath = __nodePath.join(__dirname, 'jsshaker.wasm32-wasi.wasm')
+const __wasmDebugFilePath = __nodePath.join(__dirname, 'jsshaker.wasm32-wasi.debug.wasm')
+
+if (__nodeFs.existsSync(__wasmDebugFilePath)) {
+  __wasmFilePath = __wasmDebugFilePath
+} else if (!__nodeFs.existsSync(__wasmFilePath)) {
+  try {
+    __wasmFilePath = __nodePath.resolve('@jsshaker/binding-wasm32-wasi')
+  } catch {
+    throw new Error('Cannot find jsshaker.wasm32-wasi.wasm file, and @jsshaker/binding-wasm32-wasi package is not installed.')
+  }
+}
+
+const { instance: __napiInstance, module: __wasiModule, napiModule: __napiModule } = __emnapiInstantiateNapiModuleSync(__nodeFs.readFileSync(__wasmFilePath), {
+  context: __emnapiContext,
+  asyncWorkPoolSize: (function() {
+    const threadsSizeFromEnv = Number(process.env.NAPI_RS_ASYNC_WORK_POOL_SIZE ?? process.env.UV_THREADPOOL_SIZE)
+    // NaN > 0 is false
+    if (threadsSizeFromEnv > 0) {
+      return threadsSizeFromEnv
+    } else {
+      return 4
+    }
+  })(),
+  reuseWorker: true,
+  wasi: __wasi,
+  onCreateWorker() {
+    const worker = new Worker(__nodePath.join(__dirname, 'wasi-worker.mjs'), {
+      env: process.env,
+    })
+    worker.onmessage = ({ data }) => {
+      __wasmCreateOnMessageForFsProxy(__nodeFs)(data)
+    }
+
+    // The main thread of Node.js waits for all the active handles before exiting.
+    // But Rust threads are never waited without `thread::join`.
+    // So here we hack the code of Node.js to prevent the workers from being referenced (active).
+    // According to https://github.com/nodejs/node/blob/19e0d472728c79d418b74bddff588bea70a403d0/lib/internal/worker.js#L415,
+    // a worker is consist of two handles: kPublicPort and kHandle.
+    {
+      const kPublicPort = Object.getOwnPropertySymbols(worker).find(s =>
+        s.toString().includes("kPublicPort")
+      );
+      if (kPublicPort) {
+        worker[kPublicPort].ref = () => {};
+      }
+
+      const kHandle = Object.getOwnPropertySymbols(worker).find(s =>
+        s.toString().includes("kHandle")
+      );
+      if (kHandle) {
+        worker[kHandle].ref = () => {};
+      }
+
+      worker.unref();
+    }
+    return worker
+  },
+  overwriteImports(importObject) {
+    importObject.env = {
+      ...importObject.env,
+      ...importObject.napi,
+      ...importObject.emnapi,
+      memory: __sharedMemory,
+    }
+    return importObject
+  },
+  beforeInit({ instance }) {
+    for (const name of Object.keys(instance.exports)) {
+      if (name.startsWith('__napi_register__')) {
+        instance.exports[name]()
+      }
+    }
+  },
+})
+module.exports = __napiModule.exports
+module.exports.shakeMultiModule = __napiModule.exports.shakeMultiModule
+module.exports.shakeSingleModule = __napiModule.exports.shakeSingleModule

package/package.json

@@ -1,9 +1,13 @@
 {
   "name": "jsshaker",
-  "version": "0.0.0",
+  "version": "0.1.1",
   "type": "commonjs",
   "main": "index.js",
   "browser": "browser.js",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/kermanx/jsshaker"
+  },
   "napi": {
     "binaryName": "jsshaker",
     "packageName": "@jsshaker/binding",
@@ -27,7 +31,15 @@
   "files": [
     "index.d.ts",
     "index.js",
-    "cli.js"
+    "cli.js",
+    "README.md",
+    "browser.js",
+    "jsshaker.wasi.cjs",
+    "jsshaker.wasi-browser.cjs",
+    "wasi-worker.mjs",
+    "wasi-worker-browser.mjs",
+    "*.wasm",
+    "*.node"
   ],
   "license": "MIT",
   "devDependencies": {
@@ -37,11 +49,11 @@
     "ohash": "^2.0.11"
   },
   "optionalDependencies": {
-    "@jsshaker/binding-linux-x64-gnu": "0.0.0",
-    "@jsshaker/binding-win32-x64-msvc": "0.0.0",
-    "@jsshaker/binding-darwin-x64": "0.0.0",
-    "@jsshaker/binding-darwin-arm64": "0.0.0",
-    "@jsshaker/binding-wasm32-wasi": "0.0.0"
+    "@jsshaker/binding-linux-x64-gnu": "0.1.1",
+    "@jsshaker/binding-win32-x64-msvc": "0.1.1",
+    "@jsshaker/binding-darwin-x64": "0.1.1",
+    "@jsshaker/binding-darwin-arm64": "0.1.1",
+    "@jsshaker/binding-wasm32-wasi": "0.1.1"
   },
   "scripts": {
     "artifacts": "napi artifacts",

package/wasi-worker-browser.mjs

@@ -0,0 +1,32 @@
+import { instantiateNapiModuleSync, MessageHandler, WASI } from '@napi-rs/wasm-runtime'
+
+const handler = new MessageHandler({
+  onLoad({ wasmModule, wasmMemory }) {
+    const wasi = new WASI({
+      print: function () {
+        // eslint-disable-next-line no-console
+        console.log.apply(console, arguments)
+      },
+      printErr: function() {
+        // eslint-disable-next-line no-console
+        console.error.apply(console, arguments)
+      },
+    })
+    return instantiateNapiModuleSync(wasmModule, {
+      childThread: true,
+      wasi,
+      overwriteImports(importObject) {
+        importObject.env = {
+          ...importObject.env,
+          ...importObject.napi,
+          ...importObject.emnapi,
+          memory: wasmMemory,
+        }
+      },
+    })
+  },
+})
+
+globalThis.onmessage = function (e) {
+  handler.handle(e)
+}

package/wasi-worker.mjs

@@ -0,0 +1,63 @@
+import fs from "node:fs";
+import { createRequire } from "node:module";
+import { parse } from "node:path";
+import { WASI } from "node:wasi";
+import { parentPort, Worker } from "node:worker_threads";
+
+const require = createRequire(import.meta.url);
+
+const { instantiateNapiModuleSync, MessageHandler, getDefaultContext } = require("@napi-rs/wasm-runtime");
+
+if (parentPort) {
+  parentPort.on("message", (data) => {
+    globalThis.onmessage({ data });
+  });
+}
+
+Object.assign(globalThis, {
+  self: globalThis,
+  require,
+  Worker,
+  importScripts: function (f) {
+    ;(0, eval)(fs.readFileSync(f, "utf8") + "//# sourceURL=" + f);
+  },
+  postMessage: function (msg) {
+    if (parentPort) {
+      parentPort.postMessage(msg);
+    }
+  },
+});
+
+const emnapiContext = getDefaultContext();
+
+const __rootDir = parse(process.cwd()).root;
+
+const handler = new MessageHandler({
+  onLoad({ wasmModule, wasmMemory }) {
+    const wasi = new WASI({
+      version: 'preview1',
+      env: process.env,
+      preopens: {
+        [__rootDir]: __rootDir,
+      },
+    });
+
+    return instantiateNapiModuleSync(wasmModule, {
+      childThread: true,
+      wasi,
+      context: emnapiContext,
+      overwriteImports(importObject) {
+        importObject.env = {
+          ...importObject.env,
+          ...importObject.napi,
+          ...importObject.emnapi,
+          memory: wasmMemory
+        };
+      },
+    });
+  },
+});
+
+globalThis.onmessage = function (e) {
+  handler.handle(e);
+};