mnote/scripts/task540-local-folder-event-bus-single-connection-smoke.js

#!/usr/bin/env node
"use strict";

const assert = require("node:assert");
const fs = require("node:fs");
const os = require("node:os");
const path = require("node:path");
const { chromium } = require("playwright");
const {
  BASE_URL,
  UI_TIMEOUT_MS,
  ensureAuthenticated,
} = require("./tree-shell-smoke-helpers");

const TASK = "task540-local-folder-event-bus-single-connection-smoke";
const OUTPUT_DIR = path.join(process.cwd(), "tmp", TASK);
const RESULT_PATH = path.join(OUTPUT_DIR, "result.json");
const CHROMIUM_EXECUTABLE_PATH = process.env.PLAYWRIGHT_CHROMIUM_EXECUTABLE_PATH
  || ["/usr/bin/google-chrome-stable", "/usr/bin/google-chrome", "/snap/bin/chromium", "/usr/bin/chromium"]
    .find((candidate) => fs.existsSync(candidate));

function fileUrl(localPath) {
  return `file://${localPath}`;
}

function localMdDocumentId(relativePath) {
  return `local-md:${relativePath.replaceAll("/", "~2F")}`;
}

function documentUrl(root, relativePath) {
  const url = new URL(`${BASE_URL}/documents/${encodeURIComponent(localMdDocumentId(relativePath))}`);
  url.searchParams.set("sourceKind", "local_folder");
  url.searchParams.set("rootUri", fileUrl(root));
  url.searchParams.set("treeView", "filetree");
  return url.toString();
}

function writeWorkspaceManifest(root, ownerId, workspaceId) {
  fs.mkdirSync(path.join(root, ".mnote"), { recursive: true });
  fs.writeFileSync(
    path.join(root, ".mnote", "workspace.json"),
    `${JSON.stringify({
      workspaceId,
      ownerId,
      createdAt: new Date().toISOString(),
      capabilities: ["local_files", "markdown_edit"],
    }, null, 2)}\n`,
    "utf8",
  );
}

async function main() {
  fs.mkdirSync(OUTPUT_DIR, { recursive: true });
  const suffix = Date.now().toString(36);
  const actorId = "mnote-e2e";
  const workspaceId = `local-ws:${actorId}:task540`;
  const root = fs.mkdtempSync(path.join(os.tmpdir(), "mnote-task540-event-bus-"));
  const rootUri = fileUrl(root);
  const relativePath = "EventBus.md";
  const resourcePath = "Attachment.bin";
  const filePath = path.join(root, relativePath);
  const resourceFilePath = path.join(root, resourcePath);
  writeWorkspaceManifest(root, actorId, workspaceId);
  fs.writeFileSync(filePath, `# Event Bus\n\ntask540-${suffix}\n`, "utf8");
  fs.writeFileSync(resourceFilePath, `task540-resource-${suffix}\n`, "utf8");

  const browser = await chromium.launch({
    headless: process.env.HEADFUL !== "1",
    ...(CHROMIUM_EXECUTABLE_PATH ? { executablePath: CHROMIUM_EXECUTABLE_PATH } : {}),
  });
  const context = await browser.newContext({
    viewport: { width: 1440, height: 960 },
    locale: "zh-CN",
    extraHTTPHeaders: {
      "x-mnote-actor-id": actorId,
      "x-mnote-actor-type": "user",
    },
  });
  await context.addInitScript(() => {
    const sources = [];
    const originalFetch = window.fetch.bind(window);
    window.__mnoteTask540StatRequests = [];
    window.__mnoteTask540FileProjectionRequests = [];
    window.__mnoteTask540SidebarProjectionRequests = [];
    window.fetch = function patchedTask540Fetch(input, init) {
      const url = typeof input === "string" ? input : String(input?.url || "");
      if (url.includes("/api/local-folder/files/stat")) {
        window.__mnoteTask540StatRequests.push(url);
      }
      if (url.includes("/api/tree/projections/file")) {
        window.__mnoteTask540FileProjectionRequests.push(url);
      }
      if (url.includes("/api/tree/projections/sidebar")) {
        window.__mnoteTask540SidebarProjectionRequests.push(url);
      }
      return originalFetch(input, init);
    };
    class FakeEventSource {
      constructor(input) {
        this.url = String(input || "");
        this.readyState = 0;
        this.listeners = new Map();
        sources.push(this);
        setTimeout(() => this.dispatch("open", {}), 0);
      }
      addEventListener(name, handler) {
        if (!this.listeners.has(name)) this.listeners.set(name, []);
        this.listeners.get(name).push(handler);
      }
      removeEventListener(name, handler) {
        const list = this.listeners.get(name) || [];
        this.listeners.set(name, list.filter((candidate) => candidate !== handler));
      }
      dispatch(name, payload) {
        const event = { data: JSON.stringify(payload || {}), lastEventId: String(payload?.revision || "") };
        (this.listeners.get(name) || []).forEach((handler) => handler(event));
      }
      close() {
        this.readyState = 2;
      }
    }
    window.EventSource = FakeEventSource;
    window.__mnoteTask540EventSources = sources;
    window.__mnoteTask540CompatEvents = [];
    window.__mnoteTask540FileChangeBatches = [];
    window.__mnoteTask540FileChangeReactions = [];
    window.addEventListener("tree:local-folder-watch-batch", (event) => {
      window.__mnoteTask540CompatEvents.push(event.detail || {});
    });
    window.addEventListener("mnote:file-change-batch", (event) => {
      window.__mnoteTask540FileChangeBatches.push(event.detail || {});
    });
    window.addEventListener("mnote:file-change-reaction", (event) => {
      window.__mnoteTask540FileChangeReactions.push(event.detail || {});
    });
  });

  const page = await context.newPage();
  const statRequests = [];
  page.on("request", (request) => {
    const url = request.url();
    if (url.includes("/api/local-folder/files/stat")) {
      statRequests.push(url);
    }
  });
  await page.route("**/api/user/access-policy**", async (route) => {
    await route.fulfill({
      status: 200,
      headers: { "content-type": "application/json" },
      body: JSON.stringify({
        ok: true,
        controlPlane: "sqlite",
        grants: [{
          id: `grant_task540_${suffix}`,
          userId: actorId,
          workspaceId,
          rootUri,
          rootPath: root,
          permission: "write",
          recursive: true,
          capabilities: ["markdown_edit"],
          source: "user",
          status: "active",
        }],
      }),
    });
  });
  await page.route("**/api/ui/preferences**", async (route) => {
    await route.fulfill({
      status: 200,
      headers: { "content-type": "application/json" },
      body: JSON.stringify({ ok: true, owner: "mnote-web", result: {} }),
    });
  });

  try {
    await ensureAuthenticated(page, context.request);
    await page.goto(documentUrl(root, relativePath), { waitUntil: "domcontentloaded", timeout: UI_TIMEOUT_MS });
    await page.waitForFunction(() => window.__mnoteLocalFolderEventBus, null, { timeout: UI_TIMEOUT_MS });
    await page.waitForFunction(() => {
      const sources = window.__mnoteTask540EventSources || [];
      return sources.filter((source) => String(source.url || "").includes("/api/local-folder/events")).length === 1;
    }, null, { timeout: UI_TIMEOUT_MS });

    await page.waitForFunction(() => window.__mnoteDocumentPaneRuntime?.openResourceInActiveTab, null, { timeout: UI_TIMEOUT_MS });
    const openedResource = await page.evaluate(async ({ rootUriValue, pathValue, workspaceIdValue }) => {
      const href = new URL("/api/local-folder/resource/read", window.location.origin);
      href.searchParams.set("rootUri", rootUriValue);
      href.searchParams.set("path", pathValue);
      return await window.__mnoteDocumentPaneRuntime.openResourceInActiveTab({
        objectIdentity: `local-resource:${pathValue}`,
        assetId: `task540:${pathValue}`,
        title: pathValue,
        fileName: pathValue,
        kind: "file",
        sourceKind: "local_folder",
        rootUri: rootUriValue,
        workspaceId: workspaceIdValue,
        path: pathValue,
        href: href.toString(),
      });
    }, { rootUriValue: rootUri, pathValue: resourcePath, workspaceIdValue: workspaceId });
    assert.equal(openedResource, true, "resource tab 应能打开");
    await page.waitForFunction(() => {
      const panel = document.querySelector('.mnote-resource-tab-panel[data-resource-path="Attachment.bin"]');
      return panel?.getAttribute("data-mnote-resource-watch-ready") === "event-bus";
    }, null, { timeout: UI_TIMEOUT_MS });

    await page.evaluate(() => {
      window.__mnoteTask540StatRequests = [];
      window.__mnoteTask540FileProjectionRequests = [];
      window.__mnoteTask540SidebarProjectionRequests = [];
    });
    await page.evaluate(({ rootUriValue, workspaceIdValue, pagePath, assetPath }) => {
      const detail = {
        source: "synthetic_page_ai_receipt",
        reason: "agent_run_receipt",
        rootUri: rootUriValue,
        payload: {
          schema: "mnote.local_folder.watch_batch.v1",
          source: "agent_run_receipt",
          rootUri: rootUriValue,
          revision: "task540-revision",
          changedPaths: [
            {
              relativePath: pagePath,
              documentId: `local-md:${pagePath.replaceAll("/", "~2F")}`,
              changeType: "modified",
              eventKind: "Modify(Data(Content))",
              observedFileVersion: "sha256:task540-observed",
              bufferFileVersion: "sha256:task540-buffer",
              selfWriteEcho: false,
            },
            { relativePath: assetPath, changeType: "modified" },
          ],
          affectedParents: [{ relativePath: "", reason: "task540" }],
        },
      };
      window.__mnoteLocalFolderEventBus.emitChangedFiles({
        source: detail.source,
        reason: detail.reason,
        rootUri: rootUriValue,
        workspaceId: workspaceIdValue,
        changedFiles: detail.payload.changedPaths,
        affectedParents: detail.payload.affectedParents,
      });
      window.__mnoteLocalFolderEventBus.emitChangedFiles({
        source: detail.source,
        reason: detail.reason,
        rootUri: rootUriValue,
        workspaceId: workspaceIdValue,
        changedFiles: detail.payload.changedPaths,
        affectedParents: detail.payload.affectedParents,
      });
    }, { rootUriValue: rootUri, workspaceIdValue: workspaceId, pagePath: relativePath, assetPath: resourcePath });

    await page.waitForFunction(() => {
      const events = window.__mnoteTask540CompatEvents || [];
      return events.some((event) => event?.source === "synthetic_page_ai_receipt" && event?.viaEventBus === true);
    }, null, { timeout: UI_TIMEOUT_MS });
    await page.waitForFunction((pathValue) => {
      const urls = window.__mnoteTask540StatRequests || [];
      return urls.some((url) => String(url).includes(encodeURIComponent(pathValue)) || String(url).includes(pathValue));
    }, resourcePath, { timeout: 3000 }).catch(() => undefined);
    await page.waitForFunction(() => {
      const urls = window.__mnoteTask540FileProjectionRequests || [];
      return urls.length >= 1;
    }, null, { timeout: UI_TIMEOUT_MS });

    const result = await page.evaluate(() => {
      const sources = (window.__mnoteTask540EventSources || [])
        .filter((source) => String(source.url || "").includes("/api/local-folder/events"))
        .map((source) => source.url);
      const compatEvents = window.__mnoteTask540CompatEvents || [];
      const resourcePanel = document.querySelector('.mnote-resource-tab-panel[data-resource-path="Attachment.bin"]');
      return {
        sources,
        busState: document.documentElement.getAttribute("data-mnote-local-folder-event-bus") || "",
        connectionCount: document.documentElement.getAttribute("data-mnote-local-folder-event-bus-connections") || "",
        lastSource: document.documentElement.getAttribute("data-mnote-local-folder-event-bus-last-source") || "",
        lastReason: document.documentElement.getAttribute("data-mnote-local-folder-event-bus-last-reason") || "",
        compatEventCount: compatEvents.length,
        lastCompatViaEventBus: Boolean(compatEvents.at(-1)?.viaEventBus),
        fileChangeService: document.documentElement.getAttribute("data-mnote-file-change-service") || "",
        fileChangeSchema: document.documentElement.getAttribute("data-mnote-file-change-service-last-schema") || "",
        fileChangeCount: document.documentElement.getAttribute("data-mnote-file-change-service-last-count") || "",
        fileChangeDroppedCount: document.documentElement.getAttribute("data-mnote-file-change-service-dropped-count") || "",
        fileChangeLastReaction: document.documentElement.getAttribute("data-mnote-file-change-service-last-reaction") || "",
        fileChangeBatchCount: (window.__mnoteTask540FileChangeBatches || []).length,
        fileChangeReactionTypes: (window.__mnoteTask540FileChangeReactions || []).map((event) => event.type),
        lastFileChangeBatch: (window.__mnoteTask540FileChangeBatches || []).at(-1) || null,
        hasPageAiSyntheticCompatEvent: compatEvents.some((event) => (
          event?.source === "synthetic_page_ai_receipt"
          && event?.reason === "agent_run_receipt"
          && event?.viaEventBus === true
        )),
        resourceWatchReady: resourcePanel?.getAttribute("data-mnote-resource-watch-ready") || "",
        browserStatRequests: window.__mnoteTask540StatRequests || [],
        fileProjectionRequests: window.__mnoteTask540FileProjectionRequests || [],
        sidebarProjectionRequests: window.__mnoteTask540SidebarProjectionRequests || [],
      };
    });

    assert.equal(result.sources.length, 1, `同一页面应只有一个 local-folder EventSource，实际 ${result.sources.length}`);
    assert.equal(result.busState, "ready", "event bus diagnostics 未 ready");
    assert.equal(result.connectionCount, "1", "event bus diagnostics connectionCount 应为 1");
    assert.equal(result.hasPageAiSyntheticCompatEvent, true, "Page AI synthetic 事件应保留 source/reason 并通过 bus 派发兼容事件");
    assert.equal(result.lastCompatViaEventBus, true, "兼容 tree:local-folder-watch-batch 必须标记 viaEventBus");
    assert.equal(result.fileChangeService, "ready", "FileChangeService diagnostics 未 ready");
    assert.equal(typeof result.lastFileChangeBatch, "object", "应暴露标准 FileChange batch");
    assert.equal(result.fileChangeSchema, "mnote.file_change_batch.v1", "应生成标准 FileChange batch");
    assert.equal(result.fileChangeCount, "2", "同 tick 两个 changed path 应进入同一个 FileChange batch");
    assert.equal(result.fileChangeDroppedCount, "0", "rootUri 内 changed path 不应被丢弃");
    assert.equal(result.fileChangeBatchCount, 1, "同 tick 两个 receipt 应合并成一次标准 FileChange batch");
    assert(
      result.fileChangeReactionTypes.includes("refresh_current_document"),
      `当前 Markdown 应生成 refresh_current_document reaction: ${JSON.stringify(result.fileChangeReactionTypes)}`,
    );
    assert(
      result.fileChangeReactionTypes.includes("refresh_resource_tab"),
      `资源文件应生成 refresh_resource_tab reaction: ${JSON.stringify(result.fileChangeReactionTypes)}`,
    );
    assert.equal(
      result.lastFileChangeBatch?.changes?.[0]?.observedFileVersion,
      "sha256:task540-observed",
      "FileChange batch 应保留 observedFileVersion",
    );
    assert.equal(
      result.lastFileChangeBatch?.changes?.[0]?.selfWriteEcho,
      false,
      "FileChange batch 应保留 selfWriteEcho",
    );
    assert.equal(result.resourceWatchReady, "event-bus", "resource tab watch 应通过 event bus 准备好");
    assert(
      result.browserStatRequests.some((url) => String(url).includes(encodeURIComponent(resourcePath)) || String(url).includes(resourcePath)),
      "resource-changed 应触发当前资源 stat 刷新",
    );
    assert.equal(result.fileProjectionRequests.length, 1, "同 tick 两个 receipt 应只触发一次 filetree parent projection 刷新");
    assert.equal(result.sidebarProjectionRequests.length, 0, "content-only receipt 不应触发 sidebar projection 刷新");

    fs.writeFileSync(RESULT_PATH, `${JSON.stringify({ ok: true, root, result, statRequests }, null, 2)}\n`, "utf8");
    console.log(`[${TASK}] ok`, RESULT_PATH);
  } finally {
    await browser.close();
  }
}

main().catch((error) => {
  fs.mkdirSync(OUTPUT_DIR, { recursive: true });
  fs.writeFileSync(RESULT_PATH, `${JSON.stringify({ ok: false, error: error.stack || String(error) }, null, 2)}\n`, "utf8");
  console.error(`[${TASK}] failed`, error);
  process.exit(1);
});