Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions slackbot/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@ def __init__(self, rtm_client, web_client) -> None:
self.client = rtm_client
self.web = web_client
self.user_id = self.web.auth_test()["user_id"]
self.priority = 10

@staticmethod
def user_has_perm(user, perm):
Expand Down Expand Up @@ -70,6 +71,13 @@ def handle(self, message, user=None, channel=None, ts=None, raw=None) -> Optiona
"""
raise NotImplementedError("abstract method")

def process_interactive(self, **payload):
return self.handle_interactive(**payload)

def handle_interactive(self, **payload) -> Optional[Union[int, tuple[int, int]]]:
# This method is currently a placeholder.
pass


class NoMatchSlackCommand(Exception):
pass
Expand Down
22 changes: 22 additions & 0 deletions slackbot/management/commands/run_bot.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,22 @@ def handle_message(self, **payload):
close_old_connections()
self.handle_message_really(**payload)

def handle_interactive(self, **payload):
processed_at_least_one = False
for p in self.processors:
try:
r = p.process_interactive(**payload)
if r:
if not isinstance(r, tuple):
r = (r,)
if MessageProcessor.PROCESSED in r:
processed_at_least_one = True
if MessageProcessor.STOP in r:
break
except Exception as e:
self.log_exception("Processor failed for interactive event: %s %s", payload, str(e))
return processed_at_least_one

def handle_reaction(self, **payload):
event = payload.get("event", {})
processed_at_least_one = False
Expand Down Expand Up @@ -125,6 +141,7 @@ def set_up(self, **payload):
self.my_name = data.get("user")

self.processors = [x(self.client, self.web) for x in MessageProcessor.__subclasses__()]
self.processors.sort(key=lambda x: x.priority)
self.stdout.write(f"Connected as {self.my_name}")
self.stdout.write(
f"Processors: {','.join(f'{x.__class__.__module__}.{x.__class__.__name__}' for x in self.processors)}"
Expand All @@ -147,6 +164,10 @@ def process_reaction(self, client: SocketModeClient, req: SocketModeRequest):
event = req.payload["event"]
if event["type"] in ("reaction_added", "reaction_removed"):
return self.handle_reaction(**req.payload)

def process_interactive(self, client: SocketModeClient, req: SocketModeRequest):
if req.type == "interactive":
return self.handle_interactive(**req.payload)

def handle(self, *args, **options):
# faster cold boot
Expand All @@ -161,6 +182,7 @@ def handle(self, *args, **options):
self.set_up()
self.client.socket_mode_request_listeners.append(self.process)
self.client.socket_mode_request_listeners.append(self.process_reaction)
self.client.socket_mode_request_listeners.append(self.process_interactive)
self.stdout.write("Connecting...\n")
self.client.connect()

Expand Down
Loading