feat: add auth file mappings and update related queries to support credential names

Author: sxjeru

app/api/overview/route.ts

@@ -20,12 +20,13 @@ const OVERVIEW_CACHE_TTL_MS = 30_000;
 const OVERVIEW_CACHE_MAX_ENTRIES = 100;
 const overviewCache = new Map<string, CachedOverview>();
 
-function makeCacheKey(input: { days?: number; model?: string | null; route?: string | null; source?: string | null; page?: number; pageSize?: number; start?: string | null; end?: string | null }) {
+function makeCacheKey(input: { days?: number; model?: string | null; route?: string | null; source?: string | null; name?: string | null; page?: number; pageSize?: number; start?: string | null; end?: string | null }) {
   return JSON.stringify({
     days: input.days ?? null,
     model: input.model ?? null,
     route: input.route ?? null,
     source: input.source ?? null,
+    name: input.name ?? null,
     page: input.page ?? null,
     pageSize: input.pageSize ?? null,
     start: input.start ?? null,
@@ -65,6 +66,7 @@ export async function GET(request: Request) {
     const model = searchParams.get("model");
     const route = searchParams.get("route");
     const source = searchParams.get("source");
+    const name = searchParams.get("name");
     const pageParam = searchParams.get("page");
     const pageSizeParam = searchParams.get("pageSize");
     const start = searchParams.get("start");
@@ -74,7 +76,7 @@ export async function GET(request: Request) {
     const pageSize = pageSizeParam ? Number.parseInt(pageSizeParam, 10) : undefined;
     const skipCacheParam = searchParams.get("skipCache");
     const skipCache = skipCacheParam === "1" || skipCacheParam === "true";
-    const cacheKey = makeCacheKey({ days, model, route, source, page, pageSize, start, end });
+    const cacheKey = makeCacheKey({ days, model, route, source, name, page, pageSize, start, end });
     if (!skipCache) {
       const cached = getCached(cacheKey);
       if (cached) {
@@ -86,6 +88,7 @@ export async function GET(request: Request) {
       model: model || undefined,
       route: route || undefined,
       source: source || undefined,
+      name: name || undefined,
       page,
       pageSize,
       start,

app/api/sync/route.ts

@@ -3,7 +3,8 @@ import { cookies } from "next/headers";
 import { eq, sql } from "drizzle-orm";
 import { config, assertEnv } from "@/lib/config";
 import { db } from "@/lib/db/client";
-import { usageRecords } from "@/lib/db/schema";
+import { authFileMappings, usageRecords } from "@/lib/db/schema";
+import { toAuthFileMappings } from "@/lib/auth-files";
 import { parseUsagePayload, toUsageRecords } from "@/lib/usage";
 
 export const runtime = "nodejs";
@@ -47,6 +48,44 @@ async function isAuthorized(request: Request) {
   return false;
 }
 
+async function syncAuthFileMappings(pulledAt: Date) {
+  const authFilesUrl = `${config.cliproxy.baseUrl.replace(/\/$/, "")}/auth-files`;
+
+  const response = await fetch(authFilesUrl, {
+    headers: {
+      Authorization: `Bearer ${config.cliproxy.apiKey}`,
+      "Content-Type": "application/json"
+    },
+    cache: "no-store"
+  });
+
+  if (!response.ok) {
+    throw new Error(`Failed to fetch auth-files: ${response.status} ${response.statusText}`);
+  }
+
+  const json = await response.json();
+  const rows = toAuthFileMappings(json, pulledAt);
+  if (rows.length === 0) return 0;
+
+  await db
+    .insert(authFileMappings)
+    .values(rows)
+    .onConflictDoUpdate({
+      target: authFileMappings.authId,
+      set: {
+        name: sql`coalesce(nullif(excluded.name, ''), ${authFileMappings.name})`,
+        label: sql`coalesce(nullif(excluded.label, ''), ${authFileMappings.label})`,
+        provider: sql`coalesce(nullif(excluded.provider, ''), ${authFileMappings.provider})`,
+        source: sql`coalesce(nullif(excluded.source, ''), ${authFileMappings.source})`,
+        email: sql`coalesce(nullif(excluded.email, ''), ${authFileMappings.email})`,
+        updatedAt: sql`coalesce(excluded.updated_at, ${authFileMappings.updatedAt})`,
+        syncedAt: pulledAt
+      }
+    });
+
+  return rows.length;
+}
+
 async function performSync(request: Request) {
   if (!config.password && !config.cronSecret && !PASSWORD) return missingPassword();
   if (!(await isAuthorized(request))) return unauthorized();
@@ -89,8 +128,23 @@ async function performSync(request: Request) {
 
   const rows = toUsageRecords(payload, pulledAt);
 
+  let authFilesSynced = 0;
+  let authFilesWarning: string | undefined;
+  try {
+    authFilesSynced = await syncAuthFileMappings(pulledAt);
+  } catch (error) {
+    authFilesWarning = "auth-files sync failed";
+    console.warn("/api/sync auth-files sync failed:", error);
+  }
+
   if (rows.length === 0) {
-    return NextResponse.json({ status: "ok", inserted: 0, message: "No usage data" });
+    return NextResponse.json({
+      status: "ok",
+      inserted: 0,
+      message: "No usage data",
+      authFilesSynced,
+      ...(authFilesWarning ? { authFilesWarning } : {})
+    });
   }
 
   let insertedRows: Array<{ id: number }>;
@@ -119,7 +173,13 @@ async function performSync(request: Request) {
     inserted = Number(fallback?.[0]?.count ?? 0);
   }
 
-  return NextResponse.json({ status: "ok", inserted, attempted: rows.length });
+  return NextResponse.json({
+    status: "ok",
+    inserted,
+    attempted: rows.length,
+    authFilesSynced,
+    ...(authFilesWarning ? { authFilesWarning } : {})
+  });
 }
 
 export async function POST(request: Request) {

app/page.tsx

@@ -23,7 +23,14 @@ const PIE_COLORS = [
 ];
 
 type OverviewMeta = { page: number; pageSize: number; totalModels: number; totalPages: number };
-type OverviewAPIResponse = { overview: UsageOverview | null; empty: boolean; days: number; timezone?: string; meta?: OverviewMeta; filters?: { models: string[]; routes: string[]; sources: string[] } };
+type OverviewAPIResponse = {
+  overview: UsageOverview | null;
+  empty: boolean;
+  days: number;
+  timezone?: string;
+  meta?: OverviewMeta;
+  filters?: { models: string[]; routes: string[]; sources: string[]; names: string[] };
+};
 
 type PriceForm = {
   model: string;
@@ -185,12 +192,15 @@ export default function DashboardPage() {
   const [modelOptions, setModelOptions] = useState<string[]>([]);
   const [routeOptions, setRouteOptions] = useState<string[]>([]);
   const [sourceOptions, setSourceOptions] = useState<string[]>([]);
+  const [nameOptions, setNameOptions] = useState<string[]>([]);
   const [filterModelInput, setFilterModelInput] = useState("");
   const [filterRouteInput, setFilterRouteInput] = useState("");
   const [filterSourceInput, setFilterSourceInput] = useState("");
+  const [filterNameInput, setFilterNameInput] = useState("");
   const [filterModel, setFilterModel] = useState<string | undefined>(undefined);
   const [filterRoute, setFilterRoute] = useState<string | undefined>(undefined);
   const [filterSource, setFilterSource] = useState<string | undefined>(undefined);
+  const [filterName, setFilterName] = useState<string | undefined>(undefined);
   const [page, setPage] = useState(1);
   const [form, setForm] = useState<PriceForm>({ model: "", inputPricePer1M: "", cachedInputPricePer1M: "", outputPricePer1M: "" });
   const [status, setStatus] = useState<string | null>(null);
@@ -720,6 +730,7 @@ export default function DashboardPage() {
         if (filterModel) params.set("model", filterModel);
         if (filterRoute) params.set("route", filterRoute);
         if (filterSource) params.set("source", filterSource);
+        if (filterName) params.set("name", filterName);
         params.set("page", String(page));
         params.set("pageSize", "500");
 
@@ -747,6 +758,7 @@ export default function DashboardPage() {
         setModelOptions(Array.from(new Set(data.filters?.models ?? [])));
         setRouteOptions(Array.from(new Set(data.filters?.routes ?? [])));
         setSourceOptions(Array.from(new Set(data.filters?.sources ?? [])));
+        setNameOptions(Array.from(new Set(data.filters?.names ?? [])));
         setAppliedDays(data.days ?? rangeDays);
       } catch (err) {
         if (!active) return;
@@ -763,7 +775,7 @@ export default function DashboardPage() {
       active = false;
       controller.abort();
     };
-  }, [rangeMode, customStart, customEnd, rangeDays, filterModel, filterRoute, filterSource, page, refreshTrigger, ready]);
+  }, [rangeMode, customStart, customEnd, rangeDays, filterModel, filterRoute, filterSource, filterName, page, refreshTrigger, ready]);
 
   const overviewData = overview;
   const showEmpty = overviewEmpty || !overview;
@@ -882,6 +894,7 @@ export default function DashboardPage() {
     setFilterModel(filterModelInput.trim() || undefined);
     setFilterRoute(filterRouteInput.trim() || undefined);
     setFilterSource(filterSourceInput.trim() || undefined);
+    setFilterName(filterNameInput.trim() || undefined);
   };
 
   const applyModelOption = (val: string) => {
@@ -902,6 +915,12 @@ export default function DashboardPage() {
     setPage(1);
   };
 
+  const applyNameOption = (val: string) => {
+    setFilterNameInput(val);
+    setFilterName(val.trim() || undefined);
+    setPage(1);
+  };
+
   const handleSubmit = async (event: FormEvent<HTMLFormElement>) => {
     event.preventDefault();
     setStatus(null);
@@ -1269,21 +1288,36 @@ export default function DashboardPage() {
               setPage(1);
             }}
           />
+          <ComboBox
+            value={filterNameInput}
+            onChange={setFilterNameInput}
+            options={nameOptions}
+            placeholder="按凭证名过滤"
+            darkMode={darkMode}
+            onSelectOption={applyNameOption}
+            onClear={() => {
+              setFilterNameInput("");
+              setFilterName(undefined);
+              setPage(1);
+            }}
+          />
           <button
             onClick={applyFilters}
             className={`rounded-lg border px-3 py-1.5 text-sm font-semibold transition ${darkMode ? "border-slate-700 bg-slate-800 text-slate-300 hover:border-slate-500" : "border-slate-300 bg-white text-slate-700 hover:border-slate-400"}`}
           >
             应用筛选
           </button>
-          {(filterModel || filterRoute || filterSource) ? (
+          {(filterModel || filterRoute || filterSource || filterName) ? (
             <button
               onClick={() => {
                 setFilterModelInput("");
                 setFilterRouteInput("");
                 setFilterSourceInput("");
+                setFilterNameInput("");
                 setFilterModel(undefined);
                 setFilterRoute(undefined);
                 setFilterSource(undefined);
+                setFilterName(undefined);
                 setPage(1);
               }}
               className={`rounded-lg border px-3 py-1.5 text-sm transition ${darkMode ? "border-slate-700 bg-slate-800 text-slate-400 hover:border-slate-500" : "border-slate-300 bg-white text-slate-600 hover:border-slate-400"}`}

app/records/page.tsx

@@ -12,6 +12,7 @@ type UsageRecord = {
   occurredAt: string;
   route: string;
   source: string;
+  credentialName: string;
   model: string;
   totalTokens: number;
   inputTokens: number;
@@ -787,8 +788,8 @@ export default function RecordsPage() {
                     </div>
                   </td>
                   <td className="px-3 py-3 first:rounded-l-lg last:rounded-r-lg">
-                    <div className="max-w-[220px] truncate text-slate-300" title={hideRouteValue ? "-" : row.source || "-"}>
-                      {hideRouteValue ? "-" : row.source || "-"}
+                    <div className="max-w-[220px] truncate text-slate-300" title={hideRouteValue ? "-" : row.credentialName || "-"}>
+                      {hideRouteValue ? "-" : row.credentialName || "-"}
                     </div>
                   </td>
                   <td className="px-3 py-3 first:rounded-l-lg last:rounded-r-lg">

drizzle/0003_add_auth_index_to_usage_records.sql

@@ -0,0 +1,7 @@
+ALTER TABLE "usage_records"
+  ADD COLUMN "auth_index" text;
+
+-- 回填历史数据中的 auth_index（从 raw.detail.auth_index 提取）
+UPDATE "usage_records"
+SET "auth_index" = NULLIF(("raw"::jsonb -> 'detail' ->> 'auth_index'), '')
+WHERE "auth_index" IS NULL;

drizzle/0004_add_auth_file_mappings.sql

@@ -0,0 +1,13 @@
+CREATE TABLE "auth_file_mappings" (
+  "auth_id" text PRIMARY KEY NOT NULL,
+  "name" text NOT NULL DEFAULT '',
+  "label" text,
+  "provider" text,
+  "source" text,
+  "email" text,
+  "updated_at" timestamp with time zone,
+  "synced_at" timestamp with time zone NOT NULL DEFAULT now()
+);
+
+CREATE INDEX "auth_file_mappings_name_idx"
+  ON "auth_file_mappings" USING btree ("name");

drizzle/meta/_journal.json

@@ -22,6 +22,20 @@
       "when": 1771065600000,
       "tag": "0002_add_source_to_usage_records",
       "breakpoints": true
+    },
+    {
+      "idx": 3,
+      "version": "7",
+      "when": 1771069200000,
+      "tag": "0003_add_auth_index_to_usage_records",
+      "breakpoints": true
+    },
+    {
+      "idx": 4,
+      "version": "7",
+      "when": 1771072800000,
+      "tag": "0004_add_auth_file_mappings",
+      "breakpoints": true
     }
   ]
 }