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

import os
import json
import sqlite3
from pathlib import Path
from nonebot.typing import Bot, Event
from aiohttp import client_exceptions
from nonebot.plugin import on_command
from nonebot.permission import SUPERUSER

from ATRI.utils.utils_error import errorRepo
from ATRI.utils.utils_request import aio_get_bytes

SetuData = on_command('setu', permission=SUPERUSER)


@SetuData.handle()
async def _(bot: Bot, event: Event, state: dict) -> None:
    msg0 = "-==ATRI Setu Data System==-\n"
    msg0 += "Upload:\n"
    msg0 += " - setu [type] [pid]\n"
    msg0 += " * type: normal, nearR18 r18\n"
    msg0 += "Delete:\n"
    msg0 += " - setu-delete [pid]"
    await SetuData.finish(msg0)


UploadSetu = on_command('setu-upload', permission=SUPERUSER)


@UploadSetu.handle()
async def _(bot: Bot, event: Event, state: dict) -> None:
    msg = str(event.message).strip().split(' ')

    if not msg[0] and not msg[1]:
        msg0 = "请检查格式奥~！\n"
        msg0 += "setu-upload [type] [pid]\n"
        msg0 += "type: normal, nearR18, r18"
        await UploadSetu.finish(msg0)

    if msg[0] not in ["noraml", "nearR18", "nearr18", "r18", "R18"]:
        msg0 = "请检查类型~！\n"
        msg0 += "type: normal, nearR18, r18"
        await UploadSetu.finish(msg0)

    s_type = msg[0]
    pid = msg[1]

    URL = f'https://api.imjad.cn/pixiv/v1/?type=illust&id={pid}'
    info = {}

    try:
        info = json.loads(await aio_get_bytes(URL))
    except client_exceptions:
        await UploadSetu.finish(errorRepo("网络请求出错"))

    info = info["response"][0]
    title = info["title"]
    tags = info["tags"]
    account = info["user"]["account"]
    name = info["user"]["name"]
    u_id = info["user"]["id"]
    user_link = f'https://www.pixiv.net/users/{u_id}'
    IMG = info["iamge_urls"]["large"]
    IMG = IMG.replace("i.pximg.net", "i.pixiv.cat")

    data_setu = (f'{pid}', f'{title}', f'{tags}', f'{account}', f'{name}',
                 f'{u_id}', f'{user_link}', f'{IMG}')

    if s_type == "nearr18":
        s_type = "nearR18"
    elif s_type == "R18":
        s_type = "r18"
    else:
        pass

    os.makedirs('ATRI/data/data_Sqlite/setu', exist_ok=True)

    if os.path.exists(f'ATRI/data/data_Sqlite/setu/{s_type}.db'):
        print('数据文件存在！')
    else:
        await DeleteSetu.finish("数据库都不在添加🔨！？罢了我现创一个")
        con = sqlite3.connect(
            Path('.') / 'ATRI' / 'data' / 'data_Sqlite' / 'setu' /
            f'{s_type}.db')
        cur = con.cursor()
        cur.execute(
            f'CREATE TABLE {s_type}(pid PID, title TITLE, tags TAGS, account ACCOUNT, name NAME, u_id UID, user_link USERLINK, img IMG, UNIQUE(pid, title, tags, account, name, u_id, user_link, img))'
        )
        con.commit()
        cur.close()
        await bot.send(event, '完成')

    con = sqlite3.connect(
        Path('.') / 'ATRI' / 'data' / 'data_Sqlite' / 'setu' / f'{s_type}.db')
    cur = con.cursor()
    cur.execute(
        f'INSERT INTO {s_type}(pid, title, tags, account, name, u_id, user_link, img) VALUES(?, ?, ?, ?, ?, ?, ?, ?)',
        data_setu)
    con.commit()
    cur.close()

    await UploadSetu.finish(f"数据上传完成~！\n涩图库[{s_type}]涩图 +1")


DeleteSetu = on_command('setu-delete', permission=SUPERUSER)


@DeleteSetu.handle()
async def _(bot: Bot, event: Event, state: dict) -> None:
    msg = str(event.message).strip().split(' ')

    if not msg[0] and not msg[1]:
        msg0 = "请检查格式奥~！\n"
        msg0 += "setu-delete [type] [pid]\n"
        msg0 += "type: normal, nearR18, r18"
        await DeleteSetu.finish(msg0)

    if msg[0] not in ["noraml", "nearR18", "nearr18", "r18", "R18"]:
        msg0 = "请检查类型~！\n"
        msg0 += "type: normal, nearR18, r18"
        await UploadSetu.finish(msg0)

    s_type = msg[0]
    pid = msg[1]

    if s_type == "nearr18":
        s_type = "nearR18"
    elif s_type == "R18":
        s_type = "r18"
    else:
        pass

    if os.path.exists(f'ATRI/data/data_Sqlite/setu/{s_type}.db'):
        print('数据文件存在！')
    else:
        await DeleteSetu.finish("数据库都不在删🔨！？")

    con = sqlite3.connect(
        Path('.') / 'ATRI' / 'data' / 'data_Sqlite' / 'setu' / f'{s_type}.db')
    cur = con.cursor()
    cur.execute(f'DELETE FROM {s_type} WHERE pid = {pid}')
    con.commit()
    con.close()

    await UploadSetu.finish(f"数据删除完成~！\n涩图库[{s_type}]涩图 -1")