bot.py

import asyncio
import contextlib

from creart import it
from pathlib import Path
from graia.saya import Saya
from graia.ariadne.app import Ariadne
from graia.scheduler import GraiaScheduler
from graiax.playwright.service import PlaywrightService
from graia.ariadne.entry import config, HttpClientConfig, WebsocketClientConfig

from core.log import logger
from core.bot_config import BotConfig
from core.announcement import base_telemetry

logger.info("BBot is starting...")

base_telemetry()

host = BotConfig.Mirai.mirai_host
app_config = config(
    BotConfig.Mirai.account,
    BotConfig.Mirai.verify_key,
    HttpClientConfig(host),
    WebsocketClientConfig(host),
)

app = Ariadne(app_config)
app.config(install_log=True)
if BotConfig.Bilibili.use_browser:
    app.launch_manager.add_service(
        PlaywrightService(
            user_data_dir=Path("data").joinpath("browser"),
            device_scale_factor=2 if BotConfig.Bilibili.mobile_style else 1.25,
            user_agent=(
                "Mozilla/5.0 (Linux; Android 10; RMX1911) AppleWebKit/537.36 "
                "(KHTML, like Gecko) Chrome/100.0.4896.127 Mobile Safari/537.36"
            )
            if BotConfig.Bilibili.mobile_style
            else "",
        )
    )
app.create(GraiaScheduler)
saya = it(Saya)


with saya.module_context():

    saya.require("function")

import function  # noqa: E402 F401

with contextlib.suppress(KeyboardInterrupt, asyncio.exceptions.CancelledError):
    app.launch_blocking()
logger.info("BBot is shutting down...")