chuanqi/protocol_bot/scripts/run_juece13_woma_kill50.py

#!/usr/bin/env python3
"""run_juece13_woma_kill50.py — 角色【新角色】→ 沃玛寺庙一层杀 50 只.

复用 AutoFightBehavior(保命三修版) + CaveEnterBehavior 传送链。
CaveEnterBehavior 引用模块常量(CAVE_MAP/_ACK_BODY/_TELEPORT_ACTS),
故 import 后覆写为沃玛参数(菜单第 3 项 @jiuxian_move_1,1,3,
run14 抓包 SC643 菜单实证)。墙集=d022.map(2026-09-30 凌晨映射定案)。

用法:
    uv run python scripts/run_juece13_woma_kill50.py -u longwu988 -p Longwu1988 \
        --zone 14570166 --name 新角色 [--kills 50] [--minutes 60] [--wait 0]
"""
from __future__ import annotations

import argparse
import logging
import sys
import time
from collections import deque

sys.path.insert(0, "scripts")
from run_juece13_mine_farthest import setup_logging, banner  # noqa: E402
from run_juece13_wugong_autofight import AutoFightBehavior, DIR_VECS  # noqa: E402
import run_juece13_wugong_farthest as cave_mod  # noqa: E402
from run_juece13_wugong_farthest import CaveEnterBehavior  # noqa: E402

# 沃玛参数覆写(CaveEnterBehavior 读模块全局)
WOMA_MAP = "沃玛寺庙一层"
cave_mod.CAVE_MAP = WOMA_MAP
# cap_woma_client.jsonl 真客户端抓包定案(2026-09-30 16:26 用户演示):
#   CS1010{index:2,UserID:10002} → CS1011 Act=@jiuxian_move_1,1,3
#   → CS1011 Act=@act_move,3 ←第二步是,3(进入按钮带地图编号)
#   → ACK 5012+5027 body='沃玛一层' ←协议图名别名, 非显示名!
# (run16d 失败根因: 第二步发了,1 + ACK 发显示名'沃玛寺庙一层', 服务器静默)
cave_mod._ACK_BODY = "沃玛一层"  # noqa: SLF001
cave_mod._TELEPORT_ACTS = (  # noqa: SLF001
    "@jiuxian_move_1,1,3", "@act_move,3")

from pbot.bot.pathfind import load_walls_from_map  # noqa: E402

import requests

from pbot.bot.runner import pump, 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

_LOG = logging.getLogger("juece13_woma50")

_PORTS: tuple[int, ...] = (8021, 8061, 8001, 8041)
_WOMA_MAP_FILE = "assets/maps/d022.map"
_WOMA_W, _WOMA_H = 250, 250

_WOMA_WALLS: set[tuple[int, int]] | None = None


def _woma_walls() -> set[tuple[int, int]]:
    """加载 d022.map 墙格(权威地形)."""
    global _WOMA_WALLS
    if _WOMA_WALLS is None:
        _WOMA_WALLS = load_walls_from_map(_WOMA_MAP_FILE)
    return _WOMA_WALLS


def _woma_farthest(start: tuple[int, int]) -> tuple[tuple[int, int], int, bool]:
    """d022 墙 BFS 最远可达点(撤离目标用)."""
    walls = _woma_walls()
    visited: set[tuple[int, int]] = {start}
    dq = deque([(start, 0)])
    far, far_d = start, 0
    while dq:
        (x, y), d = dq.popleft()
        if d > far_d:
            far, far_d = (x, y), d
        for dx, dy in DIR_VECS:
            nxt = (x + dx, y + dy)
            if nxt in visited or not (0 <= nxt[0] < _WOMA_W
                                      and 0 <= nxt[1] < _WOMA_H):
                continue
            if nxt in walls:
                continue
            visited.add(nxt)
            dq.append((nxt, d + 1))
    return far, far_d, len(visited) < 50


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)
    _ = ap.add_argument("--kills", type=int, default=50)
    _ = ap.add_argument("--minutes", type=float, default=60.0)
    _ = ap.add_argument("--wait", type=float, default=0.0,
                        help="进图后原地静默秒数(仅心跳+收帧)")
    _ = ap.add_argument("--login-port", type=int, default=0)
    return ap.parse_args()


def main() -> int:
    """登录 → 沃玛寺庙一层 → [静默] → 自动战斗至 N 杀或超时 → 退出."""
    a = parse_args()
    setup_logging()
    banner("[0] 沃玛自动战斗: 区=%d 角色=%s 目标=%d杀 时限=%.0fmin 静默=%.0fs",
           a.zone, a.name, a.kills, a.minutes, a.wait)

    http = requests.Session()
    http.trust_env = False
    http.verify = False
    jwt, uid = login(Account(username=a.user, password=a.password))
    _ = enter_game_client(http, jwt, ZoneId(a.zone))
    ports = (a.login_port,) if a.login_port else _PORTS
    _, sid, real_name, ticket_key, game_port = login_gate_with_retry(
        ZoneId(a.zone), jwt, uid, a.name, ports)
    _LOG.info("[票据] sid=%d 角色=%s 游戏端口=%d", sid, real_name, game_port)

    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)", world.me.map_name, world.me.x,
              world.me.y)
    try:
        from pbot.game.heal import enable_auto_heal
        state = None
        for _ in range(2):
            state = enable_auto_heal(conn, sid, want_on=True, world=world)
            if state == "on":
                break
            _LOG.warning("[喝药✗] 状态=%s — 重试一轮", state)
            time.sleep(1.0)
        if state != "on":
            _LOG.error("[喝药✗✗] 高危图无喝药保护 — 终止, 拒绝送死")
            return 1
        _LOG.info("[喝药✓] on")

        if world.me.map_name != WOMA_MAP:
            banner("[1] 传送沃玛寺庙")
            bh = CaveEnterBehavior(sid=sid, kill_target=0)
            r = run(conn, world, updater, bh, duration_s=300.0)
            if world.me.map_name != WOMA_MAP:
                _LOG.error("[✗] 传送失败 result=%s", r.value)
                return 1
            time.sleep(3.0)
            _ = pump(conn, world, updater, 2.0)

        if a.wait > 0:
            banner("[2] 原地静默 %.0f 秒", a.wait)
            deadline = time.time() + a.wait
            while time.time() < deadline:
                _ = pump(conn, world, updater, 0.5)
            _LOG.info("[静默✓] hp=%d/%d 视野怪=%d 位置=(%d,%d) map=%s",
                      world.me.hp, world.me.max_hp,
                      len(world.monsters), world.me.x, world.me.y,
                      world.me.map_name)

        banner("[3] 自动战斗: 目标 %d 杀", a.kills)
        af = AutoFightBehavior(
            sid=sid, map_name=WOMA_MAP, walls=_woma_walls(),
            farthest_fn=_woma_farthest, kill_goal=a.kills)
        r = run(conn, world, updater, af, duration_s=a.minutes * 60)
        _LOG.info("[结束] result=%s 击杀=%d/%d 腿=%d 弹回=%d 位置=(%d,%d) "
                  "hp=%d/%d map=%s",
                  r.value, af.kills, a.kills, af.legs, af.rejected,
                  world.me.x, world.me.y, world.me.hp, world.me.max_hp,
                  world.me.map_name)
        return 0 if af.kills >= a.kills 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)