From 938a180affb56edf76f37a33f337eeb2d85234ca Mon Sep 17 00:00:00 2001
From: jZonG <jzongcode@gmail.com>
Date: Tue, 27 May 2025 17:15:55 +0800
Subject: [PATCH] MCP server create & update

---
 .../[appId]/overview/cardView.tsx             |  1 -
 .../components/tools/mcp/mcp-service-card.tsx | 41 ++++++++++++-------
 web/service/use-tools.ts                      |  9 ++++
 3 files changed, 36 insertions(+), 15 deletions(-)

diff --git a/web/app/(commonLayout)/app/(appDetailLayout)/[appId]/overview/cardView.tsx b/web/app/(commonLayout)/app/(appDetailLayout)/[appId]/overview/cardView.tsx
index 98e65235f7..05799bcac5 100644
--- a/web/app/(commonLayout)/app/(appDetailLayout)/[appId]/overview/cardView.tsx
+++ b/web/app/(commonLayout)/app/(appDetailLayout)/[appId]/overview/cardView.tsx
@@ -141,7 +141,6 @@ const CardView: FC<ICardViewProps> = ({ appId, isInPanel, className }) => {
       {isInPanel && appDetail.mode === 'workflow' && (
         <MCPServiceCard
           appInfo={appDetail}
-          onGenerateCode={onGenerateCode}
         />
       )}
     </div>
diff --git a/web/app/components/tools/mcp/mcp-service-card.tsx b/web/app/components/tools/mcp/mcp-service-card.tsx
index 96afc7cc73..66709d699f 100644
--- a/web/app/components/tools/mcp/mcp-service-card.tsx
+++ b/web/app/components/tools/mcp/mcp-service-card.tsx
@@ -9,7 +9,6 @@ import {
 } from '@/app/components/base/icons/src/vender/other'
 import Button from '@/app/components/base/button'
 import Tooltip from '@/app/components/base/tooltip'
-import { asyncRunSafe } from '@/utils'
 import Switch from '@/app/components/base/switch'
 import Divider from '@/app/components/base/divider'
 import CopyFeedback from '@/app/components/base/copy-feedback'
@@ -21,23 +20,26 @@ import Indicator from '@/app/components/header/indicator'
 import MCPServerModal from '@/app/components/tools/mcp/mcp-server-modal'
 import { useAppWorkflow } from '@/service/use-workflow'
 import {
+  useInvalidateMCPServerDetail,
   useMCPServerDetail,
+  useRefreshMCPServerCode,
+  useUpdateMCPServer,
 } from '@/service/use-tools'
 import { BlockEnum } from '@/app/components/workflow/types'
 import cn from '@/utils/classnames'
 
 export type IAppCardProps = {
   appInfo: AppDetailResponse & Partial<AppSSO>
-  onGenerateCode?: () => Promise<void>
 }
 
 function MCPServiceCard({
   appInfo,
-  onGenerateCode,
 }: IAppCardProps) {
   const { t } = useTranslation()
+  const { mutateAsync: updateMCPServer } = useUpdateMCPServer()
+  const { mutateAsync: refreshMCPServerCode, isPending: genLoading } = useRefreshMCPServerCode()
+  const invalidateMCPServerDetail = useInvalidateMCPServerDetail()
   const { isCurrentWorkspaceManager, isCurrentWorkspaceEditor } = useAppContext()
-  const [genLoading, setGenLoading] = useState(false)
   const [showConfirmDelete, setShowConfirmDelete] = useState(false)
   const [showMCPServerModal, setShowMCPServerModal] = useState(false)
 
@@ -61,23 +63,34 @@ function MCPServiceCard({
   }, [currentWorkflow])
 
   const onGenCode = async () => {
-    if (onGenerateCode) {
-      setGenLoading(true)
-      await asyncRunSafe(onGenerateCode())
-      setGenLoading(false)
-    }
+    await refreshMCPServerCode(detail?.id || '')
+    invalidateMCPServerDetail(appInfo.id)
   }
 
   const onChangeStatus = async (state: boolean) => {
+    setActivated(state)
     if (state) {
-      if (!serverPublished) {
-        setActivated(true)
+      if (!serverPublished)
         setShowMCPServerModal(true)
-      }
-      // TODO handle server activation
+
+      await updateMCPServer({
+        appID: appInfo.id,
+        id: id || '',
+        description: detail?.description || '',
+        parameters: detail?.parameters || {},
+        status: 'active',
+      })
+      invalidateMCPServerDetail(appInfo.id)
     }
     else {
-      // TODO handle server activation
+      await updateMCPServer({
+        appID: appInfo.id,
+        id: id || '',
+        description: detail?.description || '',
+        parameters: detail?.parameters || {},
+        status: 'inactive',
+      })
+      invalidateMCPServerDetail(appInfo.id)
     }
   }
 
diff --git a/web/service/use-tools.ts b/web/service/use-tools.ts
index 9c5363e23e..917fe5ae99 100644
--- a/web/service/use-tools.ts
+++ b/web/service/use-tools.ts
@@ -231,6 +231,15 @@ export const useUpdateMCPServer = () => {
   })
 }
 
+export const useRefreshMCPServerCode = () => {
+  return useMutation({
+    mutationKey: [NAME_SPACE, 'refresh-mcp-server-code'],
+    mutationFn: (appID: string) => {
+      return get<MCPServerDetail>(`apps/${appID}/server/refresh`)
+    },
+  })
+}
+
 export const useBuiltinProviderInfo = (providerName: string) => {
   return useQuery({
     queryKey: [NAME_SPACE, 'builtin-provider-info', providerName],