feat: support settings

2026-05-23 08:25:57 +00:00 · 2026-01-20 23:43:21 +08:00
parent 3191a3845f
commit 10d253f461
25 changed files with 1355 additions and 217 deletions
@@ -0,0 +1,44 @@
+import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
+
+import { loadMCPConfig, updateMCPConfig } from "./api";
+
+export function useMCPConfig() {
+  const { data, isLoading, error } = useQuery({
+    queryKey: ["mcpConfig"],
+    queryFn: () => loadMCPConfig(),
+  });
+  return { config: data, isLoading, error };
+}
+
+export function useEnableMCPServer() {
+  const queryClient = useQueryClient();
+  const { config } = useMCPConfig();
+  return useMutation({
+    mutationFn: async ({
+      serverName,
+      enabled,
+    }: {
+      serverName: string;
+      enabled: boolean;
+    }) => {
+      if (!config) {
+        throw new Error("MCP config not found");
+      }
+      if (!config.mcp_servers[serverName]) {
+        throw new Error(`MCP server ${serverName} not found`);
+      }
+      await updateMCPConfig({
+        mcp_servers: {
+          ...config.mcp_servers,
+          [serverName]: {
+            ...config.mcp_servers[serverName],
+            enabled,
+          },
+        },
+      });
+    },
+    onSuccess: () => {
+      void queryClient.invalidateQueries({ queryKey: ["mcpConfig"] });
+    },
+  });
+}