from nonebot.adapters import Bot, Event
from nonebot.plugin import PluginMetadata
from nonebot_plugin_alconna import (
    Alconna,
    AlconnaQuery,
    Args,
    Arparma,
    Match,
    Option,
    Query,
    Subcommand,
    UniMessage,
    UniMsg,
    on_alconna,
    store_true,
)
from nonebot_plugin_uninfo import Uninfo

from zhenxun.configs.utils import BaseBlock, PluginExtraData, RegisterConfig
from zhenxun.services.log import logger
from zhenxun.utils.depends import UserName
from zhenxun.utils.enum import BlockType, PluginType
from zhenxun.utils.exception import GoodsNotFound
from zhenxun.utils.message import MessageUtils
from zhenxun.utils.platform import PlatformUtils

from ._data_source import ShopManage, gold_rank

__plugin_meta__ = PluginMetadata(
    name="商店",
    description="商店系统[金币回收计划]",
    usage="""
    商品操作
    指令：
        我的金币
        我的道具
        使用道具 [名称/Id]
        购买道具 [名称/Id]
        金币排行 ?[num=10]
        金币总排行 ?[num=10]
    """.strip(),
    extra=PluginExtraData(
        author="HibiKier",
        version="0.1",
        plugin_type=PluginType.NORMAL,
        menu_type="商店",
        limits=[BaseBlock(check_type=BlockType.GROUP)],
        configs=[
            RegisterConfig(
                key="style",
                value="zhenxun",
                help="商店样式类型，[normal, zhenxun]",
                default_value="zhenxun",
            )
        ],
    ).to_dict(),
)

from .goods_register import *  # noqa: F403

_matcher = on_alconna(
    Alconna(
        "商店",
        Option("--all", action=store_true),
        Subcommand("my-cost", help_text="我的金币"),
        Subcommand("my-props", help_text="我的道具"),
        Subcommand("buy", Args["name?", str]["num?", int], help_text="购买道具"),
        Subcommand("use", Args["name?", str]["num?", int], help_text="使用道具"),
        Subcommand("gold-list", Args["num?", int], help_text="金币排行"),
    ),
    priority=5,
    block=True,
)

_matcher.shortcut(
    "我的金币",
    command="商店",
    arguments=["my-cost"],
    prefix=True,
)

_matcher.shortcut(
    "我的道具",
    command="商店",
    arguments=["my-props"],
    prefix=True,
)

_matcher.shortcut(
    "购买道具(?P<name>.*?)",
    command="商店",
    arguments=["buy", "{name}"],
    prefix=True,
)

_matcher.shortcut(
    "使用道具(?P<name>.*?)",
    command="商店",
    arguments=["use", "{name}"],
    prefix=True,
)

_matcher.shortcut(
    "金币排行",
    command="商店",
    arguments=["gold-list"],
    prefix=True,
)

_matcher.shortcut(
    r"金币总排行",
    command="商店",
    arguments=["--all", "gold-list"],
    prefix=True,
)


@_matcher.assign("$main")
async def _(session: Uninfo, arparma: Arparma):
    image = await ShopManage.get_shop_image()
    logger.info("查看商店", arparma.header_result, session=session)
    await MessageUtils.build_message(image).send()


@_matcher.assign("my-cost")
async def _(session: Uninfo, arparma: Arparma):
    logger.info("查看金币", arparma.header_result, session=session)
    gold = await ShopManage.my_cost(
        session.user.id, PlatformUtils.get_platform(session)
    )
    await MessageUtils.build_message(f"你的当前余额: {gold}").send(reply_to=True)


@_matcher.assign("my-props")
async def _(session: Uninfo, arparma: Arparma, nickname: str = UserName()):
    logger.info("查看道具", arparma.header_result, session=session)
    if image := await ShopManage.my_props(
        session.user.id,
        nickname,
        PlatformUtils.get_platform(session),
    ):
        await MessageUtils.build_message(image.pic2bytes()).finish(reply_to=True)
    return await MessageUtils.build_message("你的道具为空捏...").send(reply_to=True)


@_matcher.assign("buy")
async def _(
    session: Uninfo,
    arparma: Arparma,
    name: Match[str],
    num: Query[int] = AlconnaQuery("num", 1),
):
    if not name.available:
        await MessageUtils.build_message(
            "请在指令后跟需要购买的道具名称或id..."
        ).finish(reply_to=True)
    logger.info(
        f"购买道具 {name}, 数量: {num}",
        arparma.header_result,
        session=session,
    )
    result = await ShopManage.buy_prop(session.user.id, name.result, num.result)
    await MessageUtils.build_message(result).send(reply_to=True)


@_matcher.assign("use")
async def _(
    bot: Bot,
    event: Event,
    message: UniMsg,
    session: Uninfo,
    arparma: Arparma,
    name: Match[str],
    num: Query[int] = AlconnaQuery("num", 1),
):
    if not name.available:
        await MessageUtils.build_message(
            "请在指令后跟需要使用的道具名称或id..."
        ).finish(reply_to=True)
    try:
        result = await ShopManage.use(
            bot, event, session, message, name.result, num.result, ""
        )
        logger.info(
            f"使用道具 {name.result}, 数量: {num.result}",
            arparma.header_result,
            session=session,
        )
        if isinstance(result, str):
            await MessageUtils.build_message(result).send(reply_to=True)
        elif isinstance(result, UniMessage):
            await result.finish(reply_to=True)
    except GoodsNotFound:
        await MessageUtils.build_message(
            f"没有找到道具 {name.result} 或道具数量不足..."
        ).send(reply_to=True)


@_matcher.assign("gold-list")
async def _(
    session: Uninfo, arparma: Arparma, num: Query[int] = AlconnaQuery("num", 10)
):
    if num.result > 50:
        await MessageUtils.build_message("排行榜人数不能超过50哦...").finish()
    gid = session.group.id if session.group else None
    if not arparma.find("all") and not gid:
        await MessageUtils.build_message(
            "私聊中无法查看 '金币排行'，请发送 '金币总排行'"
        ).finish()
    if arparma.find("all"):
        gid = None
    result = await gold_rank(session, gid, num.result)
    logger.info(
        "查看金币排行",
        arparma.header_result,
        session=session,
    )
    await MessageUtils.build_message(result).send(reply_to=True)
