impel-cli 0.16.4 → 0.17.0

package/src/commands/sessions.js

@@ -0,0 +1,114 @@
+import { spawn } from "node:child_process";
+
+import { parseFlags } from "../args.js";
+import { collectSessionHook, flushCollectedSession, readHookInput, sessionOutboxStatus } from "../sessionCollector.js";
+import { loadConfig, redactSecretText } from "../config.js";
+import { impelCliInvocation } from "../selfInvocation.js";
+
+const SPEC = {
+  provider: { type: "string" },
+  surface: { type: "string" },
+  tenant: { type: "string" },
+  session: { type: "string" },
+  "impel-managed-session-hook-v1": { type: "boolean" },
+};
+
+function startDetachedFlush({ provider, tenant, session }) {
+  const invocation = impelCliInvocation([
+    "sessions",
+    "flush",
+    "--provider",
+    provider,
+    "--tenant",
+    tenant,
+    "--session",
+    session,
+    "--impel-managed-session-hook-v1",
+  ]);
+  const child = spawn(invocation.command, invocation.args, {
+    detached: true,
+    stdio: "ignore",
+    env: { ...process.env, IMPEL_SESSIONS_FLUSH_CHILD: "1" },
+    windowsHide: true,
+  });
+  child.once("error", () => {});
+  child.unref();
+}
+
+export async function cmdSessions(argv) {
+  const [action, ...rest] = argv;
+  if (action === "hook") {
+    try {
+      const { flags, positionals } = parseFlags(rest, SPEC);
+      if (positionals.length > 0) throw new Error("unexpected positional arguments");
+      const input = await readHookInput();
+      let config = null;
+      try {
+        config = loadConfig();
+      } catch {
+        // Capture remains available even while the user's config needs repair.
+      }
+      await collectSessionHook({
+        provider: flags.provider,
+        surface: flags.surface,
+        tenantId: flags.tenant,
+        input,
+        config,
+        flush: false,
+      });
+      if (config && (config.tenantId === flags.tenant || process.env.IMPEL_SESSIONS_DEV_ORG_ID)) {
+        startDetachedFlush({
+          provider: flags.provider,
+          tenant: flags.tenant,
+          session: String(input.session_id || ""),
+        });
+      }
+    } catch (error) {
+      // Session persistence is observational. A collector outage must never
+      // block a model turn or change a provider hook's decision semantics.
+      if (process.env.IMPEL_SESSIONS_DEBUG === "1") {
+        console.error(`impel sessions hook: ${redactSecretText(error?.message || error)}`);
+      }
+    }
+    return;
+  }
+  const { flags, positionals } = parseFlags(rest, SPEC);
+  if (positionals.length > 0) {
+    console.error("impel sessions: unexpected positional arguments");
+    process.exitCode = 1;
+    return;
+  }
+  const config = loadConfig();
+  if (action === "flush") {
+    if (!flags.provider || !flags.session || !flags.tenant) return;
+    if (!["claude_code", "codex"].includes(flags.provider)) return;
+    if (!config || flags.tenant !== config.tenantId && !process.env.IMPEL_SESSIONS_DEV_ORG_ID) return;
+    const deadline = Date.now() + 90_000;
+    while (Date.now() < deadline) {
+      const result = await flushCollectedSession({
+        tenantId: flags.tenant,
+        provider: flags.provider,
+        sessionKey: flags.session,
+        config,
+      });
+      if (result.pending === 0) break;
+      await new Promise((resolve) => setTimeout(resolve, result.error || result.busy ? 1000 : 250));
+    }
+    return;
+  }
+  if (action === "status") {
+    if (!flags.provider || !flags.session || !flags.tenant) {
+      console.error("impel sessions status requires --provider, --session, and --tenant");
+      process.exitCode = 1;
+      return;
+    }
+    console.log(JSON.stringify(sessionOutboxStatus({
+      tenantId: flags.tenant,
+      provider: flags.provider,
+      sessionKey: flags.session,
+    }), null, 2));
+    return;
+  }
+  console.error("impel sessions: use `hook` (managed internally) or `status`");
+  process.exitCode = 1;
+}