patch-hover-interaction.py

#!/usr/bin/env python3
"""patch-hover-interaction.py — 给 Kimi Work 桌宠页面加"悬停互动"

桌宠组件的 workspace/index.html 是单文件 IIFE。本脚本在 IIFE 结束前注入一段代码:
鼠标悬停在宠物身上时,从可用的空闲彩蛋动作里随机挑一个播放(优先 idle_random_1
招手 / idle_random_3 眨眼互动,素材里没有的会自动跳过),播完一个循环后回到待机。
可重复运行(幂等,以 hover-wave-interaction 标记判断)。

用法:
    python3 scripts/patch-hover-interaction.py [--appdata <Kimi应用数据目录> | <index.html 路径>]
"""
import argparse
import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).parent))
from petlib import blueprint_dir, bump_widget_updated_at, find_pet_widget_id

MARKER = "hover-wave-interaction"

INJECTION = r"""
    /* hover-wave-interaction: pet waves back when the pointer hovers it */
    let hoverWaveTimer = 0;
    let hoverWaveCooldownUntil = 0;

    function triggerHoverWave() {
      if (!manifest || isRiveManifest(manifest)) return;
      if (interactionStateActive || preferredStateName !== 'idle') return;
      if (stateName !== 'idle' && stateName !== 'idle_random_2') return;
      const nowTime = Date.now();
      if (nowTime < hoverWaveCooldownUntil) return;
      const wavePool = ['idle_random_1', 'idle_random_3']
        .filter((name) => Boolean(manifest.states[name]));
      if (wavePool.length === 0) return;
      const waveName = wavePool[Math.floor(Math.random() * wavePool.length)];
      const waveState = manifest.states[waveName];
      hoverWaveCooldownUntil = nowTime + 6000;
      interactionStateActive = true;
      clearIdleVariantTimer();
      setState(waveName);
      clearTimeout(hoverWaveTimer);
      hoverWaveTimer = setTimeout(() => {
        interactionStateActive = false;
        setState(preferredStateName);
        scheduleIdleVariant();
      }, Math.round((waveState.frames * 1000) / waveState.fps));
    }

    if (pet) {
      pet.addEventListener('pointerenter', triggerHoverWave);
    }
    const hoverWaveOriginalPointerPresence = handlePointerPresence;
    handlePointerPresence = function (inside) {
      hoverWaveOriginalPointerPresence(inside);
      if (inside === true) triggerHoverWave();
    };
"""

# 在 IIFE 结束(最后一个 "})();" + "</script>")之前注入,与既有补丁兼容
ANCHOR = "\n  })();\n  </script>"


def main() -> None:
    parser = argparse.ArgumentParser(description="给 Kimi Work 桌宠注入悬停交互")
    parser.add_argument("path", nargs="?", default=None, help="index.html 路径(直接指定页面文件)")
    parser.add_argument("--appdata", default=None, help="Kimi 应用数据目录(路径不符时手动指定)")
    args = parser.parse_args()

    if args.path:
        path = Path(args.path)
    else:
        base = blueprint_dir(args.appdata)
        widget_id = find_pet_widget_id(base)
        path = base / "widgets" / widget_id / "workspace/index.html"
    if not path.is_file():
        sys.exit(f"未找到桌宠页面 {path},请确认 Kimi Work 桌面端已运行过,"
                 f"或用 --appdata 指定应用数据目录")

    html = path.read_text(encoding="utf-8")

    if MARKER in html:
        print("已打过补丁,跳过:", path)
        return
    idx = html.rfind(ANCHOR)
    if idx < 0:
        sys.exit("未找到注入锚点,桌宠页面结构可能已更新,请人工检查 index.html")

    patched = html[:idx] + INJECTION + html[idx:]
    path.write_text(patched, encoding="utf-8")
    bump_widget_updated_at(
        blueprint_dir(args.appdata), find_pet_widget_id(blueprint_dir(args.appdata))
    )
    print("✅ 已注入悬停交互:", path)
    print("   重启 Kimi 应用后,鼠标悬停桌宠即可看到它随机回以招手或眨眼互动(6 秒冷却)。")


if __name__ == "__main__":
    main()