@treeseed/core 0.4.6 → 0.4.7

package/dist/services/worker.js

@@ -0,0 +1,108 @@
+#!/usr/bin/env node
+import { fileURLToPath } from "node:url";
+import { createGatewayClient, createQueueClient, resolveWorkerConfig } from "./common.js";
+async function managerRequest(baseUrl, path, body) {
+  const response = await fetch(`${baseUrl}${path}`, {
+    method: "POST",
+    headers: {
+      accept: "application/json",
+      "content-type": "application/json"
+    },
+    body: JSON.stringify(body)
+  });
+  const payload = await response.json().catch(() => ({}));
+  if (!response.ok) {
+    throw new Error(typeof payload.error === "string" ? payload.error : `Manager request failed with ${response.status}.`);
+  }
+  return payload;
+}
+async function runWorkerCycle() {
+  const gateway = createGatewayClient();
+  const queue = createQueueClient();
+  const config = resolveWorkerConfig();
+  if (!gateway || !queue) {
+    throw new Error("Worker requires TREESEED_GATEWAY_BASE_URL, TREESEED_GATEWAY_BEARER_TOKEN, CLOUDFLARE_ACCOUNT_ID, TREESEED_QUEUE_ID, and TREESEED_QUEUE_PULL_TOKEN.");
+  }
+  const pulled = await queue.pull({
+    batchSize: config.batchSize,
+    visibilityTimeoutMs: config.visibilityTimeoutMs
+  });
+  if (pulled.messages.length === 0) {
+    return { ok: true, processed: 0 };
+  }
+  let processed = 0;
+  for (const message of pulled.messages) {
+    try {
+      await gateway.requestJson(`/tasks/${encodeURIComponent(message.body.taskId)}/claim`, {
+        body: {
+          workerId: config.workerId,
+          leaseSeconds: config.leaseSeconds
+        }
+      });
+      const context = await managerRequest(
+        config.managerBaseUrl,
+        "/internal/context/resolve-task",
+        { taskId: message.body.taskId }
+      );
+      const task = context.payload.task;
+      const payload = task && typeof task.payloadJson === "string" ? JSON.parse(task.payloadJson) : {};
+      await gateway.requestJson(`/tasks/${encodeURIComponent(message.body.taskId)}/progress`, {
+        body: {
+          workerId: config.workerId,
+          state: "running",
+          appendEvent: {
+            kind: "worker_started",
+            data: { workerId: config.workerId, queueAttempt: message.attempts }
+          }
+        }
+      });
+      await gateway.requestJson(`/tasks/${encodeURIComponent(message.body.taskId)}/complete`, {
+        body: {
+          output: {
+            workerId: config.workerId,
+            queueAttempt: message.attempts,
+            payload
+          },
+          summary: {
+            status: "completed",
+            workerId: config.workerId
+          }
+        }
+      });
+      await queue.ack([message.leaseId]);
+      processed += 1;
+    } catch (error) {
+      const retryDelaySeconds = Math.min(300, Math.max(15, message.attempts * 30));
+      await gateway.requestJson(`/tasks/${encodeURIComponent(message.body.taskId)}/fail`, {
+        body: {
+          errorMessage: error instanceof Error ? error.message : String(error),
+          retryable: true,
+          nextVisibleAt: new Date(Date.now() + retryDelaySeconds * 1e3).toISOString()
+        }
+      }).catch(() => null);
+      await queue.retry([{ leaseId: message.leaseId, delaySeconds: retryDelaySeconds }]);
+    }
+  }
+  return { ok: true, processed };
+}
+async function startWorkerLoop() {
+  const config = resolveWorkerConfig();
+  for (; ; ) {
+    try {
+      await runWorkerCycle();
+    } catch (error) {
+      process.stderr.write(`${error instanceof Error ? error.message : String(error)}
+`);
+    }
+    await new Promise((resolve) => setTimeout(resolve, config.pollIntervalMs));
+  }
+}
+const currentFile = fileURLToPath(import.meta.url);
+const entryFile = process.argv[1] ?? "";
+if (entryFile === currentFile) {
+  await startWorkerLoop();
+}
+export {
+  runWorkerCycle,
+  startWorkerLoop
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@treeseed/core",
-  "version": "0.4.6",
+  "version": "0.4.7",
   "description": "Treeseed integrated platform starter for Astro/Starlight web runtimes and Hono API runtimes.",
   "license": "AGPL-3.0-only",
   "repository": {
@@ -29,6 +29,7 @@
     "access": "public"
   },
   "bin": {
+    "treeseed-agents": "./dist/agents/cli.js",
     "treeseed-workspace-bootstrap": "./dist/scripts/workspace-bootstrap.js"
   },
   "scripts": {
@@ -39,6 +40,10 @@
     "dev": "node ./scripts/run-ts.mjs ./scripts/dev-platform.ts",
     "dev:web": "node ./scripts/run-ts.mjs ./scripts/dev-platform.ts --surface web",
     "dev:api": "node ./scripts/run-ts.mjs ./scripts/dev-platform.ts --surface api",
+    "dev:manager": "node ./scripts/run-ts.mjs ./src/services/manager.ts",
+    "dev:worker": "node ./scripts/run-ts.mjs ./src/services/worker.ts",
+    "dev:workday-start": "node ./scripts/run-ts.mjs ./src/services/workday-start.ts",
+    "dev:workday-report": "node ./scripts/run-ts.mjs ./src/services/workday-report.ts",
     "dev:watch": "node ./scripts/run-ts.mjs ./scripts/dev-platform.ts --watch",
     "starlight:patch": "node ./scripts/run-ts.mjs ./scripts/patch-starlight-content-path.ts",
     "precheck": "npm run starlight:patch",
@@ -62,7 +67,7 @@
     "release:publish": "node ./scripts/run-ts.mjs ./scripts/publish-package.ts"
   },
   "dependencies": {
-    "@treeseed/sdk": "^0.4.5",
+    "@treeseed/sdk": "^0.4.6",
     "@astrojs/check": "^0.9.8",
     "@astrojs/cloudflare": "^12.6.13",
     "@astrojs/sitemap": "3.7.0",
@@ -107,6 +112,26 @@
       "types": "./dist/api.d.ts",
       "default": "./dist/api.js"
     },
+    "./agent": {
+      "types": "./dist/agent.d.ts",
+      "default": "./dist/agent.js"
+    },
+    "./agent/cli": {
+      "types": "./dist/agents/cli.d.ts",
+      "default": "./dist/agents/cli.js"
+    },
+    "./runtime-types": {
+      "types": "./dist/agents/runtime-types.d.ts",
+      "default": "./dist/agents/runtime-types.js"
+    },
+    "./contracts/messages": {
+      "types": "./dist/agents/contracts/messages.d.ts",
+      "default": "./dist/agents/contracts/messages.js"
+    },
+    "./contracts/run": {
+      "types": "./dist/agents/contracts/run.d.ts",
+      "default": "./dist/agents/contracts/run.js"
+    },
     "./railway": {
       "types": "./dist/railway.d.ts",
       "default": "./dist/railway.js"
@@ -134,6 +159,22 @@
     "./tenant": "./dist/tenant/bridge.js",
     "./scripts/dev-platform": "./dist/scripts/dev-platform.js",
     "./scripts/workspace-bootstrap": "./dist/scripts/workspace-bootstrap.js",
+    "./services/manager": {
+      "types": "./dist/services/manager.d.ts",
+      "default": "./dist/services/manager.js"
+    },
+    "./services/worker": {
+      "types": "./dist/services/worker.d.ts",
+      "default": "./dist/services/worker.js"
+    },
+    "./services/workday-start": {
+      "types": "./dist/services/workday-start.d.ts",
+      "default": "./dist/services/workday-start.js"
+    },
+    "./services/workday-report": {
+      "types": "./dist/services/workday-report.d.ts",
+      "default": "./dist/services/workday-report.js"
+    },
     "./api/app": {
       "types": "./dist/api/app.d.ts",
       "default": "./dist/api/app.js"