97 lines
2.1 KiB
Python
97 lines
2.1 KiB
Python
"""The public interface for shoggoth's baked-in types."""
|
|
|
|
|
|
import typing as t
|
|
from uuid import UUID, uuid4
|
|
|
|
|
|
class TypeVariable(t.NamedTuple):
|
|
name: str
|
|
id: UUID = uuid4()
|
|
|
|
|
|
class ArrayExpr(t.NamedTuple):
|
|
child: t.Any
|
|
|
|
|
|
class SumExpr(t.NamedTuple):
|
|
children: t.List[t.Any]
|
|
|
|
|
|
class ProductExpr(t.NamedTuple):
|
|
children: t.Mapping[str, t.Any]
|
|
|
|
|
|
# FIXME: How exactly
|
|
class StructExpr(t.NamedTuple):
|
|
name: str
|
|
type_params: list
|
|
field_names: t.List[str]
|
|
children: t.List[t.Any]
|
|
|
|
|
|
####################################################################################################
|
|
####################################################################################################
|
|
####################################################################################################
|
|
|
|
class FunctionRef(t.NamedTuple):
|
|
raw: str
|
|
type_params: list
|
|
name: str
|
|
args: list
|
|
ret: list
|
|
|
|
@staticmethod
|
|
def parse_list(l):
|
|
return [e for e in l.split(",") if e]
|
|
|
|
@classmethod
|
|
def parse(cls, raw: str):
|
|
vars, name, args, ret = raw.split(";")
|
|
return cls(
|
|
raw,
|
|
cls.parse_list(vars),
|
|
name,
|
|
cls.parse_list(args),
|
|
cls.parse_list(ret)
|
|
)
|
|
|
|
|
|
class Closure(t.NamedTuple):
|
|
# Note that a closure over a closure is equivalent to a single closure which extends the captured stack fragment, so
|
|
# there's no need for a union here as we can simply convert nested closures.
|
|
funref: FunctionRef
|
|
frag: t.List[t.Any]
|
|
|
|
|
|
class FunctionSignature(t.NamedTuple):
|
|
raw: str
|
|
type_params: list
|
|
name: str
|
|
args: list
|
|
ret: list
|
|
|
|
@staticmethod
|
|
def parse_list(l):
|
|
return [e for e in l.split(",") if e]
|
|
|
|
@classmethod
|
|
def parse(cls, raw: str):
|
|
vars, name, args, ret = raw.split(";")
|
|
return cls(
|
|
raw,
|
|
cls.parse_list(vars),
|
|
name,
|
|
cls.parse_list(args),
|
|
cls.parse_list(ret)
|
|
)
|
|
|
|
|
|
class Struct(t.NamedTuple):
|
|
name: str
|
|
type_params: list
|
|
children: t.Mapping[str, t.Any]
|
|
|
|
|
|
class Identifier(t.NamedTuple):
|
|
name: str
|