@dhf-claude/grix 0.1.8 → 0.1.10

package/cli/prod-runner.test.js

@@ -0,0 +1,195 @@
+import test from "node:test";
+import assert from "node:assert/strict";
+import path from "node:path";
+import {
+  buildRuntimeArgs,
+  createManagedCommandEnv,
+  resolveRuntimeTarget,
+} from "./runtime-targets.js";
+import { runProdSwitch, stopRuntimeTarget } from "./prod-runner.js";
+
+test("createManagedCommandEnv removes connection overrides", () => {
+  const env = createManagedCommandEnv({
+    KEEP_ME: "1",
+    GRIX_CLAUDE_WS_URL: "ws://stale",
+    GRIX_CLAUDE_ENDPOINT: "ws://stale-endpoint",
+    GRIX_CLAUDE_AGENT_ID: "stale-agent",
+    GRIX_CLAUDE_API_KEY: "stale-key",
+    GRIX_CLAUDE_OUTBOUND_TEXT_CHUNK_LIMIT: "4096",
+    GRIX_CLAUDE_TEXT_CHUNK_LIMIT: "4096",
+  });
+
+  assert.equal(env.KEEP_ME, "1");
+  assert.equal("GRIX_CLAUDE_WS_URL" in env, false);
+  assert.equal("GRIX_CLAUDE_ENDPOINT" in env, false);
+  assert.equal("GRIX_CLAUDE_AGENT_ID" in env, false);
+  assert.equal("GRIX_CLAUDE_API_KEY" in env, false);
+  assert.equal("GRIX_CLAUDE_OUTBOUND_TEXT_CHUNK_LIMIT" in env, false);
+  assert.equal("GRIX_CLAUDE_TEXT_CHUNK_LIMIT" in env, false);
+});
+
+test("resolveRuntimeTarget builds the expected prod args", () => {
+  const target = resolveRuntimeTarget("prod", {
+    homeDir: "/tmp/grix-home",
+  });
+
+  assert.deepEqual(buildRuntimeArgs(target), [
+    "--data-dir",
+    path.join("/tmp/grix-home", ".claude", "grix-claude-daemon"),
+    "--ws-url",
+    "wss://clawpool.dhf.pub/v1/agent-api/ws?agent_id=2035513096339984384",
+    "--agent-id",
+    "2035513096339984384",
+    "--api-key",
+    "ak_2035513096339984384_wyIkUkt9FyEZFJGHyvWHPu7ZOXkpj0KM",
+  ]);
+});
+
+test("stopRuntimeTarget stops installed service before killing a remaining daemon pid", async () => {
+  const calls = [];
+  const target = {
+    name: "dev",
+    dataDir: "/tmp/grix-dev",
+  };
+
+  const result = await stopRuntimeTarget(target, {
+    serviceManager: {
+      async status({ dataDir }) {
+        calls.push(["status", dataDir]);
+        return { installed: true };
+      },
+      async stop({ dataDir }) {
+        calls.push(["stop", dataDir]);
+      },
+    },
+    async inspectDaemonProcessStateImpl({ dataDir }) {
+      calls.push(["inspect", dataDir]);
+      return { running: true, pid: 4321 };
+    },
+    async terminateProcessTreeImpl(pid, { platform }) {
+      calls.push(["terminate", pid, platform]);
+    },
+    async waitForProcessExitImpl(pid, { timeoutMs }) {
+      calls.push(["wait", pid, timeoutMs]);
+      return true;
+    },
+    platform: "darwin",
+  });
+
+  assert.deepEqual(calls, [
+    ["status", "/tmp/grix-dev"],
+    ["stop", "/tmp/grix-dev"],
+    ["inspect", "/tmp/grix-dev"],
+    ["terminate", 4321, "darwin"],
+    ["wait", 4321, 5000],
+  ]);
+  assert.equal(result.serviceStopped, true);
+  assert.equal(result.processTerminated, true);
+});
+
+test("runProdSwitch stops dev and installs prod service when prod is not installed", async () => {
+  const serviceCalls = [];
+  const cliCalls = [];
+  const prints = [];
+  const prodDir = path.join("/tmp/grix-home", ".claude", "grix-claude-daemon");
+  const devDir = path.join("/tmp/grix-home", ".claude", "grix-claude-daemon-dev");
+
+  await runProdSwitch({
+    GRIX_CLAUDE_AGENT_ID: "stale-agent",
+    KEEP_ME: "1",
+  }, {
+    homeDir: "/tmp/grix-home",
+    print(line) {
+      prints.push(String(line));
+    },
+    serviceManager: {
+      async status({ dataDir }) {
+        serviceCalls.push(["status", dataDir]);
+        return { installed: dataDir === devDir ? false : false };
+      },
+    },
+    async inspectDaemonProcessStateImpl({ dataDir }) {
+      serviceCalls.push(["inspect", dataDir]);
+      if (dataDir === devDir) {
+        return { running: true, pid: 2222 };
+      }
+      return { running: false, pid: 0 };
+    },
+    async terminateProcessTreeImpl(pid) {
+      serviceCalls.push(["terminate", pid]);
+    },
+    async waitForProcessExitImpl(pid) {
+      serviceCalls.push(["wait", pid]);
+      return true;
+    },
+    async runCliImpl(argv, env, deps) {
+      cliCalls.push({ argv, env, deps });
+      return 0;
+    },
+  });
+
+  assert.deepEqual(serviceCalls, [
+    ["status", devDir],
+    ["inspect", devDir],
+    ["terminate", 2222],
+    ["wait", 2222],
+    ["status", prodDir],
+    ["inspect", prodDir],
+    ["status", prodDir],
+  ]);
+  assert.equal(cliCalls.length, 1);
+  assert.equal(cliCalls[0].argv[0], "install");
+  assert.equal(cliCalls[0].argv.includes(prodDir), true);
+  assert.equal("GRIX_CLAUDE_AGENT_ID" in cliCalls[0].env, false);
+  assert.equal(cliCalls[0].env.KEEP_ME, "1");
+  assert.ok(cliCalls[0].deps.serviceManager);
+  assert.match(prints.join("\n"), /dev 已停止/u);
+  assert.match(prints.join("\n"), /安装并启动后台服务/u);
+});
+
+test("runProdSwitch starts existing prod service after cleaning old prod runtime", async () => {
+  const cliCalls = [];
+  const prodDir = path.join("/tmp/grix-home-2", ".claude", "grix-claude-daemon");
+  const devDir = path.join("/tmp/grix-home-2", ".claude", "grix-claude-daemon-dev");
+  const installStateByDir = new Map([
+    [devDir, true],
+    [prodDir, true],
+  ]);
+
+  await runProdSwitch({}, {
+    homeDir: "/tmp/grix-home-2",
+    print() {},
+    serviceManager: {
+      async status({ dataDir }) {
+        return { installed: installStateByDir.get(dataDir) === true };
+      },
+      async stop() {},
+    },
+    async inspectDaemonProcessStateImpl({ dataDir }) {
+      if (dataDir === prodDir) {
+        return { running: true, pid: 3333 };
+      }
+      return { running: false, pid: 0 };
+    },
+    async terminateProcessTreeImpl() {},
+    async waitForProcessExitImpl() {
+      return true;
+    },
+    async runCliImpl(argv) {
+      cliCalls.push(argv);
+      return 0;
+    },
+  });
+
+  assert.deepEqual(cliCalls, [[
+    "start",
+    "--data-dir",
+    prodDir,
+    "--ws-url",
+    "wss://clawpool.dhf.pub/v1/agent-api/ws?agent_id=2035513096339984384",
+    "--agent-id",
+    "2035513096339984384",
+    "--api-key",
+    "ak_2035513096339984384_wyIkUkt9FyEZFJGHyvWHPu7ZOXkpj0KM",
+  ]]);
+});

package/cli/runtime-targets.js

@@ -0,0 +1,66 @@
+import os from "node:os";
+import path from "node:path";
+
+const MANAGED_ENV_KEYS = Object.freeze([
+  "GRIX_CLAUDE_WS_URL",
+  "GRIX_CLAUDE_ENDPOINT",
+  "GRIX_CLAUDE_AGENT_ID",
+  "GRIX_CLAUDE_API_KEY",
+  "GRIX_CLAUDE_OUTBOUND_TEXT_CHUNK_LIMIT",
+  "GRIX_CLAUDE_TEXT_CHUNK_LIMIT",
+]);
+
+const RUNTIME_PROFILES = Object.freeze({
+  dev: Object.freeze({
+    dataDirName: "grix-claude-daemon-dev",
+    wsUrl: "ws://127.0.0.1:27189/v1/agent-api/ws?agent_id=2035251418226495488",
+    agentId: "2035251418226495488",
+    apiKey: "ak_2035251418226495488__Y-1s6Sy1dBsEJhuv_GLTI-n4PitFlYg",
+  }),
+  prod: Object.freeze({
+    dataDirName: "grix-claude-daemon",
+    wsUrl: "wss://clawpool.dhf.pub/v1/agent-api/ws?agent_id=2035513096339984384",
+    agentId: "2035513096339984384",
+    apiKey: "ak_2035513096339984384_wyIkUkt9FyEZFJGHyvWHPu7ZOXkpj0KM",
+  }),
+});
+
+function normalizeRuntimeName(name) {
+  return String(name ?? "").trim().toLowerCase();
+}
+
+export function createManagedCommandEnv(env = process.env) {
+  const nextEnv = { ...env };
+  for (const key of MANAGED_ENV_KEYS) {
+    delete nextEnv[key];
+  }
+  return nextEnv;
+}
+
+export function resolveRuntimeTarget(name, { homeDir = os.homedir() } = {}) {
+  const normalizedName = normalizeRuntimeName(name);
+  const profile = RUNTIME_PROFILES[normalizedName];
+  if (!profile) {
+    throw new Error(`unknown runtime target: ${name}`);
+  }
+  return {
+    name: normalizedName,
+    dataDir: path.join(homeDir, ".claude", profile.dataDirName),
+    wsUrl: profile.wsUrl,
+    agentId: profile.agentId,
+    apiKey: profile.apiKey,
+  };
+}
+
+export function buildRuntimeArgs(target) {
+  return [
+    "--data-dir",
+    target.dataDir,
+    "--ws-url",
+    target.wsUrl,
+    "--agent-id",
+    target.agentId,
+    "--api-key",
+    target.apiKey,
+  ];
+}