zhenxun_bot/zhenxun/plugins/parse_bilibili/__init__.py

import re
import time

import ujson as json
from nonebot import on_message
from nonebot.plugin import PluginMetadata
from nonebot_plugin_alconna import Hyper, Image, UniMsg
from nonebot_plugin_session import EventSession

from zhenxun.configs.path_config import TEMP_PATH
from zhenxun.configs.utils import PluginExtraData, RegisterConfig, Task
from zhenxun.models.task_info import TaskInfo
from zhenxun.services.log import logger
from zhenxun.utils.enum import PluginType
from zhenxun.utils.http_utils import AsyncHttpx
from zhenxun.utils.message import MessageUtils

from .information_container import InformationContainer
from .parse_url import parse_bili_url

__plugin_meta__ = PluginMetadata(
    name="B站内容解析",
    description="B站内容解析",
    usage="""
    usage：
        被动监听插件，解析B站视频、直播、专栏，支持小程序卡片及文本链接，5分钟内不解析相同内容
    """.strip(),
    extra=PluginExtraData(
        author="leekooyo",
        version="0.1",
        plugin_type=PluginType.DEPENDANT,
        menu_type="其他",
        configs=[
            RegisterConfig(
                module="_task",
                key="DEFAULT_BILIBILI_PARSE",
                value=True,
                default_value=True,
                help="被动 B站转发解析 进群默认开关状态",
                type=bool,
            )
        ],
        tasks=[Task(module="bilibili_parse", name="b站转发解析")],
    ).dict(),
)


async def _rule(session: EventSession) -> bool:
    return not await TaskInfo.is_block("bilibili_parse", session.id3 or session.id2)


_matcher = on_message(priority=1, block=False, rule=_rule)

_tmp = {}


@_matcher.handle()
async def _(session: EventSession, message: UniMsg):
    information_container = InformationContainer()
    # 判断文本消息内容是否相关
    match = None
    # 判断文本消息和小程序的内容是否指向一个b站链接
    get_url = None
    # 判断文本消息是否包含视频相关内容
    vd_flag = False
    # 设定时间阈值，阈值之下不会解析重复内容
    repet_second = 300
    # 尝试解析小程序消息
    data = message[0]
    if isinstance(data, Hyper) and data.raw:
        try:
            data = json.loads(data.raw)
        except (IndexError, KeyError):
            data = None
        if data:
            # 获取相关数据
            meta_data = data.get("meta", {})
            news_value = meta_data.get("news", {})
            detail_1_value = meta_data.get("detail_1", {})
            qqdocurl_value = detail_1_value.get("qqdocurl", {})
            jumpUrl_value = news_value.get("jumpUrl", {})
            get_url = (qqdocurl_value if qqdocurl_value else jumpUrl_value).split("?")[
                0
            ]
    # 解析文本消息
    elif msg := message.extract_plain_text():
        # 消息中含有视频号
        if "bv" in msg.lower() or "av" in msg.lower():
            match = re.search(r"((?=(?:bv|av))([A-Za-z0-9]+))", msg, re.IGNORECASE)
            vd_flag = True

        # 消息中含有b23的链接，包括视频、专栏、动态、直播
        elif "https://b23.tv" in msg:
            match = re.search(r"https://b23\.tv/[^?\s]+", msg, re.IGNORECASE)

        # 检查消息中是否含有直播、专栏、动态链接
        elif any(
            keyword in msg
            for keyword in [
                "https://live.bilibili.com/",
                "https://www.bilibili.com/read/",
                "https://www.bilibili.com/opus/",
                "https://t.bilibili.com/",
            ]
        ):
            pattern = r"https://(live|www\.bilibili\.com/read|www\.bilibili\.com/opus|t\.bilibili\.com)/[^?\s]+"
            match = re.search(pattern, msg)

    # 匹配成功，则获取链接
    if match:
        if vd_flag:
            number = match.group(1)
            get_url = f"https://www.bilibili.com/video/{number}"
        else:
            get_url = match.group()

    if get_url:
        # 将链接统一发送给处理函数
        data = await parse_bili_url(get_url, information_container)
        if data.vd_info:
            # 判断一定时间内是否解析重复内容，或者是第一次解析
            if (
                data.vd_url in _tmp.keys()
                and time.time() - _tmp[data.vd_url] > repet_second
            ) or data.vd_url not in _tmp.keys():
                pic = data.vd_info.get("pic", "")  # 封面
                aid = data.vd_info.get("aid", "")  # av号
                title = data.vd_info.get("title", "")  # 标题
                author = data.vd_info.get("owner", {}).get("name", "")  # UP主
                reply = data.vd_info.get("stat", {}).get("reply", "")  # 回复
                favorite = data.vd_info.get("stat", {}).get("favorite", "")  # 收藏
                coin = data.vd_info.get("stat", {}).get("coin", "")  # 投币
                like = data.vd_info.get("stat", {}).get("like", "")  # 点赞
                danmuku = data.vd_info.get("stat", {}).get("danmaku", "")  # 弹幕
                ctime = data.vd_info["ctime"]
                date = time.strftime("%Y-%m-%d", time.localtime(ctime))
                logger.info(
                    f"解析bilibili转发 {data.vd_url}", "b站解析", session=session
                )
                _tmp[data.vd_url] = time.time()
                _path = TEMP_PATH / f"{aid}.jpg"
                await AsyncHttpx.download_file(pic, _path)
                await MessageUtils.build_message(
                    [
                        _path,
                        f"av{aid}\n标题：{title}\nUP：{author}\n上传日期：{date}\n回复：{reply}，收藏：{favorite}，投币：{coin}\n点赞：{like}，弹幕：{danmuku}\n{data.vd_url}",
                    ]
                ).send()

        elif data.live_info:
            if (
                data.live_url in _tmp.keys()
                and time.time() - _tmp[data.live_url] > repet_second
            ) or data.live_url not in _tmp.keys():
                uid = data.live_info.get("uid", "")  # 主播uid
                title = data.live_info.get("title", "")  # 直播间标题
                description = data.live_info.get(
                    "description", ""
                )  # 简介，可能会出现标签
                user_cover = data.live_info.get("user_cover", "")  # 封面
                keyframe = data.live_info.get("keyframe", "")  # 关键帧画面
                live_time = data.live_info.get("live_time", "")  # 开播时间
                area_name = data.live_info.get("area_name", "")  # 分区
                parent_area_name = data.live_info.get("parent_area_name", "")  # 父分区
                logger.info(
                    f"解析bilibili转发 {data.live_url}", "b站解析", session=session
                )
                _tmp[data.live_url] = time.time()
                await MessageUtils.build_message(
                    [
                        Image(url=user_cover),
                        f"开播用户：https://space.bilibili.com/{uid}\n开播时间：{live_time}\n直播分区：{parent_area_name}——>{area_name}\n标题：{title}\n简介：{description}\n直播截图：\n",
                        Image(url=keyframe),
                        f"{data.live_url}",
                    ]
                ).send()
        elif data.image_info:
            if (
                data.image_url in _tmp.keys()
                and time.time() - _tmp[data.image_url] > repet_second
            ) or data.image_url not in _tmp.keys():
                logger.info(
                    f"解析bilibili转发 {data.image_url}", "b站解析", session=session
                )
                _tmp[data.image_url] = time.time()
                await data.image_info.send()