#!/usr/bin/env python3
from os import listdir, mkdir, path
from time import gmtime, strftime
from argparse import ArgumentParser
from configparser import ConfigParser


def handle_args():
    parser = ArgumentParser()
    parser.add_argument('config', help='location of strike.ini file (or a directory containing it)')
    return parser.parse_args()

def read_config(location):
    config = ConfigParser()
    if path.isdir(location):
        location = location + '/strike.ini'
    if not path.exists(location):
        raise FileNotFoundError(f'Config file not found at {location}.')
    config.read(location)
    return path.dirname(path.abspath(location)), config

def handle_file(location):
    meta = dict(
        title = '.'.join(path.basename(location).split('.')[:-1]),
        date = strftime('%a, %d %b %Y %H:%M:%S GMT', gmtime(path.getmtime(location)))
    )
    with open(location, 'r') as fin:
        contents = fin.read()
    delim = '---\n'
    if contents.startswith(delim):
        parts = contents.split(delim, maxsplit=2)
        meta.update(dict((key.strip(), value.strip())
            for key, value in (line.split('=') for line in parts[1].splitlines())
        ))
        contents = parts[2]
    return contents, meta

def apply_template(content, metadata, template="{content}"):
    return template.format(
        content = content,
        meta = metadata
    )

def main():
    args = handle_args()
    basedir, config = read_config(args.config)

    input_dir = path.join(basedir, config['Input']['directory'])
    exclusions = config['Input']['excludes'].splitlines()
    default_template = path.join(basedir, config['Templates']['default'])
    output_dir = path.join(basedir, config['Output']['directory'])

    try: mkdir(output_dir)
    except FileExistsError: pass

    with open(default_template) as t:
        template = t.read()

    for file in listdir(input_dir):
        if file in exclusions:
            continue 
        file_contents, metadata = handle_file(path.join(input_dir, file))
        output = apply_template(file_contents, metadata, template)
        with open(path.join(output_dir, file), 'w') as fout:
            fout.write(output)

if __name__ == '__main__':
    import sys
    sys.exit(main())