From 96cbdf8b51f2a1e85d6b612325e080b8decdf7c4 Mon Sep 17 00:00:00 2001
From: xudan <staysky25@outlook.com>
Date: Thu, 11 Sep 2025 15:27:30 +0800
Subject: [PATCH] =?UTF-8?q?refactor:=20=E6=9B=B4=E6=96=B0=E4=B8=8A?=
 =?UTF-8?q?=E4=B8=8B=E6=96=87=E8=8F=9C=E5=8D=95=E5=92=8C=E6=9C=BA=E5=99=A8?=
 =?UTF-8?q?=E4=BA=BA=E8=8F=9C=E5=8D=95=E7=BB=84=E4=BB=B6=EF=BC=8C=E6=94=B9?=
 =?UTF-8?q?=E4=B8=BA=E9=80=9A=E8=BF=87=E6=9C=BA=E5=99=A8=E4=BA=BAID?=
 =?UTF-8?q?=E4=BC=A0=E9=80=92=E4=BF=A1=E6=81=AF=EF=BC=8C=E6=B7=BB=E5=8A=A0?=
 =?UTF-8?q?editor=20token=EF=BC=8C=E4=BC=98=E5=8C=96=E6=9C=BA=E5=99=A8?=
 =?UTF-8?q?=E4=BA=BA=E4=BF=A1=E6=81=AF=E8=8E=B7=E5=8F=96=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/components/context-menu/context-menu.vue | 16 +++++-----
 src/components/context-menu/robot-menu.vue   | 33 +++++++++++++-------
 src/components/robot-groups.vue              |  3 +-
 src/pages/movement-supervision.vue           |  4 +--
 4 files changed, 35 insertions(+), 21 deletions(-)

diff --git a/src/components/context-menu/context-menu.vue b/src/components/context-menu/context-menu.vue
index 72f7bbe..2884e9a 100644
--- a/src/components/context-menu/context-menu.vue
+++ b/src/components/context-menu/context-menu.vue
@@ -34,8 +34,9 @@
 
         <!-- 机器人菜单 -->
         <RobotMenu
-          v-else-if="menuType === 'robot' && robotInfo"
-          :robot-info="robotInfo"
+          v-else-if="menuType === 'robot' && robotId"
+          :robot-id="robotId"
+          :token="token"
           @action-complete="handleActionComplete"
           @custom-image="handleCustomImage"
         />
@@ -52,10 +53,10 @@
 
 <script setup lang="ts">
 import { message } from 'ant-design-vue';
-import { computed, defineAsyncComponent, ref } from 'vue';
+import { computed, defineAsyncComponent, type InjectionKey, ref, type ShallowRef } from 'vue';
 
-import type { RobotInfo } from '../../apis/robot';
 import type { StorageLocationInfo } from '../../services/context-menu';
+import type { EditorService } from '../../services/editor.service';
 
 // 使用动态导入避免 TypeScript 错误
 const DefaultMenu = defineAsyncComponent(() => import('./default-menu.vue'));
@@ -68,14 +69,15 @@ interface Props {
   y?: number;
   menuType?: 'default' | 'storage' | 'storage-background' | 'robot' | 'point' | 'area';
   storageLocations?: StorageLocationInfo[];
-  robotInfo?: RobotInfo;
+  robotId?: string; // 改为传递机器人ID
+  token?: InjectionKey<ShallowRef<EditorService>>; // 添加 editor token
   apiBaseUrl?: string;
 }
 
 interface Emits {
   (e: 'close'): void;
   (e: 'actionComplete', data: any): void;
-  (e: 'customImage', data: { robotInfo: RobotInfo }): void;
+  (e: 'customImage', data: { robotInfo: any }): void; // 使用 any 类型，因为 RobotInfo 不再直接导入
 }
 
 const props = withDefaults(defineProps<Props>(), {
@@ -190,7 +192,7 @@ const getActionDisplayName = (action: string): string => {
 };
 
 // 处理自定义图片事件
-const handleCustomImage = (data: { robotInfo: RobotInfo }) => {
+const handleCustomImage = (data: { robotInfo: any }) => {
   console.log('打开自定义图片设置:', data.robotInfo);
   emit('customImage', data);
   // 不关闭菜单，只有关闭按钮才能关闭
diff --git a/src/components/context-menu/robot-menu.vue b/src/components/context-menu/robot-menu.vue
index 8354cdd..7965336 100644
--- a/src/components/context-menu/robot-menu.vue
+++ b/src/components/context-menu/robot-menu.vue
@@ -126,7 +126,7 @@
 
 <script setup lang="ts">
 import { message } from 'ant-design-vue';
-import { ref } from 'vue';
+import { computed, inject, type InjectionKey, ref, type ShallowRef } from 'vue';
 
 import type { RobotInfo } from '../../apis/robot';
 import type { RobotAction } from '../../services/context-menu/robot-menu.service';
@@ -135,10 +135,12 @@ import {
   getRobotStatusColor,
   getRobotStatusText
 } from '../../services/context-menu/robot-menu.service';
+import type { EditorService } from '../../services/editor.service';
 import RobotImageSettingsModal from '../modal/robot-image-settings-modal.vue';
 
 interface Props {
-  robotInfo?: RobotInfo;
+  robotId?: string; // 改为传递机器人ID而不是完整的机器人信息
+  token?: InjectionKey<ShallowRef<EditorService>>; // 添加 editor token
 }
 
 interface Emits {
@@ -150,6 +152,15 @@ const props = defineProps<Props>();
 
 const emit = defineEmits<Emits>();
 
+// 注入 editor service
+const editor = props.token ? inject(props.token) : null;
+
+// 根据 robotId 获取机器人信息
+const robotInfo = computed<RobotInfo | null>(() => {
+  if (!props.robotId || !editor?.value) return null;
+  return editor.value.getRobotById(props.robotId) || null;
+});
+
 // 图片设置模态框状态
 const imageSettingsVisible = ref(false);
 const selectedRobotName = ref('');
@@ -171,40 +182,40 @@ const getBatteryClass = (batteryLevel: number) => {
 
 // 处理机器人操作
 const handleRobotAction = async (action: RobotAction | 'custom_image', actionName: string) => {
-  if (!props.robotInfo) return;
+  if (!robotInfo.value) return;
   
   // 处理自定义图片操作
   if (action === 'custom_image') {
-    if (!props.robotInfo?.label) {
+    if (!robotInfo.value?.label) {
       message.error('未找到机器人信息');
       return;
     }
     
-    console.log('打开机器人图片设置:', props.robotInfo);
+    console.log('打开机器人图片设置:', robotInfo.value);
     
     // 打开模态框，不关闭右键菜单
-    selectedRobotName.value = props.robotInfo.label;
+    selectedRobotName.value = robotInfo.value.label;
     imageSettingsVisible.value = true;
     
     // 只触发自定义图片事件，不触发操作完成事件
     emit('customImage', {
-      robotInfo: props.robotInfo
+      robotInfo: robotInfo.value
     });
     return;
   }
   
   try {
-    console.log(`执行机器人操作: ${action} (${actionName})`, props.robotInfo);
+    console.log(`执行机器人操作: ${action} (${actionName})`, robotInfo.value);
     
     // 使用服务函数执行操作
-    const result = await executeRobotAction(action, props.robotInfo);
+    const result = await executeRobotAction(action, robotInfo.value);
     
     console.log('机器人操作结果:', result);
     
     // 发送操作完成事件
     emit('actionComplete', {
       action,
-      robot: props.robotInfo,
+      robot: robotInfo.value,
       success: result.success
     });
   } catch (error) {
@@ -212,7 +223,7 @@ const handleRobotAction = async (action: RobotAction | 'custom_image', actionNam
     
     emit('actionComplete', {
       action,
-      robot: props.robotInfo,
+      robot: robotInfo.value,
       success: false
     });
   }
diff --git a/src/components/robot-groups.vue b/src/components/robot-groups.vue
index d9fc14f..5890ca3 100644
--- a/src/components/robot-groups.vue
+++ b/src/components/robot-groups.vue
@@ -160,7 +160,8 @@ const toRemoveRobots = () =>
     :x="contextMenuState.x"
     :y="contextMenuState.y"
     :menu-type="contextMenuState.menuType"
-    :robot-info="contextMenuState.robotInfo"
+    :robot-id="contextMenuState.robotInfo?.id"
+    :token="token"
     @close="handleContextMenuClose"
   />
 
diff --git a/src/pages/movement-supervision.vue b/src/pages/movement-supervision.vue
index 01e2f39..b55e06c 100644
--- a/src/pages/movement-supervision.vue
+++ b/src/pages/movement-supervision.vue
@@ -465,8 +465,8 @@ const handleGlobalKeydown = (event: KeyboardEvent) => {
     :y="contextMenuState.y"
     :menu-type="contextMenuState.menuType"
     :storage-locations="contextMenuState.storageLocations"
-    :robot-info="contextMenuState.robotInfo"
-    :editor="editor"
+    :robot-id="contextMenuState.robotInfo?.id"
+    :token="EDITOR_KEY"
     @close="handleCloseContextMenu"
     @action-complete="handleActionComplete"
   />