@highstate/worker-sdk 0.9.16

package/dist/highstate.manifest.json

@@ -0,0 +1,5 @@
+{
+  "sourceHashes": {
+    "./dist/index.js": 2136108195
+  }
+}

package/dist/index.js

@@ -0,0 +1,87 @@
+import { EventEmitter } from 'node:events';
+import { createInterface } from 'node:readline/promises';
+import { WorkerServiceDefinition } from '@highstate/api/worker.v1';
+import { createClientFactory, createChannel } from 'nice-grpc';
+import { createAuthenticationMiddleware } from '@highstate/api';
+
+// src/index.ts
+var Worker = class _Worker {
+  constructor(projectId, workerId, apiKey, apiUrl) {
+    this.workerId = workerId;
+    const authMiddleware = createAuthenticationMiddleware(apiKey, projectId);
+    this.clientFactory = createClientFactory().use(authMiddleware);
+    this.channel = createChannel(apiUrl);
+  }
+  eventEmitter = new EventEmitter();
+  clientFactory;
+  channel;
+  onUnitRegistration(handler) {
+    const handle = async (instanceId, params) => {
+      try {
+        await handler(instanceId, params);
+      } catch (error) {
+        console.error(`Error handling unit registration for instance ${instanceId}:`, error);
+      }
+    };
+    this.eventEmitter.on(
+      "unitRegistration",
+      (instanceId, params) => void handle(instanceId, params)
+    );
+  }
+  onUnitDeregistration(handler) {
+    const handle = async (instanceId) => {
+      try {
+        await handler(instanceId);
+      } catch (error) {
+        console.error(`Error handling unit deregistration for instance ${instanceId}:`, error);
+      }
+    };
+    this.eventEmitter.on("unitDeregistration", (instanceId) => void handle(instanceId));
+  }
+  createClient(definition) {
+    return this.clientFactory.create(definition, this.channel);
+  }
+  async start() {
+    const workerClient = this.createClient(WorkerServiceDefinition);
+    for await (const { event } of workerClient.connect({ workerId: this.workerId })) {
+      switch (event?.$case) {
+        case "unitRegistration": {
+          this.eventEmitter.emit(
+            "unitRegistration",
+            event.value.instanceId,
+            event.value.params
+          );
+          break;
+        }
+        case "unitDeregistration": {
+          this.eventEmitter.emit("unitDeregistration", event.value.instanceId);
+          break;
+        }
+      }
+    }
+  }
+  static async create() {
+    for await (const line of createInterface({ input: process.stdin })) {
+      let data;
+      try {
+        data = JSON.parse(line);
+      } catch (error) {
+        throw new Error("Failed to parse worker input", { cause: error });
+      }
+      if (!data.apiKey || !data.apiUrl || !data.projectId || !data.workerId) {
+        throw new Error(
+          "Worker input must contain apiKey, apiUrl, projectId, and workerId properties"
+        );
+      }
+      if (typeof data.apiKey !== "string" || typeof data.apiUrl !== "string" || typeof data.projectId !== "string" || typeof data.workerId !== "string") {
+        throw new Error("Worker input properties must be strings");
+      }
+      return new _Worker(data.projectId, data.workerId, data.apiKey, data.apiUrl);
+    }
+    throw new Error("Failed to read worker input from stdin");
+  }
+};
+
+export { Worker };
+//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;AAYa,IAAA,MAAA,GAAN,MAAM,OAA0E,CAAA;AAAA,EAK7E,WACN,CAAA,SAAA,EACiB,QACjB,EAAA,MAAA,EACA,MACA,EAAA;AAHiB,IAAA,IAAA,CAAA,QAAA,GAAA,QAAA;AAIjB,IAAM,MAAA,cAAA,GAAiB,8BAA+B,CAAA,MAAA,EAAQ,SAAS,CAAA;AAEvE,IAAA,IAAA,CAAK,aAAgB,GAAA,mBAAA,EAAsB,CAAA,GAAA,CAAI,cAAc,CAAA;AAC7D,IAAK,IAAA,CAAA,OAAA,GAAU,cAAc,MAAM,CAAA;AAAA;AACrC,EAdiB,YAAA,GAAe,IAAI,YAAa,EAAA;AAAA,EAChC,aAAA;AAAA,EACA,OAAA;AAAA,EAcjB,mBAAmB,OAAwE,EAAA;AACzF,IAAM,MAAA,MAAA,GAAS,OAAO,UAAA,EAAoB,MAAoB,KAAA;AAC5D,MAAI,IAAA;AACF,QAAM,MAAA,OAAA,CAAQ,YAAY,MAAM,CAAA;AAAA,eACzB,KAAO,EAAA;AACd,QAAA,OAAA,CAAQ,KAAM,CAAA,CAAA,8CAAA,EAAiD,UAAU,CAAA,CAAA,CAAA,EAAK,KAAK,CAAA;AAAA;AACrF,KACF;AAEA,IAAA,IAAA,CAAK,YAAa,CAAA,EAAA;AAAA,MAChB,kBAAA;AAAA,MACA,CAAC,UAAoB,EAAA,MAAA,KAAoB,KAAK,MAAA,CAAO,YAAY,MAAM;AAAA,KACzE;AAAA;AACF,EAEA,qBAAqB,OAAuD,EAAA;AAC1E,IAAM,MAAA,MAAA,GAAS,OAAO,UAAuB,KAAA;AAC3C,MAAI,IAAA;AACF,QAAA,MAAM,QAAQ,UAAU,CAAA;AAAA,eACjB,KAAO,EAAA;AACd,QAAA,OAAA,CAAQ,KAAM,CAAA,CAAA,gDAAA,EAAmD,UAAU,CAAA,CAAA,CAAA,EAAK,KAAK,CAAA;AAAA;AACvF,KACF;AAEA,IAAK,IAAA,CAAA,YAAA,CAAa,GAAG,oBAAsB,EAAA,CAAC,eAAuB,KAAK,MAAA,CAAO,UAAU,CAAC,CAAA;AAAA;AAC5F,EAEA,aAAuD,UAAsB,EAAA;AAC3E,IAAA,OAAO,IAAK,CAAA,aAAA,CAAc,MAAO,CAAA,UAAA,EAAY,KAAK,OAAO,CAAA;AAAA;AAC3D,EAEA,MAAM,KAAuB,GAAA;AAC3B,IAAM,MAAA,YAAA,GAAe,IAAK,CAAA,YAAA,CAAa,uBAAuB,CAAA;AAE9D,IAAiB,WAAA,MAAA,EAAE,KAAM,EAAA,IAAK,YAAa,CAAA,OAAA,CAAQ,EAAE,QAAU,EAAA,IAAA,CAAK,QAAS,EAAC,CAAG,EAAA;AAC/E,MAAA,QAAQ,OAAO,KAAO;AAAA,QACpB,KAAK,kBAAoB,EAAA;AACvB,UAAA,IAAA,CAAK,YAAa,CAAA,IAAA;AAAA,YAChB,kBAAA;AAAA,YACA,MAAM,KAAM,CAAA,UAAA;AAAA,YACZ,MAAM,KAAM,CAAA;AAAA,WACd;AACA,UAAA;AAAA;AACF,QACA,KAAK,oBAAsB,EAAA;AACzB,UAAA,IAAA,CAAK,YAAa,CAAA,IAAA,CAAK,oBAAsB,EAAA,KAAA,CAAM,MAAM,UAAU,CAAA;AACnE,UAAA;AAAA;AACF;AACF;AACF;AACF,EAEA,aAAa,MAEX,GAAA;AACA,IAAA,WAAA,MAAiB,QAAQ,eAAgB,CAAA,EAAE,OAAO,OAAQ,CAAA,KAAA,EAAO,CAAG,EAAA;AAClE,MAAI,IAAA,IAAA;AACJ,MAAI,IAAA;AACF,QAAO,IAAA,GAAA,IAAA,CAAK,MAAM,IAAI,CAAA;AAAA,eACf,KAAO,EAAA;AACd,QAAA,MAAM,IAAI,KAAM,CAAA,8BAAA,EAAgC,EAAE,KAAA,EAAO,OAAO,CAAA;AAAA;AAGlE,MAAI,IAAA,CAAC,IAAK,CAAA,MAAA,IAAU,CAAC,IAAA,CAAK,MAAU,IAAA,CAAC,IAAK,CAAA,SAAA,IAAa,CAAC,IAAA,CAAK,QAAU,EAAA;AACrE,QAAA,MAAM,IAAI,KAAA;AAAA,UACR;AAAA,SACF;AAAA;AAGF,MAAA,IACE,OAAO,IAAA,CAAK,MAAW,KAAA,QAAA,IACvB,OAAO,IAAK,CAAA,MAAA,KAAW,QACvB,IAAA,OAAO,KAAK,SAAc,KAAA,QAAA,IAC1B,OAAO,IAAA,CAAK,aAAa,QACzB,EAAA;AACA,QAAM,MAAA,IAAI,MAAM,yCAAyC,CAAA;AAAA;AAG3D,MAAO,OAAA,IAAI,QAAgB,IAAK,CAAA,SAAA,EAAW,KAAK,QAAU,EAAA,IAAA,CAAK,MAAQ,EAAA,IAAA,CAAK,MAAM,CAAA;AAAA;AAGpF,IAAM,MAAA,IAAI,MAAM,wCAAwC,CAAA;AAAA;AAE5D","file":"index.js","sourcesContent":["import { EventEmitter } from \"node:events\"\nimport { createInterface } from \"node:readline/promises\"\nimport { WorkerServiceDefinition } from \"@highstate/api/worker.v1\"\nimport {\n  Channel,\n  createChannel,\n  createClientFactory,\n  type ClientFactory,\n  type CompatServiceDefinition,\n} from \"nice-grpc\"\nimport { createAuthenticationMiddleware } from \"@highstate/api\"\n\nexport class Worker<TParams extends Record<string, unknown> = Record<string, unknown>> {\n  private readonly eventEmitter = new EventEmitter()\n  private readonly clientFactory: ClientFactory\n  private readonly channel: Channel\n\n  private constructor(\n    projectId: string,\n    private readonly workerId: string,\n    apiKey: string,\n    apiUrl: string,\n  ) {\n    const authMiddleware = createAuthenticationMiddleware(apiKey, projectId)\n\n    this.clientFactory = createClientFactory().use(authMiddleware)\n    this.channel = createChannel(apiUrl)\n  }\n\n  onUnitRegistration(handler: (instanceId: string, params: TParams) => Promise<void> | void) {\n    const handle = async (instanceId: string, params: TParams) => {\n      try {\n        await handler(instanceId, params)\n      } catch (error) {\n        console.error(`Error handling unit registration for instance ${instanceId}:`, error)\n      }\n    }\n\n    this.eventEmitter.on(\n      \"unitRegistration\",\n      (instanceId: string, params: TParams) => void handle(instanceId, params),\n    )\n  }\n\n  onUnitDeregistration(handler: (instanceId: string) => Promise<void> | void) {\n    const handle = async (instanceId: string) => {\n      try {\n        await handler(instanceId)\n      } catch (error) {\n        console.error(`Error handling unit deregistration for instance ${instanceId}:`, error)\n      }\n    }\n\n    this.eventEmitter.on(\"unitDeregistration\", (instanceId: string) => void handle(instanceId))\n  }\n\n  createClient<TService extends CompatServiceDefinition>(definition: TService) {\n    return this.clientFactory.create(definition, this.channel)\n  }\n\n  async start(): Promise<void> {\n    const workerClient = this.createClient(WorkerServiceDefinition)\n\n    for await (const { event } of workerClient.connect({ workerId: this.workerId })) {\n      switch (event?.$case) {\n        case \"unitRegistration\": {\n          this.eventEmitter.emit(\n            \"unitRegistration\",\n            event.value.instanceId,\n            event.value.params as TParams,\n          )\n          break\n        }\n        case \"unitDeregistration\": {\n          this.eventEmitter.emit(\"unitDeregistration\", event.value.instanceId)\n          break\n        }\n      }\n    }\n  }\n\n  static async create<TParams extends Record<string, unknown> = Record<string, unknown>>(): Promise<\n    Worker<TParams>\n  > {\n    for await (const line of createInterface({ input: process.stdin })) {\n      let data: Record<string, unknown>\n      try {\n        data = JSON.parse(line) as Record<string, unknown>\n      } catch (error) {\n        throw new Error(\"Failed to parse worker input\", { cause: error })\n      }\n\n      if (!data.apiKey || !data.apiUrl || !data.projectId || !data.workerId) {\n        throw new Error(\n          \"Worker input must contain apiKey, apiUrl, projectId, and workerId properties\",\n        )\n      }\n\n      if (\n        typeof data.apiKey !== \"string\" ||\n        typeof data.apiUrl !== \"string\" ||\n        typeof data.projectId !== \"string\" ||\n        typeof data.workerId !== \"string\"\n      ) {\n        throw new Error(\"Worker input properties must be strings\")\n      }\n\n      return new Worker<TParams>(data.projectId, data.workerId, data.apiKey, data.apiUrl)\n    }\n\n    throw new Error(\"Failed to read worker input from stdin\")\n  }\n}\n"]}

package/package.json

@@ -0,0 +1,33 @@
+{
+  "name": "@highstate/worker-sdk",
+  "version": "0.9.16",
+  "type": "module",
+  "files": [
+    "dist",
+    "src"
+  ],
+  "exports": {
+    ".": {
+      "types": "./src/index.ts",
+      "default": "./dist/index.js"
+    }
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "scripts": {
+    "build": "highstate build"
+  },
+  "devDependencies": {
+    "@highstate/cli": "^0.9.16"
+  },
+  "dependencies": {
+    "@bufbuild/protobuf": "^2.6.0",
+    "@highstate/api": "^0.9.16",
+    "long": "^5.3.2",
+    "nice-grpc": "^2.1.12",
+    "nice-grpc-common": "^2.0.2",
+    "protobufjs": "^7.5.3"
+  },
+  "gitHead": "458d6f1f9f6d4aec0ba75a2b2c4c01408cb9c8df"
+}

package/src/index.ts

@@ -0,0 +1,113 @@
+import { EventEmitter } from "node:events"
+import { createInterface } from "node:readline/promises"
+import { WorkerServiceDefinition } from "@highstate/api/worker.v1"
+import {
+  Channel,
+  createChannel,
+  createClientFactory,
+  type ClientFactory,
+  type CompatServiceDefinition,
+} from "nice-grpc"
+import { createAuthenticationMiddleware } from "@highstate/api"
+
+export class Worker<TParams extends Record<string, unknown> = Record<string, unknown>> {
+  private readonly eventEmitter = new EventEmitter()
+  private readonly clientFactory: ClientFactory
+  private readonly channel: Channel
+
+  private constructor(
+    projectId: string,
+    private readonly workerId: string,
+    apiKey: string,
+    apiUrl: string,
+  ) {
+    const authMiddleware = createAuthenticationMiddleware(apiKey, projectId)
+
+    this.clientFactory = createClientFactory().use(authMiddleware)
+    this.channel = createChannel(apiUrl)
+  }
+
+  onUnitRegistration(handler: (instanceId: string, params: TParams) => Promise<void> | void) {
+    const handle = async (instanceId: string, params: TParams) => {
+      try {
+        await handler(instanceId, params)
+      } catch (error) {
+        console.error(`Error handling unit registration for instance ${instanceId}:`, error)
+      }
+    }
+
+    this.eventEmitter.on(
+      "unitRegistration",
+      (instanceId: string, params: TParams) => void handle(instanceId, params),
+    )
+  }
+
+  onUnitDeregistration(handler: (instanceId: string) => Promise<void> | void) {
+    const handle = async (instanceId: string) => {
+      try {
+        await handler(instanceId)
+      } catch (error) {
+        console.error(`Error handling unit deregistration for instance ${instanceId}:`, error)
+      }
+    }
+
+    this.eventEmitter.on("unitDeregistration", (instanceId: string) => void handle(instanceId))
+  }
+
+  createClient<TService extends CompatServiceDefinition>(definition: TService) {
+    return this.clientFactory.create(definition, this.channel)
+  }
+
+  async start(): Promise<void> {
+    const workerClient = this.createClient(WorkerServiceDefinition)
+
+    for await (const { event } of workerClient.connect({ workerId: this.workerId })) {
+      switch (event?.$case) {
+        case "unitRegistration": {
+          this.eventEmitter.emit(
+            "unitRegistration",
+            event.value.instanceId,
+            event.value.params as TParams,
+          )
+          break
+        }
+        case "unitDeregistration": {
+          this.eventEmitter.emit("unitDeregistration", event.value.instanceId)
+          break
+        }
+      }
+    }
+  }
+
+  static async create<TParams extends Record<string, unknown> = Record<string, unknown>>(): Promise<
+    Worker<TParams>
+  > {
+    for await (const line of createInterface({ input: process.stdin })) {
+      let data: Record<string, unknown>
+      try {
+        data = JSON.parse(line) as Record<string, unknown>
+      } catch (error) {
+        throw new Error("Failed to parse worker input", { cause: error })
+      }
+
+      if (!data.apiKey || !data.apiUrl || !data.projectId || !data.workerId) {
+        throw new Error(
+          "Worker input must contain apiKey, apiUrl, projectId, and workerId properties",
+        )
+      }
+
+      if (
+        typeof data.apiKey !== "string" ||
+        typeof data.apiUrl !== "string" ||
+        typeof data.projectId !== "string" ||
+        typeof data.workerId !== "string"
+      ) {
+        throw new Error("Worker input properties must be strings")
+      }
+
+      return new Worker<TParams>(data.projectId, data.workerId, data.apiKey, data.apiUrl)
+    }
+
+    throw new Error("Failed to read worker input from stdin")
+  }
+}