支持设备短按向家长留言

talkingq-url/services/device_target_cache.py

@@ -1,19 +1,51 @@
 import asyncio
+from dataclasses import dataclass
 from typing import Dict, Optional
 
 
+VOICE_TARGET_KIND_DEVICE = "device"
+VOICE_TARGET_KIND_PARENT = "parent"
+
+
+@dataclass(frozen=True)
+class VoiceTarget:
+    kind: str
+    target_device_id: Optional[str] = None
+    uuid: Optional[str] = None
+    audio_cache_key: Optional[str] = None
+
+
 class DeviceTargetCache:
     def __init__(self):
-        self.targets: Dict[str, str] = {}
+        self.targets: Dict[str, VoiceTarget] = {}
         self.lock = asyncio.Lock()
 
     async def get_target(self, device_id: str) -> Optional[str]:
+        async with self.lock:
+            target = self.targets.get(device_id)
+            if not target or target.kind != VOICE_TARGET_KIND_DEVICE:
+                return None
+            return target.target_device_id
+
+    async def get_voice_target(self, device_id: str) -> Optional[VoiceTarget]:
         async with self.lock:
             return self.targets.get(device_id)
 
     async def set_target(self, device_id: str, target_device_id: str):
         async with self.lock:
-            self.targets[device_id] = target_device_id
+            self.targets[device_id] = VoiceTarget(
+                kind=VOICE_TARGET_KIND_DEVICE,
+                target_device_id=target_device_id,
+                audio_cache_key=target_device_id,
+            )
+
+    async def set_parent_target(self, device_id: str, uuid: str):
+        async with self.lock:
+            self.targets[device_id] = VoiceTarget(
+                kind=VOICE_TARGET_KIND_PARENT,
+                uuid=uuid,
+                audio_cache_key=f"parent:{device_id}",
+            )
 
     async def remove_target(self, device_id: str):
         async with self.lock:
@@ -26,4 +58,4 @@ class DeviceTargetCache:
 
 
 # 单例实例
-device_target_cache = DeviceTargetCache()
+device_target_cache = DeviceTargetCache()