Handle the submission of the "New Hunt" modal

[turbot] / turbot_lambda / turbot_lambda.py

diff --git a/turbot_lambda/turbot_lambda.py b/turbot_lambda/turbot_lambda.py
index b6f4a36049360a4e154ac6bf1e6ca3217df8af69..9d878af8c7e05a92b58846e96a20c9c54fd682c5 100644 (file)
--- a/turbot_lambda/turbot_lambda.py
+++ b/turbot_lambda/turbot_lambda.py
@@ -4,11 +4,11 @@ import boto3
 import requests
 import json
 import os
+from types import SimpleNamespace
 
 import turbot.actions
 import turbot.commands
 import turbot.events
-import turbot.views
 
 ssm = boto3.client('ssm')
 
@@ -23,6 +23,12 @@ response = ssm.get_parameter(Name='SLACK_BOT_TOKEN', WithDecryption=True)
 slack_bot_token = response['Parameter']['Value']
 slack_client = WebClient(slack_bot_token)
 
+db = boto3.resource('dynamodb')
+
+turb = SimpleNamespace()
+turb.slack_client = slack_client
+turb.db = db
+
 def error(message):
     """Generate an error response for a Slack request
 
@@ -65,12 +71,12 @@ def turbot_lambda(event, context):
     content_type = headers['content-type']
 
     if (content_type == "application/json"):
-        return turbot_event_handler(event, context)
+        return turbot_event_handler(turb, event, context)
     if (content_type == "application/x-www-form-urlencoded"):
-        return turbot_interactive_or_slash_command(event, context)
+        return turbot_interactive_or_slash_command(turb, event, context)
     return error("Unknown content-type: {}".format(content_type))
 
-def turbot_event_handler(event, context):
+def turbot_event_handler(turb, event, context):
     """Handler for all subscribed Slack events"""
 
     body = json.loads(event['body'])
@@ -78,12 +84,12 @@ def turbot_event_handler(event, context):
     type = body['type']
 
     if type == 'url_verification':
-        return url_verification_handler(body)
+        return url_verification_handler(turb, body)
     if type == 'event_callback':
-        return event_callback_handler(body)
+        return event_callback_handler(turb, body)
     return error("Unknown event type: {}".format(type))
 
-def url_verification_handler(body):
+def url_verification_handler(turb, body):
 
     # First, we have to properly respond to url_verification
     # challenges or else Slack won't let us configure our URL as an
@@ -95,14 +101,14 @@ def url_verification_handler(body):
         'body': challenge
     }
 
-def event_callback_handler(body):
+def event_callback_handler(turb, body):
     type = body['event']['type']
 
     if type in turbot.events.events:
-        return turbot.events.events[type](slack_client, body)
+        return turbot.events.events[type](turb, body)
     return error("Unknown event type: {}".format(type))
 
-def turbot_interactive_or_slash_command(event, context):
+def turbot_interactive_or_slash_command(turb, event, context):
     """Handler for Slack interactive things (buttons, shortcuts, etc.)
     as well as slash commands.
 
@@ -115,12 +121,12 @@ def turbot_interactive_or_slash_command(event, context):
     # The difference is that an interactive thingy has a 'payload'
     # while a slash command has a 'command'
     if 'payload' in body:
-        return turbot_interactive(json.loads(body['payload'][0]))
+        return turbot_interactive(turb, json.loads(body['payload'][0]))
     if 'command' in body:
-        return turbot_slash_command(body)
+        return turbot_slash_command(turb, body)
     return error("Unrecognized event (neither interactive nor slash command)")
 
-def turbot_interactive(payload):
+def turbot_interactive(turb, payload):
     """Handler for Slack interactive requests
 
     These are the things that come from a user interacting with a button
@@ -130,10 +136,12 @@ def turbot_interactive(payload):
     type = payload['type']
 
     if type == 'block_actions':
-        return turbot_block_action(payload)
+        return turbot_block_action(turb, payload)
+    if type == 'view_submission':
+        return turbot.actions.view_submission(turb, payload)
     return error("Unrecognized interactive type: {}".format(type))
 
-def turbot_block_action(payload):
+def turbot_block_action(turb, payload):
     """Handler for Slack interactive block actions
 
     Specifically, those that have a payload type of 'block_actions'"""
@@ -153,10 +161,10 @@ def turbot_block_action(payload):
             atype in turbot.actions.actions
             and avalue in turbot.actions.actions[atype]
     ):
-        return turbot.actions.actions[atype][avalue](payload)
+        return turbot.actions.actions[atype][avalue](turb, payload)
     return error("Unknown action of type/value: {}/{}".format(atype, avalue))
 
-def turbot_slash_command(body):
+def turbot_slash_command(turb, body):
     """Implementation for Slack slash commands.
 
     This parses the request and arguments and farms out to
@@ -167,6 +175,6 @@ def turbot_slash_command(body):
     args = body['text'][0]
 
     if command in turbot.commands.commands:
-        return turbot.commands.commands[command](slack_client, body, args)
+        return turbot.commands.commands[command](turb, body, args)
 
     return error("Command {} not implemented".format(command))