fix(check-server): invalidate workload list with the group and refactor (#1190)

* fix(check-server): invalidate workload list with the group

* chore: move stopping to handle status from the api

* refactor: remove type assertions that are not needed anymore

Author: samuv

renderer/src/common/hooks/use-check-server-status.tsx

@@ -50,7 +50,9 @@ export function useCheckServerStatus() {
 
       if (isServerReady) {
         await queryClient.invalidateQueries({
-          queryKey: getApiV1BetaWorkloadsQueryKey({ query: { all: true } }),
+          queryKey: getApiV1BetaWorkloadsQueryKey({
+            query: { all: true, group: groupName },
+          }),
         })
 
         // Also invalidate the specific server query to ensure form updates

renderer/src/features/mcp-servers/components/actions-mcp-server.tsx

@@ -7,10 +7,10 @@ function getStatusText(status: CoreWorkload['status']) {
   if (status === 'running') return 'Running'
   if (status === 'starting') return 'Starting'
   if (status === 'stopped') return 'Stopped'
+  if (status === 'stopping') return 'Stopping'
   if (status === 'error') return 'Error'
   // add it for UI purposes, the BE cannot handle it for mvp
   if (status === 'updating') return 'Updating'
-  if (status === 'stopping') return 'Stopping'
   if (status === 'deleting') return 'Deleting'
   if (status === 'restarting') return 'Restarting'
   return 'Unknown'

renderer/src/features/mcp-servers/hooks/use-mutation-restart-server.ts

@@ -119,7 +119,7 @@ export function useMutationRestartServerAtStartup() {
           return statusResponses.map((response, index) => ({
             name: names[index],
             status: response.status || 'unknown',
-          })) as CoreWorkload[]
+          }))
         },
         serverNames,
         'running'

renderer/src/features/mcp-servers/hooks/use-mutation-update-workload.ts

@@ -16,7 +16,8 @@ export const useMutationUpdateWorkload = () => {
         query: { all: true, group: body.group },
       })
       await queryClient.cancelQueries({ queryKey })
-      const previousServersList = queryClient.getQueryData(queryKey)
+      const previousServersList =
+        queryClient.getQueryData<V1WorkloadListResponse>(queryKey)
 
       queryClient.setQueryData(queryKey, (old: V1WorkloadListResponse) => {
         const newWorkloads = old?.workloads?.map((server) =>
@@ -29,15 +30,16 @@ export const useMutationUpdateWorkload = () => {
       })
       return { queryKey, previousServersList }
     },
-    onError: (_error, _variables, context) => {
-      if (context?.queryKey && context?.previousServersList) {
-        queryClient.setQueryData(context?.queryKey, context.previousServersList)
+    onError: (_error, _variables, onMutateResult) => {
+      if (onMutateResult?.queryKey && onMutateResult?.previousServersList) {
+        queryClient.setQueryData(
+          onMutateResult?.queryKey,
+          onMutateResult.previousServersList
+        )
       }
     },
-    onSettled: async (_data, _error, variables, cachedResult) => {
-      const workloads =
-        (cachedResult?.previousServersList as V1WorkloadListResponse)
-          ?.workloads ?? []
+    onSettled: async (_data, _error, variables, onMutateResult) => {
+      const workloads = onMutateResult?.previousServersList?.workloads ?? []
       // Only refetch workloads without cached data, as servers may be temporarily unavailable during restart.
       if (
         !workloads.some((workload) => workload.name === variables.path.name)