fix local folder resource lifecycle and workspace switch

This commit is contained in:
lix-2026
2026-05-24 02:23:48 +08:00
parent 97953101a2
commit 6d9d8ce9cd
9 changed files with 374 additions and 65 deletions
@@ -4,7 +4,6 @@
const assert = require("node:assert");
const fs = require("node:fs");
const fsp = require("node:fs/promises");
const os = require("node:os");
const path = require("node:path");
const { chromium } = require("playwright");
@@ -12,6 +11,7 @@ const ROOT = path.resolve(__dirname, "..");
const BASE_URL = (process.env.MNOTE_UI_BASE_URL || process.env.MNOTE_WEB_SMOKE_BASE_URL || "http://127.0.0.1:3000").replace(/\/+$/, "");
const UI_TIMEOUT_MS = Number(process.env.MNOTE_SMOKE_UI_TIMEOUT_MS || 30_000);
const OUT_DIR = path.join(ROOT, "tmp", "task441-local-folder-cloud-switch-smoke");
const MANAGED_MY_SPACE_ROOT = "/mnt/Data1T/Mnote_data/users/mnote-e2e/workspaces/my-space";
function fileUrl(filePath) {
return `file://${filePath.split(path.sep).map((part, index) => (index === 0 ? "" : encodeURIComponent(part))).join("/")}`;
@@ -42,42 +42,53 @@ async function readCloudState(page) {
return page.evaluate(() => {
const workspaceNode = document.querySelector("#sidebar-file-tree-root[data-workspace-id], #sidebar-tree-root[data-workspace-id], [data-workspace-id]");
const workspaceId = workspaceNode instanceof HTMLElement ? workspaceNode.getAttribute("data-workspace-id") || "" : "";
const url = new URL(window.location.href);
const sourceKind = document.body?.getAttribute("data-mnote-source-kind") || url.searchParams.get("sourceKind") || "";
const rootUri = document.body?.getAttribute("data-mnote-root-uri") || url.searchParams.get("rootUri") || "";
const docRows = Array.from(document.querySelectorAll("#sidebar-file-tree-root .tree-row[data-row-id^='doc:']")).map((row) => ({
rowId: row.getAttribute("data-row-id") || "",
title: row.querySelector(".tree-link-title")?.textContent?.trim() || "",
}));
return {
url: window.location.href,
sourceKind,
rootUri,
workspaceId,
storageValue: window.localStorage.getItem("mnote.workspace.lastCloudWorkspaceId") || "",
docRows,
localMarkdownRows: Array.from(document.querySelectorAll("#sidebar-file-tree-root .tree-row[data-row-id^='local:markdown:']")).map((row) => ({
rowId: row.getAttribute("data-row-id") || "",
title: row.querySelector(".tree-link-title")?.textContent?.trim() || "",
})),
localRows: document.querySelectorAll("#sidebar-file-tree-root .tree-row[data-row-id^='local:']").length,
rootText: document.querySelector("#sidebar-file-tree-root")?.textContent?.slice(0, 500) || "",
};
});
}
async function ensureCloudDocument(page) {
function isHomeWorkspaceState(state) {
return state.sourceKind === "local_folder" && state.workspaceId.startsWith("local-ws:");
}
async function ensureHomeWorkspace(page) {
await openFileTree(page);
let state = await readCloudState(page);
if (state.workspaceId && state.workspaceId !== "default" && state.docRows.length > 0) {
return { workspaceId: state.workspaceId, rowId: state.docRows[0].rowId };
}
if (isHomeWorkspaceState(state)) return state;
const previousPathname = new URL(page.url()).pathname;
await page.getByRole("button", { name: "新建页面" }).click({ timeout: UI_TIMEOUT_MS });
await page.waitForURL((url) => url.pathname.startsWith("/documents/") && url.pathname !== previousPathname, { timeout: UI_TIMEOUT_MS });
const documentId = new URL(page.url()).pathname.split("/").filter(Boolean).pop();
assert(documentId, "新建云空间页面后 URL 缺少 documentId");
await switchCloud(page);
await openFileTree(page);
await page.waitForFunction(
(id) => Boolean(document.querySelector(`#sidebar-file-tree-root .tree-row[data-row-id="doc:${CSS.escape(id)}"]`)),
documentId,
{ timeout: UI_TIMEOUT_MS },
);
state = await readCloudState(page);
assert(state.workspaceId && state.workspaceId !== "default", `云空间 workspaceId 不应为空或 default: ${JSON.stringify(state)}`);
return { workspaceId: state.workspaceId, rowId: `doc:${documentId}` };
assert(isHomeWorkspaceState(state), `切回我的空间后应进入默认本地 workspace: ${JSON.stringify(state)}`);
return state;
}
async function ensureHomeWorkspaceTarget(page) {
const state = await ensureHomeWorkspace(page);
return {
workspaceId: state.workspaceId,
rootUri: state.rootUri,
rowId: state.localMarkdownRows[0]?.rowId || "",
};
}
async function openLocalFolder(page, root) {
@@ -102,55 +113,52 @@ async function switchCloud(page) {
await page.locator('[data-testid="mnote-workspace-source-menu"]').waitFor({ state: "visible", timeout: UI_TIMEOUT_MS });
await page.locator('[data-testid="mnote-switch-cloud-workspace"]').click({ timeout: UI_TIMEOUT_MS });
await page.waitForURL(
(url) => url.pathname === "/" && url.searchParams.get("sourceKind") === "convex_workspace" && !url.searchParams.has("rootUri"),
(url) => url.pathname === "/" && url.searchParams.get("mnoteHome") === "1" && !url.searchParams.has("sourceKind") && !url.searchParams.has("rootUri"),
{ timeout: UI_TIMEOUT_MS },
);
}
async function main() {
await fsp.mkdir(OUT_DIR, { recursive: true });
const localRoot = fs.mkdtempSync(path.join(os.tmpdir(), "mnote-cloud-switch-local-"));
fs.writeFileSync(path.join(localRoot, "README.md"), "# Local README\n", "utf8");
fs.mkdirSync(MANAGED_MY_SPACE_ROOT, { recursive: true });
let localRoot = "";
const browser = await chromium.launch({ headless: process.env.HEADFUL !== "1" });
const context = await browser.newContext({ viewport: { width: 1440, height: 960 } });
const page = await context.newPage();
const result = { ok: false, baseUrl: BASE_URL, localRoot, cloudTarget: null, localState: null, returnedState: null };
const result = { ok: false, baseUrl: BASE_URL, localRoot, homeTarget: null, localState: null, returnedState: null };
try {
await quickLogin(page);
await page.goto(BASE_URL, { waitUntil: "domcontentloaded", timeout: UI_TIMEOUT_MS });
result.cloudTarget = await ensureCloudDocument(page);
result.homeTarget = await ensureHomeWorkspaceTarget(page);
localRoot = fs.mkdtempSync(path.join(MANAGED_MY_SPACE_ROOT, "task441-local-"));
result.localRoot = localRoot;
fs.writeFileSync(path.join(localRoot, "README.md"), "# Local README\n", "utf8");
await openLocalFolder(page, localRoot);
result.localState = await readCloudState(page);
assert(result.localState.storageValue === result.cloudTarget.workspaceId, `进入本地文件夹前应记住真实云空间 workspaceId: ${JSON.stringify(result)}`);
assert(
!result.localState.storageValue || !result.localState.storageValue.startsWith("local-ws:"),
`进入普通本地文件夹不应把我的空间 local-ws 写成云 workspace: ${JSON.stringify(result)}`,
);
await switchCloud(page);
await page.waitForURL((url) => url.searchParams.get("workspaceId") === result.cloudTarget.workspaceId, {
timeout: UI_TIMEOUT_MS,
});
await openFileTree(page);
await page.waitForFunction(
(rowId) => Boolean(document.querySelector(`#sidebar-file-tree-root .tree-row[data-row-id="${CSS.escape(rowId)}"]`)),
result.cloudTarget.rowId,
{ timeout: UI_TIMEOUT_MS },
);
result.returnedState = await readCloudState(page);
assert.equal(result.returnedState.workspaceId, result.cloudTarget.workspaceId, "切回空间后应恢复 workspaceId");
assert.equal(result.returnedState.storageValue, result.cloudTarget.workspaceId, "lastCloudWorkspaceId 不应被 default 污染");
assert.equal(result.returnedState.workspaceId, result.homeTarget.workspaceId, "切回我的空间后应恢复默认本地 workspaceId");
assert.equal(result.returnedState.sourceKind, "local_folder", "切回我的空间后仍应走 local_folder 主链");
assert.equal(result.returnedState.rootUri, result.homeTarget.rootUri, "切回我的空间后应恢复默认 my-space rootUri");
assert.notEqual(result.returnedState.rootUri, fileUrl(localRoot), "切回我的空间后不能继续停留在刚打开的普通本地文件夹");
assert.notEqual(result.returnedState.storageValue, "default", "lastCloudWorkspaceId 不能是 synthetic default");
assert.equal(result.returnedState.localRows, 0, "切回云空间后不应保留本地文件夹 row");
assert(
result.returnedState.docRows.some((row) => row.rowId === result.cloudTarget.rowId),
`切回云空间后原云空间页面 row 应恢复: ${JSON.stringify(result.returnedState)}`,
);
assert(!result.returnedState.storageValue.startsWith("local-ws:"), "lastCloudWorkspaceId 不能是 local-ws:*");
assert(result.returnedState.localRows > 0, `切回我的空间后 File Tree 应恢复本地 row: ${JSON.stringify(result.returnedState)}`);
result.ok = true;
} finally {
await fsp.writeFile(path.join(OUT_DIR, "result.json"), `${JSON.stringify(result, null, 2)}\n`, "utf8");
await browser.close().catch(() => {});
fs.rmSync(localRoot, { recursive: true, force: true });
if (localRoot) fs.rmSync(localRoot, { recursive: true, force: true });
}
console.log(JSON.stringify(result, null, 2));