summaryrefslogtreecommitdiff
path: root/ATRI/plugins/rich/__init__.py
blob: 7f9369e0c4e337bef8e3700f8c4ae14255bccf95 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
#!/usr/bin/env python3
# -*- coding:utf-8 -*-
'''
File: __init__.py
Created Date: 2021-02-07 09:34:23
Author: Kyomotoi
Email: Kyomotoiowo@gmail.com
License: GPLv3
Project: https://github.com/Kyomotoi/ATRI
--------
Last Modified: Sunday, 7th March 2021 3:14:28 pm
Modified By: Kyomotoi (kyomotoiowo@gmail.com)
--------
Copyright (c) 2021 Kyomotoi
'''

import re
import json
from aiohttp.client import ClientSession

from nonebot.adapters.cqhttp import Bot, MessageEvent
from nonebot.plugin import on_message

from ATRI.utils.request import get_bytes
from ATRI.utils.list import count_list, del_list_aim
from ATRI.rule import (
    is_in_banlist,
    is_in_dormant,
)

from .data_source import dec


waiting_list = []


bilibili_rich = on_message(
    rule=is_in_banlist() & is_in_dormant()
)

@bilibili_rich.handle()
async def _bilibili_rich(bot: Bot, event: MessageEvent) -> None:
    global waiting_list
    msg = str(event.raw_message).replace("\\", "")
    user = event.user_id
    bv = False
    
    if count_list(waiting_list, user) == 5:
        waiting_list = del_list_aim(waiting_list, user)
        return

    waiting_list.append(user)
    
    if "qqdocurl" not in msg:
        if "av" in msg:
            av = re.findall(r"(av\d+)", msg)[0].replace('av', '')
        else:
            try:
                bv = re.findall(r"(BV\w+)", msg)
                av = str(dec(bv[0]))
            except:
                return
    else:
        patt = r"(?:(?:https?|ftp):\/\/)?[\w/\-?=%.]+\.[\w/\-&?=%.]+"
        bv_url = re.findall(patt, msg)
        bv_url = bv_url[3]
        async with ClientSession() as session:
            async with session.get(
                url=bv_url) as r:
                bv = re.findall(r"(BV\w+)", str(r.url))
                av = dec(bv[0])
    
    if not bv:
        if "av" in msg:
            av = re.findall(r"(av\d+)", msg)[0].replace('av', '')
        else:
            return
    
    try:
        URL = f"https://api.kyomotoi.moe/api/bilibili/v2/?aid={av}"
    except:
        return
    data = json.loads(await get_bytes(URL))['data']
    repo = (
        f"{av} INFO:\n"
        f"Title: {data['title']}\n"
        f"Bid: {data['bvid']}\n"
        f"View: {data['stat']['view']} Like: {data['stat']['like']}\n"
        f"Coin: {data['stat']['coin']} Share: {data['stat']['share']}\n"
        f"Link: {data['short_link']}\n"
        "にまねげぴのTencent rich!"
    )
    await bilibili_rich.finish(repo)