cli.py

from typer import Typer, Argument
from visitors.string_rep_visitor import FormatVisitor
from visitors.semantic_check_visitor import SemanticCheckerVisitor
from visitors.high_level_ir_generator_visitor import TzScriptToHighLevelIrVisitor
from utils import run_tzscript_ast_building_pipeline
from visitors.michelson_generator_visitor import MichelsonGenerator
from visitors.hl_string_repre import HLReprVisitor
from visitors.code_runner import CodeRunnerVisitor

import typer


app = Typer()


def process(script: str, num_steps=3):
    with typer.progressbar(length=num_steps) as progress:
        # Tokenize Script
        print("\nBuilding TZScript AST", end="")
        ast = run_tzscript_ast_building_pipeline(script)
        failed = False
        print("\nPerforming Semantic Check", end="")
        semantic_checker = SemanticCheckerVisitor()
        semantic_result = semantic_checker.visit(ast)
        if len(semantic_result) > 0:
            failed = True
            print("\nSomething Went Wrong")
            for err in semantic_result:
                print(err)
        else:
            print("... OK")
        if not failed:
            progress.update(1)

            print("\nGenerating Intermediate Representation", end="")
            high_level_ir = TzScriptToHighLevelIrVisitor()
            ir = high_level_ir.visit(ast)
            print("...OK")
            progress.update(1)

            return ast, ir, progress
        return None, None, None


@app.command()
def build(file: str = Argument("", help="tzscript file to be parsed"),
          out_file: str = Argument(None, help='michelson file to be generated')):
    """ generates the .tz michelson script from the tzscript file specified """

    with open(file, "r", encoding='utf-8') as f:
        script = f.read()

    ast, ir, progress = process(script, 3)

    print("\nGenerating Michelson Code", end="")
    # TODO uncomment this when code generation is working OK
    visit_generator = MichelsonGenerator()
    visit_generator.visit(ir)
    # michelson_result = michelson_geneator.result
    michelson_result = visit_generator.code
    if out_file is None:
        out_file = file[:file.find(".tzs")]+".tz"
    with open(out_file, "w") as f:
        f.write(michelson_result)
    print(f"\nGenerated {out_file}")
    progress.update(1)


@app.command()
def run(file: str = Argument("", help="tzscript file to be parsed")):
    """ Runs AST generated by the Parser, calling the last entry defined (this behavior is restricted to rhis option) """

    with open(file, "r", encoding='utf-8') as f:
        script = f.read()

    ast, ir, progress = process(script, 2)
    if ast is not None:
        print("\nRunning Code\n", end="")

        runner = CodeRunnerVisitor()
        runner.visit(ast)


@app.command()
def represent(file: str = Argument("", help="tzscript file to be parsed"),
              out_file: str = Argument(None, help='michelson file to be generated')):
    """ generates the .tzs.repr the string representation of the code in the input """

    # with typer.progressbar(length=total) as progress:
    with open(file, "r") as f:
        script = f.read()

    ast, ir, progress = process(script, 4)
    if ast is not None:
        print("\nGenerating String representation Code for base AST")
        string_repr = FormatVisitor()
        result = str(string_repr.visit(ast))
        if out_file is None:
            out_file = file[:file.find(".tzs")]+".tzs.rep"
        with open(out_file, "w") as f:

            f.write(result)
        progress.update(1)

        print("\nGenerating String representation Code for intermediate AST")
        hl_repr = HLReprVisitor()
        result = hl_repr.visit(ir)
        if out_file is None:
            out_file = file[:file.find(".tzs")]+".tzs.rep"
        with open(out_file, "a+") as f:
            f.write("\n\n")
            f.write(str(result))
        progress.update(1)
        print(f"\nGenerated {out_file}")


if __name__ == "__main__":
    app()