Merge pull request #90 from codeflash-ai/initial-mount-file-progress

Show initial mount file indexing progress

Author: aseembits93

apps/desktop/src-tauri/src/main.rs

@@ -225,10 +225,19 @@ struct MountSummary {
     status: String,
     root_exists: bool,
     entity_count: usize,
+    hydration_progress: Option<MountHydrationProgress>,
     pending_change_count: usize,
     provider: Option<ProviderRuntimeSummary>,
 }
 
+#[derive(Clone, Serialize)]
+#[serde(rename_all = "camelCase")]
+struct MountHydrationProgress {
+    indexed_files: usize,
+    remaining_files: usize,
+    total_files: usize,
+}
+
 #[derive(Clone, Serialize)]
 #[serde(rename_all = "camelCase")]
 struct ProviderRuntimeSummary {
@@ -3937,6 +3946,7 @@ fn degraded_snapshot(message: String) -> DesktopSnapshot {
             status: "error".to_string(),
             root_exists: false,
             entity_count: 0,
+            hydration_progress: None,
             pending_change_count: 0,
             provider: None,
         },
@@ -4122,6 +4132,7 @@ fn mount_summary_with_pending_change_count(
             status: "not_mounted".to_string(),
             root_exists: false,
             entity_count: 0,
+            hydration_progress: None,
             pending_change_count: 0,
             provider: None,
         };
@@ -4133,6 +4144,7 @@ fn mount_summary_with_pending_change_count(
         .unwrap_or("ready");
     let access_root = mount_access_root(mount);
     let root_exists = mount_root_exists_for_desktop_summary(mount, &access_root);
+    let entities = store.and_then(|store| store.list_entities(&mount.mount_id).ok());
 
     MountSummary {
         mount_id: mount.mount_id.0.clone(),
@@ -4160,10 +4172,13 @@ fn mount_summary_with_pending_change_count(
         read_only: mount.read_only,
         status: mount_status.to_string(),
         root_exists,
-        entity_count: store
-            .and_then(|store| store.list_entities(&mount.mount_id).ok())
+        entity_count: entities
+            .as_ref()
             .map(|entities| entities.len())
             .unwrap_or(0),
+        hydration_progress: entities
+            .as_ref()
+            .and_then(|entities| mount_hydration_progress(entities)),
         pending_change_count: pending_change_count.unwrap_or_else(|| {
             store
                 .map(|store| pending_changes_for_mount(store, state_root, &mount.mount_id))
@@ -4175,6 +4190,33 @@ fn mount_summary_with_pending_change_count(
     }
 }
 
+fn mount_hydration_progress(entities: &[EntityRecord]) -> Option<MountHydrationProgress> {
+    let mut indexed_files = 0usize;
+    let mut remaining_files = 0usize;
+
+    for entity in entities {
+        if entity.kind != EntityKind::Page {
+            continue;
+        }
+        match entity.hydration {
+            HydrationState::Hydrated | HydrationState::Dirty | HydrationState::Conflicted => {
+                indexed_files += 1;
+            }
+            HydrationState::Stub => {
+                remaining_files += 1;
+            }
+            HydrationState::Virtual => {}
+        }
+    }
+
+    let total_files = indexed_files + remaining_files;
+    (total_files > 0).then_some(MountHydrationProgress {
+        indexed_files,
+        remaining_files,
+        total_files,
+    })
+}
+
 fn mount_live_mode_summary(
     store: &SqliteStateStore,
     mount: Option<&MountConfig>,
@@ -11603,6 +11645,7 @@ mod tests {
         let summary = super::mount_summary(None, Path::new("."), None, None, None);
 
         assert!(Path::new(&summary.local_path).is_absolute());
+        assert!(summary.hydration_progress.is_none());
     }
 
     #[test]
@@ -11793,6 +11836,95 @@ mod tests {
         assert!(snapshot.recent_files.is_empty());
     }
 
+    #[test]
+    fn desktop_snapshot_reports_hydratable_file_progress() {
+        let temp = TestTempDir::new("desktop-file-progress");
+        let mut store = SqliteStateStore::open(temp.path().to_path_buf()).expect("open store");
+        let connection = test_connection("workspace-1", "CodeFlash");
+        let mount = MountConfig::new(
+            MountId::new("notion-main"),
+            "notion",
+            temp.path().join("codeflash-wiki"),
+        )
+        .with_connection_id(connection.connection_id.clone())
+        .projection(ProjectionMode::LinuxFuse);
+
+        store.save_connection(connection).expect("save connection");
+        store.save_mount(mount).expect("save mount");
+
+        for (remote_id, title, path, hydration) in [
+            (
+                "hydrated-page",
+                "Hydrated",
+                "Hydrated/page.md",
+                HydrationState::Hydrated,
+            ),
+            (
+                "dirty-page",
+                "Dirty",
+                "Dirty/page.md",
+                HydrationState::Dirty,
+            ),
+            (
+                "conflicted-page",
+                "Conflicted",
+                "Conflicted/page.md",
+                HydrationState::Conflicted,
+            ),
+            ("stub-page", "Stub", "Stub/page.md", HydrationState::Stub),
+            (
+                "virtual-page",
+                "Virtual",
+                "Virtual/page.md",
+                HydrationState::Virtual,
+            ),
+        ] {
+            store
+                .save_entity(
+                    EntityRecord::new(
+                        MountId::new("notion-main"),
+                        RemoteId::new(remote_id),
+                        EntityKind::Page,
+                        title,
+                        path,
+                    )
+                    .with_hydration(hydration),
+                )
+                .expect("save page entity");
+        }
+
+        for (remote_id, kind, path) in [
+            ("directory-1", EntityKind::Directory, "Directory"),
+            ("database-1", EntityKind::Database, "Database"),
+            ("asset-1", EntityKind::Asset, "asset.png"),
+        ] {
+            store
+                .save_entity(
+                    EntityRecord::new(
+                        MountId::new("notion-main"),
+                        RemoteId::new(remote_id),
+                        kind,
+                        remote_id,
+                        path,
+                    )
+                    .with_hydration(HydrationState::Stub),
+                )
+                .expect("save non-page entity");
+        }
+
+        let snapshot = super::load_desktop_snapshot_from_store(&store, temp.path())
+            .expect("load snapshot from test store");
+        let progress = snapshot
+            .mount
+            .hydration_progress
+            .expect("file progress is reported");
+
+        assert_eq!(snapshot.mount.entity_count, 8);
+        assert_eq!(progress.indexed_files, 3);
+        assert_eq!(progress.remaining_files, 1);
+        assert_eq!(progress.total_files, 4);
+    }
+
     #[test]
     fn desktop_snapshot_prefers_active_granola_over_stale_notion_mount() {
         let temp = TestTempDir::new("desktop-granola-first");
@@ -16152,6 +16284,11 @@ fn sample_snapshot() -> DesktopSnapshot {
         status: "ready".to_string(),
         root_exists: true,
         entity_count: 42,
+        hydration_progress: Some(MountHydrationProgress {
+            indexed_files: 16,
+            remaining_files: 4,
+            total_files: 20,
+        }),
         pending_change_count: 3,
         provider: None,
     };

apps/desktop/src/App.tsx

@@ -42,6 +42,8 @@ import { useEffect, useMemo, useRef, useState } from "react";
 import {
   compactPath,
   mountEntityCountLabel,
+  mountFileIndexProgressLabel,
+  mountFileIndexProgressValue,
   mountAccessLabel,
   mountRows,
   mountStatusLabel,
@@ -382,6 +384,11 @@ const sampleMount: MountSummary = {
   status: "ready",
   rootExists: true,
   entityCount: 24,
+  hydrationProgress: {
+    indexedFiles: 16,
+    remainingFiles: 4,
+    totalFiles: 20,
+  },
   pendingChangeCount: 3,
   provider: null,
 };
@@ -414,14 +421,14 @@ const sampleSnapshot: DesktopSnapshot = {
     connector: "notion",
     workspaceName: "CodeFlash",
     accountLabel: "saurabh@codeflash.ai",
-    status: "ready",
+    status: "active",
   },
   connections: [
     {
       connector: "notion",
       workspaceName: "CodeFlash",
       accountLabel: "saurabh@codeflash.ai",
-      status: "ready",
+      status: "active",
     },
   ],
   mount: sampleMount,
@@ -4006,6 +4013,7 @@ function CurrentWorkspacePanel({
   const [pullState, setPullState] = useState<"idle" | "pulling" | "success" | "error">("idle");
   const accountLabel = snapshot.connection.accountLabel.trim();
   const showAccount = accountLabel.length > 0 && accountLabel !== snapshot.connection.workspaceName;
+  const fileProgressLabel = mountFileIndexProgressLabel(snapshot.mount);
 
   async function openFolder() {
     setActionError("");
@@ -4144,7 +4152,7 @@ function CurrentWorkspacePanel({
       <div className="workspace-facts">
         <span>Permission: {snapshot.mount.readOnly ? "Read only" : "Edit enabled"}</span>
         <span>Projection: {snapshot.mount.projection}</span>
-        <span>Indexed: {mountEntityCountLabel(snapshot.mount)}</span>
+        <span>{fileProgressLabel ?? `Indexed: ${mountEntityCountLabel(snapshot.mount)}`}</span>
         {showAccount && <span>Account: {accountLabel}</span>}
       </div>
 
@@ -4204,6 +4212,7 @@ function MountDetailView({
   } = useMountLiveModeController(snapshot, onRefresh);
   const liveModeAppliesToSource = isActiveMount && mount.connector === "notion";
   const sourceSyncMode = sourceSyncModeLabel(snapshot.liveMode, liveModeAppliesToSource);
+  const fileProgressValue = mountFileIndexProgressValue(mount);
 
   async function openFolder() {
     setActionError("");
@@ -4449,6 +4458,7 @@ function MountDetailView({
           <SettingRow title="Location" value={mount.localPath} />
           <SettingRow title="Projection" value={mount.projection} />
           <SettingRow title="Mounted content" value={`${mount.entityCount} items`} />
+          {fileProgressValue && <SettingRow title="Files indexed" value={fileProgressValue} />}
           <SettingRow title="Root exists" value={mount.rootExists ? "Yes" : "No"} />
         </div>
       </section>

apps/desktop/src/dark-theme-ui.test.js

@@ -36,6 +36,13 @@ describe("dark theme UI contrast", () => {
     expect(styles).not.toMatch(/:root\[data-theme="dark"\] \.tray-live-mode-control,[\s\S]*?\.file-row\.expanded/s);
   });
 
+  it("keeps disabled mount detail buttons from taking hover colors", () => {
+    expect(styles).toMatch(/\.mount-details-button:hover:not\(:disabled\)\s*\{/);
+    expect(styles).toMatch(
+      /:root\[data-theme="dark"\] \.mount-details-button:hover:not\(:disabled\),/,
+    );
+  });
+
   it("uses readable control tokens for Sources page actions", () => {
     expect(styles).toMatch(/\.secondary-button\s*\{[\s\S]*?background:\s*var\(--control-bg\);[\s\S]*?color:\s*var\(--control-text\);/s);
     expect(styles).toMatch(/\.mount-details-button\s*\{[\s\S]*?background:\s*var\(--control-bg\);[\s\S]*?color:\s*var\(--control-text\);/s);

apps/desktop/src/mounts.test.ts

@@ -2,6 +2,7 @@ import { describe, expect, it } from "vitest";
 import {
   compactPath,
   mountEntityCountLabel,
+  mountFileIndexProgressLabel,
   mountAccessLabel,
   mountRows,
   mountStatusLabel,
@@ -96,6 +97,68 @@ describe("mount display helpers", () => {
     expect(mountEntityCountLabel(mount({ entityCount: 260 }))).toBe("260 items");
   });
 
+  it("labels hydratable file indexing progress", () => {
+    expect(
+      mountFileIndexProgressLabel(
+        mount({
+          hydrationProgress: {
+            indexedFiles: 1,
+            remainingFiles: 2,
+            totalFiles: 3,
+          },
+        }),
+      ),
+    ).toBe("Indexed: 1 of 3 files, 2 left");
+    expect(
+      mountFileIndexProgressLabel(
+        mount({
+          hydrationProgress: {
+            indexedFiles: 1,
+            remainingFiles: 0,
+            totalFiles: 1,
+          },
+        }),
+      ),
+    ).toBe("Indexed: 1 of 1 file");
+    expect(mountFileIndexProgressLabel(mount({}))).toBeNull();
+  });
+
+  it("uses hydratable file progress as source card content while files remain", () => {
+    const rows = mountRows(
+      [
+        mount({
+          hydrationProgress: {
+            indexedFiles: 12,
+            remainingFiles: 68,
+            totalFiles: 80,
+          },
+        }),
+      ],
+      mount({}),
+      "notion-main",
+    );
+
+    expect(rows[0].content).toBe("Indexed: 12 of 80 files, 68 left");
+  });
+
+  it("keeps compact source card content once file indexing is complete", () => {
+    const rows = mountRows(
+      [
+        mount({
+          hydrationProgress: {
+            indexedFiles: 80,
+            remainingFiles: 0,
+            totalFiles: 80,
+          },
+        }),
+      ],
+      mount({}),
+      "notion-main",
+    );
+
+    expect(rows[0].content).toBe("24 items, 3 pending");
+  });
+
   it("compacts long paths from the middle so filenames remain visible", () => {
     expect(compactPath("/home/ada/Locality/notion/Engineering/Roadmap 2026/page.md", 42)).toBe(
       "/.../Engineering/Roadmap 2026/page.md",