JSON-Lang/compile.py

31 lines
1008 B
Python

from emis_funky_funktions import *
from typing import Collection, Sequence, TypeAlias
from ir import BUILTIN_SUBSTITUTIONS, Expression, ReplHole, subst_all
from ir.genir import json_to_ir, PatternParseProblem, BranchTypesDiffer, UndefinedVariable
from ir.types_ import BUILTINS_CONTEXT, UnificationError
from repl import evaluate
from opt import optimize_to_fixpoint, all_optimizations
import json
from dataclasses import dataclass
from operator import add
def main():
import sys
match sys.argv:
case [_, file]:
# TODO handle this
expr, ty, substs = unwrap_r(json_to_ir(json.loads(open(sys.argv[1]).read()), BUILTINS_CONTEXT))
expr_with_builtins = subst_all(BUILTIN_SUBSTITUTIONS, expr)
expr_with_optimization = optimize_to_fixpoint(all_optimizations, expr_with_builtins)
result = evaluate(expr_with_optimization)
if isinstance(result, ReplHole):
print(result.render())
else:
print(result.codegen())
case _:
raise Exception('TODO')
if __name__ == '__main__':
main()