Uni-Lab-OS/unilabos/app/register.py

import time
from typing import Any, Dict, Optional, Tuple

from unilabos.utils.log import logger
from unilabos.utils.tools import normalize_json as _normalize_device


def normalize_model_for_upload(model_dict: dict) -> dict:
    """将 Registry YAML 的 model 字段映射为后端 DeviceModel 结构化格式。

    保留所有原始字段，额外做以下标准化：
    1. 自动推断 format（如果 YAML 未指定）
    2. 将 children_mesh 扁平字段映射为结构化 children_mesh 对象
    """
    if not model_dict:
        return model_dict

    result = {**model_dict}

    # 自动推断 format
    if "format" not in result and result.get("path"):
        path = result["path"]
        if path.endswith(".xacro"):
            result["format"] = "xacro"
        elif path.endswith(".urdf"):
            result["format"] = "urdf"
        elif path.endswith(".stl"):
            result["format"] = "stl"
        elif path.endswith((".gltf", ".glb")):
            result["format"] = "gltf"

    # 将 children_mesh 扁平字段 → 结构化 children_mesh 对象
    if "children_mesh" in result and isinstance(result["children_mesh"], str):
        cm_path = result.pop("children_mesh")
        cm_tf = result.pop("children_mesh_tf", None)
        cm_oss = result.pop("children_mesh_path", None)
        result["children_mesh"] = {
            "path": cm_oss or cm_path,
            "format": "stl" if cm_path.endswith(".stl") else "gltf",
            "default_visible": True,
        }
        if cm_tf and len(cm_tf) >= 3:
            result["children_mesh"]["local_offset"] = cm_tf[:3]
        if cm_tf and len(cm_tf) >= 6:
            result["children_mesh"]["local_rotation"] = cm_tf[3:6]

    return result


def register_devices_and_resources(lab_registry, gather_only=False) -> Optional[Tuple[Dict[str, Any], Dict[str, Any]]]:
    """
    注册设备和资源到服务器（仅支持HTTP）
    """

    from unilabos.app.web.client import http_client

    logger.info("[UniLab Register] 开始注册设备和资源...")

    devices_to_register = {}
    for device_info in lab_registry.obtain_registry_device_info():
        normalized = _normalize_device(device_info)
        # 标准化 model 字段
        if normalized.get("model"):
            normalized["model"] = normalize_model_for_upload(normalized["model"])
        devices_to_register[device_info["id"]] = normalized
        logger.trace(f"[UniLab Register] 收集设备: {device_info['id']}")

    resources_to_register = {}
    for resource_info in lab_registry.obtain_registry_resource_info():
        # 标准化 model 字段
        if resource_info.get("model"):
            resource_info["model"] = normalize_model_for_upload(resource_info["model"])
        resources_to_register[resource_info["id"]] = resource_info
        logger.trace(f"[UniLab Register] 收集资源: {resource_info['id']}")

    if gather_only:
        return devices_to_register, resources_to_register

    if devices_to_register:
        try:
            start_time = time.time()
            response = http_client.resource_registry(
                {"resources": list(devices_to_register.values())},
                tag="device_registry",
            )
            cost_time = time.time() - start_time
            res_data = response.json() if response.status_code == 200 else {}
            skipped = res_data.get("data", {}).get("skipped", False)
            if skipped:
                logger.info(
                    f"[UniLab Register] 设备注册跳过（内容未变化）"
                    f" {len(devices_to_register)} 个 {cost_time:.3f}s"
                )
            elif response.status_code in [200, 201]:
                logger.info(f"[UniLab Register] 成功注册 {len(devices_to_register)} 个设备 {cost_time:.3f}s")
            else:
                logger.error(f"[UniLab Register] 设备注册失败: {response.status_code}, {response.text} {cost_time:.3f}s")
        except Exception as e:
            logger.error(f"[UniLab Register] 设备注册异常: {e}")

    if resources_to_register:
        try:
            start_time = time.time()
            response = http_client.resource_registry(
                {"resources": list(resources_to_register.values())},
                tag="resource_registry",
            )
            cost_time = time.time() - start_time
            res_data = response.json() if response.status_code == 200 else {}
            skipped = res_data.get("data", {}).get("skipped", False)
            if skipped:
                logger.info(
                    f"[UniLab Register] 资源注册跳过（内容未变化）"
                    f" {len(resources_to_register)} 个 {cost_time:.3f}s"
                )
            elif response.status_code in [200, 201]:
                logger.info(f"[UniLab Register] 成功注册 {len(resources_to_register)} 个资源 {cost_time:.3f}s")
            else:
                logger.error(f"[UniLab Register] 资源注册失败: {response.status_code}, {response.text} {cost_time:.3f}s")
        except Exception as e:
            logger.error(f"[UniLab Register] 资源注册异常: {e}")