@downcity/city 1.1.39 → 1.1.47

package/src/terminal/core/session.ts

@@ -0,0 +1,450 @@
+/**
+ * Session 与 server 配置持久化模块。
+ *
+ * 关键说明（中文）
+ * - server 是一等资源，必须显式配置后 CLI 才进入身份菜单
+ * - 不再注入默认 server；没有 server 时必须先添加
+ * - user session 仍然按 base_url 隔离保存
+ * - admin_secret_key 直接属于 server 配置，不再作为独立 session 维护
+ */
+
+import fs from "node:fs";
+import os from "node:os";
+import path from "node:path";
+import crypto from "node:crypto";
+import { normalizeBaseUrl } from "./env.js";
+
+const DIR = path.join(os.homedir(), ".downcity");
+const CONFIG_FILE = path.join(DIR, "config.json");
+const LEGACY_CONFIG_FILE = path.join(DIR, "product-client-config.json");
+const LEGACY_USER_FILE = path.join(DIR, "product-client-user.json");
+const LEGACY_ADMIN_FILE = path.join(DIR, "product-client-admin.json");
+
+// ============================================================
+// 类型
+// ============================================================
+
+export interface AdminSession {
+  /** 当前 server 的 server URL */
+  base_url: string;
+  /** 当前 server 的 admin secret key */
+  admin_token: string;
+}
+
+export interface UserSession {
+  /** 当前登录目标的 server server URL */
+  base_url: string;
+  /** 用户邮箱 */
+  email: string;
+  /** 用户 ID */
+  user_id: string;
+  /** 当前 product ID */
+  product_id: string;
+  /** user token */
+  user_token: string;
+}
+
+export interface ServerProfile {
+  /** 展示名称 */
+  name: string;
+  /** InfraRuntime 服务地址 */
+  base_url: string;
+  /** 该 server 对应的 admin secret key */
+  admin_secret_key: string;
+}
+
+export interface ClientConfig {
+  /** 当前激活的 server URL */
+  active_server_url?: string;
+  /** 已保存的 server 列表 */
+  servers: ServerProfile[];
+  /** 当前选择的模型 ID */
+  model: string;
+  /** 上次使用的身份 */
+  last_identity?: "admin" | "user";
+}
+
+// ============================================================
+// Config 读写
+// ============================================================
+
+/**
+ * 从磁盘读取 config，并把旧结构迁移到新的 server 模型。
+ */
+export function readConfig(): ClientConfig {
+  const raw = readJSON<Record<string, unknown>>(CONFIG_FILE)
+    ?? readJSON<Record<string, unknown>>(LEGACY_CONFIG_FILE)
+    ?? {};
+  const servers = readServersFromConfig(raw);
+  const active_server_url = readActiveServerURL(raw, servers);
+
+  return {
+    active_server_url,
+    servers,
+    model: typeof raw.model === "string" ? raw.model : "",
+    last_identity: raw.last_identity === "admin" || raw.last_identity === "user"
+      ? raw.last_identity
+      : undefined,
+  };
+}
+
+/**
+ * 写入 config 到磁盘。
+ */
+export function writeConfig(config: ClientConfig): void {
+  const normalizedServers = normalizeServers(config.servers);
+  const active = normalizedServers.find((server) => server.base_url === config.active_server_url)
+    ? config.active_server_url
+    : normalizedServers[0]?.base_url;
+
+  writeJSON(CONFIG_FILE, {
+    active_server_url: active,
+    servers: normalizedServers,
+    model: config.model ?? "",
+    last_identity: config.last_identity,
+  });
+}
+
+// ============================================================
+// Server 管理
+// ============================================================
+
+/**
+ * 读取当前激活的 server。
+ */
+export function readActiveServer(): ServerProfile | undefined {
+  const config = readConfig();
+  return config.servers.find((server) => server.base_url === config.active_server_url);
+}
+
+/**
+ * 设置当前激活的 server。
+ */
+export function setActiveServer(baseUrl: string): void {
+  const config = readConfig();
+  const normalizedBaseUrl = normalizeBaseUrl(baseUrl);
+  if (!config.servers.find((server) => server.base_url === normalizedBaseUrl)) {
+    throw new Error(`Unknown server: ${normalizedBaseUrl}`);
+  }
+  writeConfig({
+    ...config,
+    active_server_url: normalizedBaseUrl,
+  });
+}
+
+/**
+ * 添加 server，并设为当前激活 server。
+ */
+export function addServer(input: {
+  base_url: string;
+  admin_secret_key: string;
+  name?: string;
+}): ServerProfile {
+  const config = readConfig();
+  const normalized = normalizeServer(input);
+  const existingIndex = config.servers.findIndex((server) => server.base_url === normalized.base_url);
+
+  if (existingIndex >= 0) {
+    config.servers[existingIndex] = normalized;
+  } else {
+    config.servers.push(normalized);
+  }
+
+  writeConfig({
+    ...config,
+    servers: config.servers,
+    active_server_url: normalized.base_url,
+  });
+
+  return normalized;
+}
+
+/**
+ * 更新已存在的 server。
+ *
+ * 关键说明（中文）
+ * - 如果 base_url 变化，会清除旧 URL 下的 user session
+ * - active server 会自动切换到新 URL
+ */
+export function updateServer(
+  currentBaseUrl: string,
+  input: {
+    base_url: string;
+    admin_secret_key: string;
+    name?: string;
+  },
+): ServerProfile {
+  const config = readConfig();
+  const normalizedCurrent = normalizeBaseUrl(currentBaseUrl);
+  const index = config.servers.findIndex((server) => server.base_url === normalizedCurrent);
+  if (index < 0) {
+    throw new Error(`Unknown server: ${normalizedCurrent}`);
+  }
+
+  const normalizedNext = normalizeServer(input);
+  const duplicateIndex = config.servers.findIndex((server) => server.base_url === normalizedNext.base_url);
+  if (duplicateIndex >= 0 && duplicateIndex !== index) {
+    throw new Error(`Server already exists: ${normalizedNext.base_url}`);
+  }
+
+  config.servers[index] = normalizedNext;
+  const activeServerUrl = config.active_server_url === normalizedCurrent
+    ? normalizedNext.base_url
+    : config.active_server_url;
+
+  writeConfig({
+    ...config,
+    servers: config.servers,
+    active_server_url: activeServerUrl,
+  });
+
+  if (normalizedCurrent !== normalizedNext.base_url) {
+    clearServerSessions(normalizedCurrent);
+  }
+
+  return normalizedNext;
+}
+
+/**
+ * 删除 server，同时清除该 URL 下的所有 session。
+ */
+export function removeServer(baseUrl: string): void {
+  const config = readConfig();
+  const normalizedBaseUrl = normalizeBaseUrl(baseUrl);
+  const nextServers = config.servers.filter((server) => server.base_url !== normalizedBaseUrl);
+  const nextActive = config.active_server_url === normalizedBaseUrl
+    ? nextServers[0]?.base_url
+    : config.active_server_url;
+
+  writeConfig({
+    ...config,
+    servers: nextServers,
+    active_server_url: nextActive,
+  });
+  clearServerSessions(normalizedBaseUrl);
+}
+
+/**
+ * 根据 URL 读取 server。
+ */
+export function readServer(baseUrl: string): ServerProfile | undefined {
+  const normalizedBaseUrl = normalizeBaseUrl(baseUrl);
+  return readConfig().servers.find((server) => server.base_url === normalizedBaseUrl);
+}
+
+// ============================================================
+// User Session（按 URL 隔离）
+// ============================================================
+
+function sessionDir(url: string): string {
+  return path.join(DIR, "servers", urlHash(url));
+}
+
+function userFile(url: string): string {
+  return path.join(sessionDir(url), "user.json");
+}
+
+export function readUserSession(url: string): UserSession | undefined {
+  return readJSON<UserSession>(userFile(url)) ?? readLegacyUserSession(url);
+}
+
+export function writeUserSession(session: UserSession): void {
+  writeJSON(userFile(session.base_url), session);
+}
+
+export function clearUserSession(url: string): void {
+  try {
+    fs.unlinkSync(userFile(url));
+  } catch {
+    // ignore
+  }
+
+  const legacy = readLegacyUserSession(url);
+  if (legacy) {
+    try {
+      fs.unlinkSync(LEGACY_USER_FILE);
+    } catch {
+      // ignore
+    }
+  }
+}
+
+/**
+ * 清除某个 server URL 下的所有 session。
+ */
+export function clearServerSessions(url: string): void {
+  clearUserSession(url);
+  clearLegacyAdminSession(url);
+}
+
+// ============================================================
+// 内部工具
+// ============================================================
+
+function readServersFromConfig(raw: Record<string, unknown>): ServerProfile[] {
+  const input = Array.isArray(raw.servers) ? raw.servers : [];
+  const servers: ServerProfile[] = [];
+
+  for (const item of input) {
+    if (!item || typeof item !== "object") continue;
+    const record = item as Record<string, unknown>;
+    const rawBaseUrl = typeof record.base_url === "string"
+      ? record.base_url
+      : typeof record.url === "string"
+        ? record.url
+        : "";
+
+    if (!rawBaseUrl.trim()) continue;
+
+    const normalizedBaseUrl = normalizeBaseUrl(rawBaseUrl);
+    const adminSecretKey = typeof record.admin_secret_key === "string"
+      ? record.admin_secret_key.trim()
+      : readLegacyAdminToken(normalizedBaseUrl);
+    const name = typeof record.name === "string" && record.name.trim()
+      ? record.name.trim()
+      : deriveServerName(normalizedBaseUrl);
+
+    const existing = servers.find((server) => server.base_url === normalizedBaseUrl);
+    if (existing) {
+      existing.name = name;
+      existing.admin_secret_key = adminSecretKey;
+      continue;
+    }
+
+    servers.push({
+      name,
+      base_url: normalizedBaseUrl,
+      admin_secret_key: adminSecretKey,
+    });
+  }
+
+  if (servers.length === 0 && typeof raw.base_url === "string" && raw.base_url.trim()) {
+    const normalizedBaseUrl = normalizeBaseUrl(raw.base_url);
+    servers.push({
+      name: deriveServerName(normalizedBaseUrl),
+      base_url: normalizedBaseUrl,
+      admin_secret_key: readLegacyAdminToken(normalizedBaseUrl),
+    });
+  }
+
+  return servers;
+}
+
+function readActiveServerURL(raw: Record<string, unknown>, servers: ServerProfile[]): string | undefined {
+  const rawActive = typeof raw.active_server_url === "string"
+    ? raw.active_server_url
+    : typeof raw.base_url === "string"
+      ? raw.base_url
+      : "";
+
+  if (rawActive.trim()) {
+    const normalized = normalizeBaseUrl(rawActive);
+    if (servers.find((server) => server.base_url === normalized)) {
+      return normalized;
+    }
+  }
+
+  return servers[0]?.base_url;
+}
+
+function normalizeServers(servers: ServerProfile[]): ServerProfile[] {
+  const normalized: ServerProfile[] = [];
+
+  for (const server of servers) {
+    const item = normalizeServer(server);
+    if (normalized.find((candidate) => candidate.base_url === item.base_url)) {
+      continue;
+    }
+    normalized.push(item);
+  }
+
+  return normalized;
+}
+
+function normalizeServer(input: {
+  base_url: string;
+  admin_secret_key: string;
+  name?: string;
+}): ServerProfile {
+  const normalizedBaseUrl = normalizeBaseUrl(input.base_url);
+  const normalizedAdminSecretKey = String(input.admin_secret_key ?? "").trim();
+  if (!normalizedAdminSecretKey) {
+    throw new Error("admin_secret_key cannot be empty");
+  }
+
+  const normalizedName = String(input.name ?? "").trim() || deriveServerName(normalizedBaseUrl);
+
+  return {
+    name: normalizedName,
+    base_url: normalizedBaseUrl,
+    admin_secret_key: normalizedAdminSecretKey,
+  };
+}
+
+function deriveServerName(baseUrl: string): string {
+  try {
+    const url = new URL(baseUrl);
+    return url.hostname || baseUrl;
+  } catch {
+    return baseUrl;
+  }
+}
+
+function readLegacyAdminToken(url: string): string {
+  const payload = readJSON<{ admin_token?: string }>(legacyAdminFile(url)) ?? readLegacyGlobalAdminSession(url);
+  return String(payload?.admin_token ?? "").trim();
+}
+
+function clearLegacyAdminSession(url: string): void {
+  try {
+    fs.unlinkSync(legacyAdminFile(url));
+  } catch {
+    // ignore
+  }
+
+  const legacy = readLegacyGlobalAdminSession(url);
+  if (legacy) {
+    try {
+      fs.unlinkSync(LEGACY_ADMIN_FILE);
+    } catch {
+      // ignore
+    }
+  }
+}
+
+function legacyAdminFile(url: string): string {
+  return path.join(sessionDir(url), "admin.json");
+}
+
+function readLegacyUserSession(url: string): UserSession | undefined {
+  const payload = readJSON<UserSession>(LEGACY_USER_FILE);
+  if (!payload?.base_url) {
+    return undefined;
+  }
+  return normalizeBaseUrl(payload.base_url) === normalizeBaseUrl(url) ? payload : undefined;
+}
+
+function readLegacyGlobalAdminSession(url: string): { admin_token?: string } | undefined {
+  const payload = readJSON<{ base_url?: string; admin_token?: string }>(LEGACY_ADMIN_FILE);
+  if (!payload?.base_url) {
+    return undefined;
+  }
+  return normalizeBaseUrl(payload.base_url) === normalizeBaseUrl(url) ? payload : undefined;
+}
+
+function urlHash(url: string): string {
+  return crypto.createHash("sha256").update(url).digest("hex").slice(0, 16);
+}
+
+function readJSON<T>(filepath: string): T | undefined {
+  try {
+    return JSON.parse(fs.readFileSync(filepath, "utf8")) as T;
+  } catch {
+    return undefined;
+  }
+}
+
+function writeJSON(filepath: string, data: unknown): void {
+  fs.mkdirSync(path.dirname(filepath), { recursive: true });
+  fs.writeFileSync(filepath, JSON.stringify(data, null, 2));
+}

package/src/terminal/core/stripe.ts

@@ -0,0 +1,37 @@
+/**
+ * Stripe 终端辅助模块。
+ *
+ * 关键说明（中文）
+ * - manager 侧不直接管理 Stripe 资源
+ * - 这里只负责基于当前 Infra server 推导可复制的固定回调地址
+ * - 这样用户在 Stripe Dashboard 里不需要手动拼 webhook URL
+ */
+
+import { normalizeBaseUrl } from "./env.js";
+
+/**
+ * Stripe 常用 endpoint 集合。
+ */
+export interface StripeEndpointSet {
+  /** 当前 InfraRuntime 对外地址 */
+  base_url: string;
+  /** Stripe webhook endpoint */
+  webhook_url: string;
+  /** 默认支付成功页 */
+  success_url: string;
+  /** 默认支付取消页 */
+  cancel_url: string;
+}
+
+/**
+ * 基于当前 InfraRuntime 地址构建 Stripe 相关 URL。
+ */
+export function buildStripeEndpoints(baseUrl: string): StripeEndpointSet {
+  const normalizedBaseUrl = normalizeBaseUrl(baseUrl);
+  return {
+    base_url: normalizedBaseUrl,
+    webhook_url: `${normalizedBaseUrl}/v1/payment.stripe/webhook`,
+    success_url: `${normalizedBaseUrl}/v1/payment.stripe/redirect/success`,
+    cancel_url: `${normalizedBaseUrl}/v1/payment.stripe/redirect/cancel`,
+  };
+}

package/src/terminal/core/ui.ts

@@ -0,0 +1,84 @@
+/**
+ * UI 工具模块。
+ *
+ * 提供 CLI 交互所需的输入/输出封装。
+ * 模型选择接受通用的 { id, name, hint } 数组，不依赖 server model 类型。
+ */
+
+import { password, select, text, isCancel, intro, log } from "@clack/prompts";
+
+export { intro, log, isCancel };
+
+// ============================================================
+// 显示函数
+// ============================================================
+
+export function show(text: string): void { log.info(text); }
+export function showError(text: string): void { log.error(text); }
+export function showSuccess(text: string): void { log.success(text); }
+
+// ============================================================
+// 交互 prompts
+// ============================================================
+
+/** 主命令菜单 */
+export async function askUserCommand(): Promise<string | undefined> {
+  const s = await select({ message: "Command", options: [
+    { label: "Chat (agent)", value: "chat", hint: "Continuous conversation with pi-agent" },
+    { label: "Models", value: "models", hint: "List and select model" },
+    { label: "Balance", value: "balance", hint: "View current balance" },
+    { label: "History", value: "history", hint: "View balance ledger" },
+    { label: "Topups", value: "topups", hint: "View recharge orders" },
+    { label: "Recharge (Stripe)", value: "recharge", hint: "Create Checkout and pay in browser" },
+    { label: "Create topup", value: "topup_create", hint: "Create a pending topup order only" },
+    { label: "Redeem code", value: "redeem_code", hint: "Redeem a one-time credit code" },
+    { label: "My profile", value: "me" },
+    { label: "List services", value: "services" },
+    { label: "Call service", value: "service" },
+    { label: "Switch to Admin", value: "switch" },
+    { label: "Logout", value: "logout" },
+    { label: "Quit", value: "quit" },
+  ]});
+  if (!s || isCancel(s)) return undefined;
+  return s as string;
+}
+
+/** 文本输入 */
+export async function askText(label: string): Promise<string | undefined> {
+  const v = await text({ message: label });
+  return (!v || isCancel(v)) ? undefined : v as string;
+}
+
+/** 密码输入 */
+export async function askSecret(label: string): Promise<string | undefined> {
+  const v = await password({ message: label });
+  return (!v || isCancel(v)) ? undefined : v as string;
+}
+
+// ============================================================
+// 模型选择（通用，不依赖 server 类型）
+// ============================================================
+
+/** 模型选项 */
+export interface ModelOption {
+  /** 模型 ID */
+  id: string;
+  /** 模型展示名 */
+  name: string;
+  /** 提示信息 */
+  hint: string;
+}
+
+/** 列出并选择模型，返回模型 id 或 undefined */
+export async function askModel(
+  models: ModelOption[],
+  currentModel: string,
+): Promise<string | undefined> {
+  const options = models.map((m) => ({
+    label: m.id === currentModel ? `★ ${m.name}` : `   ${m.name}`,
+    value: m.id,
+    hint: m.hint,
+  }));
+  const s = await select({ message: "Select model (★ current)", options });
+  return (!s || isCancel(s)) ? undefined : String(s);
+}