# -*- coding: utf-8 -*-
"""加「跨网络省流量」开关。

问题（实测出来的）：
    经隧道传一个 256KB 的屏幕分块，上行要 9.4 秒。
    因为经中转的那条线上，一帧画面的 JSON（含 base64）约 350KB，
    而这条 4G 线路的上行只有 ~37 KB/s。
    → 屏幕镜像会卡成幻灯片。

做法：
    开这个开关就把屏幕流降一档：
        分辨率 480 -> 360
        画质   55  -> 40
    一帧从 ~350KB 降到 ~60KB，速度差 5 倍以上。
    局域网直连时不用开（本来就不慢）。

    关掉时**还原成原来的值** —— 所以先把原值存下来。
"""
from __future__ import annotations

import sys
from pathlib import Path

FEATURES = Path(r"C:\Users\shu\Desktop\deepseekagent\LanRemote\windows"
                r"\lanremote\features.py")

DEFS_OLD = '''        add("net_traffic", "实时网速", "action", "网络")'''

DEFS_NEW = '''        add("net_traffic", "实时网速", "action", "网络")
        add("relay_saver", "跨网络省流量", "toggle", "网络",
            on=bool(self.store.get("relay_saver")))'''

IMPL_ANCHOR = '''    def _do_net_traffic(self, params) -> str:'''

IMPL_NEW = '''    def _do_relay_saver(self, params) -> str:
        """跨网络省流量：把屏幕流降一档。

        实测：经隧道上行只有 ~37 KB/s，而 480p/55 的一帧约 350KB，
        传一帧要 9 秒 —— 屏幕镜像等于不能用。
        降到 360p/40 之后一帧约 60KB，快 5 倍以上。

        关掉时会**还原成开之前的设置**，所以先把原值存下来。
        """
        on = self._truthy(params.get("on"))
        if on:
            saved = self.store.get("relay_saver_saved")
            if not isinstance(saved, dict) or not saved:
                # 第一次开：记下原值，方便关的时候还原
                saved = {
                    "width": self.store.get("screen_max_width"),
                    "quality": self.store.get("screen_quality"),
                }
                self.store.set("relay_saver_saved", saved)
            self.store.set("relay_saver", True)
            self.store.set("screen_max_width", 360)
            self.store.set("screen_quality", 40)
            note = (f"已开启省流量：屏幕 360p / 画质 40"
                    f"（原来 {(saved or {}).get('width') or 480}p / "
                    f"{(saved or {}).get('quality') or 55}）")
        else:
            saved = self.store.get("relay_saver_saved")
            self.store.set("relay_saver", False)
            if isinstance(saved, dict) and saved:
                if saved.get("width"):
                    self.store.set("screen_max_width", saved["width"])
                if saved.get("quality"):
                    self.store.set("screen_quality", saved["quality"])
                note = (f"已关闭省流量，还原为 {saved.get('width')}p / "
                        f"画质 {saved.get('quality')}")
                self.store.set("relay_saver_saved", {})
            else:
                note = "已关闭省流量（没有记录到原设置，保持当前值）"

        # 正在推流的话要重建采集器，新设置才生效
        try:
            self.actions.reset_capturer()
        except Exception:                                 # noqa: BLE001
            pass
        return note

''' + IMPL_ANCHOR


def main() -> int:
    text = FEATURES.read_text(encoding="utf-8")
    ok = True

    if "relay_saver" in text:
        print("  已经加过省流量开关")
        return 0

    for old, new, tag in ((DEFS_OLD, DEFS_NEW, "功能声明"),
                          (IMPL_ANCHOR, IMPL_NEW, "实现")):
        if old not in text:
            print(f"  [失败] {tag} —— 锚点没找到")
            ok = False
            continue
        text = text.replace(old, new, 1)
        print(f"  {tag}")

    FEATURES.write_text(text, encoding="utf-8")

    import ast
    try:
        ast.parse(text)
    except SyntaxError as exc:
        print(f"  语法错误: {exc}")
        ok = False
    return 0 if ok else 1


if __name__ == "__main__":
    sys.exit(main())