1
// 手动实现 WebSocket 连接逻辑
2
const serverUrl = 'ws://localhost:9100'
3
const socket = new WebSocket(serverUrl)
4

5
socket.onopen = () => {
6
  console.log('[CesiumBridge] Connected to MCP server at', serverUrl)
7
}
8

9
socket.onmessage = async (event) => {
10
  try {
11
    const data = JSON.parse(event.data)
12
    console.log('[CesiumBridge] Received message:', data)
13

14
    const action = data.action || data.method
15
    if (action) {
16
      console.log(`[CesiumBridge] Executing command: ${action}`, data.params)
17

18
      // 调用 bridge 执行 AI 发来的指令
19
      const result = await bridge.execute({
20
        action: action,
21
        params: data.params || {},
22
      })
23

24
      console.log('[CesiumBridge] Execution result:', result)
25

26
      // 返回结果给服务器
27
      // 关键：必须回传相同的 id，以便服务器识别这是哪个请求的响应
28
      const response = {
29
        type: 'result',
30
        id: data.id,
31
        success: result.success,
32
        data: result.data,
33
        error: result.error,
34
        message: result.message,
35
      }
36

37
      socket.send(JSON.stringify(response))
38
    }
39
  } catch (e) {
40
    console.error('[CesiumBridge] Error processing message:', e)
41
  }
42
}
43

44
socket.onclose = () => {
45
  console.warn('[CesiumBridge] Disconnected from MCP server')
46
}

1
const MAP_TOOLS = [
2
  {
3
    type: 'function',
4
    function: {
5
      name: 'flyTo',
6
      description: '将地图相机平滑飞行到指定坐标',
7
      parameters: {
8
        type: 'object',
9
        properties: {
10
          longitude: { type: 'number' },
11
          latitude: { type: 'number' },
12
          height: { type: 'number', default: 5000 },
13
        },
14
        required: ['longitude', 'latitude'],
15
      },
16
    },
17
  },
18
  {
19
    type: 'function',
20
    function: {
21
      name: 'addMarker',
22
      description: '在地图指定位置添加 HTML 标记',
23
      parameters: {
24
        type: 'object',
25
        properties: {
26
          longitude: { type: 'number' },
27
          latitude: { type: 'number' },
28
          label: { type: 'string' },
29
        },
30
        required: ['longitude', 'latitude', 'label'],
31
      },
32
    },
33
  },
34
]

1
async function handleCommandExecution(msg: any) {
2
  // 1. 原生工具执行 (Tool Call)
3
  if (msg.tool_calls && msg.tool_calls.length > 0) {
4
    for (const tool of msg.tool_calls) {
5
      await bridge.execute({
6
        action: tool.function.name,
7
        params: tool.function.arguments,
8
      })
9
      // 反馈执行结果给上下文...
10
    }
11
  }
12
  // 2. 备选方案：文本正则解析指令 (针对不支持工具的模型)
13
  else {
14
    const jsonMatch = msg.content.match(/```json\s*([\s\S]*?)\s*```/)
15
    if (jsonMatch) {
16
      const cmds = JSON.parse(jsonMatch[1].trim())
17
      for (const cmd of cmds) await bridge.execute(cmd)
18
    }
19
  }
20
}