"""AMT 数据库管理脚本 - 交互式管理 Tool 表"""

import json
import os
import sys

from peewee import SqliteDatabase, Model, CharField, TextField

db_path = os.path.join(os.path.dirname(__file__), "data.db")
db = SqliteDatabase(db_path)


class BaseModel(Model):
    class Meta:
        database = db


class Tool(BaseModel):
    code = CharField(max_length=4, unique=True)
    step = TextField()
    desp = TextField(default="")


def init_db():
    db.connect()
    db.create_tables([Tool])


def list_tools():
    tools = Tool.select().order_by(Tool.code)
    if not tools:
        print("\n  (空)")
        return
    print()
    for t in tools:
        print(f"  [{t.code}] {t.desp or '(无描述)'}")


def show_tool():
    code = input("输入 code: ").strip()
    tool = Tool.get_or_none(Tool.code == code)
    if tool is None:
        print(f"  code {code} 不存在")
        return
    print(f"\n  code: {tool.code}")
    print(f"  desp: {tool.desp or '(无描述)'}")
    print(f"  step:")
    print(json.dumps(json.loads(tool.step), indent=2, ensure_ascii=False))


def add_tool():
    code = input("输入 code (4位数字): ").strip()
    if len(code) != 4 or not code.isdigit():
        print("  code 必须是4位数字")
        return
    if Tool.get_or_none(Tool.code == code):
        print(f"  code {code} 已存在，请用编辑功能修改")
        return

    desp = input("输入描述: ").strip()
    print("输入 step JSON (输入空行结束):")
    lines = []
    while True:
        line = input()
        if line == "":
            break
        lines.append(line)

    raw = "\n".join(lines)
    try:
        parsed = json.loads(raw)
    except json.JSONDecodeError as e:
        print(f"  JSON 解析失败: {e}")
        return

    Tool.create(code=code, step=json.dumps(parsed, ensure_ascii=False), desp=desp)
    print(f"  已添加 [{code}]")


def edit_tool():
    code = input("输入要编辑的 code: ").strip()
    tool = Tool.get_or_none(Tool.code == code)
    if tool is None:
        print(f"  code {code} 不存在")
        return

    print(f"  当前描述: {tool.desp or '(无描述)'}")
    desp = input("新描述 (回车跳过): ").strip()
    if desp:
        tool.desp = desp

    print(f"  当前 step:")
    print(json.dumps(json.loads(tool.step), indent=2, ensure_ascii=False))
    choice = input("修改 step? (y/N): ").strip().lower()
    if choice == "y":
        print("输入新的 step JSON (输入空行结束):")
        lines = []
        while True:
            line = input()
            if line == "":
                break
            lines.append(line)
        raw = "\n".join(lines)
        try:
            parsed = json.loads(raw)
        except json.JSONDecodeError as e:
            print(f"  JSON 解析失败: {e}")
            return
        tool.step = json.dumps(parsed, ensure_ascii=False)

    tool.save()
    print(f"  已更新 [{code}]")


def delete_tool():
    code = input("输入要删除的 code: ").strip()
    tool = Tool.get_or_none(Tool.code == code)
    if tool is None:
        print(f"  code {code} 不存在")
        return

    print(f"  [{tool.code}] {tool.desp or '(无描述)'}")
    confirm = input("确认删除? (y/N): ").strip().lower()
    if confirm == "y":
        tool.delete_instance()
        print(f"  已删除 [{code}]")
    else:
        print("  已取消")


MENU = {
    "1": ("列出所有", list_tools),
    "2": ("查看详情", show_tool),
    "3": ("添加", add_tool),
    "4": ("编辑", edit_tool),
    "5": ("删除", delete_tool),
}


def main():
    init_db()
    print("=== AMT 数据库管理 ===")
    while True:
        print()
        for k, (label, _) in MENU.items():
            print(f"  {k}. {label}")
        print("  q. 退出")

        choice = input("\n> ").strip().lower()
        if choice == "q":
            break
        action = MENU.get(choice)
        if action:
            try:
                action[1]()
            except (KeyboardInterrupt, EOFError):
                print()
        else:
            print("  无效选项")

    db.close()


if __name__ == "__main__":
    main()