#!/usr/bin/env python3
# -*- encoding: utf-8 -*-
'''
@File    :   __init__.py
@Time    :   2020/11/07 14:36:53
@Author  :   Kyomotoi
@Contact :   kyomotoiowo@gmail.com
@Github  :   https://github.com/Kyomotoi
@License :   Copyright © 2018-2020 Kyomotoi, All Rights Reserved.
'''
__author__ = 'kyomotoi'

import re
import json
import sqlite3
from pathlib import Path
from random import randint

from nonebot.log import logger
from nonebot.permission import SUPERUSER
from nonebot.adapters.cqhttp import Bot, Event
from nonebot.plugin import on_message, on_command, on_regex

from utils.utils_yml import load_yaml
from utils.utils_error import errorRepo
from utils.utils_history import getMessage
from utils.utils_translate import toSimpleString
from utils.utils_rule import check_banlist, check_switch
from utils.utils_request import aio_get_bytes, request_get
from utils.utils_img import compress_image, aio_download_pics

from .body import resultRepo

CONFIG_PATH = Path('.') / 'config.yml'
config = load_yaml(CONFIG_PATH)

plugin_name_0 = "anime-pic-search"
key_SauceNAO = config['api']['SauceNaoKEY']

SaucenaoSearch = on_command('以图搜图',
                            rule=check_banlist() & check_switch(plugin_name_0))


@SaucenaoSearch.handle()  # type: ignore
async def _(bot: Bot, event: Event, state: dict) -> None:
    user = str(event.user_id)
    group = str(event.group_id)

    state["user"] = user
    state["group"] = group

    img = str(event.message).strip()

    if img:
        state["img_url"] = img


@SaucenaoSearch.got("img_url", prompt="请发送一张目标图片")  # type: ignore
async def _(bot: Bot, event: Event, state: dict) -> None:
    img = state["img_url"]
    img = re.findall(r"(http://.*?)]", img)

    if len(img):
        pass
    else:
        await SaucenaoSearch.reject("请发送一张目标图片，而非文字或其他非图片成分( -'`-; )")

    await bot.send(event, "别急！正在找图！")

    await SaucenaoSearch.finish(resultRepo(state['user'], key_SauceNAO,
                                           img[0]))


SaucenaoSearch_repo = on_message(rule=check_banlist()
                                 & check_switch(plugin_name_0))


@SaucenaoSearch_repo.handle()  # type: ignore
async def _(bot: Bot, event: Event, state: dict) -> None:
    group = str(event.group_id)
    msg = str(event.message)

    if "[CQ:reply" in msg:
        if "搜图" in msg or "识图" in msg:
            if group == "None":
                await SaucenaoSearch_repo.finish("ごめんなさい...\n该功能只对群聊开放哦~~")

            try:
                repo_info = re.findall(r"CQ:reply,id=([0-9]\S+)]", msg)
                msg_id = repo_info[0]
            except Exception:
                logger.error("Get message_id ERROR!")
                await SaucenaoSearch_repo.finish(errorRepo('定位消息内容失败'))
                return

            aim = getMessage(msg_id)[f"{msg_id}"]["message"]
            img = img = re.findall(r"(http://.*?)]", aim)

            if len(img):
                pass
            else:
                await SaucenaoSearch_repo.finish('这消息内貌似没图片呢...')

            await bot.send(event, "别急！正在找图！")

            await SaucenaoSearch.finish(
                resultRepo(state['user'], key_SauceNAO, img[0]))


plugin_name_1 = "anime-vid-search"
AnimeSearch = on_command('以图搜番',
                         rule=check_banlist() & check_switch(plugin_name_1))


@AnimeSearch.handle()  # type: ignore
async def _(bot: Bot, event: Event, state: dict) -> None:
    user = str(event.user_id)
    group = str(event.group_id)

    state["user"] = user
    state["group"] = group

    img = str(event.message).strip()

    if img:
        state["img_url"] = img


@AnimeSearch.got("img_url", prompt="请发送一张目标图片")  # type: ignore
async def _(bot: Bot, event: Event, state: dict) -> None:
    img = state["img_url"]
    img = re.findall(r"(http://.*?)]", img)

    if len(img):
        pass
    else:
        await SaucenaoSearch.reject("请发送一张目标图片，而非文字或其他非图片成分( -'`-; )")

    await bot.send(event, "别急！正在搜索！")
    req = None

    URL = f'https://trace.moe/api/search?url={img[0]}'
    try:
        req = await aio_get_bytes(URL)
    except:
        await AnimeSearch.finish(errorRepo("请求数据失败"))

    data = json.loads(req.decode())

    try:
        d = {}

        for i in range(len(data['docs'])):
            if data['docs'][i]['title_chinese'] in d:
                d[data['docs'][i]
                  ['title_chinese']][0] += data['docs'][i]['similarity']

            else:
                m = data['docs'][i]['at'] / 60
                s = data['docs'][i]['at'] % 60

                if data['docs'][i]['episode'] == '':
                    n = 1

                else:
                    n = data['docs'][i]['episode']

                d[toSimpleString(data['docs'][i]['title_chinese'])] = [
                    data['docs'][i]['similarity'], f'第{n}集',
                    f'{int(m)}分{int(s)}秒处'
                ]
    except:
        await AnimeSearch.finish(errorRepo("处理数据失败"))

    result = sorted(
        d.items(),  # type: ignore
        key=lambda x: x[1],
        reverse=True)

    t = 0
    msg0 = f'[CQ:at,qq={state["user"]}]\n根据所提供的图片按照相似度找到{len(d)}个结果:'  # type: ignore

    for i in result:
        t += 1
        lk = ('%.2f%%' % (i[1][0] * 100))
        msg = (
            f'\n——————————\n({t})\n相似度：{lk}\n动漫名：《{i[0]}》\n时间点：{i[1][1]} {i[1][2]}'
        )
        msg0 += msg

    await AnimeSearch.finish(msg0)


plugin_name_2 = "anime-setu"
key_LoliconAPI = config['api']['LoliconAPI']
setu_type = 1  # setu-type: 1(local), 2(url: https://api.lolicon.app/#/setu) default: 1(local)

setus = on_regex(
    r"来[点丶张份副个幅][涩色瑟][图圖]|[涩色瑟][图圖]来|[涩色瑟][图圖][gkd|GKD|搞快点]|[gkd|GKD|搞快点][涩色瑟][图圖]",
    rule=check_banlist() & check_switch(plugin_name_1))


@setus.handle()  # type: ignore
async def _setu(bot: Bot, event: Event, state: dict) -> None:
    group = str(event.group_id)

    res = randint(1, 5)

    if setu_type == 1:

        con = sqlite3.connect(
            Path('.') / 'ATRI' / 'data' / 'data_Sqlite' / 'setu' /
            'nearR18.db')
        cur = con.cursor()
        msg = cur.execute('SELECT * FROM nearR18 ORDER BY RANDOM() limit 1;')

        for i in msg:
            pid = i[0]
            title = i[1]
            img = i[7]

            msg0 = "setu info:\n"
            msg0 += f"Title: {title}\n"
            msg0 += f"Pid: {pid}\n"
            msg0 += f"[CQ:image,file=file:///{compress_image(await aio_download_pics(img))}]"

            if 1 <= res < 5:
                await setus.finish(msg0)

            elif res == 5:
                await bot.send(event, "我找到涩图了！但我发给主人了\nο(=•ω＜=)ρ⌒☆")

                for sup in config['bot']['superusers']:
                    await bot.send_private_msg(
                        user_id=sup,
                        message=
                        f"主人，从群{group}来的涩图！热乎着！\nTitle: {title}\nPid: {pid}\n[CQ:image,file=file:///{compress_image(await aio_download_pics(img))}]"
                    )

    else:
        params = {"apikey": key_LoliconAPI, "r18": "0", "num": "1"}

        data = {}

        try:
            data = json.loads(
                request_get('https://api.lolicon.app/setu/', params))
        except Exception:
            await setus.finish(errorRepo("请求数据失败，也可能为接口调用次数达上限"))

        msg0 = "setu info:\n"
        msg0 += f'Title: {data["data"][0]["title"]}\n'
        msg0 += f'Pid: {data["data"][0]["pid"]}\n'
        msg0 += f'[CQ:image,file=file:///{compress_image(await aio_download_pics(data["data"][0]["url"]))}]'

        if 1 <= res < 5:
            await setus.finish(msg0)

        elif res == 5:
            await bot.send(event, "我找到涩图了！但我发给主人了\nο(=•ω＜=)ρ⌒☆")

            for sup in config['bot']['superusers']:
                await bot.send_private_msg(
                    user_id=sup,
                    message=
                    f'主人，从群{group}来的涩图！热乎着！\nTitle: {data["data"][0]["title"]}\nPid: {data["data"][0]["pid"]}\n[CQ:image,file=file:///{compress_image(await aio_download_pics(data["data"][0]["url"]))}]'
                )


setuType = on_command("setu-type", permission=SUPERUSER)


@setuType.handle()  # type: ignore
async def _(bot: Bot, event: Event, state: dict) -> None:
    global setu_type
    msg = str(event.message).strip()

    if msg:
        pass
    else:
        msg0 = "-==ATRI Setu Type Control System==-\n"
        msg0 += "**Tips: For SUPERUSERS**\n"
        msg0 += "┌Usage: setu-type {type}\n"
        msg0 += "└Type:\n"
        msg0 += "  ├local\n"
        msg0 += "  └url"

        await setuType.finish(msg0)

    if msg == "local":
        setu_type = 1

    elif msg == "url":
        setu_type = 2

    else:
        await setuType.finish("请检查类型是否输入正确嗷！")

    await setuType.finish("Type conversion completed!")


# @scheduler.scheduled_job(
#     "cron",
#     minute=45,
#     bot=Bot,
#     event=Event,
#     state=dict
#     )
# async def _(bot: Bot, event: Event, state: dict) -> None:
#     group = str(event.group_id)

#     if banList(group=group):
#         if checkSwitch(plugin_name_2, group):
#             # group_list = await bot.get_group_list()
#             # group = sample(group_list, 1)
#             # group = group['group_id']

#             if setu_type == 1:

#                 con = sqlite3.connect(Path('.') / 'ATRI' / 'data' / 'data_Sqlite' / 'setu' / 'nearR18.db')
#                 cur = con.cursor()
#                 msg = cur.execute('SELECT * FROM nearR18 ORDER BY RANDOM() limit 1;')

#                 for i in msg:
#                     pid = i[0]
#                     title = i[1]
#                     img = i[7]

#                     msg0 = f"setu info:\n"
#                     msg0 += f"Title: {title}\n"
#                     msg0 += f"Pid: {pid}\n"
#                     msg0 += f"[CQ:image,file=file:///{compress_image(await aio_download_pics(img))}]"

#                     await setu.finish(msg0)

#             else:
#                 params = {
#                     "apikey": key_LoliconAPI,
#                     "r18": "0",
#                     "num": "1"
#                 }

#                 data = {}

#                 try:
#                     data = json.loads(request_get('https://api.lolicon.app/setu/', params))
#                 except Exception:
#                     await setu.finish(errorRepo("请求数据失败，也可能为接口调用次数达上限"))

#                 msg0 = f"setu info:\n"
#                 msg0 += f'Title: {data["data"][0]["title"]}\n'
#                 msg0 += f'Pid: {data["data"][0]["pid"]}\n'
#                 msg0 += f'[CQ:image,file=file:///{compress_image(await aio_download_pics(data["data"][0]["url"]))}]'

#                 await setu.finish(msg0)