Slack Bolt Socket Mode应用开发中的自动重载实现

import os
from slack_bolt import App
from slack_bolt.adapter.socket_mode import SocketModeHandler

# 初始化Slack Bolt应用
app = App(token="<bot token>")

@app.command("/hello-socket-mode")
def handle_some_command(ack, body, logger):
    ack()
    print('testing slash command')
    logger.info(body)

if __name__ == "__main__":
    # 创建一个App-Level Token，并具有connections:write权限
    handler = SocketModeHandler(app, "<app token>")
    handler.start() # 这会阻塞主线程，启动Socket Mode连接

from uvicorn import run

if __name__ == "__main__":
    run("main:app", host="0.0.0.0", port=3000, reload=True, log_level="info")

import os
from fastapi import FastAPI
from slack_bolt.adapter.socket_mode import SocketModeHandler
from slack_bolt.app import App

# 1. 设置环境变量
# 强烈建议通过环境变量管理敏感信息
BOT_TOKEN = os.environ.get("SLACK_BOT_TOKEN")
APP_TOKEN = os.environ.get("SLACK_APP_TOKEN")
SIGNING_SECRET = os.environ.get("SLACK_SIGNING_SECRET") # Socket Mode下可能不需要，但保留以防万一

if not all([BOT_TOKEN, APP_TOKEN]):
    raise ValueError("请设置 SLACK_BOT_TOKEN 和 SLACK_APP_TOKEN 环境变量")

# 2. 初始化Slack Bolt应用
# 在Socket Mode下，signing_secret通常不是必需的，但为了兼容性可以保留
app = App(token=BOT_TOKEN, signing_secret=SIGNING_SECRET)

# 3. 初始化FastAPI应用
# FastAPI将作为Uvicorn的主ASGI应用
api = FastAPI()

# 4. 建立Slack Bolt Socket Mode连接
# 这一步至关重要，它会启动一个后台线程或协程来处理Socket Mode连接
# 注意：connect()方法是非阻塞的，允许主程序继续执行
SocketModeHandler(app, APP_TOKEN).connect()

# 5. 定义Slack Bolt事件监听器
# 这些监听器将通过Socket Mode接收事件
@app.message("hello")
def message_hello(message, say):
    """
    监听包含“hello”的消息，并回复用户
    """
    say(f"Hey there <@{message['user']}>!")
    print(f"Received 'hello' from user {message['user']}")

@app.command("/hello-socket-mode")
def handle_some_command(ack, body, logger):
    """
    处理 /hello-socket-mode 斜杠命令
    """
    ack() # 立即确认命令
    print('Slash command /hello-socket-mode received')
    logger.info(body)
    # 可以在此处添加更多业务逻辑，例如回复用户
    # app.client.chat_postMessage(channel=body['channel_id'], text="Hello from Socket Mode!")


# 6. 定义FastAPI端点（可选，但推荐用于健康检查）
# 即使您的Slack Bolt应用不直接处理HTTP请求，拥有一个FastAPI端点也很有用
# 例如，用于健康检查或提供调试信息
@api.get("/")
async def root():
    """
    提供一个简单的根路径健康检查
    """
    return {"status": "OK", "message": "FastAPI is running, Slack Bolt Socket Mode connected."}

# 注意：这里没有 if __name__ == "__main__": 块，因为Uvicorn会直接加载并运行 `api` 对象

SLACK_BOT_TOKEN="xoxb-YOUR-BOT-TOKEN" \
SLACK_APP_TOKEN="xapp-YOUR-APP-TOKEN" \
uvicorn main:api --reload --host 0.0.0.0 --port 4000 --log-level info