#!/usr/bin/env python3

import argparse
import base64
import glob
import hashlib
import hmac
import io
import json
import os
import markdown
import random
import zipfile

import puzzles

def write_kv_pairs(ziphandle, filename, kv):
    """ Write out a sorted map to file
    :param ziphandle: a zipfile object
    :param filename: The filename to write within the zipfile object
    :param kv:  the map to write out
    :return:
    """
    filehandle = io.StringIO()
    for key in sorted(kv.keys()):
        if type(kv[key])  == type([]):
            for val in kv[key]:
                filehandle.write("%s: %s%s" % (key, val, os.linesep))
        else:
            filehandle.write("%s: %s%s" % (key, kv[key], os.linesep))
    filehandle.seek(0)
    ziphandle.writestr(filename, filehandle.read())
   
if __name__ == '__main__':        
    parser = argparse.ArgumentParser(description='Build a puzzle category')
    parser.add_argument('seed', help='contest seed')
    parser.add_argument('puzzledir', nargs='+', help='Directory of puzzle source')
    parser.add_argument('outdir', help='Output directory')
    args = parser.parse_args()


    for puzzledir in args.puzzledir:
        puzzles = {}
        secrets = {}


        categoryname = os.path.basename(puzzledir.strip(os.sep))
        
        # build puzzle seed
        puzzle_seed = hashlib.new('sha1')
        puzzle_seed.update(categoryname.encode('utf-8'))
        puzzle_seed.update(args.seed.encode('utf-8'))
        puzzle_seed = puzzle_seed.hexdigest()

        # create zipfile
        zipfilename = os.path.join(args.outdir, "%s.zip" % categoryname)
        if os.path.exists(zipfilename):
    # assume blank state for now
            pass
    #        # open and gather some state
    #        zf = zipfile.ZipFile(zipfilename, 'a')
        else:
            # create and read in state
            zf = zipfile.ZipFile(zipfilename, 'w')

        # read in puzzle details (will be pflarr in future)
        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 = puzzles.Puzzle(open(puzzlePath))
            puzzles[points] = puzzle

        # build mapping, answers, and summary
        mapping = {}
        answers = {}
        summary = {}
        for points in sorted(puzzles):
            puzzle = puzzles[points]
            hashmap = hashlib.sha1(args.seed.encode('utf-8'))
            hashmap.update(categoryname.encode('utf-8'))
            hashmap.update(str(points).encode('utf-8'))
            mapping[points] = hashmap.hexdigest()
            answers[points] = puzzle.answers
            summary[points] = puzzle.fields['summary']

        # write mapping, answers, and summary
        write_kv_pairs(zf, os.path.join(categoryname, 'map.txt'), mapping)
        write_kv_pairs(zf, os.path.join(categoryname, 'answers.txt'), answers)
        write_kv_pairs(zf, os.path.join(categoryname, 'summary.txt'), summary)

        # write out puzzles
        for points in sorted(puzzles):
            puzzle = puzzles[points]
            puzzledir = os.path.join(categoryname, 'content', mapping[points])
            # build/write json
            ziphandle.writestr(filename, filehandle.read())
            zf.writestr(os.path.join(puzzledir, 'puzzle.json'), \
                                    json.dumps(puzzle.publish()))
            # write associated files

#vim:py