from argparse import ArgumentParser

from chordpro.base import Song
from chordpro.renderers.html import render as html_render, get_options as html_get_options
from chordpro.renderers.text import render as text_render


def get_args():
    """Parse the command line arguments"""
    parser = ArgumentParser()
    parser.add_argument('input', metavar='FILENAME', help='Input ChordPro file')
    parser.add_argument('-o', '--output', metavar='FILENAME', help='Output to a file')
    parser.add_argument('-f', '--format', metavar='FORMAT', choices=['html', 'text'], default='html',
                        help='Output format')
    parser.add_argument('-p', '--param', metavar='KEY=VALUE', action='append', help='Parameter to send to renderer')
    return parser.parse_args()


def main():
    """The command line entrypoint"""
    args = get_args()
    render_params = {kv.split('=')[0]: kv.split('=')[-1] for kv in args.param} if args.param else {}
    # Prepare the params
    for key, value in render_params.items():
        if value.lower() == "true":
            render_params[key] = True
        elif value.lower() == "false":
            render_params[key] = False
        elif value.isdigit():
            render_params[key] = int(value)

    song = Song(args.input)
    if args.format == 'text':
        output = text_render(song, **render_params)
    else:
        output = html_render(song, **render_params)
    if args.output:
        with open(args.output, 'w') as html_file:
            html_file.write(output)
    else:
        print(output)