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

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

import {
  canonicalProjectedResourceMediaType,
  resolveProjectCharacterAnimationCapability,
  resolveProjectResourceEditCapability,
  resolveResourceEditRequestIdentity,
  RESOURCE_EDIT_PROMPT_TRUNCATED_NOTICE,
  resourceEditPromptMaxLength,
  truncateResourceEditPrompt,
} from '../src/view/project-development/resourceEditModel';
import type { ProjectResource } from '../src/view/project-development/resourceProjectionModel';

function resource(
  input: Partial<ProjectResource> &
    Pick<ProjectResource, 'id' | 'category' | 'path'>,
): ProjectResource {
  return {
    subtype: 'task-artifact',
    label: input.path,
    mediaType: '',
    sourceLabel: '测试来源',
    taskTitle: null,
    manifestAssetId: null,
    producerTaskId: 'task-1',
    externalResourceId: null,
    referenceResourceIds: [],
    dependencies: [],
    dependencyDepth: 0,
    ...input,
  };
}

describe('全类型资源编辑能力模型', () => {
  it('把现役资源稳定分流到图片派生、媒体派生、文本派生和版本分支', () => {
    const cases: Array<{
      resource: ProjectResource;
      route: 'derive';
      editKind?: string;
    }> = [
      {
        resource: resource({
          id: 'asset:png',
          category: 'unclassified',
          path: 'assets/hero.png',
          mediaType: 'image/png',
          manifestAssetId: 'png',
        }),
        route: 'derive',
        editKind: 'image-reference',
      },
      {
        resource: resource({
          id: 'task:art:hero.webp',
          category: 'unclassified',
          path: 'hero.webp',
          mediaType: '美术产物',
        }),
        route: 'derive',
        editKind: 'image-reference',
      },
      {
        resource: resource({
          id: 'asset:gif',
          category: 'unclassified',
          path: 'assets/hero.gif',
          mediaType: 'image/gif',
          manifestAssetId: 'gif',
        }),
        route: 'derive',
        editKind: 'image-reference',
      },
      {
        resource: resource({
          id: 'asset:svg',
          category: 'unclassified',
          path: 'assets/icon.svg',
          mediaType: 'image/svg+xml',
          manifestAssetId: 'svg',
        }),
        route: 'derive',
        editKind: 'svg',
      },
      {
        resource: resource({
          id: 'task:art:intro.mp4',
          category: 'unclassified',
          path: 'assets/intro.mp4',
          mediaType: '美术产物',
        }),
        route: 'derive',
        editKind: 'video',
      },
      {
        resource: resource({
          id: 'asset:bgm',
          category: 'audio',
          path: 'audio/theme.ogg',
          subtype: 'background-music',
          mediaType: 'audio/ogg',
          manifestAssetId: 'bgm',
        }),
        route: 'derive',
        editKind: 'background-music',
      },
      {
        resource: resource({
          id: 'asset:sfx',
          category: 'audio',
          path: 'audio/hit.wav',
          subtype: 'sound-effect',
          mediaType: 'audio/wav',
          manifestAssetId: 'sfx',
        }),
        route: 'derive',
        editKind: 'sound-effect',
      },
      {
        resource: resource({
          id: 'task:code:main.ts',
          category: 'document',
          path: 'src/main.ts',
          mediaType: '项目文档',
        }),
        route: 'derive',
        editKind: 'text',
      },
      {
        resource: resource({
          id: 'agent-result:design:message-1',
          category: 'document',
          path: '专业 Agent 文本回执',
          subtype: 'agent-result',
          producerTaskId: null,
          content: '回执',
        }),
        route: 'derive',
        editKind: 'agent-result',
      },
      {
        resource: resource({
          id: 'version:v1',
          category: 'version',
          path: '项目版本 · v1',
          subtype: 'project-version',
          producerTaskId: null,
          version: {
            versionId: 'v1',
            parentVersionId: null,
            projectRevision: 1,
            resourceBindings: [],
            createdReason: 'initial',
            createdAt: 1,
            label: '版本 1',
            childVersionIds: [],
          },
        }),
        route: 'derive',
        editKind: 'version',
      },
    ];

    for (const expected of cases) {
      const capability = resolveProjectResourceEditCapability(
        expected.resource,
      );
      expect(capability.route, expected.resource.id).toBe(expected.route);
      expect(capability.editKind, expected.resource.id).toBe(expected.editKind);
    }
  });

  it('用路径补齐任务产物缺失的标准媒体类型', () => {
    expect(
      canonicalProjectedResourceMediaType(
        resource({
          id: 'task:art:clip.mov',
          category: 'unclassified',
          path: 'assets/clip.mov',
          mediaType: '美术产物',
        }),
      ),
    ).toBe('video/quicktime');
    expect(
      canonicalProjectedResourceMediaType(
        resource({
          id: 'task:audio:effect.wav',
          category: 'audio',
          path: 'audio/effect.wav',
          mediaType: '美术产物',
        }),
      ),
    ).toBe('audio/wav');
    expect(
      canonicalProjectedResourceMediaType(
        resource({
          id: 'task:code:main.ts',
          category: 'document',
          path: 'src/main.ts',
          mediaType: '项目文档',
        }),
      ),
    ).toBe('text/typescript');
  });

  it('按现役媒体接口限制编辑提示词长度', () => {
    expect(resourceEditPromptMaxLength('background-music')).toBe(140);
    expect(resourceEditPromptMaxLength('sound-effect')).toBe(1_900);
    expect(resourceEditPromptMaxLength('video')).toBe(4_000);
    expect(resourceEditPromptMaxLength('character-animation')).toBe(4_000);
    expect(resourceEditPromptMaxLength('text')).toBe(32_000);
  });

  it('超上限的润色结果按类型截断并给出替代文案，边界上不动内容', () => {
    const exactlyLimit = '音'.repeat(140);
    expect(
      truncateResourceEditPrompt(exactlyLimit, 'background-music'),
    ).toEqual({
      text: exactlyLimit,
      notice: null,
    });
    const overLimit = `${exactlyLimit}多出来的一个字`;
    expect(truncateResourceEditPrompt(overLimit, 'background-music')).toEqual({
      text: exactlyLimit,
      notice: RESOURCE_EDIT_PROMPT_TRUNCATED_NOTICE,
    });
    expect(RESOURCE_EDIT_PROMPT_TRUNCATED_NOTICE).toBe('已按长度上限截断');
    // 上限按类型走，不是一刀切。
    expect(
      truncateResourceEditPrompt('字'.repeat(200), 'sound-effect').notice,
    ).toBeNull();
    expect(
      truncateResourceEditPrompt('字'.repeat(1_901), 'sound-effect').text
        .length,
    ).toBe(1_900);
    expect(
      truncateResourceEditPrompt('字'.repeat(32_001), 'text').text.length,
    ).toBe(32_000);
  });

  it('请求身份只在提示词没变时复用', () => {
    const first = resolveResourceEditRequestIdentity(null, '原始提示词');
    expect(first.prompt).toBe('原始提示词');
    expect(first.operationId).toBeTruthy();
    expect(first.idempotencyKey).toBeTruthy();

    // 同句重试：同一份身份，命中同一 operation 账本。
    expect(resolveResourceEditRequestIdentity(first, '原始提示词')).toBe(first);

    // 提示词变了（改了、或润色回填过）：必须重铸，否则 Rust 的
    // request_fingerprint（含 prompt）会把重试判成「已绑定到不同资源编辑请求」。
    const second = resolveResourceEditRequestIdentity(first, '润色后的提示词');
    expect(second).not.toBe(first);
    expect(second.operationId).not.toBe(first.operationId);
    expect(second.idempotencyKey).not.toBe(first.idempotencyKey);
    expect(second.prompt).toBe('润色后的提示词');
  });

  it('图片资源可进入角色动画派生入口', () => {
    const capability = resolveProjectCharacterAnimationCapability(
      resource({
        id: 'asset:hero',
        category: 'unclassified',
        path: 'assets/hero.png',
        mediaType: 'image/png',
        manifestAssetId: 'hero',
      }),
    );
    expect(capability?.editKind).toBe('character-animation');
    expect(capability?.generationMode).toBe('derive');
    expect(
      resolveProjectCharacterAnimationCapability(
        resource({
          id: 'asset:bgm',
          category: 'audio',
          path: 'audio/theme.mp3',
          mediaType: 'audio/mpeg',
          manifestAssetId: 'bgm',
        }),
      ),
    ).toBeNull();
  });
});