#!/usr/bin/python

import sys
import argparse
import logging

import c3
import ppci
import codegen
import codegenarm
from optimize import optimize
import outstream
import hexfile

logformat='%(asctime)s|%(levelname)s|%(name)s|%(message)s'

def logLevel(s):
    numeric_level = getattr(logging, s.upper(), None)
    if not isinstance(numeric_level, int):
        raise ValueError('Invalid log level: {}'.format(s))
    return numeric_level

# Parse arguments:
parser = argparse.ArgumentParser(description='lcfos Compiler')
# Input:
parser.add_argument('source', type=argparse.FileType('r'), \
  help='the source file to build', nargs="+")
parser.add_argument('-i', '--import', type=argparse.FileType('r'), \
  help='Possible import module', action='append')

parser.add_argument('--dumpir', action='store_true', help="Dump IR-code")
parser.add_argument('--dumpasm', action='store_true', help="Dump ASM-code")
parser.add_argument('--optimize', action='store_true', help="Optimize")
parser.add_argument('--target', help="Backend selection")
parser.add_argument('-o', '--output', help='Output file', metavar='filename')
parser.add_argument('--hexfile', help='Output hexfile', type=argparse.FileType('w'))
parser.add_argument('--log', help='Log level (INFO,DEBUG)', type=logLevel)


def zcc(srcs, imps, outs, diag, dumpir=False):
    """
        Compile sources into output stream.
        Sources is an iterable of open files.
    """
    logging.info('Zcc started')
    # Front end:
    c3b = c3.Builder(diag)
    for ircode in c3b.build(srcs, imps):
        if not ircode:
            return

        # Optimization passes:
        optimize(ircode)
        
        if dumpir:
            ircode.dump()

        # TODO select target here!
        # Code generation:
        codegenarm.ArmCodeGenerator(outs).generate(ircode)
    return c3b.ok


def main(args):
    logging.basicConfig(format=logformat, level=args.log)
    src = args.source
    imps = getattr(args, 'import')
    diag = ppci.DiagnosticsManager()
    outs = outstream.TextOutputStream()

    # Invoke compiler:
    res = zcc(src, imps, outs, diag, dumpir=args.dumpir)
    if not res:
        diag.printErrors(src)
        return 1

    if args.dumpasm:
        outs.dump()

    code_bytes = outs.sections['code'].to_bytes()
    if args.output:
        output_filename = args.output
        with open(output_filename, 'wb') as f:
            f.write(code_bytes)

    if args.hexfile:
        logging.info('Creating hexfile')
        hf = hexfile.HexFile()
        hf.addRegion(0x08000000, code_bytes)
        hf.save(args.hexfile)
    return 0


if __name__ == '__main__':
    arguments = parser.parse_args()
    sys.exit(main(arguments))