main.py

import os
import sys
from string import digits
from typing import Dict

from compiler.execution.executor import Executor
from compiler.contextual_analysis.visitor import Visitor
from compiler.parsing.top_down_parser import TopDownParser
from editor import Editor

VALID_INPUTS = digits + "*/+-^. "


def show_help():
    print("-f <FILEPATH> | Execute file")
    print("-h            | Help")


def config() -> Dict:
    switch = {
        "-f": None,
        "-h": show_help,
    }
    for i in range(1, len(sys.argv)):
        if sys.argv[i] in switch.keys():
            if switch[sys.argv[i]] is None:
                switch[sys.argv[i]] = sys.argv[i + 1]
            else:
                switch[sys.argv[i]]()
    return switch


def file_to_text(path: str) -> str:
    with open(path, "r") as file:
        return file.read()


def run(text: str):
    p = TopDownParser()
    v = Visitor()
    e = Executor()
    parsed = p.parse(text)
    parsed.visit(v)
    parsed.execute(e)


if __name__ == "__main__":
    if "--dev" not in sys.argv:
        sys.tracebacklimit = 0
    c: Dict = config()
    if c["-f"] is not None:
        if not(c["-f"].endswith(".skm")):
            raise Exception("File needs to have .skm ending.")
        code = file_to_text(c["-f"])
        run(code)
    else:
        while True:
            editor = Editor()
            code = editor.text
            os.system("cls")
            print(code)
            run(code)
            input("\nEnter    Write new code")