main.py

from hwc.hplio import loadProgram, writeAssembly, writeWhitespace
from hwc.compiler import compileProgram
from hwc.whitespace import translateAssembly
from sys import argv

def printUsage():

    message = '\n'
    message += "             HaPyLi -> Whitespace Compiler             \n"
    message += "                   By Kevin Gundlach                   \n"               
    message += "Usage: python main.py [-asm] <input file> <output file>\n"
    print(message)
    
    
if __name__ == "__main__":
    
    if len(argv) == 3:
        inputFile = argv[1]
        outputFile = argv[2]
        asmFlag = False
    elif len(argv) == 4:
        option = argv[1]
        inputFile = argv[2]
        outputFile = argv[3]
        asmFlag = (option == "-asm")
    else:
        printUsage()
        exit()
        
    program = loadProgram(inputFile)
    assembly = compileProgram(program)
    
    if asmFlag:
        writeAssembly(outputFile, assembly)
    else:
        whitespace = translateAssembly(assembly)
        writeWhitespace(outputFile, whitespace)
        
    print("Success!")