KimiSwitch Dev
ace9842556
- 账单/用量查询：balance:kimi（Moonshot 开放平台余额）、NewAPI 中转站余额（accessToken+userId）、Kimi/智谱/MiniMax 套餐档位（5小时/7天） - managed:kimi-code 走 Kimi Code OAuth 登录凭据查询订阅用量，access token 过期自动 refresh 并写回凭据（单 flight 锁 + 写回前重读防 CLI 竞态） - 全屏「配置用量查询」页面：启用开关 / 自动检测 vs NewAPI 模板 / 超时与自动查询间隔 / 测试查询 - 供应商卡片紧凑用量行（切换使用左侧）+ 多档明细行；managed 供应商列表置顶（仅展示层） - 用量显示中英文适配（five_hour→5小时、weekly_limit→7天，Rust 错误串本地化映射） - 修复：usage_kinds serde rename 导致账单入口不显示；测试查询因 save 触发 loading 翻页卸载面板导致结果丢失（silent save）
93 lines
2.8 KiB
TypeScript
93 lines
2.8 KiB
TypeScript
import { useCallback, useEffect, useRef, useState } from "react";
|
|
import { invoke } from "@tauri-apps/api/core";
|
|
import { useTranslation } from "../i18n";
|
|
import type { Agent, Config } from "../types";
|
|
|
|
interface UseConfigReturn {
|
|
config: Config | null;
|
|
dirty: boolean;
|
|
error: string | null;
|
|
loading: boolean;
|
|
refresh: (opts?: { silent?: boolean }) => Promise<void>;
|
|
save: (opts?: { silent?: boolean }) => Promise<void>;
|
|
updateConfig: (updater: (config: Config) => Config) => void;
|
|
}
|
|
|
|
export function useConfig(agent: Agent): UseConfigReturn {
|
|
const { t } = useTranslation();
|
|
const [config, setConfig] = useState<Config | null>(null);
|
|
const configRef = useRef(config);
|
|
const [dirty, setDirty] = useState(false);
|
|
const [error, setError] = useState<string | null>(null);
|
|
const [loading, setLoading] = useState(false);
|
|
|
|
useEffect(() => {
|
|
configRef.current = config;
|
|
}, [config]);
|
|
|
|
const refresh = useCallback(async (opts?: { silent?: boolean }) => {
|
|
// silent: skip the loading flip — App returns a bare loading screen while
|
|
// `loading` is true, which would unmount any open panel (and swallow its
|
|
// in-flight async results).
|
|
if (!opts?.silent) setLoading(true);
|
|
setError(null);
|
|
try {
|
|
const loadedConfig = await invoke<Config>("load_agent_config_command", {
|
|
agent,
|
|
});
|
|
setConfig(loadedConfig);
|
|
setDirty(false);
|
|
} catch (err) {
|
|
const message = err instanceof Error ? err.message : String(err);
|
|
setError(message);
|
|
console.error("useConfig refresh failed:", message);
|
|
} finally {
|
|
if (!opts?.silent) setLoading(false);
|
|
}
|
|
}, [agent]);
|
|
|
|
useEffect(() => {
|
|
refresh();
|
|
}, [refresh]);
|
|
|
|
const updateConfig = useCallback((updater: (config: Config) => Config) => {
|
|
// Compute next state synchronously from the ref (always current) and set
|
|
// the ref BEFORE setConfig. Previously the ref was updated inside the
|
|
// setConfig updater, which React 18 defers — so save() reading the ref
|
|
// immediately after would get a stale value.
|
|
const prev = configRef.current;
|
|
if (prev) {
|
|
const next = updater(prev);
|
|
configRef.current = next;
|
|
setConfig(next);
|
|
}
|
|
setDirty(true);
|
|
}, []);
|
|
|
|
const save = useCallback(async (opts?: { silent?: boolean }) => {
|
|
const current = configRef.current;
|
|
if (!current) return;
|
|
if (!opts?.silent) setLoading(true);
|
|
setError(null);
|
|
try {
|
|
await invoke("save_agent_config_command", { agent, config: current });
|
|
await refresh({ silent: opts?.silent });
|
|
} catch (err) {
|
|
const raw = err instanceof Error ? err.message : String(err);
|
|
setError(raw);
|
|
} finally {
|
|
if (!opts?.silent) setLoading(false);
|
|
}
|
|
}, [refresh, t, agent]);
|
|
|
|
return {
|
|
config,
|
|
dirty,
|
|
error,
|
|
loading,
|
|
refresh,
|
|
save,
|
|
updateConfig,
|
|
};
|
|
}
|