@tokenbuddy/tokenbuddy 1.0.36 → 1.0.38

package/src/model-index.ts

@@ -1,278 +0,0 @@
-import { createModuleLogger } from "@tokenbuddy/logging";
-import { isBuyerVisibleRegistrySeller, type RegistrySeller } from "./seller-catalog.js";
-
-const logger = createModuleLogger("tb-proxyd:model-index");
-
-/**
- * Normalize a model id for index lookup. Trims whitespace and lowercases
- * the value so that `claude-sonnet-4-5` and `Claude-Sonnet-4-5 ` resolve to
- * the same entry. v1.2 model matching is case-insensitive.
- */
-function normalizeModelId(modelId: string): string {
-  return modelId.trim().toLowerCase();
-}
-
-/**
- * Result of resolving a model id against the index.
- *
- * - `empty` is returned when the index has been built but no seller serves
- *   the requested model. The caller should treat this as "no compatible
- *   seller" and surface the empty result to the user.
- * - `sellers` is a snapshot array of the matching registry entries. Order
- *   follows the registry's `defaultSeller` preference then declaration order.
- */
-/**
- * `ModelIndex.resolve()` 的返回结果。
- * - `empty` 等价于 `matched = false && sellers = []`，路由层应当视为"无可用 seller"。
- * - `sellers` 是按 `defaultSeller` 优先 + 声明顺序排序后的快照数组。
- */
-export interface ModelIndexResolution {
-  /** 原始请求的模型 ID（未归一化） */
-  modelId: string;
-  /** 索引里是否至少有一个 seller 命中（已归一化匹配） */
-  matched: boolean;
-  /** 命中的 seller 列表（已按 default + 声明顺序排序的快照） */
-  sellers: RegistrySeller[];
-  /** 当前索引里 `models` 字段缺失的 seller 数（用于诊断和告警） */
-  missingModelsFlag: number;
-}
-
-interface ModelIndexInternals {
-  byModel: Map<string, RegistrySeller[]>;
-  bySeller: Map<string, RegistrySeller>;
-  registryVersion: number;
-  registryFetchedAt: number;
-  defaultSellerId?: string;
-  missingModelsCount: number;
-  totalSellers: number;
-}
-
-/**
- * In-memory index mapping `modelId -> sellers[]` from a fetched registry
- * snapshot. The index is rebuilt atomically on every `rebuild()` call so
- * callers always observe a consistent snapshot (Node is single-threaded, but
- * the rebuild path copies data before swapping the internal maps).
- */
-export class ModelIndex {
-  private internals: ModelIndexInternals = emptyInternals();
-
-  /**
-   * Atomically replace the index contents with a new registry snapshot.
-   * Sellers that do not declare `models` are kept in the `bySeller` reverse
-   * map so they can still be addressed by id, but they are excluded from
-   * `sellersFor()` results. The count of such sellers is exposed via
-   * `missingModelsCount` and via the `models_refresh.seller_missing_models`
-   * log event so operators can fix upstream registry payloads.
-   */
-  rebuild(sellers: RegistrySeller[], opts: { registryVersion?: number; defaultSellerId?: string } = {}): void {
-    const byModel = new Map<string, RegistrySeller[]>();
-    const bySeller = new Map<string, RegistrySeller>();
-    let missingModels = 0;
-
-    for (const seller of sellers) {
-      if (!seller || !seller.id) {
-        continue;
-      }
-      if (!isBuyerVisibleRegistrySeller(seller)) {
-        continue;
-      }
-      // v1.2 registry schema 用 "anthropic_messages" 作为协议名（OpenAI / ClawTip
-      // / 外部 client 都用这个），但 buyer 内部 `endpointProtocol` 对 /v1/messages
-      // 返 "messages"（更短、更易读）。在 modelIndex 重建时做 alias 映射，让两边
-      // 都能匹配到同一 seller。alias 只在内存里生效，不回写 registry。
-      if (
-        Array.isArray(seller.supportedProtocols) &&
-        seller.supportedProtocols.includes("anthropic_messages") &&
-        !seller.supportedProtocols.includes("messages")
-      ) {
-        seller.supportedProtocols = [...seller.supportedProtocols, "messages"];
-      }
-      bySeller.set(seller.id, seller);
-      if (!Array.isArray(seller.models) || seller.models.length === 0) {
-        missingModels += 1;
-        logger.warn("models_refresh.seller_missing_models", "registry seller entry missing models array; excluded from model index", {
-          sellerId: seller.id,
-          sellerUrl: seller.url
-        });
-        continue;
-      }
-      for (const raw of seller.models) {
-        if (typeof raw !== "string" || raw.trim() === "") {
-          continue;
-        }
-        const key = normalizeModelId(raw);
-        const bucket = byModel.get(key);
-        if (bucket) {
-          bucket.push(seller);
-        } else {
-          byModel.set(key, [seller]);
-        }
-      }
-    }
-
-    this.internals = {
-      byModel,
-      bySeller,
-      registryVersion: opts.registryVersion ?? 0,
-      registryFetchedAt: Date.now(),
-      defaultSellerId: opts.defaultSellerId,
-      missingModelsCount: missingModels,
-      totalSellers: bySeller.size
-    };
-
-    logger.info("models_refresh.rebuilt", "model index rebuilt from registry snapshot", {
-      registryVersion: this.internals.registryVersion,
-      sellerCount: this.internals.totalSellers,
-      modelCount: byModel.size,
-      missingModels: missingModels,
-      defaultSellerId: opts.defaultSellerId ?? null
-    });
-  }
-
-  /**
-   * Resolve sellers for a given model id, optionally filtered by protocol
-   * and payment method. Returns a snapshot; mutating the result does not
-   * affect the index.
-   */
-  sellersFor(modelId: string, filter?: { protocol?: string; paymentMethod?: string }): RegistrySeller[] {
-    const key = normalizeModelId(modelId);
-    const bucket = this.internals.byModel.get(key);
-    if (!bucket) {
-      return [];
-    }
-    if (!filter) {
-      return bucket.slice();
-    }
-    return bucket.filter((seller) => matchesFilter(seller, filter));
-  }
-
-  /**
-   * Resolve a single model id with diagnostic metadata. Used by route-failover
-   * to log structured "no compatible seller" events without losing the
-   * missing-models count.
-   */
-  resolve(modelId: string, filter?: { protocol?: string; paymentMethod?: string }): ModelIndexResolution {
-    return {
-      modelId,
-      matched: this.internals.byModel.has(normalizeModelId(modelId)),
-      sellers: this.sellersFor(modelId, filter),
-      missingModelsFlag: this.internals.missingModelsCount
-    };
-  }
-
-  /**
-   * Returns the registry seller entry by id, or `undefined` if the seller is
-   * not present in the latest snapshot. Used by the route-failover to look up
-   * a seller for token bookkeeping even when the requested model is unknown.
-   */
-  getSeller(sellerId: string): RegistrySeller | undefined {
-    return this.internals.bySeller.get(sellerId);
-  }
-
-  /**
-   * List every model id known to the index. Intended for diagnostics
-   * (`tb doctor`) and CLI completion; not used on the hot path.
-   */
-  knownModelIds(): string[] {
-    return Array.from(this.internals.byModel.keys());
-  }
-
-  /**
-   * Drop stale seller entries from the reverse map when the registry has not
-   * surfaced them within `staleAfterMs`. Model-keyed buckets are derived from
-   * the same `bySeller` set so dropping sellers implicitly drops their
-   * model associations. Returns the number of removed entries.
-   */
-  prune(lastSeenAt: Map<string, number>, staleAfterMs: number, now: number = Date.now()): number {
-    if (staleAfterMs <= 0) {
-      return 0;
-    }
-    const cutoff = now - staleAfterMs;
-    const toRemove: string[] = [];
-    for (const [sellerId, seen] of lastSeenAt.entries()) {
-      if (seen < cutoff && this.internals.bySeller.has(sellerId)) {
-        toRemove.push(sellerId);
-      }
-    }
-    if (toRemove.length === 0) {
-      return 0;
-    }
-
-    const byModel = new Map<string, RegistrySeller[]>();
-    for (const [key, sellers] of this.internals.byModel.entries()) {
-      const filtered = sellers.filter((seller) => !toRemove.includes(seller.id));
-      if (filtered.length > 0) {
-        byModel.set(key, filtered);
-      }
-    }
-    const bySeller = new Map<string, RegistrySeller>();
-    for (const [sellerId, seller] of this.internals.bySeller.entries()) {
-      if (!toRemove.includes(sellerId)) {
-        bySeller.set(sellerId, seller);
-      }
-    }
-
-    this.internals = {
-      ...this.internals,
-      byModel,
-      bySeller,
-      registryFetchedAt: now
-    };
-
-    logger.info("models_refresh.pruned", "stale sellers pruned from model index", {
-      removedCount: toRemove.length,
-      remainingSellers: bySeller.size,
-      remainingModels: byModel.size
-    });
-    return toRemove.length;
-  }
-
-  /**
-   * Snapshot of the internal counters for diagnostics. Cheap to call; does
-   * not copy the maps.
-   */
-  stats(): {
-    sellerCount: number;
-    modelCount: number;
-    missingModelsCount: number;
-    registryVersion: number;
-    registryFetchedAt: number;
-    defaultSellerId?: string;
-  } {
-    return {
-      sellerCount: this.internals.totalSellers,
-      modelCount: this.internals.byModel.size,
-      missingModelsCount: this.internals.missingModelsCount,
-      registryVersion: this.internals.registryVersion,
-      registryFetchedAt: this.internals.registryFetchedAt,
-      defaultSellerId: this.internals.defaultSellerId
-    };
-  }
-}
-
-function emptyInternals(): ModelIndexInternals {
-  return {
-    byModel: new Map(),
-    bySeller: new Map(),
-    registryVersion: 0,
-    registryFetchedAt: 0,
-    missingModelsCount: 0,
-    totalSellers: 0
-  };
-}
-
-function matchesFilter(seller: RegistrySeller, filter: { protocol?: string; paymentMethod?: string }): boolean {
-  if (filter.protocol) {
-    const protocols = seller.supportedProtocols ?? [];
-    if (!protocols.includes(filter.protocol)) {
-      return false;
-    }
-  }
-  if (filter.paymentMethod) {
-    const methods = seller.paymentMethods ?? [];
-    if (!methods.includes(filter.paymentMethod)) {
-      return false;
-    }
-  }
-  return true;
-}

package/src/package-update.ts

@@ -1,311 +0,0 @@
-import * as fs from "fs";
-import * as os from "os";
-import * as path from "path";
-import { execFileSync, spawn } from "child_process";
-import { fileURLToPath } from "url";
-
-export const TOKENBUDDY_LAUNCHD_LABEL = "com.tokenbuddy.proxyd";
-const DEFAULT_PACKAGE_NAME = "tokenbuddy";
-
-export interface InstalledPackageManifest {
-  name: string;
-  version: string;
-}
-
-export interface PackageUpdateCheck {
-  packageName: string;
-  currentVersion: string;
-  latestVersion: string;
-  updateAvailable: boolean;
-  registryUrl: string;
-  installCommand: string;
-}
-
-export interface PackageInstallResult {
-  attempted: boolean;
-  succeeded: boolean;
-  command: string;
-  args: string[];
-  error?: string;
-}
-
-export interface PackageRestartResult {
-  attempted: boolean;
-  restarted: boolean;
-  method: "launchd";
-  scheduled?: boolean;
-  plistPath?: string;
-  target?: string;
-  error?: string;
-}
-
-export interface PackageUpdateResult {
-  check: PackageUpdateCheck;
-  install: PackageInstallResult;
-  restart: PackageRestartResult;
-}
-
-export interface CheckPackageUpdateDeps {
-  fetch?: typeof fetch;
-  env?: NodeJS.ProcessEnv;
-  argv?: string[];
-  cwd?: string;
-}
-
-export interface RunPackageUpdateDeps extends CheckPackageUpdateDeps {
-  npmCommand?: string;
-  runNpmInstall?: (command: string, args: string[]) => void;
-  restartProxyd?: (controlPort: number) => Promise<PackageRestartResult>;
-}
-
-export interface RunPackageUpdateOptions {
-  apply: boolean;
-  controlPort: number;
-}
-
-interface RegistryDocument {
-  "dist-tags"?: {
-    latest?: unknown;
-  };
-}
-
-function currentModuleDir(): string {
-  if (typeof __dirname !== "undefined") {
-    return __dirname;
-  }
-
-  const stack = new Error().stack || "";
-  const fileUrlMatch = stack.match(/(file:\/\/\/[^)\n]+\/package-update\.js):\d+:\d+/);
-  if (fileUrlMatch) {
-    return path.dirname(fileURLToPath(fileUrlMatch[1]));
-  }
-
-  const filePathMatch = stack.match(/(\/[^)\n]+\/package-update\.(?:js|ts)):\d+:\d+/);
-  if (filePathMatch) {
-    return path.dirname(filePathMatch[1]);
-  }
-
-  return process.cwd();
-}
-
-function candidatePackageRoots(argv: string[] = process.argv, cwd = process.cwd()): string[] {
-  return [
-    argv[1],
-    path.join(cwd, "packages", "tokenbuddy-cli"),
-    path.resolve(currentModuleDir(), ".."),
-    cwd,
-  ].filter((candidate): candidate is string => Boolean(candidate));
-}
-
-export function readInstalledPackageManifest(
-  argv: string[] = process.argv,
-  cwd = process.cwd(),
-): InstalledPackageManifest {
-  const seen = new Set<string>();
-  for (const candidateRoot of candidatePackageRoots(argv, cwd)) {
-    let current = fs.existsSync(candidateRoot) ? fs.realpathSync(candidateRoot) : candidateRoot;
-    if (!fs.existsSync(current)) continue;
-    if (!fs.statSync(current).isDirectory()) {
-      current = path.dirname(current);
-    }
-    while (!seen.has(current)) {
-      seen.add(current);
-      const packageJsonPath = path.join(current, "package.json");
-      if (fs.existsSync(packageJsonPath)) {
-        const packageJson = JSON.parse(fs.readFileSync(packageJsonPath, "utf8")) as { name?: unknown; version?: unknown };
-        if (typeof packageJson.version === "string" && packageJson.version.length > 0) {
-          const name = typeof packageJson.name === "string" && packageJson.name.length > 0
-            ? packageJson.name
-            : DEFAULT_PACKAGE_NAME;
-          if (name === DEFAULT_PACKAGE_NAME || name === "@tokenbuddy/tokenbuddy") {
-            return { name, version: packageJson.version };
-          }
-        }
-      }
-      const parent = path.dirname(current);
-      if (parent === current) break;
-      current = parent;
-    }
-  }
-  return { name: DEFAULT_PACKAGE_NAME, version: "0.0.0" };
-}
-
-function registryUrl(packageName: string): string {
-  return `https://registry.npmjs.org/${encodeURIComponent(packageName)}`;
-}
-
-function packageNameForUpdate(manifest: InstalledPackageManifest, env: NodeJS.ProcessEnv = process.env): string {
-  const override = env.TOKENBUDDY_UPDATE_PACKAGE_NAME?.trim();
-  if (override) {
-    return override;
-  }
-  return manifest.name || DEFAULT_PACKAGE_NAME;
-}
-
-function parseSemver(value: string): [number, number, number] | null {
-  const match = value.trim().match(/^(\d+)\.(\d+)\.(\d+)(?:[-+].*)?$/);
-  if (!match) {
-    return null;
-  }
-  return [Number(match[1]), Number(match[2]), Number(match[3])];
-}
-
-function isVersionGreater(left: string, right: string): boolean {
-  const parsedLeft = parseSemver(left);
-  const parsedRight = parseSemver(right);
-  if (!parsedLeft || !parsedRight) {
-    return left !== right;
-  }
-  for (let index = 0; index < parsedLeft.length; index += 1) {
-    if (parsedLeft[index] > parsedRight[index]) return true;
-    if (parsedLeft[index] < parsedRight[index]) return false;
-  }
-  return false;
-}
-
-export async function checkPackageUpdate(deps: CheckPackageUpdateDeps = {}): Promise<PackageUpdateCheck> {
-  const env = deps.env ?? process.env;
-  const manifest = readInstalledPackageManifest(deps.argv, deps.cwd);
-  const packageName = packageNameForUpdate(manifest, env);
-  const url = registryUrl(packageName);
-  const fetchImpl = deps.fetch ?? fetch;
-  const res = await fetchImpl(url);
-  if (!res.ok) {
-    throw new Error(`npm registry returned HTTP ${res.status} for ${packageName}`);
-  }
-  const registry = await res.json() as RegistryDocument;
-  const latestVersion = registry["dist-tags"]?.latest;
-  if (typeof latestVersion !== "string" || latestVersion.length === 0) {
-    throw new Error(`npm registry response for ${packageName} is missing dist-tags.latest`);
-  }
-  const installCommand = `npm install -g ${packageName}@${latestVersion}`;
-  return {
-    packageName,
-    currentVersion: manifest.version,
-    latestVersion,
-    updateAvailable: isVersionGreater(latestVersion, manifest.version),
-    registryUrl: url,
-    installCommand,
-  };
-}
-
-function npmCommand(env: NodeJS.ProcessEnv, override?: string): string {
-  return override || env.TOKENBUDDY_UPDATE_NPM_COMMAND || "npm";
-}
-
-function defaultRunNpmInstall(command: string, args: string[]): void {
-  execFileSync(command, args, {
-    stdio: ["ignore", "pipe", "pipe"],
-    maxBuffer: 8 * 1024 * 1024,
-  });
-}
-
-function commandErrorMessage(error: unknown): string {
-  const withOutput = error as { message?: string; stderr?: Buffer; stdout?: Buffer };
-  const stderr = withOutput.stderr?.toString("utf8").trim();
-  const stdout = withOutput.stdout?.toString("utf8").trim();
-  return stderr || stdout || (error instanceof Error ? error.message : String(error));
-}
-
-export async function runPackageUpdate(
-  options: RunPackageUpdateOptions,
-  deps: RunPackageUpdateDeps = {},
-): Promise<PackageUpdateResult> {
-  const check = await checkPackageUpdate(deps);
-  const command = npmCommand(deps.env ?? process.env, deps.npmCommand);
-  const args = ["install", "-g", `${check.packageName}@${check.latestVersion}`];
-  const install: PackageInstallResult = {
-    attempted: false,
-    succeeded: false,
-    command,
-    args,
-  };
-  const restart: PackageRestartResult = {
-    attempted: false,
-    restarted: false,
-    method: "launchd",
-  };
-
-  if (!options.apply || !check.updateAvailable) {
-    return {
-      check,
-      install,
-      restart,
-    };
-  }
-
-  install.attempted = true;
-  try {
-    (deps.runNpmInstall ?? defaultRunNpmInstall)(command, args);
-    install.succeeded = true;
-  } catch (error: unknown) {
-    install.error = commandErrorMessage(error);
-    return { check, install, restart };
-  }
-
-  if (!deps.restartProxyd) {
-    restart.error = "tb-proxyd restart runner is not configured";
-    return { check, install, restart };
-  }
-
-  return {
-    check,
-    install,
-    restart: await deps.restartProxyd(options.controlPort),
-  };
-}
-
-function launchdUserDomain(): string {
-  if (typeof process.getuid === "function") {
-    return `gui/${process.getuid()}`;
-  }
-  return "gui/501";
-}
-
-function launchdServiceTarget(label: string): string {
-  return `${launchdUserDomain()}/${label}`;
-}
-
-export function scheduleLaunchAgentRestart(
-  deps: {
-    platform?: NodeJS.Platform;
-    homeDir?: string;
-    existsSync?: (filePath: string) => boolean;
-    spawn?: typeof spawn;
-  } = {},
-): PackageRestartResult {
-  const platform = deps.platform ?? process.platform;
-  const homeDir = deps.homeDir ?? os.homedir();
-  const plistPath = path.join(homeDir, "Library", "LaunchAgents", `${TOKENBUDDY_LAUNCHD_LABEL}.plist`);
-  const base = {
-    attempted: false,
-    restarted: false,
-    method: "launchd" as const,
-    plistPath,
-  };
-  if (platform !== "darwin") {
-    return {
-      ...base,
-      error: "tb-proxyd restart is only supported for the macOS LaunchAgent service.",
-    };
-  }
-  if (!(deps.existsSync ?? fs.existsSync)(plistPath)) {
-    return {
-      ...base,
-      error: "LaunchAgent plist is missing. Run `tb init` to install tb-proxyd as a service first.",
-    };
-  }
-
-  const target = launchdServiceTarget(TOKENBUDDY_LAUNCHD_LABEL);
-  const child = (deps.spawn ?? spawn)("sh", ["-c", `sleep 0.35; launchctl kickstart -k ${target}`], {
-    detached: true,
-    stdio: "ignore",
-  });
-  child.unref();
-  return {
-    ...base,
-    attempted: true,
-    scheduled: true,
-    target,
-  };
-}