import os
import json
from pathlib import Path


WIFE_DIR = Path(".") / "ATRI" / "data" / "database" / "wife"
MERRY_LIST_PATH = WIFE_DIR / "merry_list.json"
os.makedirs(WIFE_DIR, exist_ok=True)


class Tsuma:
    @staticmethod
    def _load_tsuma() -> dict:
        try:
            return json.loads(MERRY_LIST_PATH.read_bytes())
        except FileNotFoundError:
            with open(MERRY_LIST_PATH, "w") as r:
                r.write(json.dumps({}, indent=4))
            return dict()

    @staticmethod
    def _store_tsuma(data: dict) -> None:
        with open(MERRY_LIST_PATH, "w") as r:
            r.write(json.dumps(data, indent=4))

    @classmethod
    def check_tsuma(cls, user: str):
        data = cls._load_tsuma()
        if user in data:
            msg = "阿，你已经有老婆惹！" f"ta是：{data[user]['lassie']['nickname']}"
            return msg, True
        else:
            return "悲——你还没老婆...", False

    @classmethod
    def inquire_tsuma(cls, user: str) -> str:
        data = cls._load_tsuma()
        if user in data:
            return f"你的老婆是：{data[user]['lassie']['nickname']} 哦~❤"
        else:
            return "悲——你还没老婆..."

    @classmethod
    def got_tsuma(cls, user: str, d: dict) -> str:
        check_repo, if_h = cls.check_tsuma(user)  # 防止出现多人同时操作导致 NTR 事件
        if if_h:
            return check_repo
        else:
            data = cls._load_tsuma()
            data[user] = {
                "nickname": d["nickname"],
                "gender": d["gender"],
                "lassie": {
                    "nickname": d["lassie"]["nickname"],
                    "qq": d["lassie"]["qq"],
                    "gender": d["lassie"]["gender"],
                },
            }
            cls._store_tsuma(data)

            data[d["lassie"]["qq"]] = {
                "nickname": d["lassie"]["nickname"],
                "gender": d["lassie"]["gender"],
                "lassie": {
                    "nickname": d["nickname"],
                    "qq": user,
                    "gender": d["gender"],
                },
            }
            cls._store_tsuma(data)

            msg = (
                f"> {d['lassie']['nickname']}({d['lassie']['qq']})\n"
                f"恭喜成为 {d['nickname']} 的老婆~⭐"
            )
            return msg

    @classmethod
    def divorce(cls, user: str) -> str:
        data = cls._load_tsuma()
        if not user in data:
            return "悲——你还没老婆。。"

        msg = f"悲——，({data[user]['nickname']})抛弃了({data[user]['lassie']['nickname']})"
        del data[user]
        cls._store_tsuma(data)
        return msg