"""搜图 —— 搜索编排(Service 层)。"""

from __future__ import annotations

from nonebot import get_driver
from nonebot.adapters.onebot.v11 import Bot, GroupMessageEvent
from nonebot.log import logger

from ..config import CHAIN_REPLY, SAUCENAO_KEY, threshold
from .image import get_image_data_ascii, get_image_data_sauce
from .state import lmtd

_driver = get_driver()
SUPERUSERS = list(_driver.config.superusers)
_NICKNAMES = list(_driver.config.nickname) if _driver.config.nickname else ["竹竹"]
NICKNAME = _NICKNAMES[0]


async def chain_reply(bot: Bot, ev: GroupMessageEvent, chain, msg):
    if not CHAIN_REPLY:
        return chain
    data = {
        "type": "node",
        "data": {
            "name": str(NICKNAME) if str(NICKNAME) else "竹竹",
            "uin": str(ev.self_id),
            "content": str(msg),
        },
    }
    chain.append(data)
    return chain


async def picfinder(bot: Bot, ev: GroupMessageEvent, image_data):
    uid = ev.user_id
    chain = []
    result = await get_image_data_sauce(image_data, SAUCENAO_KEY)
    image_data_report = result[0]
    simimax = result[1]
    if "Index #" in image_data_report:
        su = next(iter(SUPERUSERS))
        await bot.send_private_msg(user_id=int(su), message="发生index解析错误")
        await bot.send_private_msg(user_id=int(su), message=image_data)
        await bot.send_private_msg(user_id=int(su), message=image_data_report)
    chain = await chain_reply(bot, ev, chain, image_data_report)

    if float(simimax) > float(threshold):
        lmtd.increase(uid)
    else:
        if simimax != 0:
            chain = await chain_reply(bot, ev, chain, "相似度过低，换用ascii2d检索中…")
        else:
            logger.error("SauceNao not found imageInfo")
            chain = await chain_reply(bot, ev, chain, "SauceNao检索失败,换用ascii2d检索中…")

        image_data_report = await get_image_data_ascii(image_data)
        if image_data_report[0]:
            chain = await chain_reply(bot, ev, chain, image_data_report[0])
            lmtd.increase(uid)
        if image_data_report[1]:
            chain = await chain_reply(bot, ev, chain, image_data_report[1])
        if not (image_data_report[0] or image_data_report[1]):
            logger.error("ascii2d not found imageInfo")
            chain = await chain_reply(bot, ev, chain, "ascii2d检索失败…")

    if CHAIN_REPLY:
        await bot.send_group_forward_msg(group_id=ev.group_id, messages=chain)