dsh-llm-codebuddy 1.3.5 → 1.3.6

package/codebuddy-auth.js

@@ -1,7 +1,10 @@
 import { spawn } from "node:child_process";
+import { createHash } from "node:crypto";
 import { setTimeout as delay } from "node:timers/promises";
 
 export const CODEBUDDY_SESSION_REF = "CODEBUDDY_LOGIN_SESSION";
+export const CODEBUDDY_SESSIONS_REF = "CODEBUDDY_LOGIN_SESSIONS";
+export const CODEBUDDY_API_KEYS_REF = "CODEBUDDY_API_KEYS";
 
 const BASE_URL = "https://copilot.tencent.com/v2/plugin";
 const USER_AGENT = "CLI/unknown CodeBuddy/2.137.1";
@@ -93,13 +96,186 @@ function openBrowser(url) {
   });
 }
 
-function normalizeAccount(account) {
+function textValue(...values) {
+  for (const value of values) {
+    if (typeof value === "string" && value.trim()) return value.trim();
+    if (typeof value === "number" && Number.isFinite(value)) return String(value);
+  }
+  return undefined;
+}
+
+function tokenClaims(token) {
+  if (typeof token !== "string") return {};
+  try {
+    const payload = token.split(".")[1];
+    if (!payload) return {};
+    const parsed = JSON.parse(Buffer.from(payload, "base64url").toString("utf8"));
+    return parsed && typeof parsed === "object" ? parsed : {};
+  } catch {
+    return {};
+  }
+}
+
+function normalizeApiKeyEntry(entry, now = Date.now()) {
+  const ref = textValue(entry?.ref);
+  if (!ref || !/^[A-Za-z_][A-Za-z0-9_]*$/.test(ref)) throw new Error("CodeBuddy API Key 引用无效");
+  const id = textValue(entry?.id) ?? `dsh:${ref}`;
+  return {
+    id,
+    ref,
+    label: textValue(entry?.label) ?? "DSH 保存的 API Key",
+    createdAt: Number.isFinite(entry?.createdAt) ? entry.createdAt : now,
+    updatedAt: Number.isFinite(entry?.updatedAt) ? entry.updatedAt : now,
+  };
+}
+
+export function createCodeBuddyApiKeyStore(entries = [], activeId) {
+  const keys = [];
+  for (const entry of entries) {
+    const normalized = normalizeApiKeyEntry(entry);
+    if (!keys.some((item) => item.id === normalized.id || item.ref === normalized.ref)) keys.push(normalized);
+  }
+  const selected = activeId === null
+    ? null
+    : typeof activeId === "string" && keys.some((entry) => entry.id === activeId) ? activeId : keys[0]?.id;
+  return { version: 1, activeId: selected, entries: keys };
+}
+
+export function upsertCodeBuddyApiKey(store, entry, now = Date.now()) {
+  const current = createCodeBuddyApiKeyStore(store?.entries ?? [], store?.activeId);
+  const incoming = normalizeApiKeyEntry({ ...entry, updatedAt: now }, now);
+  const index = current.entries.findIndex((item) => item.id === incoming.id || item.ref === incoming.ref);
+  if (index >= 0) incoming.createdAt = current.entries[index].createdAt;
+  const entries = index >= 0
+    ? current.entries.map((item, position) => position === index ? incoming : item)
+    : [...current.entries, incoming];
+  return { version: 1, activeId: incoming.id, entries };
+}
+
+export function serializeCodeBuddyApiKeys(store) {
+  const normalized = createCodeBuddyApiKeyStore(store?.entries ?? [], store?.activeId);
+  return JSON.stringify(normalized);
+}
+
+export function parseCodeBuddyApiKeys(value) {
+  let parsed;
+  try {
+    parsed = JSON.parse(value);
+  } catch (error) {
+    throw new Error("CodeBuddy API Key 列表已损坏，请重新配置", { cause: error });
+  }
+  if (!Array.isArray(parsed?.entries)) throw new Error("CodeBuddy API Key 列表格式无效，请重新配置");
+  return createCodeBuddyApiKeyStore(parsed.entries, parsed.activeId);
+}
+
+export function codeBuddyApiKeyEntries(store) {
+  return (store?.entries ?? []).map(({ id, ref, label, createdAt, updatedAt }) => ({ id, ref, label, createdAt, updatedAt }));
+}
+
+function normalizeAccount(account, auth) {
+  // The account endpoint has returned both a flat object and wrapped objects
+  // across CodeBuddy versions. Keep all known wrappers in the lookup so a newly
+  // added account gets the same display metadata as an imported account.
+  const sources = [
+    account,
+    account?.account,
+    account?.user,
+    account?.userInfo,
+    account?.profile,
+    account?.data,
+  ].filter((source) => source && typeof source === "object");
+  const read = (...keys) => textValue(...sources.flatMap((source) => keys.map((key) => source[key])));
+  const claims = tokenClaims(auth?.accessToken);
+  const userId = read("userId", "uid", "user_id", "id") ?? textValue(claims.userId, claims.uid, claims.user_id, claims.sub);
+  const enterpriseId = read("enterpriseId", "tenantId", "enterprise_id", "tenant_id")
+    ?? textValue(claims.enterpriseId, claims.tenantId, claims.enterprise_id, claims.tenant_id);
+  const email = read("email", "mail", "emailAddress") ?? textValue(claims.email, claims.mail);
+  const uin = read("uin", "phoneNumber", "phone", "mobile", "mobilePhone")
+    ?? textValue(claims.uin, claims.phoneNumber, claims.phone_number, claims.mobile);
+  const type = read("type", "accountType", "account_type");
+  const displayName = read("displayName", "name", "nickname", "username", "accountName")
+    ?? uin
+    ?? textValue(claims.displayName, claims.name, claims.nickname, claims.username, claims.preferred_username)
+    ?? email;
   return {
-    ...(account?.userId || account?.uid ? { userId: account.userId ?? account.uid } : {}),
-    ...(account?.enterpriseId || account?.tenantId ? { enterpriseId: account.enterpriseId ?? account.tenantId } : {}),
+    ...(userId ? { userId } : {}),
+    ...(enterpriseId ? { enterpriseId } : {}),
+    ...(email ? { email } : {}),
+    ...(uin ? { uin } : {}),
+    ...(type ? { type } : {}),
+    ...(displayName ? { displayName } : {}),
   };
 }
 
+export function codeBuddySessionId(session) {
+  const account = normalizeAccount(session?.account, session?.auth);
+  if (account.userId) return `user:${account.userId}`;
+  if (account.email) return `email:${account.email}`;
+  if (account.enterpriseId) return `enterprise:${account.enterpriseId}`;
+  const refreshToken = session?.auth?.refreshToken ?? session?.auth?.accessToken;
+  if (!refreshToken) throw new Error("CodeBuddy 登录会话缺少账号标识和令牌");
+  return `token:${createHash("sha256").update(refreshToken).digest("hex").slice(0, 24)}`;
+}
+
+export function codeBuddySessionLabel(session) {
+  const account = normalizeAccount(session?.account, session?.auth);
+  return account.displayName ?? account.email ?? account.userId ?? account.enterpriseId ?? `账号 ${codeBuddySessionId(session).slice(-8)}`;
+}
+
+export function normalizeCodeBuddySessionEntry(session, now = Date.now()) {
+  const normalized = {
+    auth: calculateExpiresAt(session?.auth),
+    account: normalizeAccount(session?.account, session?.auth),
+  };
+  if (!normalized.auth.accessToken || !normalized.auth.refreshToken) throw new Error("CodeBuddy 登录会话无效");
+  const id = typeof session?.id === "string" && session.id.trim() ? session.id : codeBuddySessionId(normalized);
+  const createdAt = Number.isFinite(session?.createdAt) ? session.createdAt : now;
+  return {
+    id,
+    label: typeof session?.label === "string" && session.label.trim() ? session.label : codeBuddySessionLabel(normalized),
+    createdAt,
+    updatedAt: Number.isFinite(session?.updatedAt) ? session.updatedAt : now,
+    ...normalized,
+  };
+}
+
+export function createCodeBuddySessionStore(entries = [], activeId) {
+  const sessions = [];
+  for (const entry of entries) {
+    const normalized = normalizeCodeBuddySessionEntry(entry);
+    if (!sessions.some((item) => item.id === normalized.id)) sessions.push(normalized);
+  }
+  const selected = typeof activeId === "string" && sessions.some((entry) => entry.id === activeId) ? activeId : sessions[0]?.id;
+  return { version: 1, activeId: selected, sessions };
+}
+
+export function upsertCodeBuddySession(store, session, now = Date.now()) {
+  const current = createCodeBuddySessionStore(store?.sessions ?? [], store?.activeId);
+  const incoming = normalizeCodeBuddySessionEntry({ ...session, updatedAt: now }, now);
+  const index = current.sessions.findIndex((entry) => entry.id === incoming.id);
+  if (index >= 0) incoming.createdAt = current.sessions[index].createdAt;
+  const sessions = index >= 0
+    ? current.sessions.map((entry, position) => position === index ? incoming : entry)
+    : [...current.sessions, incoming];
+  return { version: 1, activeId: incoming.id, sessions };
+}
+
+export function activeCodeBuddySession(store) {
+  return store?.sessions?.find((entry) => entry.id === store.activeId) ?? store?.sessions?.[0];
+}
+
+export function codeBuddySessionAccounts(store) {
+  return (store?.sessions ?? []).map(({ id, label, account, createdAt, updatedAt }) => ({
+    id,
+    label,
+    accountName: label,
+    userId: account?.userId ?? null,
+    account,
+    createdAt,
+    updatedAt,
+  }));
+}
+
 export async function loginCodeBuddy(onAuthUrl, signal) {
   const state = await request("/auth/state?platform=CLI", {
     method: "POST",
@@ -129,7 +305,7 @@ export async function loginCodeBuddy(onAuthUrl, signal) {
     60_000,
     signal,
   );
-  return { auth, account: normalizeAccount(account) };
+  return { auth, account: normalizeAccount(account, auth) };
 }
 
 export async function refreshCodeBuddySession(session, signal) {
@@ -148,12 +324,18 @@ export async function refreshCodeBuddySession(session, signal) {
   const fresh = calculateExpiresAt(auth);
   const merged = { ...session.auth, ...fresh, refreshToken: fresh?.refreshToken ?? session.auth.refreshToken };
   if (!merged.accessToken) throw new Error("CodeBuddy 刷新接口没有返回访问令牌");
-  return { auth: merged, account: normalizeAccount(session.account) };
+  return { auth: merged, account: normalizeAccount(session.account, merged) };
 }
 
 export function serializeCodeBuddySession(session) {
   if (!session?.auth?.accessToken || !session?.auth?.refreshToken) throw new Error("CodeBuddy 登录会话无效");
-  return JSON.stringify({ auth: calculateExpiresAt(session.auth), account: normalizeAccount(session.account) });
+  return JSON.stringify({ auth: calculateExpiresAt(session.auth), account: normalizeAccount(session.account, session.auth) });
+}
+
+export function serializeCodeBuddySessions(store) {
+  const normalized = createCodeBuddySessionStore(store?.sessions ?? [], store?.activeId);
+  if (normalized.sessions.length === 0) throw new Error("CodeBuddy 登录账号列表为空");
+  return JSON.stringify(normalized);
 }
 
 export function parseCodeBuddySession(value) {
@@ -164,7 +346,22 @@ export function parseCodeBuddySession(value) {
     throw new Error("CodeBuddy 登录凭据已损坏，请重新登录", { cause: error });
   }
   if (!session?.auth?.accessToken || !session?.auth?.refreshToken) throw new Error("CodeBuddy 登录凭据不完整，请重新登录");
-  return { auth: calculateExpiresAt(session.auth), account: normalizeAccount(session.account) };
+  return { auth: calculateExpiresAt(session.auth), account: normalizeAccount(session.account, session.auth) };
+}
+
+export function parseCodeBuddySessions(value) {
+  let parsed;
+  try {
+    parsed = JSON.parse(value);
+  } catch (error) {
+    throw new Error("CodeBuddy 登录账号列表已损坏，请重新登录", { cause: error });
+  }
+  if (Array.isArray(parsed?.sessions)) return createCodeBuddySessionStore(parsed.sessions, parsed.activeId);
+  if (parsed?.auth) {
+    const session = parseCodeBuddySession(value);
+    return createCodeBuddySessionStore([session], codeBuddySessionId(session));
+  }
+  throw new Error("CodeBuddy 登录账号列表格式无效，请重新登录");
 }
 
 export function sessionNeedsRefresh(session, now = Date.now()) {

package/codebuddy-credits.js

@@ -0,0 +1,441 @@
+const DEFAULT_BILLING_HOST = "https://www.codebuddy.cn";
+const BROWSER_USER_AGENT = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/151.0.0.0 Safari/537.36";
+const ENTERPRISE_EDITIONS = new Set(["ultimate", "exclusive"]);
+const REMAINING_FIELDS = [
+  "SlicePeriodCapacityRemainPrecise",
+  "SlicePeriodCapacityRemain",
+  "CycleCapacityRemainPrecise",
+  "CycleCapacityRemain",
+  "CapacityRemainPrecise",
+  "CapacityRemain",
+  "RemainPrecise",
+  "Remain",
+  "Remaining",
+  "Balance",
+];
+const TOTAL_FIELDS = [
+  "SlicePeriodCapacitySizePrecise",
+  "SlicePeriodCapacitySize",
+  "CycleCapacitySizePrecise",
+  "CycleCapacitySize",
+  "CycleCapacityPrecise",
+  "CycleCapacity",
+  "CapacityPrecise",
+  "Capacity",
+  "TotalCapacityPrecise",
+  "TotalCapacity",
+  "PackageCapacity",
+  "Quota",
+  "Amount",
+];
+const EXPIRY_FIELDS = [
+  "DeductionEndTime",
+  "ExpiredTime",
+  "SlicePeriodEndTime",
+  "PackageEndTime",
+  "EndTime",
+  "CycleEndTime",
+  "ExpireTime",
+  "ExpirationTime",
+  "ValidEndTime",
+  "ValidPeriodEndTime",
+  "EndAt",
+  "ExpireAt",
+];
+const LABEL_FIELDS = ["PackageName", "PackageTypeName", "AccountName", "ProductName", "Name", "RuleName", "Description"];
+const MAX_USAGE_PAGES = 100;
+const PAGE_SIZE = 100;
+
+function normalizeHost(value) {
+  const fallback = new URL(DEFAULT_BILLING_HOST);
+  if (typeof value !== "string" || !value.trim()) return fallback.origin;
+  try {
+    const candidate = new URL(value.includes("://") ? value : `https://${value}`);
+    if (candidate.protocol !== "https:") return fallback.origin;
+    const host = candidate.hostname.toLowerCase();
+    if (!["codebuddy.cn", "www.codebuddy.cn", "workbuddy.cn", "www.workbuddy.cn"].includes(host)) return fallback.origin;
+    return candidate.origin;
+  } catch {
+    return fallback.origin;
+  }
+}
+
+function billingHost(session) {
+  return normalizeHost(session?.auth?.domain);
+}
+
+function authToken(session) {
+  const token = typeof session?.auth?.accessToken === "string" ? session.auth.accessToken.trim() : "";
+  if (!token) throw new Error("CodeBuddy 登录令牌为空");
+  return token;
+}
+
+function billingHeaders(session, host, enterpriseId) {
+  const headers = {
+    accept: "application/json, text/plain, */*",
+    "content-type": "application/json",
+    "x-client-platform": "web",
+    origin: host,
+    referer: `${host}/profile/plans-usage`,
+    authorization: `Bearer ${authToken(session)}`,
+    "user-agent": BROWSER_USER_AGENT,
+  };
+  const domain = typeof session?.auth?.domain === "string" ? session.auth.domain.trim() : "";
+  if (domain) headers["x-domain"] = domain;
+  const userId = typeof session?.account?.userId === "string" ? session.account.userId.trim() : "";
+  if (userId) headers["x-user-id"] = userId;
+  if (enterpriseId) {
+    headers["x-enterprise-id"] = String(enterpriseId);
+    headers["x-tenant-id"] = String(enterpriseId);
+  }
+  return headers;
+}
+
+function timeoutSignal(timeoutMs, externalSignal) {
+  if (externalSignal) return externalSignal;
+  if (typeof AbortSignal?.timeout === "function") return AbortSignal.timeout(timeoutMs);
+  const controller = new AbortController();
+  setTimeout(() => controller.abort(), timeoutMs).unref?.();
+  return controller.signal;
+}
+
+async function readJson(response, action) {
+  const raw = await response.text();
+  if (!raw.trim()) throw new Error(`${action}返回空响应（HTTP ${response.status}）`);
+  let payload;
+  try {
+    payload = JSON.parse(raw);
+  } catch (error) {
+    throw new Error(`${action}返回了无法解析的数据`, { cause: error });
+  }
+  if (!response.ok) throw new Error(`${action} HTTP ${response.status}: ${raw.slice(0, 160)}`);
+  if (payload?.code !== undefined && payload.code !== null && payload.code !== 0) {
+    throw new Error(`${action}失败（${payload.msg ?? payload.message ?? `code=${payload.code}`}）`);
+  }
+  return payload;
+}
+
+function firstNumber(value, fields) {
+  for (const field of fields) {
+    const raw = value?.[field];
+    if (raw === undefined || raw === null || raw === "") continue;
+    const number = Number(raw);
+    if (Number.isFinite(number)) return number;
+  }
+  return null;
+}
+
+function parseTimestamp(value) {
+  if (value === undefined || value === null || value === "") return null;
+  if (typeof value === "number" || /^\d+(?:\.\d+)?$/.test(String(value).trim())) {
+    const number = Number(value);
+    if (!Number.isFinite(number)) return null;
+    return number < 1e12 ? Math.round(number * 1000) : Math.round(number);
+  }
+  const parsed = Date.parse(String(value).replace(/^(\d{4}-\d\d-\d\d)\s+/, "$1T"));
+  return Number.isFinite(parsed) ? parsed : null;
+}
+
+function firstTimestamp(value, fields) {
+  for (const field of fields) {
+    const parsed = parseTimestamp(value?.[field]);
+    if (parsed !== null) return parsed;
+  }
+  return null;
+}
+
+function firstText(value, fields) {
+  for (const field of fields) {
+    const text = value?.[field];
+    if (typeof text === "string" && text.trim()) return text.trim();
+  }
+  return "";
+}
+
+function extractAccounts(payload) {
+  const candidates = [
+    payload?.data?.Response?.Data?.Accounts,
+    payload?.data?.data?.Response?.Data?.Accounts,
+    payload?.data?.accounts,
+    payload?.data?.data?.accounts,
+    payload?.Response?.Data?.Accounts,
+  ];
+  return candidates.find(Array.isArray) ?? [];
+}
+
+function extractCreditSegments(accounts, source = "积分") {
+  return (Array.isArray(accounts) ? accounts : [])
+    .flatMap((account) => {
+      const details = Array.isArray(account?.SlicePeriodUsageDetails) && account.SlicePeriodUsageDetails.length
+        ? account.SlicePeriodUsageDetails.map((detail) => ({ ...account, ...detail }))
+        : [account];
+      return details.map((item) => {
+        const remaining = firstNumber(item, REMAINING_FIELDS);
+        if (remaining === null || remaining <= 0) return null;
+        const total = firstNumber(item, TOTAL_FIELDS);
+        return {
+          remaining: Number(remaining.toFixed(2)),
+          total: Number((total === null ? remaining : Math.max(total, remaining)).toFixed(2)),
+          expiresAt: firstTimestamp(item, EXPIRY_FIELDS),
+          source: firstText(item, LABEL_FIELDS) || source,
+          packageCode: item?.PackageCode ? String(item.PackageCode) : "",
+        };
+      });
+    })
+    .filter(Boolean);
+}
+
+function sortSegments(segments) {
+  return (Array.isArray(segments) ? segments : [])
+    .filter((segment) => segment && Number(segment.remaining) > 0)
+    .map((segment) => ({
+      remaining: Number(Number(segment.remaining).toFixed(2)),
+      total: Number(Number(segment.total || segment.remaining).toFixed(2)),
+      expiresAt: segment.expiresAt === null || segment.expiresAt === undefined ? null : Number(segment.expiresAt),
+      source: String(segment.source || "积分"),
+      packageCode: String(segment.packageCode || ""),
+    }))
+    .sort((left, right) => {
+      if (left.expiresAt === null && right.expiresAt !== null) return 1;
+      if (left.expiresAt !== null && right.expiresAt === null) return -1;
+      return (left.expiresAt || 0) - (right.expiresAt || 0);
+    });
+}
+
+function mergeSegments(segments) {
+  const merged = new Map();
+  for (const segment of Array.isArray(segments) ? segments : []) {
+    if (!segment || Number(segment.remaining) <= 0) continue;
+    const key = [segment.packageCode || segment.source || "积分", segment.expiresAt ?? "unknown"].join("|");
+    const previous = merged.get(key);
+    if (previous) {
+      previous.remaining += Number(segment.remaining) || 0;
+      previous.total += Number(segment.total || segment.remaining) || 0;
+    } else {
+      merged.set(key, {
+        remaining: Number(segment.remaining) || 0,
+        total: Number(segment.total || segment.remaining) || 0,
+        expiresAt: segment.expiresAt === undefined ? null : segment.expiresAt,
+        source: String(segment.source || "积分"),
+        packageCode: String(segment.packageCode || ""),
+      });
+    }
+  }
+  return sortSegments(Array.from(merged.values()));
+}
+
+function buildCreditResourceBody(now = new Date()) {
+  const end = new Date(now.getTime());
+  end.setFullYear(end.getFullYear() + 101);
+  const format = (date) => {
+    const pad = (value) => String(value).padStart(2, "0");
+    return `${date.getFullYear()}-${pad(date.getMonth() + 1)}-${pad(date.getDate())} ${pad(date.getHours())}:${pad(date.getMinutes())}:${pad(date.getSeconds())}`;
+  };
+  return {
+    PageNumber: 1,
+    PageSize: 100,
+    ProductCode: "p_tcaca",
+    Status: [0, 3],
+    PackageEndTimeRangeBegin: format(now),
+    PackageEndTimeRangeEnd: format(end),
+  };
+}
+
+async function postJson(url, session, body, action, options = {}) {
+  const response = await (options.fetchImpl || globalThis.fetch)(url, {
+    method: "POST",
+    headers: billingHeaders(session, options.host, options.enterpriseId),
+    body: JSON.stringify(body),
+    signal: timeoutSignal(options.timeoutMs ?? 12_000, options.signal),
+  });
+  return readJson(response, action);
+}
+
+function enterpriseUsage(payload) {
+  const candidates = [payload?.data, payload?.data?.data, payload?.data?.Response?.Data, payload];
+  const data = candidates.find((value) => value && typeof value === "object" && ("limitNum" in value || "LimitNum" in value));
+  if (!data) return null;
+  const limitNum = Number(data.limitNum ?? data.LimitNum);
+  if (!Number.isFinite(limitNum)) return null;
+  const reset = firstTimestamp(data, ["cycleResetTime", "CycleResetTime", "CycleResetTimeMs"]);
+  if (limitNum === -1) return { unlimited: true, credits: null, total: null, count: 0, segments: [], cycleResetTime: reset };
+  const credit = Number(data.credit ?? data.Credit);
+  const used = Number.isFinite(credit) ? credit : 0;
+  const credits = Math.max(0, limitNum - used);
+  return {
+    unlimited: false,
+    credits: Number(credits.toFixed(2)),
+    total: Number(limitNum.toFixed(2)),
+    count: 1,
+    segments: sortSegments([{ remaining: credits, total: limitNum, expiresAt: reset, source: "企业配额" }]),
+    cycleResetTime: reset,
+  };
+}
+
+async function queryPersonalCredits(session, host, options = {}) {
+  const payload = await postJson(`${host}/v2/billing/meter/get-user-resource`, session, buildCreditResourceBody(), "CodeBuddy 积分接口", { ...options, host });
+  const accounts = extractAccounts(payload);
+  let credits = 0;
+  for (const account of accounts) {
+    const remaining = firstNumber(account, [
+      "CycleCapacityRemainPrecise",
+      "CycleCapacityRemain",
+      "CapacityRemainPrecise",
+      "CapacityRemain",
+    ]);
+    if (remaining !== null) credits += remaining;
+  }
+  return {
+    credits: Number(credits.toFixed(2)),
+    count: accounts.length,
+    totalDosage: payload?.data?.Response?.Data?.TotalDosage ?? payload?.data?.data?.Response?.Data?.TotalDosage ?? null,
+    segments: mergeSegments(extractCreditSegments(accounts)),
+    unlimited: false,
+    cycleResetTime: null,
+  };
+}
+
+async function resolveEnterpriseId(session, host, options = {}) {
+  const uid = session?.account?.userId;
+  if (!uid) return "";
+  const response = await (options.fetchImpl || globalThis.fetch)(`${host}/console/accounts`, {
+    method: "GET",
+    headers: billingHeaders(session, host),
+    signal: timeoutSignal(8_000, options.signal),
+  });
+  const payload = await readJson(response, "CodeBuddy 企业账号接口");
+  const accounts = payload?.data?.accounts;
+  const first = Array.isArray(accounts) ? accounts[0] : undefined;
+  return typeof first?.enterpriseId === "string" ? first.enterpriseId.trim() : "";
+}
+
+async function queryEnterpriseCredits(session, host, enterpriseId, options = {}) {
+  const payload = await postJson(`${host}/v2/billing/meter/get-enterprise-user-usage`, session, {}, "CodeBuddy 企业积分接口", {
+    ...options,
+    host,
+    enterpriseId,
+  });
+  const parsed = enterpriseUsage(payload);
+  if (!parsed) throw new Error("CodeBuddy 企业积分接口返回数据无法解析");
+  return parsed;
+}
+
+function pad2(value) {
+  return String(value).padStart(2, "0");
+}
+
+function formatLocalDateTime(date) {
+  return `${date.getFullYear()}-${pad2(date.getMonth() + 1)}-${pad2(date.getDate())} ${pad2(date.getHours())}:${pad2(date.getMinutes())}:${pad2(date.getSeconds())}`;
+}
+
+function localDateString(date) {
+  return `${date.getFullYear()}-${pad2(date.getMonth() + 1)}-${pad2(date.getDate())}`;
+}
+
+function parseUsageTime(value) {
+  return parseTimestamp(value);
+}
+
+function usageRows(payload) {
+  const data = payload?.data;
+  const rows = data && Array.isArray(data.data) ? data.data : data && Array.isArray(data.rows) ? data.rows : [];
+  const total = Number(data?.total);
+  return { rows, total: Number.isSafeInteger(total) && total >= 0 ? total : rows.length };
+}
+
+async function queryTodayUsage(session, host, options = {}) {
+  const now = new Date();
+  const start = new Date(now.getTime());
+  start.setHours(0, 0, 0, 0);
+  const fetchImpl = options.fetchImpl || globalThis.fetch;
+  const records = [];
+  let expectedTotal = null;
+  let fetched = 0;
+  for (let pageNum = 1; pageNum <= MAX_USAGE_PAGES; pageNum += 1) {
+    const payload = await postJson(`${host}/billing/meter/get-user-request-usage`, session, {
+      startTime: formatLocalDateTime(start),
+      endTime: formatLocalDateTime(now),
+      pageNum,
+      pageSize: PAGE_SIZE,
+    }, "CodeBuddy 今日请求量接口", { ...options, host, fetchImpl, timeoutMs: options.usageTimeoutMs ?? 8_000 });
+    const page = usageRows(payload);
+    if (expectedTotal === null) expectedTotal = page.total;
+    if (!page.rows.length) break;
+    for (const row of page.rows) {
+      const requestTime = parseUsageTime(row?.requestTime ?? row?.RequestTime ?? row?.createdAt);
+      const credit = Number(row?.credit ?? row?.Credit ?? 0);
+      if (requestTime === null || !Number.isFinite(credit) || credit < 0) continue;
+      records.push({ requestTime, credit });
+    }
+    fetched += page.rows.length;
+    if (fetched >= expectedTotal || page.rows.length < PAGE_SIZE) break;
+  }
+  const used = records.reduce((sum, record) => sum + record.credit, 0);
+  return {
+    date: localDateString(now),
+    used: Number(used.toFixed(2)),
+    count: records.length,
+    synced: true,
+  };
+}
+
+async function retry(task, attempts = 2) {
+  let lastError;
+  for (let attempt = 1; attempt <= attempts; attempt += 1) {
+    try {
+      return await task();
+    } catch (error) {
+      lastError = error;
+      if (attempt < attempts) await new Promise((resolve) => setTimeout(resolve, 250 * attempt));
+    }
+  }
+  throw lastError;
+}
+
+export async function fetchCodeBuddyCredits(session, options = {}) {
+  const host = billingHost(session);
+  const account = session?.account && typeof session.account === "object" ? session.account : {};
+  let creditResult;
+  let creditError = null;
+  try {
+    const enterpriseId = typeof account.enterpriseId === "string" ? account.enterpriseId.trim() : "";
+    const enterpriseEdition = typeof account.type === "string" && ENTERPRISE_EDITIONS.has(account.type.toLowerCase());
+    if (enterpriseId || enterpriseEdition) {
+      const resolvedId = enterpriseId || await retry(() => resolveEnterpriseId(session, host, options));
+      if (!resolvedId) throw new Error("企业账号缺少 enterpriseId");
+      creditResult = await retry(() => queryEnterpriseCredits(session, host, resolvedId, options));
+    } else {
+      creditResult = await retry(() => queryPersonalCredits(session, host, options));
+    }
+  } catch (error) {
+    creditError = error instanceof Error ? error.message : String(error);
+    creditResult = { credits: null, count: 0, totalDosage: null, segments: [], unlimited: false, cycleResetTime: null };
+  }
+
+  let todayUsage;
+  let todayUsageError = null;
+  try {
+    todayUsage = await retry(() => queryTodayUsage(session, host, options));
+  } catch (error) {
+    todayUsageError = error instanceof Error ? error.message : String(error);
+  }
+  return {
+    ...creditResult,
+    creditError,
+    todayUsage: todayUsage ?? null,
+    todayUsageError,
+  };
+}
+
+export const __testing = Object.freeze({
+  billingHost,
+  buildCreditResourceBody,
+  enterpriseUsage,
+  extractAccounts,
+  extractCreditSegments,
+  formatLocalDateTime,
+  mergeSegments,
+  normalizeHost,
+  queryTodayUsage,
+  sortSegments,
+});