Merge upstream/main into main

Resolved conflicts in src-tauri/src/lib.rs: - Kept both: open_provider_terminal (my feature) and universal provider commands (upstream) 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
2026-07-29 01:25:33 +08:00 · 2025-12-29 09:14:42 +08:00
parent 3dcbe313be 7fdaeacb5b
commit 390839a8d5
84 changed files with 6164 additions and 1498 deletions
@@ -1,10 +1,11 @@
 export type { AppId } from "./types";
-export { providersApi } from "./providers";
+export { providersApi, universalProvidersApi } from "./providers";
 export { settingsApi } from "./settings";
 export { mcpApi } from "./mcp";
 export { promptsApi } from "./prompts";
 export { usageApi } from "./usage";
 export { vscodeApi } from "./vscode";
+export { proxyApi } from "./proxy";
 export * as configApi from "./config";
 export type { ProviderSwitchEvent } from "./providers";
 export type { Prompt } from "./prompts";
@@ -1,6 +1,10 @@
 import { invoke } from "@tauri-apps/api/core";
 import { listen, type UnlistenFn } from "@tauri-apps/api/event";
-import type { Provider } from "@/types";
+import type {
+  Provider,
+  UniversalProvider,
+  UniversalProvidersMap,
+} from "@/types";
 import type { AppId } from "./types";

 export interface ProviderSortUpdate {
@@ -71,3 +75,44 @@ export const providersApi = {
    return await invoke("open_provider_terminal", { providerId, app: appId });
  },
 };
+
+// ============================================================================
+// 统一供应商（Universal Provider）API
+// ============================================================================
+
+export const universalProvidersApi = {
+  /**
+   * 获取所有统一供应商
+   */
+  async getAll(): Promise<UniversalProvidersMap> {
+    return await invoke("get_universal_providers");
+  },
+
+  /**
+   * 获取单个统一供应商
+   */
+  async get(id: string): Promise<UniversalProvider | null> {
+    return await invoke("get_universal_provider", { id });
+  },
+
+  /**
+   * 添加或更新统一供应商
+   */
+  async upsert(provider: UniversalProvider): Promise<boolean> {
+    return await invoke("upsert_universal_provider", { provider });
+  },
+
+  /**
+   * 删除统一供应商
+   */
+  async delete(id: string): Promise<boolean> {
+    return await invoke("delete_universal_provider", { id });
+  },
+
+  /**
+   * 手动同步统一供应商到各应用
+   */
+  async sync(id: string): Promise<boolean> {
+    return await invoke("sync_universal_provider", { id });
+  },
+};
@@ -0,0 +1,95 @@
+import { invoke } from "@tauri-apps/api/core";
+import type {
+  ProxyConfig,
+  ProxyStatus,
+  ProxyServerInfo,
+  ProxyTakeoverStatus,
+  GlobalProxyConfig,
+  AppProxyConfig,
+} from "@/types/proxy";
+
+export const proxyApi = {
+  // ========== 代理服务器控制 API ==========
+
+  // 启动代理服务器
+  async startProxyServer(): Promise<ProxyServerInfo> {
+    return invoke("start_proxy_server");
+  },
+
+  // 停止代理服务器并恢复配置
+  async stopProxyWithRestore(): Promise<void> {
+    return invoke("stop_proxy_with_restore");
+  },
+
+  // 获取代理服务器状态
+  async getProxyStatus(): Promise<ProxyStatus> {
+    return invoke("get_proxy_status");
+  },
+
+  // 检查代理服务器是否正在运行
+  async isProxyRunning(): Promise<boolean> {
+    return invoke("is_proxy_running");
+  },
+
+  // 检查是否处于接管模式
+  async isLiveTakeoverActive(): Promise<boolean> {
+    return invoke("is_live_takeover_active");
+  },
+
+  // 代理模式下切换供应商
+  async switchProxyProvider(
+    appType: string,
+    providerId: string,
+  ): Promise<void> {
+    return invoke("switch_proxy_provider", { appType, providerId });
+  },
+
+  // ========== 接管状态 API ==========
+
+  // 获取各应用接管状态
+  async getProxyTakeoverStatus(): Promise<ProxyTakeoverStatus> {
+    return invoke("get_proxy_takeover_status");
+  },
+
+  // 为指定应用开启/关闭接管
+  async setProxyTakeoverForApp(
+    appType: string,
+    enabled: boolean,
+  ): Promise<void> {
+    return invoke("set_proxy_takeover_for_app", { appType, enabled });
+  },
+
+  // ========== Legacy 代理配置 API (兼容) ==========
+
+  // 获取代理配置（旧版 v2 兼容接口）
+  async getProxyConfig(): Promise<ProxyConfig> {
+    return invoke("get_proxy_config");
+  },
+
+  // 更新代理配置（旧版 v2 兼容接口）
+  async updateProxyConfig(config: ProxyConfig): Promise<void> {
+    return invoke("update_proxy_config", { config });
+  },
+
+  // ========== v3+ 全局/应用级配置 API ==========
+
+  // 获取全局代理配置
+  async getGlobalProxyConfig(): Promise<GlobalProxyConfig> {
+    return invoke("get_global_proxy_config");
+  },
+
+  // 更新全局代理配置
+  async updateGlobalProxyConfig(config: GlobalProxyConfig): Promise<void> {
+    return invoke("update_global_proxy_config", { config });
+  },
+
+  // 获取指定应用的代理配置
+  async getProxyConfigForApp(appType: string): Promise<AppProxyConfig> {
+    return invoke("get_proxy_config_for_app", { appType });
+  },
+
+  // 更新指定应用的代理配置
+  async updateProxyConfigForApp(config: AppProxyConfig): Promise<void> {
+    return invoke("update_proxy_config_for_app", { config });
+  },
+};
@@ -1,3 +1,4 @@
 export * from "./queryClient";
 export * from "./queries";
 export * from "./mutations";
+export * from "./proxy";
@@ -0,0 +1,239 @@
+import { useQuery, useMutation, useQueryClient } from "@tanstack/react-query";
+import { proxyApi } from "@/lib/api/proxy";
+import { toast } from "sonner";
+import { useTranslation } from "react-i18next";
+import type { GlobalProxyConfig, AppProxyConfig } from "@/types/proxy";
+
+// ========== 代理服务器状态 Hooks ==========
+
+/**
+ * 获取代理服务器状态
+ */
+export function useProxyStatus() {
+  return useQuery({
+    queryKey: ["proxyStatus"],
+    queryFn: () => proxyApi.getProxyStatus(),
+    refetchInterval: 5000, // 每 5 秒刷新一次
+  });
+}
+
+/**
+ * 检查代理服务器是否运行
+ */
+export function useIsProxyRunning() {
+  return useQuery({
+    queryKey: ["proxyRunning"],
+    queryFn: () => proxyApi.isProxyRunning(),
+    refetchInterval: 2000,
+  });
+}
+
+/**
+ * 检查是否处于接管模式
+ */
+export function useIsLiveTakeoverActive() {
+  return useQuery({
+    queryKey: ["liveTakeoverActive"],
+    queryFn: () => proxyApi.isLiveTakeoverActive(),
+    refetchInterval: 2000,
+  });
+}
+
+/**
+ * 获取各应用接管状态
+ */
+export function useProxyTakeoverStatus() {
+  return useQuery({
+    queryKey: ["proxyTakeoverStatus"],
+    queryFn: () => proxyApi.getProxyTakeoverStatus(),
+    refetchInterval: 2000,
+  });
+}
+
+// ========== 代理服务器控制 Hooks ==========
+
+/**
+ * 启动代理服务器
+ */
+export function useStartProxyServer() {
+  const queryClient = useQueryClient();
+
+  return useMutation({
+    mutationFn: () => proxyApi.startProxyServer(),
+    onSuccess: () => {
+      queryClient.invalidateQueries({ queryKey: ["proxyStatus"] });
+      queryClient.invalidateQueries({ queryKey: ["proxyRunning"] });
+      queryClient.invalidateQueries({ queryKey: ["liveTakeoverActive"] });
+      queryClient.invalidateQueries({ queryKey: ["proxyTakeoverStatus"] });
+    },
+  });
+}
+
+/**
+ * 停止代理服务器
+ */
+export function useStopProxyServer() {
+  const queryClient = useQueryClient();
+
+  return useMutation({
+    mutationFn: () => proxyApi.stopProxyWithRestore(),
+    onSuccess: () => {
+      queryClient.invalidateQueries({ queryKey: ["proxyStatus"] });
+      queryClient.invalidateQueries({ queryKey: ["proxyRunning"] });
+      queryClient.invalidateQueries({ queryKey: ["liveTakeoverActive"] });
+      queryClient.invalidateQueries({ queryKey: ["proxyTakeoverStatus"] });
+    },
+  });
+}
+
+/**
+ * 设置应用接管状态
+ */
+export function useSetProxyTakeoverForApp() {
+  const queryClient = useQueryClient();
+
+  return useMutation({
+    mutationFn: ({ appType, enabled }: { appType: string; enabled: boolean }) =>
+      proxyApi.setProxyTakeoverForApp(appType, enabled),
+    onSuccess: () => {
+      queryClient.invalidateQueries({ queryKey: ["proxyTakeoverStatus"] });
+      queryClient.invalidateQueries({ queryKey: ["liveTakeoverActive"] });
+    },
+  });
+}
+
+/**
+ * 代理模式下切换供应商
+ */
+export function useSwitchProxyProvider() {
+  const queryClient = useQueryClient();
+  const { t } = useTranslation();
+
+  return useMutation({
+    mutationFn: ({
+      appType,
+      providerId,
+    }: {
+      appType: string;
+      providerId: string;
+    }) => proxyApi.switchProxyProvider(appType, providerId),
+    onSuccess: (_, variables) => {
+      queryClient.invalidateQueries({ queryKey: ["proxyStatus"] });
+      queryClient.invalidateQueries({
+        queryKey: ["providers", variables.appType],
+      });
+    },
+    onError: (error: Error) => {
+      toast.error(t("proxy.switchFailed", { error: error.message }));
+    },
+  });
+}
+
+// ========== Legacy 代理配置 Hooks (兼容) ==========
+
+/**
+ * 获取代理配置（旧版）
+ */
+export function useProxyConfig() {
+  const queryClient = useQueryClient();
+  const { t } = useTranslation();
+
+  const { data: config, isLoading } = useQuery({
+    queryKey: ["proxyConfig"],
+    queryFn: () => proxyApi.getProxyConfig(),
+  });
+
+  const updateMutation = useMutation({
+    mutationFn: proxyApi.updateProxyConfig,
+    onSuccess: () => {
+      toast.success(t("proxy.settings.toast.saved"), { closeButton: true });
+      queryClient.invalidateQueries({ queryKey: ["proxyConfig"] });
+      queryClient.invalidateQueries({ queryKey: ["proxyStatus"] });
+    },
+    onError: (error: Error) => {
+      toast.error(
+        t("proxy.settings.toast.saveFailed", { error: error.message }),
+      );
+    },
+  });
+
+  return {
+    config,
+    isLoading,
+    updateConfig: updateMutation.mutateAsync,
+    isUpdating: updateMutation.isPending,
+  };
+}
+
+// ========== v3+ 全局/应用级配置 Hooks ==========
+
+/**
+ * 获取全局代理配置
+ */
+export function useGlobalProxyConfig() {
+  return useQuery({
+    queryKey: ["globalProxyConfig"],
+    queryFn: () => proxyApi.getGlobalProxyConfig(),
+  });
+}
+
+/**
+ * 更新全局代理配置
+ */
+export function useUpdateGlobalProxyConfig() {
+  const queryClient = useQueryClient();
+  const { t } = useTranslation();
+
+  return useMutation({
+    mutationFn: (config: GlobalProxyConfig) =>
+      proxyApi.updateGlobalProxyConfig(config),
+    onSuccess: () => {
+      toast.success(t("proxy.settings.toast.saved"), { closeButton: true });
+      queryClient.invalidateQueries({ queryKey: ["globalProxyConfig"] });
+      queryClient.invalidateQueries({ queryKey: ["proxyConfig"] });
+      queryClient.invalidateQueries({ queryKey: ["proxyStatus"] });
+    },
+    onError: (error: Error) => {
+      toast.error(
+        t("proxy.settings.toast.saveFailed", { error: error.message }),
+      );
+    },
+  });
+}
+
+/**
+ * 获取指定应用的代理配置
+ */
+export function useAppProxyConfig(appType: string) {
+  return useQuery({
+    queryKey: ["appProxyConfig", appType],
+    queryFn: () => proxyApi.getProxyConfigForApp(appType),
+    enabled: !!appType,
+  });
+}
+
+/**
+ * 更新指定应用的代理配置
+ */
+export function useUpdateAppProxyConfig() {
+  const queryClient = useQueryClient();
+  const { t } = useTranslation();
+
+  return useMutation({
+    mutationFn: (config: AppProxyConfig) =>
+      proxyApi.updateProxyConfigForApp(config),
+    onSuccess: (_, variables) => {
+      toast.success(t("proxy.settings.toast.saved"), { closeButton: true });
+      queryClient.invalidateQueries({
+        queryKey: ["appProxyConfig", variables.appType],
+      });
+      queryClient.invalidateQueries({ queryKey: ["proxyConfig"] });
+      queryClient.invalidateQueries({ queryKey: ["circuitBreakerConfig"] });
+    },
+    onError: (error: Error) => {
+      toast.error(
+        t("proxy.settings.toast.saveFailed", { error: error.message }),
+      );
+    },
+  });
+}