integrations.discord.events.handler

  1"""
  2integrations/discord/events/handler.py
  3"""
  4
  5import logging
  6import os
  7import sys
  8from typing import TYPE_CHECKING
  9
 10import integrations.discord.events.audioop as _audioop
 11import libs.dispatcher
 12from libs.types import MessageStatus
 13
 14if TYPE_CHECKING:
 15    from integrations.discord.adapter import ServiceAdapter
 16
 17
 18def main(adapter: "ServiceAdapter") -> None:
 19    """
 20    メイン処理
 21
 22    Args:
 23        adapter (ServiceAdapter): アダプタインターフェース
 24
 25    Raises:
 26        ModuleNotFoundError: ライブラリ未インストール
 27
 28    """
 29    try:
 30        sys.modules["audioop"] = _audioop
 31        import discord
 32        from discord.ext import commands
 33    except ModuleNotFoundError as err:
 34        raise ModuleNotFoundError(err.msg) from None
 35
 36    # ログレベル変更
 37    for name in logging.Logger.manager.loggerDict:
 38        if name.startswith(("discord_", "discord")):
 39            logging.getLogger(name).setLevel(logging.WARNING)
 40
 41    intents = discord.Intents.default()
 42    intents.message_content = True
 43    intents.messages = True
 44    bot = commands.Bot(intents=intents)
 45    adapter.api.bot = bot
 46
 47    @bot.event
 48    async def on_ready() -> None:
 49        logging.info("login: %s", bot.user)
 50        adapter.conf.bot_name = bot.user
 51
 52    @bot.event
 53    async def on_message(message: discord.Message) -> None:
 54        if message.author.bot:
 55            return
 56
 57        adapter.api.response = message
 58
 59        m = adapter.parser()
 60        m.data.status = MessageStatus.APPEND
 61        m.parser(message)
 62
 63        libs.dispatcher.by_keyword(m)
 64
 65    @bot.event
 66    async def on_raw_message_edit(payload: discord.RawMessageUpdateEvent) -> None:
 67        channel = bot.get_channel(payload.channel_id)
 68        if channel is None or not isinstance(channel, discord.TextChannel):
 69            return
 70
 71        try:
 72            message = await channel.fetch_message(payload.message_id)
 73        except discord.NotFound:
 74            return  # メッセージが既に削除されていた場合
 75        except discord.Forbidden:
 76            return  # 権限不足
 77        except discord.HTTPException:
 78            return  # Discord API 側の一時的エラーなど
 79
 80        assert isinstance(message, discord.Message)
 81
 82        if message.author.bot:
 83            return
 84
 85        adapter.api.response = message
 86
 87        m = adapter.parser()
 88        m.data.status = MessageStatus.CHANGED
 89        m.parser(message)
 90
 91        libs.dispatcher.by_keyword(m)
 92
 93    @bot.event
 94    async def on_message_delete(message: discord.Message) -> None:
 95        if message.author.bot:
 96            return
 97
 98        adapter.api.response = message
 99
100        m = adapter.parser()
101        m.data.status = MessageStatus.DELETED
102        m.parser(message)
103
104        libs.dispatcher.by_keyword(m)
105
106    @bot.slash_command(name=adapter.conf.slash_command)  # type: ignore[no-untyped-call, untyped-decorator]
107    async def slash_command(ctx: discord.ApplicationContext, command: str) -> None:
108        adapter.api.response = ctx
109
110        m = adapter.parser()
111        m.status.command_flg = True
112        m.data.text = command
113        m.data.status = MessageStatus.APPEND
114        m.data.thread_ts = "0"
115
116        libs.dispatcher.by_keyword(m)
117
118    bot.run(token=os.environ["DISCORD_TOKEN"])