mirror of https://github.com/dirtbags/moth.git
85 lines
2.1 KiB
Python
85 lines
2.1 KiB
Python
|
#!/usr/bin/python3
|
||
|
|
|
||
|
|
import hmac
|
||
|
|
import base64
|
||
|
|
import argparse
|
||
|
|
import glob
|
||
|
|
import json
|
||
|
|
import os
|
||
|
|
import markdown
|
||
|
|
import random
|
||
|
|
|
||
|
|
messageChars = b'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ'
|
||
|
|
|
||
|
|
def djb2hash(buf):
|
||
|
|
h = 5381
|
||
|
|
for c in buf:
|
||
|
|
h = ((h * 33) + c) & 0xffffffff
|
||
|
|
return h
|
||
|
|
|
||
|
|
class Puzzle:
|
||
|
|
def __init__(self, stream):
|
||
|
|
self.message = bytes(random.choice(messageChars) for i in range(20))
|
||
|
|
self.fields = {}
|
||
|
|
self.answers = []
|
||
|
|
self.hashes = []
|
||
|
|
|
||
|
|
body = []
|
||
|
|
header = True
|
||
|
|
for line in stream:
|
||
|
|
if header:
|
||
|
|
line = line.strip()
|
||
|
|
if not line.strip():
|
||
|
|
header = False
|
||
|
|
continue
|
||
|
|
key, val = line.split(':', 1)
|
||
|
|
key = key.lower()
|
||
|
|
val = val.strip()
|
||
|
|
self._add_field(key, val)
|
||
|
|
else:
|
||
|
|
body.append(line)
|
||
|
|
self.body = ''.join(body)
|
||
|
|
|
||
|
|
def _add_field(self, key, val):
|
||
|
|
if key == 'answer':
|
||
|
|
h = djb2hash(val.encode('utf8'))
|
||
|
|
self.answers.append(val)
|
||
|
|
self.hashes.append(h)
|
||
|
|
else:
|
||
|
|
self.fields[key] = val
|
||
|
|
|
||
|
|
def publish(self):
|
||
|
|
obj = {
|
||
|
|
'author': self.fields['author'],
|
||
|
|
'hashes': self.hashes,
|
||
|
|
'body': markdown.markdown(self.body),
|
||
|
|
}
|
||
|
|
return obj
|
||
|
|
|
||
|
|
def secrets(self):
|
||
|
|
obj = {
|
||
|
|
'answers': self.answers,
|
||
|
|
'summary': self.fields['summary'],
|
||
|
|
}
|
||
|
|
return obj
|
||
|
|
|
||
|
|
|
||
|
|
parser = argparse.ArgumentParser(description='Build a puzzle category')
|
||
|
|
parser.add_argument('puzzledir', nargs='+', help='Directory of puzzle source')
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
for puzzledir in args.puzzledir:
|
||
|
|
puzzles = {}
|
||
|
|
secrets = {}
|
||
|
|
for puzzlePath in glob.glob(os.path.join(puzzledir, "*.moth")):
|
||
|
|
filename = os.path.basename(puzzlePath)
|
||
|
|
points, ext = os.path.splitext(filename)
|
||
|
|
points = int(points)
|
||
|
|
puzzle = Puzzle(open(puzzlePath))
|
||
|
|
puzzles[points] = puzzle
|
||
|
|
|
||
|
|
for points in sorted(puzzles):
|
||
|
|
puzzle = puzzles[points]
|
||
|
|
print(puzzle.secrets())
|
||
|
|
|