import re
from random import choice

from nonebot.matcher import Matcher
from nonebot.params import CommandArg, ArgPlainText
from nonebot.adapters.onebot.v11 import Bot, MessageEvent, Message, MessageSegment
from nonebot.adapters.onebot.v11.helpers import (
    extract_image_urls,
    Cooldown,
    autorevoke_send,
)

from ATRI import conf
from ATRI.service import Service
from ATRI.permission import MASTER

from .data_source import Setu


plugin = Service("涩图").document("hso!").main_cmd("/setu")


random_setu = plugin.on_command(
    "来张涩图", "来张随机涩图，冷却2分钟", aliases={"涩图来", "来点涩图", "来份涩图"}, priority=5
)


@random_setu.handle([Cooldown(120)])
async def _():
    ...


@random_setu.handle()
async def _(bot: Bot, event: MessageEvent):
    setu, setu_data = await Setu.new()
    setu_info = f"Title: {setu_data.title}\nPid: {setu_data.pid}"
    await bot.send(event, setu_info)

    try:
        await autorevoke_send(bot, event, setu)
    except Exception:
        await random_setu.send("hso (发不出")
        await random_setu.send(f"自己动手: {setu_data.url}")


@random_setu.got("r_rush_after_think", prompt="看完不来点感想么-w-")
async def _(think: str = ArgPlainText("r_rush_after_think")):
    is_repo = will_think(think)
    if not is_repo:
        await random_setu.finish()
    else:
        await random_setu.finish(is_repo)


tag_setu = plugin.on_regex(r"来[张点丶份](.*?)的?[涩色🐍]图", "根据提供的tag查找涩图，冷却2分钟", priority=6)


@tag_setu.handle([Cooldown(120)])
async def _(bot: Bot, event: MessageEvent):
    msg = str(event.get_message()).strip()
    pattern = r"来[张点丶份](.*?)的?[涩色🐍]图"
    tag = re.findall(pattern, msg)[0]
    setu, setu_data = await Setu.new(tag)
    if not setu_data.url:
        await tag_setu.finish("没有合适的涩图呢...")

    setu_info = f"Title: {setu_data.title}\nPid: {setu_data.pid}"
    await bot.send(event, setu_info)

    try:
        await autorevoke_send(bot, event, setu)
    except Exception:
        await random_setu.send("hso (发不出")
        await random_setu.send(f"自己动手: {setu_data.url}")
        return


@tag_setu.got("t_rush_after_think", prompt="看完不来点感想么-w-")
async def _(think: str = ArgPlainText("t_rush_after_think")):
    is_repo = will_think(think)
    if not is_repo:
        await random_setu.finish()
    else:
        await random_setu.finish(is_repo)


_catcher_max_file_size = 128
_catcher_disab_gif = False


setu_catcher = plugin.on_message("涩图嗅探", "涩图嗅探器", block=False)


@setu_catcher.handle()
async def _(bot: Bot, event: MessageEvent):
    args = extract_image_urls(event.message)
    if not args:
        return
    else:
        hso = list()
        for i in args:
            try:
                data = await Setu(i).detecter(
                    _catcher_max_file_size, _catcher_disab_gif
                )
            except Exception:
                return
            if data > 0.7:
                hso.append(data)

        hso.sort(reverse=True)

        if not hso:
            return
        elif len(hso) == 1:
            u_repo = f"hso! 涩值：{'{:.2%}'.format(hso[0])}\n不行我要发给别人看"
            s_repo = (
                f"涩图来咧！\n{MessageSegment.image(args[0])}\n涩值: {'{:.2%}'.format(hso[0])}"
            )

        else:
            u_repo = f"hso! 最涩的达到：{'{:.2%}'.format(hso[0])}\n不行我一定要发给别人看"

            ss = list()
            for s in args:
                ss.append(MessageSegment.image(s))
            ss = "\n".join(map(str, ss))
            s_repo = f"多张涩图来咧！\n{ss}\n最涩的达到：{'{:.2%}'.format(hso[0])}"

        await bot.send(event, u_repo)
        for superuser in conf.BotConfig.superusers:
            await bot.send_private_msg(user_id=superuser, message=s_repo)


nsfw_checker = plugin.cmd_as_group("nsfw", "涩值检测")


@nsfw_checker.got("nsfw_img", "图呢？")
async def _(bot: Bot, event: MessageEvent):
    args = extract_image_urls(event.message)
    if not args:
        await nsfw_checker.reject("请发送图片而不是其他东西！！")

    hso = await Setu(args[0]).detecter(_catcher_max_file_size, _catcher_disab_gif)
    if not hso:
        await nsfw_checker.finish("图不行，不测！")

    resu = f"涩值：{'{:.2%}'.format(hso)}\n"
    if hso >= 0.75:
        resu += "hso! 不行我要发给别人看"
        repo = f"涩图来咧！\n{MessageSegment.image(args[0])}\n涩值：{'{:.2%}'.format(hso)}"
        for superuser in conf.BotConfig.superusers:
            await bot.send_private_msg(user_id=superuser, message=repo)

    elif 0.75 > hso >= 0.5:
        resu += "嗯。可冲"
    else:
        resu += "还行8"

    await nsfw_checker.finish(resu)


catcher_setting = plugin.cmd_as_group("nsfw.size", "涩图检测图片文件大小设置", permission=MASTER)


@catcher_setting.handle()
async def _(matcher: Matcher, args: Message = CommandArg()):
    msg = args.extract_plain_text()
    if msg:
        matcher.set_arg("catcher_set", args)


@catcher_setting.got("catcher_set", "数值呢? (1对应1kb, 默认128)")
async def _(msg: str = ArgPlainText("catcher_set")):
    global _catcher_max_file_size
    try:
        _catcher_max_file_size = int(msg)
    except Exception:
        await catcher_setting.reject("请发送阿拉伯数字～！")

    repo = f"好诶！涩图检测文件最小值已设为：{_catcher_max_file_size}kb"
    await catcher_setting.finish(repo)


animation_checker = plugin.cmd_as_group("nsfw.gif", "对动图的检测开关", permission=MASTER)


@animation_checker.handle()
async def _(event: MessageEvent):
    global _catcher_disab_gif
    if _catcher_disab_gif:
        _catcher_disab_gif = False
    else:
        _catcher_disab_gif = True
    await animation_checker.finish(
        f"已{'禁用' if _catcher_disab_gif else '启用'}对 gif 的涩值检测"
    )


_ag_l = ["涩图来", "来点涩图", "来份涩图"]
_ag_patt = r"来[张点丶份](.*?)的[涩色🐍]图"

_nice_patt = r"[hH好][sS涩色][oO哦]|[嗯恩摁社蛇🐍射]了|(硬|石更)了|[牛🐂][牛🐂]要炸了|[炼恋]起来|开?导"
_nope_patt = r"不够[涩色]|就这|不行|不彳亍|一般|这也[是叫算]|[?？]|就这|爬|爪巴"
_again_patt = r"再来一张|不够"

_nice_repo = ["w", "好诶！", "ohh", "(///w///)", "🥵", "我也"]
_nope_repo = ["那你来发", "爱看不看", "你看不看吧", "看这种类型的涩图，是一件多么美妙的事情"]
_again_repo = ["没了...", "自己找去"]


def will_think(msg: str) -> str:
    if msg in _ag_l:
        return str()

    ag_jud = re.findall(_ag_patt, msg)
    if ag_jud:
        return str()

    nice_jud = re.findall(_nice_patt, msg)
    nope_jud = re.findall(_nope_patt, msg)
    again_jud = re.findall(_again_patt, msg)

    if nice_jud:
        return choice(_nice_repo)
    elif nope_jud:
        return choice(_nope_repo)
    elif again_jud:
        return choice(_again_repo)
    else:
        return str()