Files
CC-Switch/src/main.tsx
T
Thefool 12b972a66e feat(usage): add automatic models.dev pricing sync (#5734)
* feat(usage): persist model pricing in local config

* feat(usage): sync selected models.dev pricing on startup

* fix(usage): address models.dev sync review feedback

* fix(usage): harden local pricing synchronization
2026-07-28 23:37:55 +08:00

153 lines
4.9 KiB
TypeScript

import React from "react";
import ReactDOM from "react-dom/client";
import App from "./App";
import { DatabaseUpgrade } from "./components/DatabaseUpgrade";
import { UpdateProvider } from "./contexts/UpdateContext";
import "./index.css";
// 导入国际化配置
import i18n from "./i18n";
import { QueryClientProvider } from "@tanstack/react-query";
import { ThemeProvider } from "@/components/theme-provider";
import { queryClient } from "@/lib/query";
import { Toaster } from "@/components/ui/sonner";
import { listen } from "@tauri-apps/api/event";
import { invoke } from "@tauri-apps/api/core";
import { message } from "@tauri-apps/plugin-dialog";
import { exit } from "@tauri-apps/plugin-process";
import { FrontendErrorBoundary } from "./components/FrontendErrorBoundary";
import {
installGlobalErrorHandlers,
reportFrontendError,
} from "./lib/frontendLogger";
import {
MODELS_DEV_SYNC_CONFIG_QUERY_KEY,
syncModelsDevPricingOnStartup,
} from "./lib/modelsDevAutoSync";
installGlobalErrorHandlers();
// 根据平台添加 body class,便于平台特定样式
try {
const ua = navigator.userAgent || "";
const plat = (navigator.platform || "").toLowerCase();
const isMac = /mac/i.test(ua) || plat.includes("mac");
if (isMac) {
document.body.classList.add("is-mac");
}
} catch {
// 忽略平台检测失败
}
// 配置加载错误payload类型
interface ConfigLoadErrorPayload {
path?: string;
error?: string;
/** "db_version_too_new" 表示数据库版本过新,渲染应用内升级恢复界面 */
kind?: string;
}
/**
* 处理配置加载失败:显示错误消息并强制退出应用
* 不给用户"取消"选项,因为配置损坏时应用无法正常运行
*/
async function handleConfigLoadError(
payload: ConfigLoadErrorPayload | null,
): Promise<void> {
const path = payload?.path ?? "~/.cc-switch/config.json";
const detail = payload?.error ?? "Unknown error";
await message(
i18n.t("errors.configLoadFailedMessage", {
path,
detail,
defaultValue:
"无法读取配置文件:\n{{path}}\n\n错误详情:\n{{detail}}\n\n请手动检查 JSON 是否有效,或从同目录的备份文件(如 config.json.bak)恢复。\n\n应用将退出以便您进行修复。",
}),
{
title: i18n.t("errors.configLoadFailedTitle", {
defaultValue: "配置加载失败",
}),
kind: "error",
},
);
await exit(1);
}
// 监听后端的配置加载错误事件:仅提醒用户并强制退出,不修改任何配置文件
try {
void listen("configLoadError", async (evt) => {
await handleConfigLoadError(evt.payload as ConfigLoadErrorPayload | null);
});
} catch (e) {
// 忽略事件订阅异常(例如在非 Tauri 环境下)
reportFrontendError("config_load_error_listener", e);
}
async function bootstrap() {
// 启动早期主动查询后端初始化错误,避免事件竞态
try {
const initError = (await invoke(
"get_init_error",
)) as ConfigLoadErrorPayload | null;
if (initError && initError.kind === "db_version_too_new") {
// 数据库版本过新:渲染应用内「升级应用」恢复界面,不进入正常 App
ReactDOM.createRoot(document.getElementById("root")!).render(
<React.StrictMode>
<FrontendErrorBoundary>
<ThemeProvider defaultTheme="system" storageKey="cc-switch-theme">
<DatabaseUpgrade payload={initError} />
<Toaster />
</ThemeProvider>
</FrontendErrorBoundary>
</React.StrictMode>,
);
return;
}
if (initError && (initError.path || initError.error)) {
await handleConfigLoadError(initError);
// 注意:不会执行到这里,因为 exit(1) 会终止进程
return;
}
} catch (e) {
// 忽略拉取错误,继续渲染
reportFrontendError("get_init_error", e);
}
ReactDOM.createRoot(document.getElementById("root")!).render(
<React.StrictMode>
<FrontendErrorBoundary>
<QueryClientProvider client={queryClient}>
<ThemeProvider defaultTheme="system" storageKey="cc-switch-theme">
<UpdateProvider>
<App />
<Toaster />
</UpdateProvider>
</ThemeProvider>
</QueryClientProvider>
</FrontendErrorBoundary>
</React.StrictMode>,
);
void syncModelsDevPricingOnStartup()
.then((result) => {
if (!result.skipped) {
return Promise.all([
queryClient.invalidateQueries({ queryKey: ["usage"] }),
queryClient.invalidateQueries({
queryKey: MODELS_DEV_SYNC_CONFIG_QUERY_KEY,
}),
]);
}
})
.catch((error) => {
// 离线或 models.dev 暂时不可用不应阻塞应用启动。
reportFrontendError("models_dev_startup_sync", error);
void queryClient.invalidateQueries({
queryKey: MODELS_DEV_SYNC_CONFIG_QUERY_KEY,
});
});
}
void bootstrap();