import type { CesiumBrowserToolsetName } from '../../packages/cesium-mcp-contracts/src/index.js'

export interface ToolRoutingEvalCase {
  id: string
  locale: 'en' | 'zh-CN'
  prompt: string
  expectedToolsets: CesiumBrowserToolsetName[]
  requiredTools: string[]
}

export const toolRoutingEvalCases: ToolRoutingEvalCase[] = [
  {
    id: 'view-en',
    locale: 'en',
    prompt: 'Fly to Tokyo and save this viewpoint',
    expectedToolsets: ['view'],
    requiredTools: ['flyTo', 'saveViewpoint'],
  },
  {
    id: 'view-zh',
    locale: 'zh-CN',
    prompt: '飞到北京并保存这个视点',
    expectedToolsets: ['view'],
    requiredTools: ['flyTo', 'saveViewpoint'],
  },
  {
    id: 'entity-en',
    locale: 'en',
    prompt: 'Add a marker and draw a polyline route',
    expectedToolsets: ['entity'],
    requiredTools: ['addMarker', 'addPolyline'],
  },
  {
    id: 'entity-zh',
    locale: 'zh-CN',
    prompt: '添加标记并绘制一条折线',
    expectedToolsets: ['entity'],
    requiredTools: ['addMarker', 'addPolyline'],
  },
  {
    id: 'layer-en',
    locale: 'en',
    prompt: 'Load a GeoJSON layer and update its style',
    expectedToolsets: ['layer'],
    requiredTools: ['addGeoJsonLayer', 'updateLayerStyle'],
  },
  {
    id: 'layer-zh',
    locale: 'zh-CN',
    prompt: '加载 GeoJSON 图层并设置分级设色',
    expectedToolsets: ['layer'],
    requiredTools: ['addGeoJsonLayer', 'updateLayerStyle'],
  },
  {
    id: 'camera-en',
    locale: 'en',
    prompt: 'Orbit around the target and then stop orbiting',
    expectedToolsets: ['camera'],
    requiredTools: ['startOrbit', 'stopOrbit'],
  },
  {
    id: 'camera-zh',
    locale: 'zh-CN',
    prompt: '环绕观察目标并停止环绕',
    expectedToolsets: ['camera'],
    requiredTools: ['startOrbit', 'stopOrbit'],
  },
  {
    id: 'entity-ext-en',
    locale: 'en',
    prompt: 'Add a rectangle and a wall',
    expectedToolsets: ['entity-ext'],
    requiredTools: ['addRectangle', 'addWall'],
  },
  {
    id: 'entity-ext-zh',
    locale: 'zh-CN',
    prompt: '添加矩形和墙体',
    expectedToolsets: ['entity-ext'],
    requiredTools: ['addRectangle', 'addWall'],
  },
  {
    id: 'animation-en',
    locale: 'en',
    prompt: 'Create an animation and track the entity',
    expectedToolsets: ['animation'],
    requiredTools: ['createAnimation', 'trackEntity'],
  },
  {
    id: 'animation-zh',
    locale: 'zh-CN',
    prompt: '创建动画并跟踪实体',
    expectedToolsets: ['animation'],
    requiredTools: ['createAnimation', 'trackEntity'],
  },
  {
    id: 'scene-en',
    locale: 'en',
    prompt: 'Enable bloom and configure scene fog',
    expectedToolsets: ['scene'],
    requiredTools: ['setPostProcess', 'setSceneOptions'],
  },
  {
    id: 'scene-zh',
    locale: 'zh-CN',
    prompt: '开启泛光并调整场景雾效',
    expectedToolsets: ['scene'],
    requiredTools: ['setPostProcess', 'setSceneOptions'],
  },
  {
    id: 'tiles-en',
    locale: 'en',
    prompt: 'Load 3D Tiles and enable edge display',
    expectedToolsets: ['tiles'],
    requiredTools: ['load3dTiles', 'setEdgeDisplayMode'],
  },
  {
    id: 'tiles-zh',
    locale: 'zh-CN',
    prompt: '加载三维瓦片并设置边缘显示',
    expectedToolsets: ['tiles'],
    requiredTools: ['load3dTiles', 'setEdgeDisplayMode'],
  },
  {
    id: 'interaction-en',
    locale: 'en',
    prompt: 'Measure an area and take a screenshot',
    expectedToolsets: ['interaction'],
    requiredTools: ['measure', 'screenshot'],
  },
  {
    id: 'interaction-zh',
    locale: 'zh-CN',
    prompt: '测量这个区域的面积并截图',
    expectedToolsets: ['interaction'],
    requiredTools: ['measure', 'screenshot'],
  },
  {
    id: 'trajectory-en',
    locale: 'en',
    prompt: 'Play this trajectory on the globe',
    expectedToolsets: ['trajectory'],
    requiredTools: ['playTrajectory'],
  },
  {
    id: 'trajectory-zh',
    locale: 'zh-CN',
    prompt: '播放一条轨迹回放',
    expectedToolsets: ['trajectory'],
    requiredTools: ['playTrajectory'],
  },
  {
    id: 'heatmap-en',
    locale: 'en',
    prompt: 'Create a heat map from these points',
    expectedToolsets: ['heatmap'],
    requiredTools: ['addHeatmap'],
  },
  {
    id: 'heatmap-zh',
    locale: 'zh-CN',
    prompt: '根据这些点位生成热力图',
    expectedToolsets: ['heatmap'],
    requiredTools: ['addHeatmap'],
  },
  {
    id: 'geolocation-en',
    locale: 'en',
    prompt: 'Geocode this address: 1 Market Street, San Francisco',
    expectedToolsets: ['geolocation'],
    requiredTools: ['geocode'],
  },
  {
    id: 'geolocation-zh',
    locale: 'zh-CN',
    prompt: '对这个地址进行地理编码：北京市东城区',
    expectedToolsets: ['geolocation'],
    requiredTools: ['geocode'],
  },
  {
    id: 'multi-tiles-interaction',
    locale: 'en',
    prompt: 'Load 3D Tiles, then take a screenshot',
    expectedToolsets: ['tiles', 'interaction'],
    requiredTools: ['load3dTiles', 'screenshot'],
  },
  {
    id: 'multi-layer-interaction',
    locale: 'zh-CN',
    prompt: '加载 GeoJSON 图层并高亮其中的重点区域',
    expectedToolsets: ['layer', 'interaction'],
    requiredTools: ['addGeoJsonLayer', 'highlight'],
  },
  {
    id: 'multi-entity-ext-animation',
    locale: 'en',
    prompt: 'Add a rectangle and animate an entity along a path',
    expectedToolsets: ['entity-ext', 'animation'],
    requiredTools: ['addRectangle', 'createAnimation'],
  },
  {
    id: 'multi-scene-camera',
    locale: 'en',
    prompt: 'Enable bloom and orbit around the target',
    expectedToolsets: ['scene', 'camera'],
    requiredTools: ['setPostProcess', 'startOrbit'],
  },
  {
    id: 'multi-geolocation-entity',
    locale: 'en',
    prompt: 'Where is the Forbidden City? Add a marker there.',
    expectedToolsets: ['geolocation', 'entity'],
    requiredTools: ['geocode', 'addMarker'],
  },
  {
    id: 'multi-heatmap-view',
    locale: 'zh-CN',
    prompt: '创建热力图并保存当前视点',
    expectedToolsets: ['heatmap', 'view'],
    requiredTools: ['addHeatmap', 'saveViewpoint'],
  },
]