Implement sidebar tree view state persistence

2026-05-27 16:32:05 +08:00
parent bd7470d03e
commit e43efacbce
15 changed files with 1875 additions and 17 deletions
@@ -4,7 +4,8 @@ use crate::error::WebError;
 use crate::routes::local_folder_source::{
    control_plane_db_path_display, create_default_local_workspace_for_actor,
    ensure_local_workspace_read_access_with_state, is_local_access_policy_admin_context,
-    load_local_folder_page_tree_snapshot, load_local_trash_entries,
+    load_local_folder_page_tree_scope_snapshot, load_local_folder_page_tree_snapshot,
+    load_local_trash_entries,
 };
 use crate::routes::snapshot_support::load_sidebar_dataset;
 use crate::routes::web_shell::{
@@ -326,7 +327,11 @@ pub async fn root_entry(
                WebError::bad_request_code("local_folder_root_required", "缺少本地文件夹 rootUri")
            })?;
        ensure_local_workspace_read_access_with_state(&state, &context, root_uri)?;
-        let page_tree_snapshot = load_local_folder_page_tree_snapshot(root_uri)?;
+        let page_tree_snapshot = if let Some(scope) = file_tree_scope {
+            load_local_folder_page_tree_scope_snapshot(root_uri, scope)?
+        } else {
+            load_local_folder_page_tree_snapshot(root_uri)?
+        };
        let workspace_id = page_tree_snapshot
            .dataset
            .get("workspace")
@@ -2860,6 +2865,55 @@ mod tests {
        );
    }

+    #[tokio::test]
+    async fn root_entry_local_folder_filetree_scope_renders_scoped_page_tree() {
+        let root = temp_root("mnote-root-local-folder-scoped-page-tree");
+        std::fs::create_dir_all(root.join("design").join("05-editor-mainline"))
+            .expect("create scoped design tree");
+        std::fs::write(root.join("Home.md"), "# Home\n").expect("write root page");
+        std::fs::write(
+            root.join("design")
+                .join("05-editor-mainline")
+                .join("Target.md"),
+            "# Target\n",
+        )
+        .expect("write scoped page");
+
+        let root_uri = format!("file://{}", root.display());
+        crate::routes::local_folder_source::initialize_local_workspace_for_actor(
+            "user_real",
+            &root_uri,
+        )
+        .expect("init local workspace");
+        let response = app_with_config("http://127.0.0.1:3100".into(), false)
+            .oneshot(
+                Request::builder()
+                    .uri(format!(
+                        "/?sourceKind=local_folder&rootUri={root_uri}&treeView=filetree&fileTreeScope=design"
+                    ))
+                    .header("x-mnote-actor-id", "user_real")
+                    .header("x-mnote-actor-type", "user")
+                    .body(Body::empty())
+                    .expect("request"),
+            )
+            .await
+            .expect("response");
+
+        let _ = std::fs::remove_dir_all(&root);
+
+        assert_eq!(response.status(), StatusCode::OK);
+        let body = to_bytes(response.into_body(), usize::MAX)
+            .await
+            .expect("body");
+        let html = String::from_utf8(body.to_vec()).expect("utf8");
+        assert!(html.contains(r#"data-node-id="local-dir:design~2F05-editor-mainline""#));
+        assert!(html.contains(r#"data-node-id="local-md:design~2F05-editor-mainline~2FTarget.md""#));
+        assert!(
+            !html.contains(r#"data-node-id="local-md:Home.md""#),
+            "星标 scoped folder 入口的 PageTree 不应回退到 workspace root 页面树"
+        );
+    }
+
    #[tokio::test]
    async fn root_entry_local_folder_without_active_page_keeps_resource_tab_host() {
        let root = temp_root("mnote-root-local-folder-no-active-page");