discord対応

.env.example

@@ -1,5 +1,7 @@
 DATABASE_URL=postgres://postgres:password@postgres:5432/
+DISCORD_API_TOKEN=
 OPENAI_API_KEY=
 SLACK_API_TOKEN=
 SLACK_SIGNING_SECRET=
 YAHOO_API_TOKEN=
+MODE=

Pipfile

@@ -32,3 +32,4 @@ gitpython = "==3.1.31"
 pandas = "==1.5.3"
 matplotlib = "==3.7.0"
 openai = "==0.27.0"
+discord = "==2.2.2"

README.md

@@ -52,6 +52,12 @@
 
     `.env.example` をコピーして使うとよいでしょう
 
+    MODEに `discord` を指定すると、DiscordのBotとして動作します。
+
+    DISCORD_API_TOKENにDiscordのBot Tokenを指定します。
+
+    DISCORD_API_TOKENには `READ MESSAGES/VIEW CHANNELS` と、 `Send Messages` の権限が必要です。
+
 6. docker composeで鳩botとPostgreSQLを起動します。
 
     ```sh

README.template.md

@@ -52,6 +52,12 @@
 
     `.env.example` をコピーして使うとよいでしょう
 
+    MODEに `discord` を指定すると、DiscordのBotとして動作します。
+
+    DISCORD_API_TOKENにDiscordのBot Tokenを指定します。
+
+    DISCORD_API_TOKENには `READ MESSAGES/VIEW CHANNELS` と、 `Send Messages` の権限が必要です。
+
 6. docker composeで鳩botとPostgreSQLを起動します。
 
     ```sh

library/clientclass.py

@@ -3,9 +3,11 @@
 """
 clientに使うclass
 """
+import asyncio
 import os
 from abc import ABCMeta, abstractmethod
 
+import discord
 from slack import WebClient
 
 import slackbot_settings as conf
@@ -108,3 +110,35 @@ def get_send_user_name(self):
     def get_type():
         """api"""
         return "api"
+
+
+class DisscordClient(BaseClient):
+    """
+    Discordを操作するClient
+    """
+
+    def __init__(self, discord_client, message: discord.Message):
+        self.client = discord_client
+        self.message = message
+
+    def post(self, message):
+        """Discordにポストする"""
+        asyncio.create_task(self.message.channel.send(message))
+
+    def upload(self, file, filename):
+        """ファイルを投稿する"""
+        asyncio.create_task(
+            self.message.channel.send(file=discord.File(file, filename=filename))
+        )
+
+    def get_send_user(self):
+        """botを呼び出したユーザーを返す"""
+        return self.message.author.id
+
+    def get_send_user_name(self):
+        return self.message.author.name
+
+    @staticmethod
+    def get_type():
+        """discord"""
+        return "discord"

run.py

@@ -3,23 +3,24 @@
 """
 BotのMain関数
 """
+import asyncio
 import logging
 import logging.config
 import sys
 from concurrent.futures import ThreadPoolExecutor
 from typing import Callable, List
 
+import discord
 from flask import Flask, escape, jsonify, request
 from slackeventsapi import SlackEventAdapter
 
 import slackbot_settings as conf
-from library.clientclass import ApiClient, SlackClient
+from library.clientclass import ApiClient, DisscordClient, SlackClient
 from library.database import Database
 from plugins import analyze
 
 app = Flask(__name__)
 
-
 slack_events_adapter = SlackEventAdapter(
     signing_secret=conf.SLACK_SIGNING_SECRET, endpoint="/slack/events", server=app
 )
@@ -151,10 +152,29 @@ def status():
     return jsonify({"message": "hato-bot is running", "version": conf.VERSION}), 200
 
 
+intents = discord.Intents.all()
+discordClient = discord.Client(intents=intents)
+
+
+@discordClient.event
+async def on_message(message):
+    if message.author == discordClient.user:
+        return
+
+    if discordClient.user in message.mentions:
+        # `message.content.split(" ", 1)[1]` は、メンション先を除いた文字列
+        analyze.analyze_message(message.content.split(" ", 1)[1])(
+            DisscordClient(discordClient, message)
+        )
+
+
 def main():
     """メイン関数"""
 
-    app.run(host="0.0.0.0", port=conf.PORT)
+    if conf.MODE == "discord":
+        discordClient.run(token=conf.DISCORD_API_TOKEN)
+    else:
+        app.run(host="0.0.0.0", port=conf.PORT)
 
 
 if __name__ == "__main__":

slackbot_settings.py

@@ -37,6 +37,11 @@
 # ChatGPT用の設定
 OPENAI_API_KEY = str(os.environ["OPENAI_API_KEY"])
 
+# Discord用の設定
+DISCORD_API_TOKEN = str(os.environ["DISCORD_API_TOKEN"])
+
+MODE = str(os.environ["MODE"])
+
 GIT_COMMIT_HASH = os.environ.get("GIT_COMMIT_HASH")
 
 VERSION = "2.4.4"