feat(host-service): tRPC query timeout + retry, fix hung file tree and git status

apps/desktop/src/renderer/hooks/host-service/useFileTree/useFileTree.ts

@@ -40,6 +40,43 @@ interface FileTreeState {
 	loadingDirectories: Set<string>;
 }
 
+interface LoadDirectoryOptions {
+	force?: boolean;
+}
+
+function applyDirectoryEntries(
+	current: FileTreeState,
+	absolutePath: string,
+	entries: FsEntry[],
+): FileTreeState {
+	const nextEntries = new Map(current.entriesByPath);
+	const nextChildren = new Map(current.childPathsByDirectory);
+	const nextLoaded = new Set(current.loadedDirectories);
+	const nextInvalidated = new Set(current.invalidatedDirectories);
+	const nextLoading = new Set(current.loadingDirectories);
+	nextLoading.delete(absolutePath);
+	nextLoaded.add(absolutePath);
+	nextInvalidated.delete(absolutePath);
+
+	for (const entry of entries) {
+		nextEntries.set(entry.absolutePath, entry);
+	}
+
+	nextChildren.set(
+		absolutePath,
+		entries.map((entry) => entry.absolutePath),
+	);
+
+	return {
+		...current,
+		childPathsByDirectory: nextChildren,
+		entriesByPath: nextEntries,
+		invalidatedDirectories: nextInvalidated,
+		loadedDirectories: nextLoaded,
+		loadingDirectories: nextLoading,
+	};
+}
+
 function createInitialState(): FileTreeState {
 	return {
 		childPathsByDirectory: new Map<string, string[]>(),
@@ -187,16 +224,15 @@ export function useFileTree({
 	);
 
 	const loadDirectory = useCallback(
-		async (absolutePath: string, force = false): Promise<void> => {
-			if (!workspaceId || !absolutePath) {
-				return;
-			}
+		async (
+			absolutePath: string,
+			options: LoadDirectoryOptions = {},
+		): Promise<void> => {
+			const { force = false } = options;
+			if (!workspaceId || !absolutePath) return;
 
 			const currentState = stateRef.current;
-			if (currentState.loadingDirectories.has(absolutePath)) {
-				return;
-			}
-
+			if (currentState.loadingDirectories.has(absolutePath)) return;
 			if (
 				!force &&
 				currentState.loadedDirectories.has(absolutePath) &&
@@ -205,65 +241,38 @@ export function useFileTree({
 				return;
 			}
 
-			updateState((current) => {
-				const nextLoading = new Set(current.loadingDirectories);
-				nextLoading.add(absolutePath);
-				return {
-					...current,
-					loadingDirectories: nextLoading,
-				};
-			});
+			const input = { workspaceId, absolutePath };
+			const cachedResult = utils.filesystem.listDirectory.getData(input);
+			if (cachedResult) {
+				updateState((current) =>
+					applyDirectoryEntries(current, absolutePath, cachedResult.entries),
+				);
+				if (!force) return;
+			}
 
-			try {
-				const result = await utils.filesystem.listDirectory.fetch({
-					workspaceId,
+			updateState((current) => ({
+				...current,
+				loadingDirectories: new Set(current.loadingDirectories).add(
 					absolutePath,
-				});
-
-				updateState((current) => {
-					const nextEntries = new Map(current.entriesByPath);
-					const nextChildren = new Map(current.childPathsByDirectory);
-					const nextLoaded = new Set(current.loadedDirectories);
-					const nextInvalidated = new Set(current.invalidatedDirectories);
-					const nextLoading = new Set(current.loadingDirectories);
-					nextLoading.delete(absolutePath);
-					nextLoaded.add(absolutePath);
-					nextInvalidated.delete(absolutePath);
-
-					for (const entry of result.entries) {
-						nextEntries.set(entry.absolutePath, entry);
-					}
-
-					nextChildren.set(
-						absolutePath,
-						result.entries.map((entry) => entry.absolutePath),
-					);
+				),
+			}));
 
-					return {
-						...current,
-						childPathsByDirectory: nextChildren,
-						entriesByPath: nextEntries,
-						invalidatedDirectories: nextInvalidated,
-						loadedDirectories: nextLoaded,
-						loadingDirectories: nextLoading,
-					};
-				});
+			try {
+				// Server-side timeout + React Query's TIMEOUT-aware retry handle
+				// hung host-service IPC; we just await the fetch and apply results.
+				const result = await utils.filesystem.listDirectory.fetch(input);
+				updateState((current) =>
+					applyDirectoryEntries(current, absolutePath, result.entries),
+				);
 			} catch (error) {
 				console.error(
 					"[workspace-client/useFileTree] Failed to load directory:",
-					{
-						absolutePath,
-						error,
-					},
+					{ absolutePath, error },
 				);
-
 				updateState((current) => {
 					const nextLoading = new Set(current.loadingDirectories);
 					nextLoading.delete(absolutePath);
-					return {
-						...current,
-						loadingDirectories: nextLoading,
-					};
+					return { ...current, loadingDirectories: nextLoading };
 				});
 			}
 		},
@@ -272,7 +281,7 @@ export function useFileTree({
 
 	const refreshPath = useCallback(
 		async (absolutePath: string): Promise<void> => {
-			await loadDirectory(absolutePath, true);
+			await loadDirectory(absolutePath, { force: true });
 		},
 		[loadDirectory],
 	);
@@ -288,10 +297,10 @@ export function useFileTree({
 			(left, right) => left.split(/[/\\]/).length - right.split(/[/\\]/).length,
 		);
 
-		await loadDirectory(rootPath, true);
+		await loadDirectory(rootPath, { force: true });
 		for (const absolutePath of expandedDirectories) {
 			if (absolutePath !== rootPath) {
-				await loadDirectory(absolutePath, true);
+				await loadDirectory(absolutePath, { force: true });
 			}
 		}
 	}, [loadDirectory, rootPath]);
@@ -347,11 +356,8 @@ export function useFileTree({
 
 	useEffect(() => {
 		updateState(() => createInitialState());
-		if (!rootPath) {
-			return;
-		}
-
-		void loadDirectory(rootPath, true);
+		if (!rootPath) return;
+		void loadDirectory(rootPath, { force: true });
 	}, [loadDirectory, rootPath, updateState]);
 
 	useWorkspaceEvent(
@@ -404,16 +410,16 @@ export function useFileTree({
 				});
 
 				if (stateRef.current.loadedDirectories.has(oldParentPath)) {
-					void loadDirectory(oldParentPath, true);
+					void loadDirectory(oldParentPath, { force: true });
 				}
 				if (stateRef.current.loadedDirectories.has(newParentPath)) {
-					void loadDirectory(newParentPath, true);
+					void loadDirectory(newParentPath, { force: true });
 				}
 				if (
 					event.isDirectory &&
 					stateRef.current.expandedDirectories.has(event.absolutePath)
 				) {
-					void loadDirectory(event.absolutePath, true);
+					void loadDirectory(event.absolutePath, { force: true });
 				}
 				return;
 			}
@@ -438,7 +444,7 @@ export function useFileTree({
 			});
 
 			if (stateRef.current.loadedDirectories.has(parentPath)) {
-				void loadDirectory(parentPath, true);
+				void loadDirectory(parentPath, { force: true });
 			}
 		},
 		Boolean(workspaceId && rootPath),

apps/desktop/src/renderer/routes/_authenticated/_dashboard/v2-workspace/$workspaceId/components/WorkspaceSidebar/WorkspaceSidebar.tsx

@@ -42,7 +42,6 @@ interface WorkspaceSidebarProps {
 	selectedFilePath?: string;
 	pendingReveal?: PendingReveal | null;
 	workspaceId: string;
-	workspaceName?: string;
 }
 
 function IconButton({
@@ -80,7 +79,6 @@ export function WorkspaceSidebar({
 	selectedFilePath,
 	pendingReveal,
 	workspaceId,
-	workspaceName,
 }: WorkspaceSidebarProps) {
 	const collections = useCollections();
 	const localState = collections.v2WorkspaceLocalState.get(workspaceId);
@@ -142,7 +140,6 @@ export function WorkspaceSidebar({
 				selectedFilePath={selectedFilePath}
 				pendingReveal={pendingReveal}
 				workspaceId={workspaceId}
-				workspaceName={workspaceName}
 				gitStatus={gitStatus.data}
 			/>
 		),

apps/desktop/src/renderer/routes/_authenticated/_dashboard/v2-workspace/$workspaceId/components/WorkspaceSidebar/components/FilesTab/FilesTab.tsx

@@ -5,7 +5,13 @@ import { toast } from "@superset/ui/sonner";
 import { Tooltip, TooltipContent, TooltipTrigger } from "@superset/ui/tooltip";
 import { workspaceTrpc } from "@superset/workspace-client";
 import type { inferRouterOutputs } from "@trpc/server";
-import { FilePlus, FolderPlus, FoldVertical, RefreshCw } from "lucide-react";
+import {
+	FilePlus,
+	FolderPlus,
+	FoldVertical,
+	Loader2,
+	RefreshCw,
+} from "lucide-react";
 import { Fragment, useCallback, useEffect, useRef, useState } from "react";
 import {
 	type FileTreeNode,
@@ -39,7 +45,6 @@ interface FilesTabProps {
 		isDirectory: boolean;
 	} | null;
 	workspaceId: string;
-	workspaceName?: string;
 	gitStatus: GitStatusData | undefined;
 }
 
@@ -209,7 +214,6 @@ export function FilesTab({
 	selectedFilePath,
 	pendingReveal,
 	workspaceId,
-	workspaceName,
 	gitStatus,
 }: FilesTabProps) {
 	const [_isRefreshing, setIsRefreshing] = useState(false);
@@ -462,10 +466,17 @@ export function FilesTab({
 		[workspaceId, deletePath],
 	);
 
-	if (!workspaceQuery.data?.worktreePath) {
+	if (!rootPath) {
 		return (
-			<div className="flex h-full items-center justify-center text-sm text-muted-foreground">
-				Workspace worktree not available
+			<div className="flex h-full items-center justify-center gap-2 text-sm text-muted-foreground">
+				{workspaceQuery.isLoading ? (
+					<>
+						<Loader2 className="size-3.5 animate-spin" />
+						<span>Loading files...</span>
+					</>
+				) : (
+					"Workspace worktree not available"
+				)}
 			</div>
 		);
 	}
@@ -503,7 +514,7 @@ export function FilesTab({
 						zIndex: 20,
 					}}
 				>
-					<span className="truncate">{workspaceName ?? "Explorer"}</span>
+					<span className="truncate">Explorer</span>
 					<div className="flex items-center gap-0.5">
 						<Tooltip>
 							<TooltipTrigger asChild>

apps/desktop/src/renderer/routes/_authenticated/_dashboard/v2-workspace/$workspaceId/page.tsx

@@ -102,7 +102,6 @@ function V2WorkspacePage() {
 		<WorkspaceContent
 			projectId={workspace.projectId}
 			workspaceId={workspace.id}
-			workspaceName={workspace.name}
 			terminalId={terminalId}
 			chatSessionId={chatSessionId}
 			focusRequestId={focusRequestId}
@@ -116,7 +115,6 @@ function V2WorkspacePage() {
 function WorkspaceContent({
 	projectId,
 	workspaceId,
-	workspaceName,
 	terminalId,
 	chatSessionId,
 	focusRequestId,
@@ -126,7 +124,6 @@ function WorkspaceContent({
 }: {
 	projectId: string;
 	workspaceId: string;
-	workspaceName: string;
 	terminalId?: string;
 	chatSessionId?: string;
 	focusRequestId?: string;
@@ -295,7 +292,6 @@ function WorkspaceContent({
 					>
 						<WorkspaceSidebar
 							workspaceId={workspaceId}
-							workspaceName={workspaceName}
 							onSelectFile={openFilePane}
 							onSelectDiffFile={openDiffPane}
 							onOpenComment={openCommentPane}