fix(workflow): retry unavailable resource lock data

Author: Bill Leoutsakos

apps/sim/app/workspace/[workspaceId]/home/components/mothership-view/components/resource-content/resource-content.tsx

@@ -7,6 +7,7 @@ import {
   FileX,
   Folder as FolderIcon,
   Library,
+  RefreshCw,
   Square,
   SquareArrowUpRight,
   Workflow as WorkflowIcon,
@@ -362,8 +363,16 @@ export function EmbeddedWorkflowActions({ workspaceId, workflowId }: EmbeddedWor
   const { data: session } = useSession()
   const hostContext = useWorkspaceHostContext()
   const { userPermissions: effectivePermissions } = useWorkspacePermissionsContext()
-  const { data: workflowMap, isLoading: isWorkflowMapLoading } = useWorkflowMap(workspaceId)
-  const { data: folderMap, isLoading: isFolderMapLoading } = useFolderMap(workspaceId)
+  const {
+    data: workflowMap,
+    isLoading: isWorkflowMapLoading,
+    refetch: refetchWorkflowMap,
+  } = useWorkflowMap(workspaceId)
+  const {
+    data: folderMap,
+    isLoading: isFolderMapLoading,
+    refetch: refetchFolderMap,
+  } = useFolderMap(workspaceId)
   const setActiveWorkflow = useWorkflowRegistry((state) => state.setActiveWorkflow)
   const { handleRunWorkflow, handleCancelExecution } = useWorkflowExecution()
   const isExecuting = useExecutionStore(
@@ -384,8 +393,14 @@ export function EmbeddedWorkflowActions({ workspaceId, workflowId }: EmbeddedWor
     !isExecuting &&
     (isUsageGateLoading || (!effectivePermissions.canRead && !effectivePermissions.isLoading))
   const isWorkflowLockDataLoading = isWorkflowMapLoading || isFolderMapLoading
+  const isWorkflowLockDataUnavailable = !workflowMap || !folderMap
   const isWorkflowLocked = isWorkflowEffectivelyLocked(workflowMap?.[workflowId], folderMap ?? {})
 
+  const handleRetryWorkflowLockData = () => {
+    if (!workflowMap) void refetchWorkflowMap()
+    if (!folderMap) void refetchFolderMap()
+  }
+
   const handleRun = async () => {
     setActiveWorkflow(workflowId)
 
@@ -456,14 +471,34 @@ export function EmbeddedWorkflowActions({ workspaceId, workflowId }: EmbeddedWor
           <p>{isExecuting ? 'Stop' : 'Run workflow'}</p>
         </Tooltip.Content>
       </Tooltip.Root>
-      <Deploy
-        activeWorkflowId={workflowId}
-        userPermissions={effectivePermissions}
-        className={RESOURCE_TAB_ICON_BUTTON_CLASS}
-        compact
-        disabled={isWorkflowLockDataLoading || isWorkflowLocked}
-        disabledTooltip={isWorkflowLockDataLoading ? 'Loading workflow lock status...' : undefined}
-      />
+      {isWorkflowLockDataUnavailable && !isWorkflowLockDataLoading ? (
+        <Tooltip.Root>
+          <Tooltip.Trigger asChild>
+            <Button
+              variant='subtle'
+              onClick={handleRetryWorkflowLockData}
+              className={RESOURCE_TAB_ICON_BUTTON_CLASS}
+              aria-label='Retry loading workflow lock status'
+            >
+              <RefreshCw className={RESOURCE_TAB_ICON_CLASS} />
+            </Button>
+          </Tooltip.Trigger>
+          <Tooltip.Content side='bottom'>
+            <p>Retry loading workflow lock status</p>
+          </Tooltip.Content>
+        </Tooltip.Root>
+      ) : (
+        <Deploy
+          activeWorkflowId={workflowId}
+          userPermissions={effectivePermissions}
+          className={RESOURCE_TAB_ICON_BUTTON_CLASS}
+          compact
+          disabled={isWorkflowLockDataLoading || isWorkflowLocked}
+          disabledTooltip={
+            isWorkflowLockDataLoading ? 'Loading workflow lock status...' : undefined
+          }
+        />
+      )}
     </>
   )
 }