Merge branch 'develop' into add_image_generate

Author: Goryudyuma

.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=

.github/workflows/deploy-hato-bot.yml

@@ -100,7 +100,7 @@ jobs:
       - name: Start docker
         env:
           DOCKER_CONTENT_TRUST: 1
-        run: bash "${GITHUB_WORKSPACE}/scripts/deploy_hato_bot/deploy_docker_image/test.sh"
+        run: docker compose up -d --wait
 
   # .python-version をDockerイメージと同期させる
   update-version-python-version:

.github/workflows/pr-test-hato-bot.yml

@@ -25,7 +25,7 @@ jobs:
         with:
           python-version: ${{ env.PYTHON_VERSION }}
           cache: pipenv
-      - name: Install pipenv
+      - name: Test
         run: bash "${GITHUB_WORKSPACE}/scripts/pr_test_hato_bot/pr_test/test.sh"
 
 concurrency:

.python-lint

@@ -467,4 +467,4 @@ valid-metaclass-classmethod-first-arg=mcs
 
 # Exceptions that will emit a warning when being caught. Defaults to
 # "Exception"
-overgeneral-exceptions=Exception
+overgeneral-exceptions=builtins.Exception

Dockerfile

@@ -38,7 +38,7 @@ RUN apt-get update && \
     apt-get remove -y git gcc libc6-dev && \
     apt-get autoremove -y && \
     apt-get clean && \
-    rm -rf /var/lib/apt/lists ~/.cache /tmp && \
+    rm -rf /var/lib/apt/lists ~/.cache /tmp /root/.npm /usr/src/app/node_modules/re2/.github/actions/*/Dockerfile && \
     find / -type f -perm /u+s -ignore_readdir_race -exec chmod u-s {} \; && \
     find / -type f -perm /g+s -ignore_readdir_race -exec chmod g-s {} \; && \
     useradd -l -m -s /bin/bash -N -u "1000" "nonroot" && \
@@ -54,5 +54,4 @@ COPY --from=commit-hash slackbot_settings.py slackbot_settings.py
 
 ENV GIT_PYTHON_REFRESH=quiet
 ENV NODE_OPTIONS="--max-old-space-size=512"
-HEALTHCHECK --interval=5s --retries=20 CMD ["curl", "-s", "-S", "-o", "/dev/null", "http://localhost:3000/status"]
 CMD ["python", "entrypoint.py"]

Pipfile

@@ -9,9 +9,9 @@ python_version = "3.11"
 [dev-packages]
 autopep8 = "==2.0.2"
 requests-mock = "==1.10.0"
-pylint = "==2.16.3"
+pylint = "==2.16.4"
 sqlfluff = "==1.4.5"
-mypy = "==1.0.1"
+mypy = "==1.1.1"
 flake8 = "==6.0.0"
 black = "*"
 isort = "*"
@@ -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"

Pipfile.lock

@@ -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.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

@@ -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"