makelore/tests/unit/coding-product-services.test.ts

// @vitest-environment node

import { mkdtemp, rm, writeFile } from 'node:fs/promises';
import { tmpdir } from 'node:os';
import path from 'node:path';
import { afterEach, describe, expect, it } from 'vitest';
import type { AgentBrowserModule } from '../../electron/agent-browser';
import {
  CodingProductHostError,
  createCodingProductHost,
} from '../../electron/api/coding-product-services';
import { CodingAttachmentStore } from '../../electron/coding-projects/attachment-store';
import { createCodingConversationStore } from '../../electron/coding-projects/conversation-store';
import {
  createCodingProjectAgent,
  createCodingProjectMetadata,
} from '../../electron/coding-projects/project-config';
import { PiProductTools } from '../../electron/coding-runtime/pi/product-tools';

const roots: string[] = [];
const conversationId = '11111111-1111-4111-8111-111111111111';

afterEach(async () => {
  await Promise.all(roots.splice(0).map((root) => rm(root, { recursive: true, force: true })));
});

async function temporaryRoot(prefix: string): Promise<string> {
  const root = await mkdtemp(path.join(tmpdir(), prefix));
  roots.push(root);
  return root;
}

async function configuredProject(): Promise<string> {
  const root = await temporaryRoot('makelore-pi-products-');
  await createCodingProjectMetadata(root, { now: '2026-08-23T00:00:00.000Z' });
  await createCodingProjectAgent(root, {
    id: 'builder',
    avatarId: 'avatar-01',
    roleName: '实现者',
    name: 'Builder',
    model: null,
    modelResolution: 'required',
    skillIds: ['agent-browser', 'grilling'],
    responsibility: {
      mission: 'Implement changes', owns: [], boundaries: [], collaborators: [], principles: [],
    },
  }, { now: '2026-08-23T00:00:00.000Z' });
  await createCodingConversationStore(root, {
    createId: () => conversationId,
    now: () => '2026-08-23T00:00:00.000Z',
  }).create({
    agentId: 'builder', title: 'PI-105', model: null, modelResolution: 'required',
  });
  return root;
}

function productTools(root: string): PiProductTools {
  return new PiProductTools({
    browser: {} as AgentBrowserModule,
    attachments: new CodingAttachmentStore(path.join(root, 'attachments')),
    bundledSkillsDir: path.resolve('resources/coding-skills'),
  });
}

describe('PI-105 product Host composition', () => {
  it('projects managed skills and a stable command catalog without raw Pi fields', async () => {
    const root = await configuredProject();
    const tools = productTools(root);
    const host = createCodingProductHost({
      getActiveProject: async () => ({ id: 'project-a', path: root }),
      productTools: tools,
      listPiCommands: async () => ({
        commands: [
          { name: 'custom', description: 'Custom Pi command', token: 'pi-secret' },
          { name: 'compact', description: 'Must not shadow Makelore' },
          { name: '../unsafe', description: 'Invalid command' },
        ],
        apiKey: 'raw-provider-secret',
      }),
    });

    const skills = await host.listSkills('builder');
    expect(skills.filter(({ selected }) => selected).map(({ id }) => id)).toEqual([
      'agent-browser', 'grilling',
    ]);
    const commands = await host.listCommands(conversationId);
    expect(commands.slice(0, 5).map(({ source }) => source)).toEqual([
      'makelore', 'makelore', 'makelore', 'makelore', 'makelore',
    ]);
    expect(commands).toContainEqual(expect.objectContaining({ name: 'custom', source: 'pi' }));
    expect(commands).toContainEqual(expect.objectContaining({ name: 'agent-browser', source: 'skill' }));
    expect(commands.filter(({ name }) => name === 'compact')).toHaveLength(1);
    const serialized = JSON.stringify({ skills, commands });
    expect(serialized).not.toContain(root);
    expect(serialized).not.toContain('pi-secret');
    expect(serialized).not.toContain('raw-provider-secret');
    expect(serialized).not.toMatch(/"(todo|share|revert|unrevert)"/);
  });

  it('reads exact-run changes from the same PiProductTools tracker instance', async () => {
    const root = await configuredProject();
    await writeFile(path.join(root, 'notes.txt'), 'baseline\n', 'utf8');
    const tools = productTools(root);
    const host = createCodingProductHost({
      getActiveProject: async () => ({ id: 'project-a', path: root }),
      productTools: tools,
    });
    await tools.beginRun({ conversationId, runId: 'run-a', projectPath: root });
    await writeFile(path.join(root, 'notes.txt'), 'changed\n', 'utf8');
    await tools.recordTouchedPaths(conversationId, 'run-a', ['notes.txt']);

    expect(await host.getChanges(conversationId)).toMatchObject({
      conversationId, runId: 'run-a', git: false,
      files: [{ path: 'notes.txt', status: 'modified', preview: 'changed\n' }],
    });
    expect(JSON.stringify(await host.getChanges(conversationId))).not.toContain(root);
  });

  it('returns typed project, Agent, and Conversation errors', async () => {
    const root = await configuredProject();
    const tools = productTools(root);
    const unavailable = createCodingProductHost({
      getActiveProject: async () => null,
      productTools: tools,
    });
    await expect(unavailable.fileStatus()).rejects.toMatchObject<CodingProductHostError>({
      status: 409, code: 'CODING_ACTIVE_PROJECT_REQUIRED',
    });

    const host = createCodingProductHost({
      getActiveProject: async () => ({ id: 'project-a', path: root }),
      productTools: tools,
    });
    await expect(host.listSkills('missing')).rejects.toMatchObject<CodingProductHostError>({
      status: 404, code: 'CODING_AGENT_NOT_FOUND',
    });
    await expect(host.listCommands('22222222-2222-4222-8222-222222222222'))
      .rejects.toMatchObject<CodingProductHostError>({
        status: 404, code: 'CODING_CONVERSATION_NOT_FOUND',
      });
  });
});