import { describe, expect, it } from 'vitest';
import type { AgentConfig } from '@/lib/orchestration/registry/types';
import { buildDirectorPrompt } from '@/lib/chat/pi/prompts';
import { buildReadSceneTool } from '@/lib/chat/pi/tools/read-scene';
import type { StatelessChatRequest } from '@/lib/types/chat';

const teacher: AgentConfig = {
  id: 'teacher-1',
  name: 'Teacher',
  role: 'teacher',
  persona: 'Teach from evidence.',
  avatar: '',
  color: '#3366ff',
  allowedActions: [],
  priority: 10,
  createdAt: new Date('2026-01-01T00:00:00Z'),
  updatedAt: new Date('2026-01-01T00:00:00Z'),
  isDefault: true,
};

function makeBody(question: string): StatelessChatRequest {
  return {
    messages: [
      {
        id: 'user-1',
        role: 'user',
        parts: [{ type: 'text', text: question }],
      },
    ],
    storeState: {
      stage: {
        id: 'stage-1',
        name: 'Quadratic Functions',
        createdAt: 1,
        updatedAt: 77,
      },
      outlines: [
        {
          id: 'outline-1',
          type: 'slide',
          title: 'Vertex form',
          description: 'The current slide introduces the vertex form.',
          keyPoints: ['y=a(x-h)^2+k'],
          order: 1,
        },
        {
          id: 'outline-2',
          type: 'slide',
          title: 'Axis of symmetry',
          description: 'The next slide derives the axis of symmetry.',
          keyPoints: ['x=h'],
          order: 2,
        },
      ],
      scenes: [
        {
          id: 'scene-1',
          outlineId: 'outline-1',
          stageId: 'stage-1',
          title: 'Vertex form',
          order: 1,
          type: 'slide',
          content: { type: 'slide', canvas: { elements: [] } as never },
        },
        {
          id: 'scene-2',
          outlineId: 'outline-2',
          stageId: 'stage-1',
          title: 'Axis of symmetry',
          order: 2,
          type: 'slide',
          updatedAt: 88,
          content: {
            type: 'slide',
            canvas: {
              elements: [
                {
                  id: 'next-slide-proof',
                  type: 'text',
                  content: 'For y=a(x-h)^2+k, the axis of symmetry is x=h.',
                  left: 40,
                  top: 80,
                  width: 480,
                  height: 60,
                },
              ],
            } as never,
          },
        },
      ],
      currentSceneId: 'scene-1',
      mode: 'autonomous',
      whiteboardOpen: false,
    },
    config: {
      agentIds: [teacher.id],
      agentConfigs: [teacher],
      sessionType: 'qa',
    },
    apiKey: '',
  } as StatelessChatRequest;
}

describe('Pi Director scene tool wiring', () => {
  it('can resolve a question whose evidence exists only on the next slide', async () => {
    const body = makeBody('下一页才会讲到的对称轴是什么？');
    const prompt = buildDirectorPrompt(body, [teacher], 4);
    const tool = buildReadSceneTool({ body });

    const result = await tool.execute('read-next', { sceneId: 'scene-2' });
    const text = result.content[0]?.type === 'text' ? result.content[0].text : '';

    expect(prompt).toContain('order=2, sceneId="scene-2"');
    expect(prompt).toContain('another scene, or the next scene');
    expect(text).toContain('the axis of symmetry is x=h');
    expect(result.details).toMatchObject({
      status: 'ok',
      sceneId: 'scene-2',
      revision: '88',
      source: 'request_start_snapshot',
    });
  });
});