"""打卡记鹿 —— 触发器(Trigger + View)层。"""

from __future__ import annotations

import base64
import io
from datetime import datetime

from nonebot import on_command
from nonebot.adapters.onebot.v11 import MessageEvent, Bot, MessageSegment
from nonebot.rule import to_me
from PIL import Image

from ..services.checkin import checkin, review

record = on_command("打卡记鹿", aliases={"打卡"}, rule=to_me())
query = on_command("查询记录", aliases={"查卡"})


def pic2b64(pic: Image) -> str:
    buf = io.BytesIO()
    pic.save(buf, format="PNG")
    base64_str = base64.b64encode(buf.getvalue()).decode()
    return "base64://" + base64_str


@record.handle()
async def do_job(bot: Bot, ev: MessageEvent):
    now = datetime.now()
    year = now.year
    month = now.month
    uid = str(ev.user_id)
    message_id = ev.message_id
    img = await checkin(uid, year, month)
    text_msg = MessageSegment.text("这是您的打卡记鹿：\n")
    image_msg = MessageSegment.image(pic2b64(img))
    mix_msg = (text_msg,)
    mix_msg += image_msg
    reply_msg = (MessageSegment.reply(message_id),)
    reply_msg += mix_msg
    await record.finish(reply_msg)


@query.handle()
async def query_job(bot: Bot, ev: MessageEvent):
    now = datetime.now()
    year = now.year
    month = now.month
    uid = str(ev.user_id)
    message_id = ev.message_id
    img_file = await review(uid, year, month)
    text_msg = MessageSegment.text("这是您的记鹿：\n")
    image_msg = MessageSegment.image(pic2b64(img_file))
    mix_msg = (text_msg,)
    mix_msg += image_msg
    reply_msg = (MessageSegment.reply(message_id),)
    reply_msg += mix_msg
    await record.finish(reply_msg)