Genarrative/apps/ai-game-creator-shell/src/features/project-workspace/projectCommandPolicy.ts

import {
  GAME_CREATION_APP_COMMANDS,
  type GameCreationAppCommandDescriptor,
} from '../../../../../packages/shared/src/contracts/gameCreationApp';
import type { PendingCommand } from '../../app/types';
import {
  isAbsoluteProjectPath,
  projectPathHasControlCharacter,
} from '../project-summary/projectSummary';

export function resolveChatProjectPath(
  localProject: { projectPath: string } | null,
) {
  const projectPath = localProject?.projectPath?.trim();
  if (
    !projectPath ||
    !isAbsoluteProjectPath(projectPath) ||
    projectPathHasControlCharacter(projectPath)
  ) {
    return null;
  }
  return projectPath;
}

export function isAgentTraceFilePath(value: string) {
  const path = value.trim();
  return (
    path === '.agent/run.latest.json' ||
    (path.startsWith('.agent/runs/') && path.endsWith('.json'))
  );
}

export function isRegisteredGameCreationCommandId(value: string) {
  return GAME_CREATION_APP_COMMANDS.some((command) => command.id === value);
}

export function isProjectPolicyConfirmableCommandId(value: string) {
  return [
    'project.index',
    'project.status',
    'project.checkpoint',
    'project.diff',
    'project.restore',
    'project.export_package',
    'project.export_list',
    'file.list',
    'file.read',
    'memory.read',
    'memory.write',
    'memory.delete',
    'asset.register',
    'asset.list',
    'task.list',
    'task.create',
    'task.update',
    'agent.run_status',
    'agent.kill',
    'agent.retry',
    'agent.resume',
    'agent.delegate',
    'agent.spawn_isolated',
    'agent.schedule_ready',
    'agent.audit',
    'agent.trace_read',
    'preview.status',
    'preview.start',
    'preview.validate',
    'preview.open',
    'preview.stop',
    'canvas.project_sync',
    'canvas.asset_import',
    'canvas.asset_generate',
    'canvas.export_import',
    'conversation.read',
    'conversation.write',
  ].includes(value);
}

export function isSafeCanvasProjectId(value: string) {
  return !!value.trim() && !projectPathHasControlCharacter(value);
}

export function isSafeCheckpointId(value: string) {
  const checkpointId = value.trim();
  return (
    !!checkpointId &&
    !projectPathHasControlCharacter(checkpointId) &&
    !checkpointId.includes('/') &&
    !checkpointId.includes('\\') &&
    !checkpointId.includes('..')
  );
}

export function needsInitializedChatProject(
  commandId: GameCreationAppCommandDescriptor['id'],
) {
  return [
    'asset.upload',
    'asset.register',
    'agent.delegate',
    'agent.kill',
    'agent.retry',
    'agent.resume',
    'command.run_limited',
    'game.generate_draft',
    'game.run_local',
    'file.list',
    'file.read',
    'project.checkpoint',
    'project.export_package',
    'project.export_list',
    'project.restore',
    'project.policy_write',
    'preview.open',
    'preview.start',
    'canvas.project_sync',
    'canvas.asset_import',
    'canvas.asset_generate',
    'canvas.export_import',
    'memory.read',
    'memory.write',
    'memory.delete',
    'conversation.read',
    'conversation.write',
  ].includes(commandId);
}

export function resolvePendingCommandProjectPath(
  command: PendingCommand,
  localProject: { projectPath: string } | null,
  draftProjectPath: string,
) {
  if (command.id === 'project.create') {
    return command.projectPath;
  }
  if (needsInitializedChatProject(command.id)) {
    return resolveChatProjectPath(localProject) ?? draftProjectPath;
  }
  return draftProjectPath;
}