Initial project setup with core files including .gitignore, LICENSE, pyproject.toml, and main application structure. Added MQTT handling and navigation server functionality for robot control.

temi_mcpserver/temiRobot/mcp_mqtt.py

@@ -0,0 +1,113 @@
+import paho.mqtt.client as mqtt
+import json
+import logging
+import threading
+from typing import Optional
+import uuid
+import threading
+import requests
+from os import getenv
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+MQTT_CLIENT_ID = f"MCPMQTT-{uuid.uuid4().hex[:8]}"
+
+def getConfig_url(array):
+    url = "http://lzwcai-demp-corp-manager:8086/system/config/getConfig"
+    data = [array]
+    try:
+        response = requests.post(url, json=data, timeout=5)
+        response.raise_for_status()
+        data = response.json()['data']
+        return data[0]['configValue']
+    except Exception as e:
+        print(f"Error fetching config for {array}: {e}")
+        return None
+
+class MQTTHandler:
+    def __init__(self):
+        self.client = mqtt.Client()
+        self.tasks = {}  # {task_id: task_data}
+        self._lock = threading.Lock()
+        
+        # 获取MQTT配置，提供默认值
+        self.mqtt_username = getenv('MQTT_USERNAME') or 'lzwc'
+        self.mqtt_password = getenv('MQTT_PASSWORD') or 'Lzwc@4187.'
+        mqtt_broker_raw = getenv('MQTT_BROKER') or 'tcp://dempmana.lzwcai.com'
+        # 移除协议前缀，只保留主机名
+        self.mqtt_broker = mqtt_broker_raw.replace('tcp://', '').replace('mqtt://', '')
+        mqtt_port_str = getenv('MQTT_PORT')
+        self.mqtt_port = int(mqtt_port_str) if mqtt_port_str else 1883
+        
+        # 记录配置信息
+        logger.info(f"MQTT配置 - Broker: {self.mqtt_broker}, Port: {self.mqtt_port}, Username: {self.mqtt_username}")
+        if not getenv('MQTT_BROKER'):
+            logger.warning("MQTT_BROKER环境变量未设置，使用默认值")
+        if not mqtt_port_str:
+            logger.warning("MQTT_PORT环境变量未设置，使用默认端口1883")
+
+        self.client.username_pw_set(self.mqtt_username, self.mqtt_password)
+        self.client.on_connect = self._on_connect
+        self.client.on_message = self._on_message
+        self.client.on_disconnect = self._on_disconnect
+        try:
+            logger.info(f"正在连接 MQTT 代理: {self.mqtt_broker}:{self.mqtt_port}")
+            self.client.connect(self.mqtt_broker, self.mqtt_port, 60)
+            self.client.loop_start()
+        except Exception as e:
+            logger.error(f"连接 MQTT 失败: {e}")
+
+
+    def _on_connect(self, client, userdata, flags, rc):
+        """MQTT 连接回调"""
+        if rc == 0:
+            logger.info("已成功连接到 MQTT 代理服务器")
+        else:
+            logger.error(f"连接 MQTT 代理服务器失败，返回码: {rc}")
+
+    def get_status(self, task_id: str = None):
+        """获取任务状态
+        Args:
+            task_id: 可选参数，指定任务ID。如果为None则返回所有任务状态
+        Returns:
+            如果指定task_id则返回该任务的状态，否则返回所有任务状态
+        """
+        with self._lock:
+            if task_id:
+                return self.tasks.get(task_id, {}).copy()
+            return {k: v.copy() for k, v in self.tasks.items()}
+
+    def _on_message(self, client, userdata, msg):
+        try:
+            payload = json.loads(msg.payload.decode())
+            logger.info(f"[MQTT] topic={msg.topic} msg: {payload}")
+            
+            # task_id = payload.get('task_id')
+            # if task_id and task_id in self.tasks:
+            #     with self._lock:
+            #         self.tasks[task_id].update({
+            #             'status': payload.get('status', 'UNKNOWN'),
+            #             'description': payload.get('description', '')
+            #         })
+                
+        except Exception as e:
+            logger.error(f"[错误] 处理消息失败: {e}")
+
+    def _on_disconnect(self, client, userdata, rc):
+        """MQTT 断开连接回调"""
+        if rc != 0:
+            logger.warning("意外断开与 MQTT 代理服务器的连接。")
+            try:
+                logger.info("尝试重新连接 MQTT...")
+                client.reconnect()
+            except Exception as e:
+                logger.error(f"重新连接 MQTT 失败: {e}")
+
+# 单例模式
+_instance: Optional[MQTTHandler] = None
+
+def get_mcpmqtt_handler() -> MQTTHandler:
+    """获取单例"""
+    global _instance
+    if _instance is None:
+        _instance = MQTTHandler()
+    return _instance