“一源双包”影分身构建器源码精讲

"""从 Example/ 下的技能源码构建 Q1/T1 双机型 .rpk 包及可审阅的发布元数据。"""

import importlib.util
import json
import shutil
import tempfile
from pathlib import Path

ROOT = Path(__file__).resolve().parents[1]
WORKSPACE = ROOT.parent
DEV_KIT = ROOT / "primehub_dev_kit"
EXAMPLE_ROOT = WORKSPACE / "Example"
OUTPUT = ROOT / "dist"

SKILL_SPECS = (
    {
        "folder": "energy_station",
        "pkg_prefix": "cans518_energy_station",
        "publish_prefix": "publish",
        "title": "社交能量补给站",
        "short_desc": "三格好心情，给每个敢动手的人充个电",
        "long_desc_tpl": (
            "离线小剧场：开场、勇气/灵感/快乐三格充电、谢幕。"
            "每格短台词配动作，无需联网或语音输入。"
            "仅适配%s，使用预设动作，运行时请留出周围1米空间。"
            "基于官方示例改编；台词和编排可在源码中修改。"
        ),
        "icon": "skill-game.svg",
    },
    {
        "folder": "universal_skill",
        "pkg_prefix": "cans518_universal_skill",
        "publish_prefix": "publish_universal",
        "title": "启元百变互动官",
        "short_desc": "迎宾问候+灵感广播+好运盲盒抽签的通用互动剧场",
        "long_desc_tpl": (
            "Q1/T1通用互动技能模板：包含元气迎宾、每日灵感金句（支持联网API与断网自动兜底）、"
            "随机好运抽签及谢幕。仅适配%s，内置形态检测与TTS防抖轮询，运行时请留出周围1米空间。"
        ),
        "icon": "skill-verify.svg",
    },
)


def build_skill(pack_module, spec_cfg: dict) -> None:
    source_dir = EXAMPLE_ROOT / spec_cfg["folder"]
    for model, model_id, model_code in (("q1", 100, "robot_q"), ("t1", 101, "robot_t")):
        with tempfile.TemporaryDirectory(prefix="qiyuan_build_") as staging:
            skill_dir = Path(staging) / (spec_cfg["folder"] + "_" + model)
            shutil.copytree(source_dir, skill_dir, ignore=shutil.ignore_patterns("__pycache__", "*.pyc"))
            script = skill_dir / "src/main.py"
            source = script.read_text(encoding="utf-8")
            assignment = 'ROBOT_MODEL = "q1"'
            if source.count(assignment) != 1:
                raise ValueError("期望恰好一个 ROBOT_MODEL 声明: " + str(script))
            script.write_text(source.replace(assignment, 'ROBOT_MODEL = "%s"' % model), encoding="utf-8")
            config = skill_dir / "cfg/pkg.yaml"
            package_name = "%s_%s_20261024" % (spec_cfg["pkg_prefix"], model)
            lines = []
            for line in config.read_text(encoding="utf-8").splitlines():
                if line.startswith("package_name:"):
                    line = "package_name: " + package_name
                elif line.startswith("robot_model:"):
                    line = "robot_model: " + model
                lines.append(line)
            config.write_text("\n".join(lines) + "\n", encoding="utf-8")
            pack_module.pack(skill_dir, OUTPUT)
        metadata = {
            "skillName": spec_cfg["title"] + " " + model.upper(),
            "version": "0.1.0",
            "skillDesc": spec_cfg["short_desc"],
            "long_description": spec_cfg["long_desc_tpl"] % model.upper(),
            "categoryIds": [104],
            "categoryCodeList": ["action"],
            "tagsIds": [127],
            "tagsCodeList": ["companion"],
            "robotModelIds": [model_id],
            "robotModelCodeList": [model_code],
            "iconUrl": spec_cfg["icon"],
        }
        publish_file = "%s_%s.json" % (spec_cfg["publish_prefix"], model)
        (OUTPUT / publish_file).write_text(
            json.dumps(metadata, ensure_ascii=False, indent=2) + "\n", encoding="utf-8"
        )


def build() -> None:
    spec = importlib.util.spec_from_file_location("primehub_pack", DEV_KIT / "pack.py")
    pack_module = importlib.util.module_from_spec(spec)
    spec.loader.exec_module(pack_module)
    OUTPUT.mkdir(exist_ok=True)
    for spec_cfg in SKILL_SPECS:
        build_skill(pack_module, spec_cfg)


if __name__ == "__main__":
    build()