#!/usr/bin/env python3
"""猜数字游戏 — 飞书/聊天交互版（文件持久化）"""

import json
import os
import random

STATE_FILE = "/tmp/guess_number_state.json"


def _load_state() -> dict:
    if os.path.exists(STATE_FILE):
        with open(STATE_FILE) as f:
            return json.load(f)
    return {"secret": None, "attempts": 0}


def _save_state(state: dict):
    with open(STATE_FILE, "w") as f:
        json.dump(state, f)


def is_valid(guess: str) -> bool:
    return len(guess) == 4 and guess.isdigit() and len(set(guess)) == 4


def start() -> str:
    secret = "".join(random.sample("0123456789", 4))
    _save_state({"secret": secret, "attempts": 0})
    return (
        "🎯 猜数字游戏开始！\n"
        "规则：我已生成一个四位不重复数字（0-9）。\n"
        "- A = 数字对且位置对\n"
        "- B = 数字对但位置不对\n\n"
        "直接输入你的猜测（4位数字）即可开始！"
    )


def guess(guess_str: str) -> str:
    state = _load_state()
    secret = state.get("secret")

    if secret is None:
        return "游戏还没开始！请说「开始游戏」"

    if not is_valid(guess_str):
        return "输入错误！请输入4位不重复数字（0-9）。例如：1234"

    a = sum(g == s for g, s in zip(guess_str, secret))
    b = sum(g in secret for g in guess_str) - a
    state["attempts"] += 1

    if a == 4:
        msg = (
            f"🎉 恭喜！猜对了！\n"
            f"答案：{secret}\n"
            f"总共猜了 {state['attempts']} 次！\n"
            f"说「再来一局」重新开始"
        )
        _save_state({"secret": None, "attempts": 0})
        return msg

    _save_state(state)
    return f"结果：{a}A{b}B（第 {state['attempts']} 次）"