@clef-sh/agent 0.1.9 → 0.1.10

package/dist/agent.cjs

@@ -184795,10 +184795,100 @@ var Daemon = class {
   }
 };
 
+// src/lifecycle/lambda-extension.ts
+var LambdaExtension = class {
+  lastRefresh = 0;
+  options;
+  startedAt;
+  constructor(options) {
+    this.options = options;
+    this.startedAt = Date.now();
+  }
+  /** Run the Lambda Extension lifecycle. */
+  async start() {
+    const { poller, server, onLog, refreshTtl, telemetry } = this.options;
+    const extensionId = await this.register();
+    onLog?.(`Registered with Lambda Extensions API (id: ${extensionId})`);
+    onLog?.(`Agent server listening at ${server.url}`);
+    if (!this.options.skipInitialFetch) {
+      if (refreshTtl === 0) {
+        await poller.fetchAndValidate();
+      } else {
+        await poller.fetchAndDecrypt();
+      }
+      onLog?.("Initial secrets loaded.");
+    }
+    this.lastRefresh = Date.now();
+    while (true) {
+      const event = await this.nextEvent(extensionId);
+      if (event.eventType === "SHUTDOWN") {
+        onLog?.("SHUTDOWN event received.");
+        poller.stop();
+        telemetry?.agentStopped({
+          reason: "lambda_shutdown",
+          uptimeSeconds: Math.round((Date.now() - this.startedAt) / 1e3)
+        });
+        try {
+          await telemetry?.stopAsync();
+        } catch {
+        }
+        await server.stop();
+        break;
+      }
+      if (event.eventType === "INVOKE") {
+        if (refreshTtl === 0) {
+          try {
+            await poller.fetchAndValidate();
+            this.lastRefresh = Date.now();
+          } catch (err) {
+            onLog?.(`Refresh failed: ${err instanceof Error ? err.message : String(err)}`);
+          }
+        } else {
+          const elapsed = (Date.now() - this.lastRefresh) / 1e3;
+          if (elapsed >= refreshTtl) {
+            try {
+              await poller.fetchAndDecrypt();
+              this.lastRefresh = Date.now();
+            } catch (err) {
+              onLog?.(`Refresh failed: ${err instanceof Error ? err.message : String(err)}`);
+            }
+          }
+        }
+      }
+    }
+  }
+  async register() {
+    const res = await fetch("http://127.0.0.1:9001/2020-01-01/extension/register", {
+      method: "POST",
+      headers: { "Lambda-Extension-Name": "clef-agent" },
+      body: JSON.stringify({ events: ["INVOKE", "SHUTDOWN"] })
+    });
+    if (!res.ok) {
+      throw new Error(`Lambda Extensions API register failed: ${res.status}`);
+    }
+    const extensionId = res.headers.get("Lambda-Extension-Identifier");
+    if (!extensionId) {
+      throw new Error("Lambda Extensions API did not return an extension ID.");
+    }
+    return extensionId;
+  }
+  async nextEvent(extensionId) {
+    const res = await fetch("http://127.0.0.1:9001/2020-01-01/extension/event/next", {
+      method: "GET",
+      headers: { "Lambda-Extension-Identifier": extensionId }
+    });
+    if (!res.ok) {
+      throw new Error(`Lambda Extensions API event/next failed: ${res.status}`);
+    }
+    return await res.json();
+  }
+};
+
 // package.json
-var version5 = "0.1.9";
+var version5 = "0.1.10";
 
 // src/main.ts
+var isLambda = !!process.env.AWS_LAMBDA_RUNTIME_API;
 async function main() {
   const config = resolveConfig();
   const jitMode = config.cacheTtl === 0;
@@ -184879,16 +184969,32 @@ async function main() {
     cacheTtl: config.cacheTtl,
     ...jitMode ? { decryptor: poller.getDecryptor(), encryptedStore } : {}
   });
-  const daemon = new Daemon({
-    poller,
-    server,
-    telemetry,
-    onLog: (msg) => console.log(`[clef-agent] ${msg}`)
-  });
+  const onLog = (msg) => console.log(`[clef-agent] ${msg}`);
   telemetry?.agentStarted({ version: version5 });
-  console.log(`[clef-agent] mode: ${jitMode ? "jit" : "cached"}`);
+  const modeLabel = jitMode ? "jit" : "cached";
+  console.log(`[clef-agent] mode: ${modeLabel}`);
   console.log(`[clef-agent] token: [set]`);
-  await daemon.start();
+  if (isLambda) {
+    console.log("[clef-agent] lifecycle: lambda-extension");
+    const extension = new LambdaExtension({
+      poller,
+      server,
+      refreshTtl: config.cacheTtl,
+      telemetry,
+      onLog,
+      skipInitialFetch: true
+    });
+    await extension.start();
+  } else {
+    console.log("[clef-agent] lifecycle: daemon");
+    const daemon = new Daemon({
+      poller,
+      server,
+      telemetry,
+      onLog
+    });
+    await daemon.start();
+  }
 }
 main().catch((err) => {
   if (err instanceof ConfigError) {