)
from turbot.channel import channel_url
from boto3.dynamodb.conditions import Key
+import turbot.sheets
import re
+def find_puzzle_for_puzzle_id(turb, hunt_id, puzzle_id):
+ """Given a hunt_id and puzzle_id, return that puzzle
+
+ Returns None if no puzzle with the given hunt_id and puzzle_id
+ exists in the database, otherwise a dictionary with all fields
+ from the puzzle's row in the database.
+ """
+
+ response = turb.table.get_item(
+ Key={
+ 'hunt_id': hunt_id,
+ 'SK': 'puzzle-{}'.format(puzzle_id)
+ })
+
+ if 'Item' in response:
+ return response['Item']
+ else:
+ return None
+
def find_puzzle_for_url(turb, hunt_id, url):
"""Given a hunt_id and URL, return the puzzle with that URL
return False
return True
+
+def puzzle_id_from_name(name):
+ return re.sub(r'[^a-zA-Z0-9_]', '', name).lower()
+
+def puzzle_update_channel_and_sheet(turb, puzzle):
+
+ channel_id = puzzle['channel_id']
+ name = puzzle['name']
+ url = puzzle.get('url', None)
+ sheet_url = puzzle.get('sheet_url', None)
+ state = puzzle.get('state', None)
+ status = puzzle['status']
+
+ topic = ''
+
+ if status == 'solved':
+ topic += "SOLVED: `{}` ".format('`, `'.join(puzzle['solution']))
+
+ topic += name
+
+ links = []
+ if url:
+ links.append("<{}|Puzzle>".format(url))
+ if sheet_url:
+ links.append("<{}|Sheet>".format(sheet_url))
+
+ if len(links):
+ topic += "({})".format(', '.join(links))
+
+ if state:
+ topic += " {}".format(state)
+
+ # Slack only allows 250 characters for a topic
+ if len(topic) > 250:
+ topic = topic[:247] + "..."
+
+ turb.slack_client.conversations_setTopic(channel=channel_id,
+ topic=topic)
+
+ # Rename the sheet to include indication of solved/solution status
+ sheet_name = puzzle['name']
+ if puzzle['status'] == 'solved':
+ sheet_name += " - Solved {}".format(", ".join(puzzle['solution']))
+
+ turbot.sheets.renameSheet(turb, puzzle['sheet_url'], sheet_name)
+
+ # Finally, rename the Slack channel to reflect the latest name and
+ # the solved status
+ #
+ # Note: We don't use puzzle['hunt_id'] here because we're keeping
+ # that as a persistent identifier in the database. Instead we
+ # create a new ID-like identifier from the current name.
+ channel_name = "{}-{}".format(
+ puzzle['hunt_id'],
+ puzzle_id_from_name(puzzle['name'])
+ )
+ if puzzle['status'] == 'solved':
+ channel_name += "-solved"
+
+ turb.slack_client.conversations_rename(
+ channel=puzzle['channel_id'],
+ name=channel_name
+ )