PagerMaid_Plugins_Pyro/luxiaoxunbs/main.py

""" PagerMaid module that 鲁小迅 报时 """

from datetime import datetime
from typing import Optional

from pyrogram import Client
from pyrogram.enums.parse_mode import ParseMode
from pyrogram.raw.functions.messages import GetStickerSet
from pyrogram.raw.types import InputStickerSetShortName
from pyrogram.raw.types.messages import StickerSet
from pyrogram.types import Document

from pagermaid.dependence import add_delete_message_job
from pagermaid.hook import Hook
from pagermaid.listener import listener
from pagermaid.enums import Message
from pagermaid.services import bot, scheduler
from pagermaid.utils import Sub
from pagermaid.utils.bot_utils import edit_delete
from pagermaid.utils.listener import check_manage_subs

lu_xiao_xun_bs_sub = Sub("luxiaoxunbs")
lu_xiao_xun_sticker: Optional[StickerSet] = None


@Hook.on_startup()
async def load_bs_sticker():
    global lu_xiao_xun_sticker
    try:
        lu_xiao_xun_sticker = await bot.invoke(
            GetStickerSet(
                stickerset=InputStickerSetShortName(short_name="luxiaoxunbs"), hash=0
            )
        )
    except Exception:
        lu_xiao_xun_sticker = None


async def get_bs_sticker():
    if not lu_xiao_xun_sticker:
        await load_bs_sticker()
    if not lu_xiao_xun_sticker:
        return None
    now = datetime.now()
    hour = now.hour - 1
    if now.minute > 30:
        hour += 1
    hour %= 12
    if hour == -1:
        hour = 11
    return Document._parse(
        bot, lu_xiao_xun_sticker.documents[hour % 12], "sticker.webp"
    )  # noqa


@scheduler.scheduled_job("cron", minute="0", id="lu_xiao_xun_bs.push")
async def lu_xiao_xun_bs_subscribe() -> None:
    sticker = await get_bs_sticker()
    if not sticker:
        return
    for gid in lu_xiao_xun_bs_sub.get_subs():
        try:
            msg = await bot.send_document(gid, sticker.file_id)
            add_delete_message_job(msg, delete_seconds=3600)
        except Exception:
            lu_xiao_xun_bs_sub.del_id(gid)


@listener(
    command="luxiaoxunbs", parameters="订阅/退订", description="整点报时，每小时定时发送，自动删除上一条消息"
)
async def lu_xiao_xun_bs(_: Client, message: Message):
    if not message.arguments:
        return await message.edit("请输入订阅/退订")
    if not check_manage_subs(message):
        return await edit_delete(message, "❌ 权限不足，无法操作整点报时", parse_mode=ParseMode.HTML)
    if message.arguments == "订阅":
        if lu_xiao_xun_bs_sub.check_id(message.chat.id):
            return await edit_delete(message, "❌ 你已经订阅了整点报时", parse_mode=ParseMode.HTML)
        lu_xiao_xun_bs_sub.add_id(message.chat.id)
        await message.edit("你已经成功订阅了整点报时")
    elif message.arguments == "退订":
        if not lu_xiao_xun_bs_sub.check_id(message.chat.id):
            return await edit_delete(message, "❌ 你还没有订阅整点报时", parse_mode=ParseMode.HTML)
        lu_xiao_xun_bs_sub.del_id(message.chat.id)
        await message.edit("你已经成功退订了整点报时")
    else:
        return await message.edit("请输入订阅/退订")