effortless-aws 0.36.1 → 0.37.0

package/dist/runtime/wrap-table-stream.js

@@ -1,7 +1,11 @@
 import {
-  createHandlerRuntime,
   createTableClient
-} from "../chunk-Q3ZDMZF3.js";
+} from "../chunk-JFNBO4K3.js";
+import "../chunk-U56MLLWP.js";
+import {
+  createHandlerRuntime
+} from "../chunk-3NKYT4OX.js";
+import "../chunk-VONNUUEN.js";
 
 // src/runtime/wrap-table-stream.ts
 import { unmarshall } from "@aws-sdk/util-dynamodb";
@@ -44,20 +48,20 @@ var wrapTableStream = (handler) => {
   const tagField = handler.__spec.tagField ?? "tag";
   const concurrency = handler.__spec.concurrency ?? 1;
   let selfClient = null;
-  const getSelfClient = () => {
+  const getSelfClient = async () => {
     if (selfClient) return selfClient;
     const raw = process.env[ENV_DEP_SELF];
     if (!raw) return void 0;
     const tableName = raw.startsWith("table:") ? raw.slice(6) : raw;
-    selfClient = createTableClient(tableName, { tagField });
+    selfClient = await createTableClient(tableName, { tagField });
     return selfClient;
   };
-  const rt = createHandlerRuntime(handler, "table", handler.__spec.lambda?.logLevel ?? "info", () => {
-    const table = getSelfClient();
+  const rt = createHandlerRuntime(handler, "table", handler.__spec.lambda?.logLevel ?? "info", async () => {
+    const table = await getSelfClient();
     return table ? { table } : {};
   });
   const handleError = handler.onError ?? (({ error }) => console.error(`[effortless:${rt.handlerName}]`, error));
-  return async (event) => {
+  const fn = async (event) => {
     const startTime = Date.now();
     rt.patchConsole();
     let ctxProps = {};
@@ -67,7 +71,8 @@ var wrapTableStream = (handler) => {
       const common = await rt.commonArgs();
       const ctx = common.ctx;
       ctxProps = ctx && typeof ctx === "object" ? { ...ctx } : {};
-      const shared = { ...ctxProps };
+      const table = await getSelfClient();
+      const shared = { ...ctxProps, ...table ? { table } : {} };
       let records;
       let sequenceNumbers;
       try {
@@ -141,6 +146,8 @@ var wrapTableStream = (handler) => {
       rt.restoreConsole();
     }
   };
+  fn.__preload = () => rt.preload();
+  return fn;
 };
 export {
   wrapTableStream

package/dist/runtime/wrap-worker.js

@@ -1,7 +1,8 @@
 import {
   buildDeps,
   buildParams
-} from "../chunk-Q3ZDMZF3.js";
+} from "../chunk-3NKYT4OX.js";
+import "../chunk-VONNUUEN.js";
 
 // src/runtime/wrap-worker.ts
 import { SQS } from "@aws-sdk/client-sqs";

package/dist/ssm-client-2XDJC3HF.js

@@ -0,0 +1,33 @@
+import {
+  lazyImport
+} from "./chunk-U56MLLWP.js";
+
+// src/runtime/ssm-client.ts
+var loadSdk = () => lazyImport("@aws-sdk/client-ssm");
+var client = null;
+var getClient = async () => {
+  if (!client) {
+    const { SSM: SSMCls } = await loadSdk();
+    client = new SSMCls({});
+  }
+  return client;
+};
+var getParameters = async (names) => {
+  const map = /* @__PURE__ */ new Map();
+  for (let i = 0; i < names.length; i += 10) {
+    const batch = names.slice(i, i + 10);
+    const result = await (await getClient()).getParameters({
+      Names: batch,
+      WithDecryption: true
+    });
+    for (const p of result.Parameters ?? []) {
+      if (p.Name && p.Value !== void 0) {
+        map.set(p.Name, p.Value);
+      }
+    }
+  }
+  return map;
+};
+export {
+  getParameters
+};

package/dist/table-client-BBLIC3EV.js

@@ -0,0 +1,7 @@
+import {
+  createTableClient
+} from "./chunk-JFNBO4K3.js";
+import "./chunk-U56MLLWP.js";
+export {
+  createTableClient
+};

package/dist/worker-client-ZS25XCC2.js

@@ -0,0 +1,69 @@
+import {
+  lazyImport
+} from "./chunk-U56MLLWP.js";
+import {
+  toSeconds
+} from "./chunk-VONNUUEN.js";
+
+// src/runtime/worker-client.ts
+var loadSqs = () => lazyImport("@aws-sdk/client-sqs");
+var loadEcs = () => lazyImport("@aws-sdk/client-ecs");
+var createWorkerClient = async (depValue) => {
+  const [{ SQS: SQSCls }, { ECSClient: ECSCls, DescribeServicesCommand, UpdateServiceCommand }] = await Promise.all([
+    loadSqs(),
+    loadEcs()
+  ]);
+  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 SQSCls({});
+  let ecsClient = null;
+  const getEcs = () => ecsClient ??= new ECSCls({});
+  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;
+    }
+  };
+};
+export {
+  createWorkerClient
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "effortless-aws",
-  "version": "0.36.1",
+  "version": "0.37.0",
   "license": "MIT",
   "type": "module",
   "description": "Code-first AWS Lambda framework. Export handlers, deploy with one command.",