server/claude-accounts.ts

Rama main · réplica de lectura del repositorio original.

// Named accounts reuse provider instances. Claude owns login and credentials;
// removing an OpenMausBot entry never removes its directory or signs it out.
import { randomUUID } from "node:crypto";
import { realpathSync } from "node:fs";
import { join } from "node:path";
import { z } from "zod";
import { DATA_DIR, persistableInstanceConfigs, type AppConfig } from "./config.ts";
import type { InstanceConfig, InstanceConfigMap } from "./contracts.ts";
import { addEndpointSchema } from "./openai-compat-endpoints.ts";
import { CLAUDE_ACCOUNT_ENV_KEYS, resolveClaudeConfigDir } from "./drivers/claude.ts";
import { resolveCli } from "./procs.ts";

const displayName = z.string().trim().min(1).max(80).refine((value) => !/\p{Cc}/u.test(value), "Name cannot contain control characters");
const configDir = z.string().trim().max(4096).refine((value) => !/\p{Cc}/u.test(value), "Directory cannot contain control characters");
export const createClaudeAccountSchema = z.object({ displayName, configDir: configDir.optional() }).strict();
export const instanceSettingsSchema = z.object({
  cli: z.string().max(4096).refine((value) => !/\p{Cc}/u.test(value), "CLI cannot contain control characters").optional(),
  displayName: displayName.optional(),
  configDir: configDir.optional(),
  tools: z.boolean().optional(),
  hidden: z.boolean().optional(),
  /** null goes back to the server's default wait. */
  replyWaitMinutes: z.number().int().min(1).max(60).nullable().optional(),
  /** An added OpenAI-compatible server's model IDs; [] goes back to its /models list. */
  models: addEndpointSchema.shape.models,
}).strict().refine((value) => Object.keys(value).length > 0, "No settings supplied");

function rawConfig(entry: InstanceConfig): Record<string, unknown> {
  return entry.config && typeof entry.config === "object" && !Array.isArray(entry.config)
    ? entry.config as Record<string, unknown> : {};
}

export function accountDirectory(entry: InstanceConfig): string {
  const raw = rawConfig(entry).configDir;
  try {
    if (raw !== undefined && typeof raw !== "string") throw new Error("Invalid saved directory");
    return resolveClaudeConfigDir(raw, { ...process.env, ...entry.environment });
  } catch {
    throw Object.assign(new Error("Use an absolute Claude configuration directory or a path beginning with ~/."), { status: 400 });
  }
}

export function configuredAccountDirectory(entry: InstanceConfig): string {
  const explicit = rawConfig(entry).configDir;
  if (explicit !== undefined && typeof explicit !== "string") return accountDirectory(entry);
  return (typeof explicit === "string" && explicit.trim()) || entry.environment?.CLAUDE_CONFIG_DIR || process.env.CLAUDE_CONFIG_DIR
    ? accountDirectory(entry) : "";
}

function directoryIdentity(path: string): string {
  let canonical = path;
  try { canonical = realpathSync.native(path); } catch {}
  return process.platform === "win32" ? canonical.toLowerCase() : canonical;
}

export function assertSeparateClaudeAccount(map: InstanceConfigMap, id: string, entry: InstanceConfig): void {
  const directory = directoryIdentity(accountDirectory(entry));
  if (Object.entries(map).some(([otherId, other]) => {
    if (otherId === id || other.driver !== "claudeAgent") return false;
    // A malformed saved account must not block adding or repairing another one.
    try { return directoryIdentity(accountDirectory(other)) === directory; } catch { return false; }
  })) {
    throw Object.assign(new Error("That Claude directory is already configured. Select the existing account or use a separate directory."), { status: 409 });
  }
}

export function newClaudeAccount(cfg: AppConfig, body: z.infer<typeof createClaudeAccountSchema>) {
  const instances = persistableInstanceConfigs(cfg);
  const instanceId = `claude-${randomUUID()}`;
  // Reuse the installed binary, not the original account's environment/auth.
  const existing = Object.values(instances).find((entry) => entry.driver === "claudeAgent");
  const cli = existing ? rawConfig(existing).cli : undefined;
  const entry: InstanceConfig = {
    driver: "claudeAgent",
    displayName: body.displayName,
    config: {
      ...(typeof cli === "string" && cli ? { cli } : {}),
      configDir: body.configDir || join(DATA_DIR, "providers", instanceId),
    },
  };
  entry.config = { ...entry.config as Record<string, unknown>, configDir: accountDirectory(entry) };
  assertSeparateClaudeAccount(instances, instanceId, entry);
  instances[instanceId] = entry;
  return { instanceId, instances };
}

/** Copyable commands quote everything as data, wrapper arguments included; the subshell restores the caller's environment. */
export function claudeSignInCommand(cli: string, directory: string, platform: NodeJS.Platform = process.platform): string {
  const resolved = resolveCli(cli, ["auth", "login"]);
  if (platform === "win32") {
    const quote = (value: string) => `'${value.replaceAll("'", "''")}'`;
    const keys = [...CLAUDE_ACCOUNT_ENV_KEYS, "CLAUDE_CONFIG_DIR"];
    return `& { $ombKeys = ${keys.map(quote).join(",")}; $ombSaved = @{}; foreach ($k in $ombKeys) { $ombSaved[$k] = [Environment]::GetEnvironmentVariable($k, 'Process'); [Environment]::SetEnvironmentVariable($k, $null, 'Process') }; try { ${directory ? `$env:CLAUDE_CONFIG_DIR = ${quote(directory)}; ` : ""}& ${[resolved.command, ...resolved.args].map(quote).join(" ")} } finally { foreach ($k in $ombKeys) { [Environment]::SetEnvironmentVariable($k, $ombSaved[$k], 'Process') } } }`;
  }
  const quote = (value: string) => `'${value.replaceAll("'", "'\"'\"'")}'`;
  return `(unset ${[...CLAUDE_ACCOUNT_ENV_KEYS, "CLAUDE_CONFIG_DIR"].join(" ")}; ${directory ? `export CLAUDE_CONFIG_DIR=${quote(directory)}; ` : ""}${[resolved.command, ...resolved.args].map(quote).join(" ")})`;
}

export function claudeAccountInfo(instanceId: string, entry: InstanceConfig, cli: string) {
  const directory = configuredAccountDirectory(entry);
  return {
    configDir: directory,
    signInCommand: claudeSignInCommand(cli, directory),
    signInShell: process.platform === "win32" ? "powershell" as const : "sh" as const,
    isDefault: instanceId === "claude",
  };
}