Drive indexing: wire it up WIP

Now it displays the sizes! Yay!

Author: vdavid

apps/desktop/src-tauri/src/file_system/listing/operations.rs

@@ -151,15 +151,19 @@ pub fn get_file_range(
         .ok_or_else(|| format!("Listing not found: {}", listing_id))?;
 
     // Filter entries if not including hidden
-    if include_hidden {
+    let mut entries: Vec<FileEntry> = if include_hidden {
         let end = (start + count).min(listing.entries.len());
-        Ok(listing.entries[start..end].to_vec())
+        listing.entries[start..end].to_vec()
     } else {
         // Need to filter and then slice
         let visible: Vec<&FileEntry> = listing.entries.iter().filter(|e| is_visible(e)).collect();
         let end = (start + count).min(visible.len());
-        Ok(visible[start..end].iter().cloned().cloned().collect())
-    }
+        visible[start..end].iter().cloned().cloned().collect()
+    };
+
+    crate::indexing::enrich_entries_with_index(&mut entries);
+
+    Ok(entries)
 }
 
 /// Gets total count of entries in a cached listing.

apps/desktop/src/lib/file-explorer/pane/DualPaneExplorer.svelte

@@ -76,6 +76,7 @@
         getSelfDragFileInfos,
         endSelfDragSession,
     } from '../drag/drag-drop'
+    import { initIndexEvents, prioritizeDir, cancelNavPriority } from '$lib/indexing/index'
     import { resolveDropTarget } from '../drag/drop-target-hit-testing'
     import DragOverlay from '../drag/DragOverlay.svelte'
     import { showOverlay, updateOverlay, hideOverlay, type OverlayFileInfo } from '../drag/drag-overlay.svelte.js'
@@ -118,6 +119,7 @@
     let unlistenDragDrop: UnlistenFn | undefined
     let unlistenDragImageSize: UnlistenFn | undefined
     let unlistenDragModifiers: UnlistenFn | undefined
+    let unlistenIndexEvents: UnlistenFn | undefined
 
     // Drag image size from the source app (macOS only, via swizzle).
     // If the source provides a large preview (like Finder), we suppress our overlay.
@@ -275,10 +277,18 @@
     // --- Unified handler functions ---
 
     function handlePathChange(pane: 'left' | 'right', path: string) {
+        const oldPath = getPanePath(pane)
         setPanePath(pane, path)
         setPaneHistory(pane, pushPath(getPaneHistory(pane), path))
         void saveAppStatus({ [paneKey(pane, 'path')]: path })
         void saveLastUsedPathForVolume(getPaneVolumeId(pane), path)
+
+        // Update index priorities: cancel old dir, prioritize new dir
+        if (oldPath !== path) {
+            void cancelNavPriority(oldPath)
+            void prioritizeDir(path, 'current_dir')
+        }
+
         containerElement?.focus()
     }
 
@@ -331,7 +341,8 @@
         volumePath: string,
         targetPath: string,
     ) {
-        void saveLastUsedPathForVolume(getPaneVolumeId(pane), getPanePath(pane))
+        const oldPath = getPanePath(pane)
+        void saveLastUsedPathForVolume(getPaneVolumeId(pane), oldPath)
 
         if (!volumes.find((v) => v.id === volumeId)) {
             volumes = await listVolumes()
@@ -343,6 +354,10 @@
             otherPanePath: getPanePath(other),
         })
 
+        // Update index priorities: cancel old dir, prioritize new dir
+        void cancelNavPriority(oldPath)
+        void prioritizeDir(pathToNavigate, 'current_dir')
+
         setPaneVolumeId(pane, volumeId)
         setPanePath(pane, pathToNavigate)
         setPaneHistory(pane, push(getPaneHistory(pane), { volumeId, path: pathToNavigate }))
@@ -652,6 +667,44 @@
         dropTargetFolderEl = null
     }
 
+    /** Ensures a path ends with '/' for correct prefix matching. */
+    function ensureTrailingSlash(path: string): string {
+        return path.endsWith('/') ? path : path + '/'
+    }
+
+    /** Called when the drive index updates directory stats. Re-fetches panes whose visible entries changed. */
+    function handleIndexDirUpdated(paths: string[]) {
+        const leftDir = ensureTrailingSlash(leftPath)
+        const rightDir = ensureTrailingSlash(rightPath)
+
+        let refreshLeft = false
+        let refreshRight = false
+
+        for (const updatedPath of paths) {
+            // An updated path is relevant if it is a direct child or deeper descendant
+            // whose parent is displayed in the pane's current directory.
+            // We check if parentOf(updatedPath) === panePath (i.e. the updated path itself
+            // is a direct child entry visible in the listing).
+            const updatedWithSlash = ensureTrailingSlash(updatedPath)
+            if (!refreshLeft && updatedWithSlash.startsWith(leftDir) && updatedWithSlash !== leftDir) {
+                refreshLeft = true
+            }
+            if (!refreshRight && updatedWithSlash.startsWith(rightDir) && updatedWithSlash !== rightDir) {
+                refreshRight = true
+            }
+            if (refreshLeft && refreshRight) break
+        }
+
+        if (refreshLeft) {
+            // eslint-disable-next-line @typescript-eslint/no-unsafe-call
+            leftPaneRef?.refreshView?.()
+        }
+        if (refreshRight) {
+            // eslint-disable-next-line @typescript-eslint/no-unsafe-call
+            rightPaneRef?.refreshView?.()
+        }
+    }
+
     function handleResizeForDevTools() {
         void recalculateWebviewOffset()
     }
@@ -780,6 +833,13 @@
             setAltHeld(event.payload.altHeld)
         })
 
+        // Listen for index directory updates to refresh panes when sizes change
+        unlistenIndexEvents = await initIndexEvents(handleIndexDirUpdated)
+
+        // Prioritize scanning the initial directories of both panes
+        void prioritizeDir(leftPath, 'current_dir')
+        void prioritizeDir(rightPath, 'current_dir')
+
         // Register drag-and-drop target handler for external and pane-to-pane drops
         unlistenDragDrop = await getCurrentWebview().onDragDropEvent((event) => {
             const { type } = event.payload
@@ -844,9 +904,16 @@
         newHistory: NavigationHistory,
         targetPath: string,
     ) {
+        const oldPath = getPanePath(pane)
         const entry = getCurrentEntry(newHistory)
         const paneRef = getPaneRef(pane)
 
+        // Update index priorities: cancel old dir, prioritize new dir
+        if (oldPath !== targetPath) {
+            void cancelNavPriority(oldPath)
+            void prioritizeDir(targetPath, 'current_dir')
+        }
+
         setPaneHistory(pane, newHistory)
         setPanePath(pane, targetPath)
         if (entry.volumeId !== getPaneVolumeId(pane)) {
@@ -907,6 +974,7 @@
         unlistenDragImageSize?.()
         unlistenDragModifiers?.()
         unlistenDragDrop?.()
+        unlistenIndexEvents?.()
         cleanupNetworkDiscovery()
         stopModifierTracking()
         window.removeEventListener('resize', handleResizeForDevTools) // No-op in non-dev, safe to always call

apps/desktop/src/lib/file-explorer/pane/FilePane.svelte

@@ -78,6 +78,7 @@
     import * as benchmark from '$lib/benchmark'
     import { handleNavigationShortcut } from '../navigation/keyboard-shortcuts'
     import { resolveValidPath } from '../navigation/path-navigation'
+    import { prioritizeDir } from '$lib/indexing/index'
 
     interface Props {
         initialPath: string
@@ -1247,6 +1248,13 @@
         if (e.key === ' ') {
             e.preventDefault()
             selection.toggleAt(cursorIndex, hasParent)
+
+            // Request index priority scan when selecting a directory
+            const entry = getEntryUnderCursor()
+            if (entry?.isDirectory && entry.name !== '..') {
+                void prioritizeDir(entry.path, 'user_selected')
+            }
+
             return true
         }
         // Cmd+A - select all (Cmd+Shift+A - deselect all)

apps/desktop/src/lib/file-explorer/selection/SelectionInfo.svelte

@@ -12,6 +12,7 @@
     } from './selection-info-utils'
     import { measureDateColumnWidth } from '../views/full-list-utils'
     import { formatFileSize, formatDateTime } from '$lib/settings/reactive-settings.svelte'
+    import { isScanning } from '$lib/indexing/index-state.svelte'
 
     interface Props {
         /** View mode: 'brief' or 'full' */
@@ -183,6 +184,9 @@
     // Selection summary mode
     // ========================================================================
 
+    // Drive index scanning state — used for stale indicator when dirs are selected
+    const scanning = $derived(isScanning())
+
     // Computed values for selection summary
     const selectedFiles = $derived(stats?.selectedFiles ?? 0)
     const selectedDirs = $derived(stats?.selectedDirs ?? 0)
@@ -195,6 +199,9 @@
     const hasDirs = $derived(totalDirs > 0)
     const hasOnlyDirs = $derived(!hasFiles && hasDirs)
 
+    // When directories are selected during scanning, sizes might be incomplete
+    const showSelectionStale = $derived(scanning && selectedDirs > 0)
+
     const sizePercentage = $derived(calculatePercentage(selectedFileSize, totalFileSize))
     const filePercentage = $derived(calculatePercentage(selectedFiles, totalFiles))
     const dirPercentage = $derived(calculatePercentage(selectedDirs, totalDirs))
@@ -232,9 +239,12 @@
         <!-- Selection summary -->
         <span class="summary-text" title={selectionSizeTooltip}>
             {#if hasOnlyDirs}
-                <!-- Only dirs, no files: can't show size -->
+                <!-- Only dirs, no files -->
                 {formatNumber(selectedDirs)} of {formatNumber(totalDirs)}
                 {pluralize(totalDirs, 'dir', 'dirs')} ({dirPercentage}%) selected.
+                {#if showSelectionStale}
+                    <span class="stale-indicator" title="Might be outdated. Currently scanning...">⚠️</span>
+                {/if}
             {:else if hasFiles}
                 <!-- Has files: show full summary -->
                 {#each selectedSizeTriads as triad, i (i)}<span class={triad.tierClass}>{triad.value}</span>{/each}
@@ -244,6 +254,9 @@
                 {pluralize(totalFiles, 'file', 'files')} ({filePercentage}%){#if hasDirs}
                     &nbsp;and {formatNumber(selectedDirs)} of {formatNumber(totalDirs)}
                     {pluralize(totalDirs, 'dir', 'dirs')} ({dirPercentage}%){/if}.
+                {#if showSelectionStale}
+                    <span class="stale-indicator" title="Might be outdated. Currently scanning...">⚠️</span>
+                {/if}
             {/if}
         </span>
     {/if}
@@ -291,4 +304,9 @@
         overflow: hidden;
         text-overflow: ellipsis;
     }
+
+    .stale-indicator {
+        font-size: 10px;
+        cursor: help;
+    }
 </style>

apps/desktop/src/lib/file-explorer/types.ts

@@ -16,6 +16,9 @@ export interface FileEntry {
     iconId: string
     /** Whether extended metadata (addedAt, openedAt) has been loaded */
     extendedMetadataLoaded: boolean
+    recursiveSize?: number
+    recursiveFileCount?: number
+    recursiveDirCount?: number
 }
 
 /** Cloud sync status for files in Dropbox/iCloud/etc. folders */

apps/desktop/src/lib/file-explorer/views/BriefList.svelte

@@ -17,8 +17,10 @@
         shouldResetCache,
         refetchIconsForEntries,
     } from './file-list-utils'
-    import { getRowHeight } from '$lib/settings/reactive-settings.svelte'
+    import { getRowHeight, formatFileSize } from '$lib/settings/reactive-settings.svelte'
     import { getSetting } from '$lib/settings/settings-store'
+    import { formatNumber, pluralize } from '../selection/selection-info-utils'
+    import { isScanning } from '$lib/indexing/index-state.svelte'
     import { extensionCacheCleared } from '$lib/icon-cache'
     import type { RenameState } from '../rename/rename-state.svelte'
 
@@ -89,6 +91,9 @@
     let cachedRange = $state({ start: 0, end: 0 })
     let isFetching = $state(false)
 
+    // Drive index scanning state — used for directory size tooltips
+    const scanning = $derived(isScanning())
+
     // ==== Layout constants ====
     // Row height is reactive based on UI density setting
     const rowHeight = $derived(getRowHeight())
@@ -418,6 +423,17 @@
         }
     })
 
+    /** Build tooltip for a directory entry showing recursive size info. */
+    function buildDirTooltip(file: FileEntry): string | undefined {
+        if (!file.isDirectory) return undefined
+        if (file.recursiveSize !== undefined) {
+            const sizeInfo = `${formatFileSize(file.recursiveSize)} · ${formatNumber(file.recursiveFileCount ?? 0)} ${pluralize(file.recursiveFileCount ?? 0, 'file', 'files')} · ${formatNumber(file.recursiveDirCount ?? 0)} ${pluralize(file.recursiveDirCount ?? 0, 'folder', 'folders')}`
+            return scanning ? `${sizeInfo} — Might be outdated` : sizeInfo
+        }
+        if (scanning) return 'Scanning...'
+        return undefined
+    }
+
     // Report visible range to parent for MCP state sync
     $effect(() => {
         // Calculate visible item range from column range
@@ -495,6 +511,7 @@
                                 class:is-under-cursor={globalIndex === cursorIndex}
                                 class:is-selected={selectedIndices.has(globalIndex)}
                                 data-drop-target-path={file.isDirectory && file.name !== '..' ? file.path : undefined}
+                                title={buildDirTooltip(file)}
                                 style="height: {rowHeight}px;"
                                 onmousedown={(e: MouseEvent) => {
                                     handleMouseDown(e, globalIndex)