from zhenxun.services.log import logger
from zhenxun.utils._build_image import BuildImage
from zhenxun.utils.image_utils import ImageTemplate

from ..config import g_pJsonManager, g_sResourcePath
from ..database import g_pSqlManager


class CShopManager:

    @classmethod
    async def getPlantShopImage(cls) -> bytes:
        """by ATTomato

        Returns:
            bytes: _description_
        """

        data_list = []
        column_name = [
            "-",
            "种子名称",
            "解锁等级",
            "种子单价",
            "收获经验",
            "收获数量",
            "成熟时间（分钟）",
            "收获次数",
            "再次成熟时间（分钟）",
            "是否可以上架交易行"
        ]

        sell = ""
        plants = g_pJsonManager.m_pPlant['plant'] # type: ignore
        for key, plant in plants.items():
            icon = ""
            icon_path = g_sResourcePath / f"plant/{key}/icon.png"
            if icon_path.exists():
                icon = (icon_path, 33, 33)

            if plant['again'] == True:
                sell = "可以"
            else:
                sell = "不可以"

            data_list.append(
                [
                    icon,
                    key,
                    plant['level'],
                    plant['price'],
                    plant['experience'],
                    plant['harvest'],
                    plant['time'],
                    plant['crop'],
                    plant['again'],
                    sell
                ]
            )

        result = await ImageTemplate.table_page(
            "种子商店",
            "购买示例：@小真寻 购买种子 大白菜 5",
            column_name,
            data_list,
        )

        return result.pic2bytes()

    @classmethod
    async def buyPlant(cls, uid: str, name: str, num: int = 1) -> str:
        if num <= 0:
            return "请输入购买数量！"

        plantInfo = None

        try:
            plantInfo = g_pJsonManager.m_pPlant['plant'][name] # type: ignore
        except Exception as e:
            return "购买出错！请检查需购买的种子名称！"

        userPlants = {}

        point = await g_pSqlManager.getUserPointByUid(uid)
        total = int(plantInfo['price']) * num

        logger.debug(f"用户：{uid}购买{name}，数量为{num}。用户农场币为{point}，购买需要{total}")

        if point < total:
            return "你的农场币不够哦~ 快速速氪金吧！"
        else:
            p = await g_pSqlManager.getUserPlantByUid(uid)

            if not p == None:
                for item in p.split(','):
                    if '|' in item:
                        plant_name, count = item.split('|', 1)  # 分割一次，避免多竖线问题
                        userPlants[plant_name] = int(count)

            if name in userPlants:
                userPlants[name] += num
            else:
                userPlants[name] = num

            plant_list = [f"{k}|{v}" for k, v in userPlants.items()]

            await g_pSqlManager.updateUserPointByUid(uid, point - total)
            await g_pSqlManager.updateUserPlantByUid(uid, ','.join(plant_list))

            return f"成功购买{name}，当前仓库数量为：{userPlants[name]}，花费{total}农场币, 剩余{point - total}农场币"


g_pShopManager = CShopManager()