Misskey対応

.env.example

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

Pipfile

@@ -33,3 +33,5 @@ pandas = "==1.5.3"
 matplotlib = "==3.7.1"
 openai = "==0.27.0"
 discord = "==2.2.2"
+"misskey.py" = "==4.1.0"
+websockets = "==10.4"

README.md

@@ -58,6 +58,12 @@
 
     DISCORD_API_TOKENには `Read Messages/View Channels` と、 `Send Messages` の権限が必要です。
 
+   MODEに `misskey` を指定すると、自分のサーバーからのメンションに限って反応するMisskeyのBotとして動作します。
+
+   MISSKEY_URLにBotのいるMisskeyサーバーのドメインを指定します。
+
+   MISSKEY_API_TOKENにMisskeyのBotのアクセストークンを指定します。
+
 6. docker composeで鳩botとPostgreSQLを起動します。
 
     ```sh

README.template.md

@@ -58,6 +58,12 @@
 
     DISCORD_API_TOKENには `Read Messages/View Channels` と、 `Send Messages` の権限が必要です。
 
+   MODEに `misskey` を指定すると、自分のサーバーからのメンションに限って反応するMisskeyのBotとして動作します。
+
+   MISSKEY_URLにBotのいるMisskeyサーバーのドメインを指定します。
+
+   MISSKEY_API_TOKENにMisskeyのBotのアクセストークンを指定します。
+
 6. docker composeで鳩botとPostgreSQLを起動します。
 
     ```sh

library/clientclass.py

@@ -142,3 +142,40 @@ def get_send_user_name(self):
     def get_type():
         """discord"""
         return "discord"
+
+
+class MisskeyClient(BaseClient):
+    """
+    Misskeyを操作するClient
+    """
+
+    def __init__(self, misskey_client, message):
+        self.client = misskey_client
+        self.message = message
+
+    def post(self, text):
+        """Discordにポストする"""
+        self._post(text=text)
+
+    def upload(self, file, filename=None):
+        """ファイルを投稿する"""
+        with open(file, "rb") as f:
+            drive_file = self.client.drive_files_create(file=f)
+            self._post(file_ids=[drive_file["id"]])
+
+    def _post(self, text=None, file_ids=None):
+        self.client.notes_create(
+            text=text, reply_id=self.message["id"], file_ids=file_ids
+        )
+
+    def get_send_user(self):
+        """botを呼び出したユーザーを返す"""
+        return self.message["user"]["username"]
+
+    def get_send_user_name(self):
+        return self.message["user"]["name"]
+
+    @staticmethod
+    def get_type():
+        """misskey"""
+        return "misskey"

run.py

@@ -3,18 +3,27 @@
 """
 BotのMain関数
 """
+import asyncio
+import json
 import logging
 import logging.config
 import sys
 from concurrent.futures import ThreadPoolExecutor
 from typing import Callable, List
 
 import discord
+import websockets
 from flask import Flask, escape, jsonify, request
+from misskey import Misskey
 from slackeventsapi import SlackEventAdapter
 
 import slackbot_settings as conf
-from library.clientclass import ApiClient, DiscordClient, SlackClient
+from library.clientclass import (
+    ApiClient,
+    DiscordClient,
+    MisskeyClient,
+    SlackClient,
+)
 from library.database import Database
 from plugins import analyze
 
@@ -172,6 +181,39 @@ def main():
 
     if conf.MODE == "discord":
         discordClient.run(token=conf.DISCORD_API_TOKEN)
+    elif conf.MODE == "misskey":
+        misskey_client = Misskey(conf.MISSKEY_DOMAIN, i=conf.MISSKEY_API_TOKEN)
+
+        async def discord_runner():
+            # pylint: disable=E1101
+            async with websockets.connect(
+                "wss://"
+                + misskey_client.address
+                + "/streaming"
+                + "?i="
+                + misskey_client.token
+            ) as ws:
+                await ws.send(
+                    json.dumps(
+                        {"type": "connect", "body": {"channel": "main", "id": "main"}}
+                    )
+                )
+                while True:
+                    data = json.loads(await ws.recv())
+                    if data["type"] == "channel" and data["body"]["type"] == "mention":
+                        note = data["body"]["body"]
+                        host = note["user"].get("host")
+                        mentions = note.get("mentions")
+                        if (
+                            (host is None or host == conf.MISSKEY_DOMAIN)
+                            and mentions
+                            and misskey_client.i()["id"] in mentions
+                        ):
+                            analyze.analyze_message(
+                                note["text"].replace("\xa0", " ").split(" ", 1)[1]
+                            )(MisskeyClient(misskey_client, note))
+
+        asyncio.get_event_loop().run_until_complete(discord_runner())
     else:
         app.run(host="0.0.0.0", port=conf.PORT)
 

slackbot_settings.py

@@ -40,6 +40,10 @@
 # Discord用の設定
 DISCORD_API_TOKEN = str(os.environ["DISCORD_API_TOKEN"])
 
+# Misskey用の設定
+MISSKEY_DOMAIN = str(os.environ["MISSKEY_DOMAIN"])
+MISSKEY_API_TOKEN = str(os.environ["MISSKEY_API_TOKEN"])
+
 MODE = str(os.environ["MODE"])
 
 GIT_COMMIT_HASH = os.environ.get("GIT_COMMIT_HASH")