"""今日老婆 —— 触发器(Trigger + View)层。"""

from __future__ import annotations

import base64
import datetime
from io import BytesIO
from random import choice

from nonebot import on_command
from nonebot.adapters.onebot.v11 import MessageEvent, Bot, MessageSegment
from nonebot.plugin import PluginMetadata
from PIL import Image

from ..services.store import get_member_list, load_group_config, write_group_config
from ..utils import download_avatar, png_resize

__plugin_meta__ = PluginMetadata(
    name="今日老婆",
    description="随机抓取群友作为老婆",
    usage="发送【今日老婆】",
    type="application",
)

wife = on_command("今日老婆", aliases={"今日老婆"})


async def get_wife_info(member_info, qq_id):
    img = await download_avatar(qq_id)
    avatar = Image.open(BytesIO(img)).convert("RGBA")
    avatar = png_resize(avatar, new_width=145, new_height=145)
    b_io = BytesIO()
    avatar.save(b_io, format="PNG")
    avatar_str = "base64://" + base64.b64encode(b_io.getvalue()).decode()
    member_name = member_info["card"] or member_info["nickname"]
    return (
        MessageSegment.text("你今天的群友老婆是:"),
        MessageSegment.image(avatar_str),
        MessageSegment.text(f"{member_name}({qq_id})"),
    )


@wife.handle()
async def wife_handle(bot: Bot, ev: MessageEvent):
    group_id = ev.group_id
    user_id = ev.user_id
    bot_id = ev.self_id
    wife_id = None
    today = str(datetime.date.today())
    config = load_group_config(group_id)

    if config is not None:
        if str(user_id) in list(config):
            if config[str(user_id)][1] == today:
                wife_id = config[str(user_id)][0]
            else:
                del config[str(user_id)]

    if wife_id is None:
        all_list = await bot.get_group_member_list(group_id=group_id)
        id_list = get_member_list(all_list)
        id_list.remove(bot_id)
        id_list.remove(user_id)
        if config is not None:
            for record_id in list(config):
                if config[record_id][1] != today:
                    del config[record_id]
                else:
                    try:
                        id_list.remove(int(config[record_id][0]))
                    except Exception:
                        del config[record_id]
        wife_id = choice(id_list)

    write_group_config(group_id, user_id, wife_id, today, config)
    member_info = await bot.get_group_member_info(group_id=group_id, user_id=wife_id)
    result = await get_wife_info(member_info, wife_id)
    await bot.send(ev, result, at_sender=True)