Update slack_bot.py

Author: haoruizhou

installer/slackbot/slack_bot.py

@@ -317,69 +317,69 @@ def process_events(client: SocketModeClient, req: SocketModeRequest):
         if event.get("type") != "message" or event.get("subtype") is not None:
             return
 
-    # Get channel type - check if it's a DM or the default channel
-    channel = event.get("channel")
-    channel_type = event.get("channel_type")
-    
-    # Allow messages from default channel or DMs (im = direct message)
-    is_dm = channel_type == "im"
-    is_default_channel = channel == DEFAULT_CHANNEL
-    
-    if not (is_dm or is_default_channel):
-        return
-
-    msg_ts = event.get("ts")
-    if msg_ts in processed_messages:
-        print(f"Skipping already processed message: {msg_ts}")
-        return
+        # Get channel type - check if it's a DM or the default channel
+        channel = event.get("channel")
+        channel_type = event.get("channel_type")
+        
+        # Allow messages from default channel or DMs (im = direct message)
+        is_dm = channel_type == "im"
+        is_default_channel = channel == DEFAULT_CHANNEL
+        
+        if not (is_dm or is_default_channel):
+            return
 
-    processed_messages.add(msg_ts)
-    if len(processed_messages) > 1000:
-        oldest_ts = sorted(processed_messages)[0]
-        processed_messages.remove(oldest_ts)
+        msg_ts = event.get("ts")
+        if msg_ts in processed_messages:
+            print(f"Skipping already processed message: {msg_ts}")
+            return
 
-    user = event.get("user")
-    bot_user_id = os.environ.get("SLACK_BOT_USER_ID", "U08P8KS8K25")
-    if user == bot_user_id:
-        print(f"Skipping message from bot itself ({bot_user_id}).")
-        return
+        processed_messages.add(msg_ts)
+        if len(processed_messages) > 1000:
+            oldest_ts = sorted(processed_messages)[0]
+            processed_messages.remove(oldest_ts)
 
-    text = event.get("text", "").strip()
-    if not text.startswith("!"):
-        return
+        user = event.get("user")
+        bot_user_id = os.environ.get("SLACK_BOT_USER_ID", "U08P8KS8K25")
+        if user == bot_user_id:
+            print(f"Skipping message from bot itself ({bot_user_id}).")
+            return
 
-    command_full = text[1:]
-    command_parts = command_full.split()
-    main_command = command_parts[0] if command_parts else ""
+        text = event.get("text", "").strip()
+        if not text.startswith("!"):
+            return
 
-    print(
-        f"Received command: '{command_full}' from user {user} "
-        f"in {'DM' if is_dm else f'channel {channel}'}"
-    )
+        command_full = text[1:]
+        command_parts = command_full.split()
+        main_command = command_parts[0] if command_parts else ""
 
-    # Get thread_ts - use the message timestamp to create/reply to thread
-    # For DMs, thread_ts is the same as msg_ts
-    thread_ts = event.get("thread_ts") or msg_ts
-    
-    # For DM responses, use the DM channel; otherwise use DEFAULT_CHANNEL
-    response_channel = channel if is_dm else DEFAULT_CHANNEL
-
-    if main_command == "location":
-        handle_location(user, thread_ts, response_channel)
-    elif main_command == "testimage":
-        handle_testimage(user, thread_ts, response_channel)
-    elif main_command == "agent":
-        handle_agent(user, command_full, thread_ts, timeout=120, channel=response_channel)
-    elif main_command == "agent-debug":
-        handle_agent(user, command_full, thread_ts, timeout=1200, channel=response_channel)
-    elif main_command == "help":
-        handle_help(user, thread_ts, response_channel)
-    else:
-        send_slack_message(
-            response_channel,
-            text=f"❓ <@{user}> Unknown command: `{text}`. Try `!help`.",
-            thread_ts=thread_ts,
+        print(
+            f"Received command: '{command_full}' from user {user} "
+            f"in {'DM' if is_dm else f'channel {channel}'}"
         )
+
+        # Get thread_ts - use the message timestamp to create/reply to thread
+        # For DMs, thread_ts is the same as msg_ts
+        thread_ts = event.get("thread_ts") or msg_ts
+        
+        # For DM responses, use the DM channel; otherwise use DEFAULT_CHANNEL
+        response_channel = channel if is_dm else DEFAULT_CHANNEL
+
+        if main_command == "location":
+            handle_location(user, thread_ts, response_channel)
+        elif main_command == "testimage":
+            handle_testimage(user, thread_ts, response_channel)
+        elif main_command == "agent":
+            handle_agent(user, command_full, thread_ts, timeout=120, channel=response_channel)
+        elif main_command == "agent-debug":
+            handle_agent(user, command_full, thread_ts, timeout=1200, channel=response_channel)
+        elif main_command == "help":
+            handle_help(user, thread_ts, response_channel)
+        else:
+            send_slack_message(
+                response_channel,
+                text=f"❓ <@{user}> Unknown command: `{text}`. Try `!help`.",
+                thread_ts=thread_ts,
+            )
     except Exception as e:
         import traceback
         print(f"❌ Error processing event: {e}")