agtest/gen.py

import textwrap
import os
from lark import Lark

from agast import *
from agtypeck import *

p = Lark(open("grammar.lark").read(), start="program", parser="lalr")

if __name__ == "__main__":
	with open("arith.ag") as f:
		data = f.read()

	cst = p.parse(data)
	# print("cst", cst)

	trans = Parser()
	ast = trans.transform(cst)
	print("ast", ast)

	res = typecheck(ast)

	if not os.path.exists("gen"):
		os.makedirs("gen")
	with open("gen/arith.py", "w") as f:
		f.write(textwrap.dedent(f"""
			from typing import Generic, TypeVar, Optional, Callable
			from lark import Lark, Transformer
			T = TypeVar('T')
			class Thunk(Generic[T]):
				''' A thunk represents a value that may be computed lazily. '''
				value: Optional[T]
				def __init__(self, func: Callable[[], T]):
					self.func = func
					self.value = None
				def get(self) -> T:
					if self.value is None:
						self.value = self.func()
					return self.value
			parser = Lark('''{res.parser_data}''')
			class Trans(Transformer[None]):
				pass
		"""))