import { test, expect } from '@playwright/test';
import { _electron as electron } from 'playwright';
import { promises as fs } from 'node:fs';
import nodePath from 'node:path';
import os from 'node:os';

const MAIN_JS = nodePath.join(process.cwd(), 'dist-electron/main/index.js');

async function launch(userData: string) {
  const app = await electron.launch({ args: ['--user-data-dir=' + userData, MAIN_JS] });
  const page = await app.firstWindow();
  await page.waitForLoadState('domcontentloaded');
  const invoke = (ch: string, args: any) => page.evaluate(({ ch, args }) => (globalThis as any).dsh.invoke(ch, args), { ch, args });
  return { app, page, invoke };
}

test('知识图谱：卡片+连线（消费 IPC.graph）', async () => {
  const tmp = await fs.mkdtemp(nodePath.join(os.tmpdir(), 'hn-graph-'));
  const userData = nodePath.join(tmp, 'userdata');
  const vault = nodePath.join(tmp, 'vault');
  await fs.mkdir(userData, { recursive: true });
  await fs.mkdir(vault, { recursive: true });

  const a = await launch(userData);
  await a.invoke('vault:switch', { root: vault });
  await a.invoke('file:create', { basePath: '', name: 'a', isFolder: false });
  await a.invoke('file:create', { basePath: '', name: 'b', isFolder: false });
  await a.invoke('doc:write', { path: 'a.md', meta: { title: 'A篇', links: ['b.md'] }, body: '# A\n' });
  await a.invoke('doc:write', { path: 'b.md', meta: { title: 'B篇' }, body: '# B\n' });

  // IPC.graph：2 节点 + a→b 边
  const gd: { nodes: { id: string; title: string; star: boolean; color: string }[]; edges: { source: string; target: string }[] } = await a.invoke('graph', null);
  expect(gd.nodes.length).toBe(2);
  expect(gd.nodes.some(n => n.title === 'A篇')).toBe(true);
  expect(gd.edges.some(e => e.source === 'a.md' && e.target === 'b.md')).toBe(true);

  // UI：先经 TitleBar 切到图谱视图（view 默认 editor），再切「知识图谱」模式
  await a.page.getByTitle('切换 编辑器/图谱').click();
  await a.page.getByRole('button', { name: '知识图谱' }).click();
  await expect(a.page.locator('svg rect')).toHaveCount(4, { timeout: 10000 });
  await a.app.close();
});