from zzcore import StdAns, mysakuya
import requests

from config import LOLIKEY


class Ans(StdAns):
    AllowGroup = [959613860, 125733077]

    def GETMSG(self):
        if self.parms[-1] == 'p':
            flash = ''
            self.parms = self.parms[0:-1]
        else:
            flash = ',type=flash'

        if len(self.parms) == 1:
            code, picurl, pid = getsetu()
        else:
            keyword = self.parms[1]
            if mysakuya(self, keyword) == False:
                return "不许你们看咲夜的涩图！！"
            code, picurl, pid = getsetu(keyword)
        if code == 0:
            self.sendmsg(f'[CQ:reply,id={self.mid}]Pixiv ID:{pid}')
            return f'[CQ:image,file={picurl}{flash}]'
        else:
            return f'[CQ:reply,id={self.mid}] 什么东西出错了，code:{code}'


def getsetu(keyword=''):
    url = 'https://api.lolicon.app/setu/'
    params = {
        'apikey': LOLIKEY,
        'keyword': keyword,
    }

    try:
        resp = requests.get(url=url, params=params, timeout=5).json()
    except:
        return 500, '',0


    picurl = ''
    pid = 0
    if resp['code'] == 0:
        picurl = "https://r.zjuyk.site/" + resp['data'][0]['url']
        pid = resp['data'][0]['pid']

    return resp['code'], picurl, pid