import requests
import json
import time
import datetime

# ====================== 你只需要改这里 ======================
# Evennia 账号（你自己创建）
EVENNIA_HOST = "localhost"
EVENNIA_PORT = 4000
ACCOUNT = ""          # 你填：比如 ai_bot1
PASSWORD = ""         # 你填密码

# 本地大模型接口（llama.cpp / Ollama / Deepseek 本地都行）
LLM_API_URL = "http://localhost:1234/v1/chat/completions"
MODEL_NAME = ""       # 你本地模型名，比如 llama-7b.gguf

# 日志文件
LOG_FILE = "ai_bot_log.txt"
# ==========================================================

def log_print(msg):
    """同时打印+写日志"""
    now = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
    line = f"[{now}] {msg}\n"
    print(line.strip())
    with open(LOG_FILE, "a", encoding="utf-8") as f:
        f.write(line)

def get_ai_response(prompt_history):
    """调用本地大模型"""
    headers = {"Content-Type": "application/json"}
    data = {
        "model": MODEL_NAME,
        "messages": prompt_history,
        "temperature": 0.7,
        "max_tokens": 256,
        "stop": ["\n"]
    }
    try:
        resp = requests.post(LLM_API_URL, json=data, timeout=20)
        resp.raise_for_status()
        return resp.json()["choices"][0]["message"]["content"].strip()
    except Exception as e:
        log_print(f"模型调用失败: {e}")
        return "look"

def build_prompt(mud_output):
    """给AI的系统提示 + 当前场景"""
    system_prompt = """
你是一个MUD游戏里的探索AI。
你只能输出**一条简单命令**，不要多余文字。
目标：探索、移动、查看、互动，正常玩MUD。
可用命令：
look, north, south, east, west, up, down, inventory, take, drop, say xxx
只输出命令本身，不要解释。
""".strip()

    return [
        {"role": "system", "content": system_prompt},
        {"role": "user", "content": f"当前场景：\n{mud_output}\n你下一步命令："}
    ]

def evennia_telnet_bot():
    """简易Telnet客户端，连Evennia"""
    import socket
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    s.settimeout(10)
    try:
        s.connect((EVENNIA_HOST, EVENNIA_PORT))
        log_print("已连接Evennia")
    except Exception as e:
        log_print(f"连接失败: {e}")
        return

    def read_chunk():
        try:
            return s.recv(8192).decode("utf-8", errors="ignore")
        except:
            return ""

    def send_cmd(cmd):
        s.sendall((cmd + "\n").encode("utf-8"))

    # 登录流程
    buffer = ""
    while True:
        buffer += read_chunk()
        if "Enter account name" in buffer or "Account:" in buffer:
            send_cmd(ACCOUNT)
            buffer = ""
            time.sleep(0.5)
        elif "Password:" in buffer:
            send_cmd(PASSWORD)
            buffer = ""
            time.sleep(0.5)
            break
        time.sleep(0.2)

    log_print("登录成功，开始AI自动探索")

    # 主循环：看场景 → AI决策 → 执行 → 记录
    last_output = ""
    while True:
        buffer += read_chunk()
        if buffer.strip():
            last_output = buffer.strip()
            log_print(f"=== MUD输出 ===\n{last_output}")
            buffer = ""

            # AI决策
            prompt = build_prompt(last_output)
            ai_cmd = get_ai_response(prompt)
            log_print(f"AI命令: {ai_cmd}")

            send_cmd(ai_cmd)
            time.sleep(1.5)
        time.sleep(0.3)

if __name__ == "__main__":
    log_print("=== AI探索机器人启动 ===")
    evennia_telnet_bot()