chuanqi/protocol_bot/scripts/run_juece13_cave30.py

"""run_juece13_cave30.py — 角色【新角色】→ 天然洞穴一层 → 杀 30 怪(全部在洞内)。

与 run_juece13_kill30.py 的差异:
- 硬约束: 30 杀必须全部发生在【天然洞穴一层】地图内
  - CaveOnlyKillBehavior 继承 KillBehavior, 三道闸门:
    1) on_enter: 非洞穴 → 只走 renwu_mov,2 任务链传送进洞, 绝不异图开战
    2) _pick_target/_current_target: 非洞穴 → 永不选怪(异图只触发换落点)
    3) _credit_kill: 异图击杀 → 记日志但不计数(只有洞内击杀进 30/30)
- 无怪应对(框架既有 + 本约束): CS5284 问服务器怪位 → 新颖度锚点跑图
  → 落点无怪 45s 重走 renwu 链换洞内落点

用法:
    uv run python scripts/run_juece13_cave30.py -u longwu988 -p Longwu1988 \
        --zone 14570166 --name 新角色 --kill 30
"""
from __future__ import annotations

import argparse
import logging
import sys
import time

import requests

from typing import override

from pbot.bot.kill import LAND_QUIET_S, KillBehavior
from pbot.bot.runner import run
from pbot.config import Account
from pbot.errors import PbotError
from pbot.game.session import enter_world
from pbot.login.box_client import enter_game_client, login
from pbot.transport.gateway import login_gate_with_retry
from pbot.types import ZoneId
from pbot.world.mirror import WorldMirror

_LOG = logging.getLogger("juece13_cave")

_PORTS: tuple[int, ...] = (8021, 8061, 8001, 8041)
CAVE_MAP: str = "天然洞穴一层"


def setup_logging() -> None:
    """全量 INFO 日志, 框架每步/每刀/每杀均有日志, 便于事后分析。"""
    logging.basicConfig(
        level=logging.INFO,
        format="%(asctime)s %(levelname)-7s %(name)s: %(message)s",
        stream=sys.stdout,
    )


def banner(msg: str, *args: object) -> None:
    """阶段分隔条日志(支持 %-style 格式化参数)。"""
    _LOG.info("======== %s ========", msg % args if args else msg)


class CaveOnlyKillBehavior(KillBehavior):
    """只在 天然洞穴一层 选怪/开战/计杀; 异图一律重走 renwu 链回洞。"""

    def _in_cave(self, world: WorldMirror) -> bool:
        """当前是否在目标洞穴图。"""
        return world.me.map_name == CAVE_MAP

    @override
    def on_enter(self, conn: GatewayConnection, world: WorldMirror) -> None:
        """入口闸门: 洞内直接开打; 异图(比奇/等待图/其他)→ renwu 链进洞。"""
        self._world = world
        self._last_step_t = time.time()
        self._declare_guaji(conn)
        if self._bound is None:
            self._bound = ((world.me.x, world.me.y), 25)
        self._load_map_if_needed(world)
        if self._in_cave(world):
            _LOG.info("[洞穴] 出生即在 %s @(%d,%d) — 静置%.1fs后开打",
                      CAVE_MAP, world.me.x, world.me.y, LAND_QUIET_S)
            self._enter("fight")
            self._phase_t = time.time() + LAND_QUIET_S
        else:
            _LOG.info("[洞穴] 当前 %s @(%d,%d) 非目标图 — 走 renwu_mov,2 传送进洞",
                      world.me.map_name, world.me.x, world.me.y)
            self._enter("renwu_3031")

    @override
    def _current_target(self, world: WorldMirror):  # noqa: ANN201
        """非洞穴 → 永不锁定目标(防切图瞬间打残影目标)。"""
        if not self._in_cave(world):
            return None
        return super()._current_target(world)

    @override
    def _pick_target(self, world: WorldMirror):  # noqa: ANN201
        """非洞穴 → 永不选怪: tick 走无怪分支 → 45s 后 renwu 换落点回洞。"""
        if not self._in_cave(world):
            return None
        return super()._pick_target(world)

    @override
    def _credit_kill(self, world: WorldMirror, aid: int, name: str,
                     signal: str) -> None:
        """计杀闸门: 只有洞内击杀计入 30/30; 异图击杀仅记日志。"""
        if not self._in_cave(world):
            _LOG.warning("[异图击杀不计] aid=%d %s 信号=%s @%s — 目标图=%s",
                         aid, name, signal, world.me.map_name, CAVE_MAP)
            _ = self._my_hits.pop(aid, None)
            return
        super()._credit_kill(world, aid, name, signal)


def parse_args() -> argparse.Namespace:
    """命令行: 账号/区/角色/目标击杀数。"""
    ap = argparse.ArgumentParser()
    _ = ap.add_argument("-u", "--user", required=True)
    _ = ap.add_argument("-p", "--password", required=True)
    _ = ap.add_argument("--zone", type=int, required=True)
    _ = ap.add_argument("--name", required=True, help="已有角色名")
    _ = ap.add_argument("--kill", type=int, default=30)
    _ = ap.add_argument("--login-port", type=int, default=0)
    return ap.parse_args()


def main() -> int:
    """登录 → 进游戏 → 喝药确认 → 洞穴限定的 30 杀。"""
    a = parse_args()
    setup_logging()
    banner("[0] 启动: 区=%d 角色=%s 目标=%d 杀(仅限 %s)",
           a.zone, a.name, a.kill, CAVE_MAP)

    http = requests.Session()
    http.trust_env = False
    http.verify = False

    banner("[1] 盒子登录")
    jwt, uid = login(Account(username=a.user, password=a.password))
    _LOG.info("[✓] uid=%d", uid)

    banner("[2] EnterGameClient zone=%d", a.zone)
    for attempt in range(4):
        try:
            _ = enter_game_client(http, jwt, ZoneId(a.zone))
            _LOG.info("[✓] 上报成功")
            break
        except Exception as e:
            if attempt == 3:
                raise
            _LOG.warning("[!] 第%d次失败(%s), 12s 后重试", attempt + 1, e)
            time.sleep(12)

    banner("[3] 登录服选角: %s", a.name)
    ports = (a.login_port,) if a.login_port else _PORTS
    key32, sid, real_name, ticket_key, game_port = login_gate_with_retry(
        ZoneId(a.zone), jwt, uid, a.name, ports)
    _LOG.info("[票据] key32=%s sid=%d 角色=%s 游戏端口=%d",
              key32.decode(errors="replace"), sid, real_name, game_port)

    banner("[4] 进游戏(CS20 → SC658 → 17帧初始化 → 出生)")
    stop: list[bool] = [False]
    conn, world, updater = enter_world(ZoneId(a.zone), real_name, ticket_key,
                                       sid, game_port, uid, stop=stop)
    _LOG.info("[✓] 出生 map=%s @(%d,%d) actor=%s",
              world.me.map_name, world.me.x, world.me.y, world.me.actor_key)
    if world.me.map_name != CAVE_MAP:
        _LOG.info("[i] 出生不在洞穴(在 %s) — 行为入口会自动走 renwu 链传送进洞",
                  world.me.map_name)

    try:
        banner("[5] 喝药保护确认")
        from pbot.game.heal import enable_auto_heal
        state = enable_auto_heal(conn, sid, want_on=True, world=world)
        if state == "on":
            _LOG.info("[✓] 喝药保护已开启(服务端 but_1 确认)")
        else:
            _LOG.warning("[!] 喝药状态=%s(不阻断, 继续任务链)", state)

        banner("[6] 洞穴杀怪启动: 目标 %d 杀 @%s", a.kill, CAVE_MAP)
        kill = CaveOnlyKillBehavior(
            sid=sid,
            kill_target=a.kill,
            renwu_act="renwu_mov,2",   # 任务二 = 天然洞穴一层(SC643 弹窗实测)
            allow_reroll=True,         # 落点无怪 → 重走 renwu 链换洞内落点
        )
        duration = max(900.0, a.kill * 150.0)
        _LOG.info("行为超时上限: %.0fs(达标即提前退出)", duration)
        result = run(conn, world, updater, kill, duration_s=duration)

        banner("[7] 结果")
        _LOG.info("result=%s 洞内击杀=%d/%d 最终 map=%s @(%d,%d)",
                  result.value, world.kills, a.kill,
                  world.me.map_name, world.me.x, world.me.y)
        top = sorted(world.frame_counts.items(), key=lambda kv: -kv[1])[:15]
        _LOG.info("收帧Top15(问题分析用): %s", top)
        ok = world.kills >= a.kill and world.me.map_name == CAVE_MAP
        _LOG.info("[%s] %s", "✓" if ok else "✗",
                  "任务完成: 30 杀全部在 天然洞穴一层"
                  if ok else "未达标(看上方日志定位)")
        return 0 if ok else 1
    finally:
        stop[0] = True
        conn.close()
        _LOG.info("连接已关闭")


if __name__ == "__main__":
    try:
        sys.exit(main())
    except PbotError as e:
        _LOG.error("失败: %s", e.detail)
        sys.exit(1)