effortless-aws 0.30.0 → 0.32.0

package/dist/{chunk-ISJC6CHC.js → chunk-HGSMOO4A.js}

@@ -474,6 +474,62 @@ var createQueueClient = (queueName) => {
   };
 };
 
+// src/runtime/worker-client.ts
+import { SQS as SQS2 } from "@aws-sdk/client-sqs";
+import { ECSClient, DescribeServicesCommand, UpdateServiceCommand } from "@aws-sdk/client-ecs";
+var createWorkerClient = (depValue) => {
+  const lastColon = depValue.lastIndexOf(":");
+  const workerName = depValue.slice(0, lastColon);
+  const idleTimeoutMs = Number(depValue.slice(lastColon + 1)) * 1e3;
+  const queueName = `${workerName}-worker`;
+  const cluster = workerName.replace(/-[^-]+$/, "");
+  const service = workerName;
+  let sqsClient = null;
+  const getSqs = () => sqsClient ??= new SQS2({});
+  let ecsClient = null;
+  const getEcs = () => ecsClient ??= new ECSClient({});
+  let resolvedQueueUrl;
+  const getQueueUrl = async () => {
+    if (resolvedQueueUrl) return resolvedQueueUrl;
+    const result = await getSqs().getQueueUrl({ QueueName: queueName });
+    resolvedQueueUrl = result.QueueUrl;
+    return resolvedQueueUrl;
+  };
+  let awakeUntil = 0;
+  const ensureRunning = async () => {
+    if (Date.now() < awakeUntil) return;
+    const resp = await getEcs().send(new DescribeServicesCommand({ cluster, services: [service] }));
+    const svc = resp.services?.[0];
+    if (svc && svc.desiredCount === 0) {
+      await getEcs().send(new UpdateServiceCommand({ cluster, service, desiredCount: 1 }));
+    }
+    awakeUntil = Date.now() + idleTimeoutMs;
+  };
+  return {
+    async send(msg, options) {
+      const queueUrl = await getQueueUrl();
+      await getSqs().sendMessage({
+        QueueUrl: queueUrl,
+        MessageBody: JSON.stringify(msg),
+        ...options?.delay ? { DelaySeconds: toSeconds(options.delay) } : {}
+      });
+      if (options?.start !== false && !options?.delay) {
+        await ensureRunning();
+      }
+    },
+    async status() {
+      const resp = await getEcs().send(new DescribeServicesCommand({ cluster, services: [service] }));
+      const svc = resp.services?.[0];
+      if (svc && svc.runningCount && svc.runningCount > 0) return "running";
+      return "idle";
+    },
+    async stop() {
+      await getEcs().send(new UpdateServiceCommand({ cluster, service, desiredCount: 0 }));
+      awakeUntil = 0;
+    }
+  };
+};
+
 // src/runtime/ssm-client.ts
 import { SSM } from "@aws-sdk/client-ssm";
 var client = null;
@@ -512,7 +568,8 @@ var DEP_FACTORIES = {
   },
   bucket: (name) => createBucketClient(name),
   mailer: () => createEmailClient(),
-  queue: (name) => createQueueClient(name)
+  queue: (name) => createQueueClient(name),
+  worker: (name) => createWorkerClient(name)
 };
 var parseDepValue = (raw) => {
   const idx = raw.indexOf(":");
@@ -678,5 +735,7 @@ export {
   createTableClient,
   AUTH_COOKIE_NAME,
   createBucketClient,
+  buildDeps,
+  buildParams,
   createHandlerRuntime
 };