@shelby-protocol/sdk 0.3.1 → 0.4.0

package/dist/node/parallel/parallel_commitments.mjs

@@ -205,6 +205,35 @@ import fs from "fs";
 import path from "path";
 import { fileURLToPath } from "url";
 import { Worker } from "worker_threads";
+
+// src/node/parallel/coverage_flush.ts
+import { takeCoverage } from "v8";
+var FLUSH_REQUEST = "__shelby_cov_flush_request__";
+var FLUSH_ACK = "__shelby_cov_flush_ack__";
+var coverageEnabled = () => Boolean(process.env.NODE_V8_COVERAGE);
+var isType = (msg, type) => typeof msg === "object" && msg !== null && msg.type === type;
+function flushWorkerCoverage(worker, timeoutMs = 2e3) {
+  if (!coverageEnabled()) return Promise.resolve();
+  return new Promise((resolve) => {
+    const onMessage = (msg) => {
+      if (isType(msg, FLUSH_ACK)) finish();
+    };
+    const finish = () => {
+      clearTimeout(timer);
+      worker.off("message", onMessage);
+      resolve();
+    };
+    const timer = setTimeout(finish, timeoutMs);
+    worker.on("message", onMessage);
+    try {
+      worker.postMessage({ type: FLUSH_REQUEST });
+    } catch {
+      finish();
+    }
+  });
+}
+
+// src/node/parallel/worker_pool.ts
 var WorkerPool = class {
   workers = [];
   queue = [];
@@ -241,6 +270,29 @@ var WorkerPool = class {
     const dir = path.dirname(currentPath);
     const ext = path.extname(currentPath);
     const workerExt = ext === ".mjs" ? ".mjs" : ext === ".js" ? ".js" : ".ts";
+    if (workerExt === ".ts") {
+      const srcSegment = `${path.sep}src${path.sep}`;
+      const srcIdx = currentPath.indexOf(srcSegment);
+      if (srcIdx !== -1) {
+        const packageRoot = currentPath.slice(0, srcIdx);
+        const relFromSrc = path.relative(path.join(packageRoot, "src"), dir);
+        return path.join(
+          packageRoot,
+          "dist",
+          "workers",
+          relFromSrc,
+          `${scriptName}.js`
+        );
+      }
+    }
+    let workerPath = path.join(dir, `${scriptName}${workerExt}`);
+    if (fs.existsSync(workerPath)) {
+      return workerPath;
+    }
+    workerPath = path.join(dir, "workers", `${scriptName}${workerExt}`);
+    if (fs.existsSync(workerPath)) {
+      return workerPath;
+    }
     return path.join(dir, `${scriptName}${workerExt}`);
   }
   initialize() {
@@ -412,9 +464,10 @@ var WorkerPool = class {
     for (const waiter of waiters) {
       waiter.reject(terminationError);
     }
-    const terminatePromises = this.workers.map((workerWrapper) => {
-      return new Promise((resolve) => {
-        const worker = workerWrapper.worker;
+    const terminatePromises = this.workers.map(async (workerWrapper) => {
+      const worker = workerWrapper.worker;
+      await flushWorkerCoverage(worker);
+      await new Promise((resolve) => {
         let resolved = false;
         const doResolve = () => {
           if (!resolved) {

package/dist/node/parallel/worker_pool.mjs

@@ -3,6 +3,35 @@ import fs from "fs";
 import path from "path";
 import { fileURLToPath } from "url";
 import { Worker } from "worker_threads";
+
+// src/node/parallel/coverage_flush.ts
+import { takeCoverage } from "v8";
+var FLUSH_REQUEST = "__shelby_cov_flush_request__";
+var FLUSH_ACK = "__shelby_cov_flush_ack__";
+var coverageEnabled = () => Boolean(process.env.NODE_V8_COVERAGE);
+var isType = (msg, type) => typeof msg === "object" && msg !== null && msg.type === type;
+function flushWorkerCoverage(worker, timeoutMs = 2e3) {
+  if (!coverageEnabled()) return Promise.resolve();
+  return new Promise((resolve) => {
+    const onMessage = (msg) => {
+      if (isType(msg, FLUSH_ACK)) finish();
+    };
+    const finish = () => {
+      clearTimeout(timer);
+      worker.off("message", onMessage);
+      resolve();
+    };
+    const timer = setTimeout(finish, timeoutMs);
+    worker.on("message", onMessage);
+    try {
+      worker.postMessage({ type: FLUSH_REQUEST });
+    } catch {
+      finish();
+    }
+  });
+}
+
+// src/node/parallel/worker_pool.ts
 var WorkerPool = class {
   workers = [];
   queue = [];
@@ -39,6 +68,29 @@ var WorkerPool = class {
     const dir = path.dirname(currentPath);
     const ext = path.extname(currentPath);
     const workerExt = ext === ".mjs" ? ".mjs" : ext === ".js" ? ".js" : ".ts";
+    if (workerExt === ".ts") {
+      const srcSegment = `${path.sep}src${path.sep}`;
+      const srcIdx = currentPath.indexOf(srcSegment);
+      if (srcIdx !== -1) {
+        const packageRoot = currentPath.slice(0, srcIdx);
+        const relFromSrc = path.relative(path.join(packageRoot, "src"), dir);
+        return path.join(
+          packageRoot,
+          "dist",
+          "workers",
+          relFromSrc,
+          `${scriptName}.js`
+        );
+      }
+    }
+    let workerPath = path.join(dir, `${scriptName}${workerExt}`);
+    if (fs.existsSync(workerPath)) {
+      return workerPath;
+    }
+    workerPath = path.join(dir, "workers", `${scriptName}${workerExt}`);
+    if (fs.existsSync(workerPath)) {
+      return workerPath;
+    }
     return path.join(dir, `${scriptName}${workerExt}`);
   }
   initialize() {
@@ -210,9 +262,10 @@ var WorkerPool = class {
     for (const waiter of waiters) {
       waiter.reject(terminationError);
     }
-    const terminatePromises = this.workers.map((workerWrapper) => {
-      return new Promise((resolve) => {
-        const worker = workerWrapper.worker;
+    const terminatePromises = this.workers.map(async (workerWrapper) => {
+      const worker = workerWrapper.worker;
+      await flushWorkerCoverage(worker);
+      await new Promise((resolve) => {
         let resolved = false;
         const doResolve = () => {
           if (!resolved) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@shelby-protocol/sdk",
-  "version": "0.3.1",
+  "version": "0.4.0",
   "private": false,
   "publishConfig": {
     "access": "public"
@@ -18,19 +18,24 @@
       "types": "./dist/node/parallel/index.d.ts",
       "import": "./dist/node/parallel/index.mjs"
     },
+    "./sp": {
+      "types": "./dist/core/sp/index.d.ts",
+      "import": "./dist/core/sp/index.mjs"
+    },
     "./browser": {
       "types": "./dist/browser/index.d.ts",
       "import": "./dist/browser/index.mjs"
     }
   },
   "dependencies": {
+    "@bufbuild/protobuf": "^2.8.0",
     "graphql": "^16.11.0",
     "graphql-request": "^7.2.0",
     "graphql-tag": "^2.12.6",
     "p-limit": "7.1.1",
     "zod": "^3.24.3",
-    "@shelby-protocol/clay-codes": "0.0.2",
-    "@shelby-protocol/reed-solomon": "0.0.1"
+    "@shelby-protocol/reed-solomon": "0.0.1",
+    "@shelby-protocol/clay-codes": "0.0.2"
   },
   "devDependencies": {
     "@aptos-labs/ts-sdk": "5.2.1",
@@ -53,12 +58,14 @@
   "scripts": {
     "lint": "biome check .",
     "fmt": "biome check . --write",
-    "generate": "graphql-codegen",
+    "generate": "graphql-codegen && (cd ../.. && pnpm exec buf generate --template buf.gen.server.yaml --path proto/rpc_server.proto)",
     "generate:watch": "graphql-codegen --watch",
+    "generate:schema": "node scripts/dump-schema.mjs",
     "verify:config": "tsx scripts/verify-config.ts",
     "build": "tsc --noEmit && (rimraf dist; tsup) && pnpm run verify:config",
     "test": "vitest dev",
     "test:once": "pnpm run build && vitest run",
+    "test:integration": "vitest run --config vitest.integration.config.ts --reporter=verbose",
     "benchmark:erasure": "tsx benchmark/erasure.ts"
   }
 }