import argparse
import yaml
import os
from urllib.parse import urlencode, quote
import webbrowser

parser = argparse.ArgumentParser(description='Generate projects for Things 3 from templates')
parser.add_argument('--options', help='List available template options', default=False, action='store_true')
parser.add_argument('--template', help='The name of the template to create')
parser.add_argument('--title', help='Override the title field in the template')
parser.add_argument('--notes', help='Override the notes field in the template')
parser.add_argument('--deadline', help='Override the deadline field in the template')
parser.add_argument('--when', help='Override the when field in the template')


def formatFiles(tup):
    fname, description = tup
    fname = fname.split('.')[0]
    return '- {}: {}'.format(fname, description) if description is not None else '- {}'.format(fname)

TEMPLATE_PATH = os.path.join(os.path.dirname(__file__), 'templates')
# Get the names of all the template files
def getTemplateOptions():
    files = os.listdir(TEMPLATE_PATH)
    descriptions = []
    for f in files:
        with open(os.path.join(TEMPLATE_PATH, f), 'r') as infile:
            s = yaml.load(infile, Loader=yaml.FullLoader)
            descriptions.append(s['description'] if 'description' in s else None)
    return list(map(formatFiles, zip(files, descriptions)))

def createThings3Project(template, args):
    projectTemplate = {
        'title': args.title if args.title is not None else template['title'],
        'notes': args.notes if args.notes is not None else template['notes'],
        'area': template['area'],
        'when': args.when if args.when is not None else template['when'],
        'tags': ','.join(template['tags']),
        'deadline': args.deadline if args.deadline is not None else template['deadline'],
        'to-dos': '\n'.join(template['todos']),
        'completed': template['completed'],
        'canceled': template['canceled'],
        'reveal': template['reveal'],
        'creation-date': template['creation_date'],
        'completion-date': template['completion_date'],
    }
    params = urlencode({ k: v for k, v in projectTemplate.items() if v is not None }, quote_via=quote)
    success = webbrowser.open_new_tab('things:///add-project?{}'.format(params))
    if not success:
        print('Failed to create new project')

def createThings3Template(template, args):
    supportedTypes = {
        'project': createThings3Project,
    }
    if template['type'] in supportedTypes:
        return supportedTypes[template['type']](template, args)
    else:
        raise Exception('Unknown template type: {}'.format(template.type))


ARGS = parser.parse_args()

if ARGS.template:
    try:
        with open(os.path.join(TEMPLATE_PATH, ARGS.template + '.yml'), 'r') as infile:
            settings = yaml.load(infile, Loader=yaml.FullLoader)
    except Exception as e:
        print('Could not load template for "{}"'.format(ARGS.title))
        print(e)

    createThings3Template(settings, ARGS)
elif ARGS.options:
    print('\n'.join(getTemplateOptions()))
else:
    parser.print_help()