From 4d1a37da3ad391ca3d0b348900a3ddcbcc033f94 Mon Sep 17 00:00:00 2001
From: Kazuhiro Sera <ksera@slack-corp.com>
Date: Wed, 23 Mar 2022 14:40:09 +0900
Subject: [PATCH] Fix #622 by adding Socket Mode healthcheck endpoint examples

---
 examples/socket_mode_async_healthcheck.py | 56 +++++++++++++++++++++++
 examples/socket_mode_healthcheck.py       | 50 ++++++++++++++++++++
 2 files changed, 106 insertions(+)
 create mode 100644 examples/socket_mode_async_healthcheck.py
 create mode 100644 examples/socket_mode_healthcheck.py

diff --git a/examples/socket_mode_async_healthcheck.py b/examples/socket_mode_async_healthcheck.py
new file mode 100644
index 000000000..26de6b786
--- /dev/null
+++ b/examples/socket_mode_async_healthcheck.py
@@ -0,0 +1,56 @@
+import logging
+import os
+from typing import Optional
+
+from slack_sdk.socket_mode.aiohttp import SocketModeClient
+
+from slack_bolt.app.async_app import AsyncApp
+from slack_bolt.adapter.socket_mode.async_handler import AsyncSocketModeHandler
+
+logging.basicConfig(level=logging.DEBUG)
+
+#
+# Socket Mode Bolt app
+#
+
+# Install the Slack app and get xoxb- token in advance
+app = AsyncApp(token=os.environ["SLACK_BOT_TOKEN"])
+socket_mode_client: Optional[SocketModeClient] = None
+
+
+@app.event("app_mention")
+async def event_test(event, say):
+    await say(f"Hi there, <@{event['user']}>!")
+
+
+#
+# Web app for hosting the healthcheck endpoint for k8s etc.
+#
+
+from aiohttp import web
+
+
+async def healthcheck(_req: web.Request):
+    if socket_mode_client is not None and socket_mode_client.is_connected():
+        return web.Response(status=200, text="OK")
+    return web.Response(status=503, text="The Socket Mode client is inactive")
+
+
+web_app = app.web_app()
+web_app.add_routes([web.get("/health", healthcheck)])
+
+
+#
+# Start the app
+#
+
+if __name__ == "__main__":
+
+    async def start_socket_mode(_web_app: web.Application):
+        handler = AsyncSocketModeHandler(app, os.environ["SLACK_APP_TOKEN"])
+        await handler.connect_async()
+        global socket_mode_client
+        socket_mode_client = handler.client
+
+    app.web_app().on_startup.append(start_socket_mode)
+    app.start(8080)
diff --git a/examples/socket_mode_healthcheck.py b/examples/socket_mode_healthcheck.py
new file mode 100644
index 000000000..b76e14b87
--- /dev/null
+++ b/examples/socket_mode_healthcheck.py
@@ -0,0 +1,50 @@
+import logging
+import os
+from slack_bolt import App
+from slack_bolt.adapter.socket_mode import SocketModeHandler
+
+logging.basicConfig(level=logging.DEBUG)
+
+#
+# Socket Mode Bolt app
+#
+
+# Install the Slack app and get xoxb- token in advance
+app = App(token=os.environ["SLACK_BOT_TOKEN"])
+socket_mode_handler = SocketModeHandler(app, os.environ["SLACK_APP_TOKEN"])
+
+
+@app.event("app_mention")
+def event_test(event, say):
+    say(f"Hi there, <@{event['user']}>!")
+
+
+#
+# Web app for hosting the healthcheck endpoint for k8s etc.
+#
+
+# pip install Flask
+from flask import Flask, make_response
+
+flask_app = Flask(__name__)
+
+
+@flask_app.route("/health", methods=["GET"])
+def slack_events():
+    if (
+        socket_mode_handler.client is not None
+        and socket_mode_handler.client.is_connected()
+    ):
+        return make_response("OK", 200)
+    return make_response("The Socket Mode client is inactive", 503)
+
+
+#
+# Start the app
+#
+# export SLACK_APP_TOKEN=xapp-***
+# export SLACK_BOT_TOKEN=xoxb-***
+
+if __name__ == "__main__":
+    socket_mode_handler.connect()  # does not block the current thread
+    flask_app.run(port=8080)