Genarrative/apps/ai-game-creator-shell/tests/resourceBookLayout.test.ts

import { describe, expect, it } from 'vitest';

import {
  buildResourceBookScenePlan,
  groupResourceBookStacks,
  RESOURCE_BOOK_OVERVIEW_STACK_LIMIT,
  resourceBookOverviewCardLayout,
  resourceBookSceneCardKey,
  selectResourceBookOverviewCards,
} from '../src/view/project-development/resourceBookLayout';
import type { ProjectResource } from '../src/view/project-development/resourceProjectionModel';

function resource(id: string): ProjectResource {
  return {
    id,
    category: 'unclassified',
    subtype: 'image',
    label: id,
    path: `assets/${id}.png`,
    mediaType: 'image/png',
    sourceLabel: '测试',
    taskTitle: null,
    manifestAssetId: id,
    producerTaskId: null,
    externalResourceId: null,
    referenceResourceIds: [],
    dependencies: [],
    dependencyDepth: 0,
  };
}

const rect = { left: 100, top: 200, width: 280, height: 210 };

describe('resourceBookOverviewCardLayout', () => {
  it('keeps the stack tilt bounded however many resources share one type', () => {
    const rotations = [0, 1, 2, 3, 12, 59].map(
      (stackIndex) =>
        resourceBookOverviewCardLayout({ rect, stackIndex, stackColumn: 0 })
          .rotation,
    );

    for (const rotation of rotations) {
      expect(Math.abs(rotation)).toBeLessThanOrEqual(3);
    }
    expect(rotations[0]).toBe(0.5);
    expect(rotations.at(-1)).toBe(rotations[2]);
  });

  it('stops offsetting deeper cards once the visible stack depth is reached', () => {
    const positions = [0, 1, 2, 3, 59].map(
      (stackIndex) =>
        resourceBookOverviewCardLayout({ rect, stackIndex, stackColumn: 0 }).y,
    );

    expect(positions[0]).toBe(rect.top + 58);
    expect(positions[1]).toBe(rect.top + 63);
    expect(positions[2]).toBe(rect.top + 68);
    expect(positions.at(-1)).toBe(positions[2]);
  });

  it('limits how many cards the overview renders per type stack', () => {
    expect(RESOURCE_BOOK_OVERVIEW_STACK_LIMIT).toBe(3);
  });

  describe('selectResourceBookOverviewCards', () => {
    const cards = ['a', 'b', 'c', 'd'].map(resource);
    const positions = new Map([
      ['a', { x: 0, y: 0 }],
      ['b', { x: 1_000, y: 0 }],
      ['c', { x: 2_000, y: 0 }],
      ['d', { x: 3_000, y: 0 }],
    ]);
    const cardSizes = new Map<string, { width: number; height: number }>();

    it('keeps the first visible cards when the view center is unknown', () => {
      expect(
        selectResourceBookOverviewCards({
          resources: cards,
          visibleResourceIds: new Set(['a', 'b', 'c', 'd']),
          positions,
          cardSizes,
          viewCenter: undefined,
        }).map((item) => item.id),
      ).toEqual(['a', 'b', 'c']);
    });

    it('prefers the cards nearest the current view center', () => {
      expect(
        selectResourceBookOverviewCards({
          resources: cards,
          visibleResourceIds: new Set(['a', 'b', 'c', 'd']),
          positions,
          cardSizes,
          viewCenter: { x: 3_000, y: 64 },
        }).map((item) => item.id),
      ).toEqual(['d', 'c', 'b']);
    });

    it('never surfaces cards hidden by the current filter', () => {
      expect(
        selectResourceBookOverviewCards({
          resources: cards,
          visibleResourceIds: new Set(['a', 'c', 'd']),
          positions,
          cardSizes,
          viewCenter: { x: 1_000, y: 64 },
        }).map((item) => item.id),
      ).toEqual(['a', 'c', 'd']);
    });
  });
});

describe('groupResourceBookStacks', () => {
  it('keeps resources of one type in a single stack', () => {
    const stacks = groupResourceBookStacks([
      resource('overview-art-0'),
      resource('overview-art-1'),
      resource('overview-art-2'),
      resource('overview-art-3'),
    ]);

    expect(stacks).toHaveLength(1);
    expect(stacks[0]?.[1]).toHaveLength(4);
  });
});

describe('buildResourceBookScenePlan', () => {
  const resources = ['art-0', 'art-1', 'art-2', 'art-3'].map(resource);
  const positions = new Map(
    resources.map((item, index) => [item.id, { x: index * 200, y: 0 }]),
  );
  const cardSizes = new Map(
    resources.map((item) => [item.id, { width: 180, height: 128 }]),
  );
  const base = {
    visibleCategoryOrder: ['unclassified'] as const,
    resourcesByCategory: new Map([['unclassified', resources]]),
    overviewRects: new Map([['unclassified', rect]]),
    positions,
    cardSizes,
    visibleResourceIds: new Set(resources.map((item) => item.id)),
    categoryViewCenters: new Map([['unclassified', { x: 0, y: 64 }]]),
    cardsReady: true,
  };

  it('renders the bounded overview stack on the steady main canvas', () => {
    const plan = buildResourceBookScenePlan({
      ...base,
      state: {
        view: 'main',
        category: null,
        phase: 'idle',
        token: 0,
      },
    });

    expect(plan).toHaveLength(1);
    expect(plan[0]!.cards).toHaveLength(RESOURCE_BOOK_OVERVIEW_STACK_LIMIT);
    expect(plan[0]!.cards[0]!.presentation).toBe('overview');
    expect(plan[0]!.cards[0]!.stackCount).toBe(
      RESOURCE_BOOK_OVERVIEW_STACK_LIMIT,
    );
    expect(plan[0]!.cards[0]!.layout).toMatchObject({
      x: rect.left + 18,
      y: rect.top + 58,
      width: 92,
      height: 64,
    });
    expect(plan[0]!.titlebarActive).toBe(false);
    expect(plan[0]!.titlebarOpacity).toBe(1);
    expect(plan[0]!.resourceCount).toBe(4);
  });

  it('keeps other categories mounted for a fade-out while entering', () => {
    const plan = buildResourceBookScenePlan({
      ...base,
      visibleCategoryOrder: ['unclassified', 'scene'],
      resourcesByCategory: new Map([
        ['unclassified', resources],
        ['scene', [resource('code-0')]],
      ]),
      overviewRects: new Map([
        ['unclassified', rect],
        ['scene', { ...rect, left: 500 }],
      ]),
      state: {
        view: 'child',
        category: 'unclassified',
        phase: 'entering',
        token: 1,
      },
    });

    const artGroup = plan.find((group) => group.category === 'unclassified')!;
    const codeGroup = plan.find((group) => group.category === 'scene')!;
    expect(artGroup.cards.every((card) => card.presentation === 'child')).toBe(
      true,
    );
    expect(artGroup.cards).toHaveLength(4);
    expect(artGroup.titlebarActive).toBe(true);
    // 其他栏目留在 DOM 里淡出，而不是直接卸载。
    expect(
      codeGroup.cards.every((card) => card.presentation === 'overview'),
    ).toBe(true);
    expect(codeGroup.titlebarActive).toBe(false);
    expect(codeGroup.titlebarOpacity).toBe(0);
  });

  it('marks cards that do not return to the stack as exiting', () => {
    const plan = buildResourceBookScenePlan({
      ...base,
      state: {
        view: 'main',
        category: null,
        phase: 'returning-main',
        token: 2,
      },
      exitingCategory: 'unclassified',
    });

    const cards = plan[0]!.cards;
    expect(
      cards.filter((card) => card.presentation === 'overview'),
    ).toHaveLength(RESOURCE_BOOK_OVERVIEW_STACK_LIMIT);
    const exiting = cards.filter((card) => card.presentation === 'exiting');
    expect(exiting).toHaveLength(1);
    // 淡出节点按子画布布局渲染，位置由 FLIP 钉在 First。
    expect(exiting[0]!.layout.x).toBe(positions.get('art-3')!.x);
  });

  it('keeps search-driven removals mounted without duplicating keys', () => {
    const plan = buildResourceBookScenePlan({
      ...base,
      state: { view: 'main', category: null, phase: 'idle', token: 0 },
      exitingCards: [
        {
          key: resourceBookSceneCardKey('art-3'),
          category: 'unclassified',
          resource: resources[3]!,
          presentation: 'exiting',
          stackIndex: 0,
          stackColumn: 0,
          stackCount: 1,
          layout: {
            x: 0,
            y: 0,
            width: 180,
            height: 128,
            dragX: 0,
            dragY: 0,
            rotation: 0,
          },
        },
        {
          key: resourceBookSceneCardKey('art-0'),
          category: 'unclassified',
          resource: resources[0]!,
          presentation: 'exiting',
          stackIndex: 0,
          stackColumn: 0,
          stackCount: 1,
          layout: {
            x: 0,
            y: 0,
            width: 180,
            height: 128,
            dragX: 0,
            dragY: 0,
            rotation: 0,
          },
        },
      ],
    });

    const keys = plan[0]!.cards.map((card) => card.key);
    expect(keys).toContain(resourceBookSceneCardKey('art-3'));
    expect(
      keys.filter((key) => key === resourceBookSceneCardKey('art-0')),
    ).toHaveLength(1);
  });
});