@dhf-claude/grix 0.1.8

package/cli/main.js

@@ -0,0 +1,369 @@
+import process from "node:process";
+import {
+  maskApiKey,
+  resolvePackageBinPath,
+  validateConfig,
+} from "./config.js";
+import { ConfigStore } from "../server/config-store.js";
+import { resolveDaemonConfigPath, resolveDaemonDataDir } from "../server/daemon/daemon-paths.js";
+import { ServiceManager } from "../server/service/service-manager.js";
+
+function usage() {
+  return `用法:
+  grix-claude [options]
+  grix-claude daemon [options]
+  grix-claude worker [options]
+  grix-claude install [options]
+  grix-claude start [options]
+  grix-claude stop [options]
+  grix-claude restart [options]
+  grix-claude status [options]
+  grix-claude uninstall [options]
+
+说明:
+  默认命令会写好 daemon 配置，并启动 grix-claude daemon。
+  daemon 是唯一对接 Grix 的常驻服务，Claude 会话由 daemon 按需拉起。
+
+  选项:
+  --ws-url <value>      Grix Agent API WebSocket 地址
+  --agent-id <value>    Agent ID
+  --api-key <value>     API Key
+  --data-dir <path>     daemon 数据目录，默认 ~/.claude/grix-claude-daemon
+  --chunk-limit <n>     单段文本长度上限，默认 1200
+  --show-claude         开发调试时把 Claude 拉到可见的 Terminal 窗口
+  --no-launch           只检查并写好配置，不启动 daemon
+  --help, -h            显示帮助
+
+第一次运行需要传完整参数。
+后续再次运行时，如果本地已经保存过配置，可以直接执行: grix-claude
+`;
+}
+
+function parseArgs(argv) {
+  const options = {
+    wsUrl: "",
+    agentId: "",
+    apiKey: "",
+    dataDir: "",
+    chunkLimit: "",
+    showClaude: false,
+    noLaunch: false,
+    help: false,
+  };
+
+  for (let index = 0; index < argv.length; index += 1) {
+    const current = argv[index];
+    if (current === "--help" || current === "-h") {
+      options.help = true;
+      continue;
+    }
+    if (current === "--no-launch") {
+      options.noLaunch = true;
+      continue;
+    }
+    if (current === "--show-claude") {
+      options.showClaude = true;
+      continue;
+    }
+    const next = argv[index + 1];
+    if (current === "--ws-url") {
+      if (!next) {
+        throw new Error("--ws-url 缺少值。");
+      }
+      options.wsUrl = next;
+      index += 1;
+      continue;
+    }
+    if (current === "--agent-id") {
+      if (!next) {
+        throw new Error("--agent-id 缺少值。");
+      }
+      options.agentId = next;
+      index += 1;
+      continue;
+    }
+    if (current === "--api-key") {
+      if (!next) {
+        throw new Error("--api-key 缺少值。");
+      }
+      options.apiKey = next;
+      index += 1;
+      continue;
+    }
+    if (current === "--data-dir") {
+      if (!next) {
+        throw new Error("--data-dir 缺少值。");
+      }
+      options.dataDir = next;
+      index += 1;
+      continue;
+    }
+    if (current === "--chunk-limit") {
+      if (!next) {
+        throw new Error("--chunk-limit 缺少值。");
+      }
+      options.chunkLimit = next;
+      index += 1;
+      continue;
+    }
+    throw new Error(`未知参数: ${current}`);
+  }
+
+  return options;
+}
+
+function print(message) {
+  process.stdout.write(`${message}\n`);
+}
+
+function printError(message) {
+  process.stderr.write(`${message}\n`);
+}
+
+function shellQuoteForDisplay(value) {
+  const text = String(value ?? "");
+  if (!text) {
+    return "''";
+  }
+  if (/^[A-Za-z0-9_./:=@-]+$/u.test(text)) {
+    return text;
+  }
+  return `'${text.replace(/'/g, `'\\''`)}'`;
+}
+
+function redactSensitiveArgs(argv) {
+  const args = Array.isArray(argv) ? argv.map((item) => String(item ?? "")) : [];
+  const redacted = [];
+  for (let index = 0; index < args.length; index += 1) {
+    const current = args[index];
+    if (current === "--api-key") {
+      redacted.push("--api-key");
+      if (index + 1 < args.length) {
+        redacted.push("******");
+        index += 1;
+      }
+      continue;
+    }
+    if (current.startsWith("--api-key=")) {
+      redacted.push("--api-key=******");
+      continue;
+    }
+    redacted.push(current);
+  }
+  return redacted;
+}
+
+function formatRunningCommand(argv) {
+  const command = ["grix-claude", ...redactSensitiveArgs(argv)];
+  return command.map((item) => shellQuoteForDisplay(item)).join(" ");
+}
+
+function formatRuntimeEntryCommand(argv) {
+  const command = [process.execPath, resolvePackageBinPath(), ...redactSensitiveArgs(argv)];
+  return command.map((item) => shellQuoteForDisplay(item)).join(" ");
+}
+
+function createServiceManager(env = process.env) {
+  return new ServiceManager({ env });
+}
+
+function buildRuntimeEnv(options, env) {
+  return {
+    ...env,
+    ...(options.dataDir ? { GRIX_CLAUDE_DAEMON_DATA_DIR: options.dataDir } : {}),
+    ...(options.showClaude ? { GRIX_CLAUDE_SHOW_CLAUDE_WINDOW: "1" } : {}),
+  };
+}
+
+function buildDaemonStatus(configStore) {
+  const config = configStore.get();
+  validateConfig(config);
+  return {
+    config,
+    configPath: configStore.filePath,
+  };
+}
+
+async function prepareDaemonConfig(options, env = process.env, { persistResolvedConfig = false } = {}) {
+  const runtimeEnv = buildRuntimeEnv(options, env);
+  const configStore = new ConfigStore(resolveDaemonConfigPath(runtimeEnv), {
+    env: runtimeEnv,
+  });
+  await configStore.load();
+  const hasCliOverrides = Boolean(
+    options.wsUrl
+    || options.agentId
+    || options.apiKey
+    || options.chunkLimit,
+  );
+  if (hasCliOverrides) {
+    await configStore.update({
+      ...(options.wsUrl ? { ws_url: options.wsUrl } : {}),
+      ...(options.agentId ? { agent_id: options.agentId } : {}),
+      ...(options.apiKey ? { api_key: options.apiKey } : {}),
+      ...(options.chunkLimit ? { outbound_text_chunk_limit: Number(options.chunkLimit) } : {}),
+    });
+  } else if (persistResolvedConfig) {
+    await configStore.update(configStore.get());
+  }
+  return {
+    runtimeEnv,
+    dataDir: resolveDaemonDataDir(runtimeEnv),
+    ...buildDaemonStatus(configStore),
+  };
+}
+
+async function runDefault(argv, env = process.env) {
+  const options = parseArgs(argv);
+  if (options.help) {
+    print(usage());
+    return 0;
+  }
+
+  const { runtimeEnv, dataDir, config, configPath } = await prepareDaemonConfig(options, env);
+
+  print(`配置已准备好。`);
+  print(`数据目录: ${dataDir}`);
+  print(`配置文件: ${configPath}`);
+  print(`Agent ID: ${config.agent_id}`);
+  print(`API Key: ${maskApiKey(config.api_key)}`);
+
+  if (options.noLaunch) {
+    print(`daemon 还没有启动。需要时直接执行 grix-claude 即可。`);
+    return 0;
+  }
+
+  print(`正在启动 daemon...`);
+  const { run } = await import("../server/daemon/main.js");
+  return run([], runtimeEnv);
+}
+
+function formatServiceStatus(status) {
+  const lines = [
+    `服务已安装: ${status.installed ? "yes" : "no"}`,
+    `服务类型: ${status.service_kind || ""}`,
+    `数据目录: ${status.data_dir || ""}`,
+    `安装状态: ${status.install_state || ""}`,
+    `daemon 状态: ${status.daemon_state || ""}`,
+  ];
+  if (status.service_id) {
+    lines.push(`服务标识: ${status.service_id}`);
+  }
+  if (status.pid) {
+    lines.push(`进程 PID: ${status.pid}`);
+  }
+  if (status.connection_state) {
+    lines.push(`连接状态: ${status.connection_state}`);
+  }
+  if (status.bridge_url) {
+    lines.push(`Bridge: ${status.bridge_url}`);
+  }
+  if (status.definition_path) {
+    lines.push(`启动项: ${status.definition_path}`);
+  }
+  return lines.join("\n");
+}
+
+async function runServiceSubcommand(name, argv, env, deps = {}) {
+  const options = parseArgs(argv);
+  if (options.help) {
+    print(usage());
+    return 0;
+  }
+  const runtimeEnv = buildRuntimeEnv(options, env);
+  const dataDir = resolveDaemonDataDir(runtimeEnv);
+  const manager = deps.serviceManager || createServiceManager(runtimeEnv);
+  const shouldPrepareConfig = ["install", "start", "restart"].includes(name);
+  const prepared = shouldPrepareConfig
+    ? await prepareDaemonConfig(options, env, { persistResolvedConfig: true })
+    : null;
+  if (name === "install") {
+    const { config, configPath } = prepared;
+    const status = await manager.install({ dataDir });
+    print(`配置文件: ${configPath}`);
+    print(`Agent ID: ${config.agent_id}`);
+    print(`API Key: ${maskApiKey(config.api_key)}`);
+    print(formatServiceStatus(status));
+    return 0;
+  }
+  if (name === "start") {
+    const { config, configPath } = prepared;
+    const status = await manager.start({ dataDir });
+    print(`配置文件: ${configPath}`);
+    print(`Agent ID: ${config.agent_id}`);
+    print(`API Key: ${maskApiKey(config.api_key)}`);
+    print(formatServiceStatus(status));
+    return 0;
+  }
+  if (name === "stop") {
+    const status = await manager.stop({ dataDir });
+    print(formatServiceStatus(status));
+    return 0;
+  }
+  if (name === "restart") {
+    const { config, configPath } = prepared;
+    const status = await manager.restart({ dataDir });
+    print(`配置文件: ${configPath}`);
+    print(`Agent ID: ${config.agent_id}`);
+    print(`API Key: ${maskApiKey(config.api_key)}`);
+    print(formatServiceStatus(status));
+    return 0;
+  }
+  if (name === "status") {
+    const status = await manager.status({ dataDir });
+    print(formatServiceStatus(status));
+    return 0;
+  }
+  if (name === "uninstall") {
+    const status = await manager.uninstall({ dataDir });
+    print(formatServiceStatus(status));
+    return 0;
+  }
+  throw new Error(`未知子命令: ${name}`);
+}
+
+async function runSubcommand(name, argv, env, deps = {}) {
+  if (name === "daemon") {
+    const { run } = await import("../server/daemon/main.js");
+    return run(argv, env);
+  }
+  if (name === "worker") {
+    const { run } = await import("../server/worker/main.js");
+    return run(argv, env);
+  }
+  if (["install", "start", "stop", "restart", "status", "uninstall"].includes(name)) {
+    return runServiceSubcommand(name, argv, env, deps);
+  }
+  throw new Error(`未知子命令: ${name}`);
+}
+
+export async function run(argv, env = process.env, deps = {}) {
+  const [first = ""] = argv;
+  if ([
+    "daemon",
+    "worker",
+    "install",
+    "start",
+    "stop",
+    "restart",
+    "status",
+    "uninstall",
+  ].includes(first)) {
+    return runSubcommand(first, argv.slice(1), env, deps);
+  }
+  return runDefault(argv, env);
+}
+
+export async function main(argv = process.argv.slice(2), env = process.env) {
+  try {
+    print(`运行命令: ${formatRunningCommand(argv)}`);
+    print(`实际入口: ${formatRuntimeEntryCommand(argv)}`);
+    const exitCode = await run(argv, env);
+    process.exitCode = exitCode;
+  } catch (error) {
+    printError(error instanceof Error ? error.message : String(error));
+    printError("");
+    printError(usage());
+    process.exitCode = 1;
+  }
+}