sera-2 1.2.0__py3-none-any.whl → 1.4.2__py3-none-any.whl

sera/make/make_typescript_model.py

@@ -1 +1,847 @@
 from __future__ import annotations
+
+from typing import Any, Callable
+
+from codegen.models import AST, PredefinedFn, Program, expr, stmt
+from codegen.models.var import DeferredVar
+from loguru import logger
+from sera.misc import (
+    assert_isinstance,
+    assert_not_null,
+    to_camel_case,
+    to_pascal_case,
+    to_snake_case,
+)
+from sera.models import (
+    Class,
+    DataProperty,
+    ObjectProperty,
+    Package,
+    Schema,
+    TsTypeWithDep,
+)
+
+
+def make_typescript_data_model(schema: Schema, target_pkg: Package):
+    """Generate TypeScript data model from the schema. The data model aligns with the public data model in Python, not the database model."""
+    app = target_pkg.app
+
+    def clone_prop(prop: DataProperty | ObjectProperty, value: expr.Expr):
+        # detect all complex types is hard, we can assume that any update to this does not mutate
+        # the original object, then it's okay.
+        return value
+
+    def make_normal(cls: Class, pkg: Package):
+        """Make a data model for the normal Python data model"""
+        if not cls.is_public:
+            # skip classes that are not public
+            return
+
+        idprop = cls.get_id_property()
+
+        program = Program()
+
+        prop_defs = []
+        prop_constructor_assigns = []
+        deser_args = []
+
+        for prop in cls.properties.values():
+            if prop.data.is_private:
+                # skip private fields as this is for APIs exchange
+                continue
+
+            propname = to_camel_case(prop.name)
+
+            if isinstance(prop, DataProperty):
+                tstype = prop.get_data_model_datatype().get_typescript_type()
+                if tstype.dep is not None:
+                    program.import_(tstype.dep, True)
+
+                if idprop is not None and prop.name == idprop.name:
+                    # use id type alias
+                    tstype = TsTypeWithDep(f"{cls.name}Id")
+
+                deser_args.append(
+                    (
+                        expr.ExprIdent(propname),
+                        PredefinedFn.attr_getter(
+                            expr.ExprIdent("data"), expr.ExprIdent(prop.name)
+                        ),
+                    )
+                )
+            else:
+                assert isinstance(prop, ObjectProperty)
+                if prop.target.db is not None:
+                    # this class is stored in the database, we store the id instead
+                    tstype = TsTypeWithDep(
+                        f"{prop.target.name}Id",
+                        f"@.models.{prop.target.get_tsmodule_name()}.{prop.target.name}.{prop.target.name}Id",
+                    )
+                    if prop.cardinality.is_star_to_many():
+                        tstype = tstype.as_list_type()
+                    deser_args.append(
+                        (
+                            expr.ExprIdent(propname),
+                            PredefinedFn.attr_getter(
+                                expr.ExprIdent("data"), expr.ExprIdent(prop.name)
+                            ),
+                        )
+                    )
+                else:
+                    # we are going to store the whole object
+                    tstype = TsTypeWithDep(
+                        prop.target.name,
+                        f"@.models.{prop.target.get_tsmodule_name()}.{prop.target.name}.{prop.target.name}",
+                    )
+                    if prop.cardinality.is_star_to_many():
+                        tstype = tstype.as_list_type()
+                        deser_args.append(
+                            (
+                                expr.ExprIdent(propname),
+                                PredefinedFn.map_list(
+                                    PredefinedFn.attr_getter(
+                                        expr.ExprIdent("data"),
+                                        expr.ExprIdent(prop.name),
+                                    ),
+                                    lambda item: expr.ExprMethodCall(
+                                        expr.ExprIdent(
+                                            assert_isinstance(
+                                                prop, ObjectProperty
+                                            ).target.name
+                                        ),
+                                        "deser",
+                                        [item],
+                                    ),
+                                ),
+                            )
+                        )
+                    else:
+                        deser_args.append(
+                            (
+                                expr.ExprIdent(propname),
+                                expr.ExprFuncCall(
+                                    PredefinedFn.attr_getter(
+                                        expr.ExprIdent(prop.target.name),
+                                        expr.ExprIdent("deser"),
+                                    ),
+                                    [
+                                        PredefinedFn.attr_getter(
+                                            expr.ExprIdent("data"),
+                                            expr.ExprIdent(prop.name),
+                                        )
+                                    ],
+                                ),
+                            )
+                        )
+
+                if tstype.dep is not None:
+                    program.import_(
+                        tstype.dep,
+                        True,
+                    )
+
+            prop_defs.append(stmt.DefClassVarStatement(propname, tstype.type))
+            prop_constructor_assigns.append(
+                stmt.AssignStatement(
+                    PredefinedFn.attr_getter(
+                        expr.ExprIdent("this"),
+                        expr.ExprIdent(propname),
+                    ),
+                    expr.ExprIdent("args." + propname),
+                )
+            )
+
+        program.root(
+            stmt.LineBreak(),
+            (
+                stmt.TypescriptStatement(
+                    f"export type {cls.name}Id = {idprop.get_data_model_datatype().get_typescript_type().type};"
+                )
+                if idprop is not None
+                else None
+            ),
+            stmt.LineBreak(),
+            lambda ast00: ast00.interface(
+                cls.name + "ConstructorArgs",
+            )(*prop_defs),
+            stmt.LineBreak(),
+            lambda ast10: ast10.class_(cls.name)(
+                *prop_defs,
+                stmt.LineBreak(),
+                lambda ast11: ast11.func(
+                    "constructor",
+                    [
+                        DeferredVar.simple(
+                            "args", expr.ExprIdent(cls.name + "ConstructorArgs")
+                        ),
+                    ],
+                )(*prop_constructor_assigns),
+                stmt.LineBreak(),
+                lambda ast12: ast12.func(
+                    "deser",
+                    [
+                        DeferredVar.simple("data", expr.ExprIdent("any")),
+                    ],
+                    expr.ExprIdent(cls.name),
+                    is_static=True,
+                    comment="Deserialize the data from the server to create a new instance of the class",
+                )(
+                    lambda ast: ast.return_(
+                        expr.ExprNewInstance(
+                            expr.ExprIdent(cls.name), [PredefinedFn.dict(deser_args)]
+                        )
+                    )
+                ),
+            ),
+        )
+
+        pkg.module(cls.name).write(program)
+
+    def make_draft(cls: Class, pkg: Package):
+        if not cls.is_public:
+            # skip classes that are not public
+            return
+
+        idprop = cls.get_id_property()
+
+        draft_clsname = "Draft" + cls.name
+
+        program = Program()
+        program.import_(f"@.models.{pkg.dir.name}.{cls.name}.{cls.name}", True)
+        program.import_("mobx.makeObservable", True)
+        program.import_("mobx.observable", True)
+        program.import_("mobx.action", True)
+
+        program.root.linebreak()
+
+        # make sure that the property stale is not in existing properties
+        if "stale" in cls.properties:
+            raise ValueError(f"Class {cls.name} already has property stale")
+
+        # information about class primary key
+        cls_pk = None
+        observable_args: list[tuple[expr.Expr, expr.ExprIdent]] = []
+        prop_defs = []
+        prop_constructor_assigns = []
+        # attrs needed for the cls.create function
+        create_args = []
+        update_args = []
+        ser_args = []
+        update_field_funcs: list[Callable[[AST], Any]] = []
+
+        for prop in cls.properties.values():
+            if prop.data.is_private:
+                # skip private fields as this is for APIs exchange
+                continue
+
+            propname = to_camel_case(prop.name)
+
+            def _update_field_func(
+                prop: DataProperty | ObjectProperty,
+                propname: str,
+                tstype: TsTypeWithDep,
+                draft_clsname: str,
+            ):
+                return lambda ast: ast(
+                    stmt.LineBreak(),
+                    lambda ast01: ast01.func(
+                        f"update{to_pascal_case(prop.name)}",
+                        [
+                            DeferredVar.simple(
+                                "value",
+                                expr.ExprIdent(tstype.type),
+                            ),
+                        ],
+                        expr.ExprIdent(draft_clsname),
+                        comment=f"Update the `{prop.name}` field",
+                    )(
+                        stmt.AssignStatement(
+                            PredefinedFn.attr_getter(
+                                expr.ExprIdent("this"), expr.ExprIdent(propname)
+                            ),
+                            expr.ExprIdent("value"),
+                        ),
+                        stmt.AssignStatement(
+                            PredefinedFn.attr_getter(
+                                expr.ExprIdent("this"), expr.ExprIdent("stale")
+                            ),
+                            expr.ExprConstant(True),
+                        ),
+                        stmt.ReturnStatement(expr.ExprIdent("this")),
+                    ),
+                )
+
+            if isinstance(prop, DataProperty):
+                tstype = prop.get_data_model_datatype().get_typescript_type()
+                if idprop is not None and prop.name == idprop.name:
+                    # use id type alias
+                    tstype = TsTypeWithDep(
+                        f"{cls.name}Id",
+                        dep=f"@.models.{pkg.dir.name}.{cls.name}.{cls.name}Id",
+                    )
+                if tstype.dep is not None:
+                    program.import_(tstype.dep, True)
+
+                # however, if this is a primary key and auto-increment, we set a different default value
+                # to be -1 to avoid start from 0
+                if (
+                    prop.db is not None
+                    and prop.db.is_primary_key
+                    and prop.db.is_auto_increment
+                ):
+                    create_propvalue = expr.ExprConstant(-1)
+                else:
+                    create_propvalue = tstype.get_default()
+
+                if prop.db is not None and prop.db.is_primary_key:
+                    # for checking if the primary key is from the database or default (create_propvalue)
+                    cls_pk = (expr.ExprIdent(propname), create_propvalue)
+
+                update_propvalue = PredefinedFn.attr_getter(
+                    expr.ExprIdent("record"), expr.ExprIdent(propname)
+                )
+
+                ser_args.append(
+                    (
+                        expr.ExprIdent(prop.name),
+                        PredefinedFn.attr_getter(
+                            expr.ExprIdent("this"), expr.ExprIdent(propname)
+                        ),
+                    )
+                )
+                if not (prop.db is not None and prop.db.is_primary_key):
+                    # skip observable for primary key as it is not needed
+                    observable_args.append(
+                        (
+                            expr.ExprIdent(propname),
+                            expr.ExprIdent("observable"),
+                        )
+                    )
+                    observable_args.append(
+                        (
+                            expr.ExprIdent(f"update{to_pascal_case(prop.name)}"),
+                            expr.ExprIdent("action"),
+                        )
+                    )
+            else:
+                assert isinstance(prop, ObjectProperty)
+                if prop.target.db is not None:
+                    # this class is stored in the database, we store the id instead
+                    tstype = TsTypeWithDep(
+                        f"{prop.target.name}Id",
+                        f"@.models.{prop.target.get_tsmodule_name()}.{prop.target.name}.{prop.target.name}Id",
+                    )
+                    if prop.cardinality.is_star_to_many():
+                        tstype = tstype.as_list_type()
+                    create_propvalue = tstype.get_default()
+                    update_propvalue = PredefinedFn.attr_getter(
+                        expr.ExprIdent("record"), expr.ExprIdent(propname)
+                    )
+                    ser_args.append(
+                        (
+                            expr.ExprIdent(prop.name),
+                            PredefinedFn.attr_getter(
+                                expr.ExprIdent("this"), expr.ExprIdent(propname)
+                            ),
+                        )
+                    )
+                else:
+                    # we are going to store the whole object
+                    tstype = TsTypeWithDep(
+                        f"Draft{prop.target.name}",
+                        f"@.models.{prop.target.get_tsmodule_name()}.Draft{prop.target.name}.Draft{prop.target.name}",
+                    )
+                    if prop.cardinality.is_star_to_many():
+                        tstype = tstype.as_list_type()
+                        create_propvalue = expr.ExprConstant([])
+                        update_propvalue = PredefinedFn.map_list(
+                            PredefinedFn.attr_getter(
+                                expr.ExprIdent("record"), expr.ExprIdent(propname)
+                            ),
+                            lambda item: expr.ExprMethodCall(
+                                expr.ExprIdent(tstype.type),
+                                "update",
+                                [item],
+                            ),
+                        )
+                        ser_args.append(
+                            (
+                                expr.ExprIdent(prop.name),
+                                PredefinedFn.map_list(
+                                    PredefinedFn.attr_getter(
+                                        expr.ExprIdent("this"), expr.ExprIdent(propname)
+                                    ),
+                                    lambda item: expr.ExprMethodCall(item, "ser", []),
+                                ),
+                            )
+                        )
+                    else:
+                        create_propvalue = expr.ExprMethodCall(
+                            expr.ExprIdent(tstype.type),
+                            "create",
+                            [],
+                        )
+                        update_propvalue = expr.ExprMethodCall(
+                            expr.ExprIdent(tstype.type),
+                            "update",
+                            [
+                                PredefinedFn.attr_getter(
+                                    expr.ExprIdent("record"), expr.ExprIdent(propname)
+                                ),
+                            ],
+                        )
+                        ser_args.append(
+                            (
+                                expr.ExprIdent(prop.name),
+                                expr.ExprMethodCall(
+                                    PredefinedFn.attr_getter(
+                                        expr.ExprIdent("this"), expr.ExprIdent(propname)
+                                    ),
+                                    "ser",
+                                    [],
+                                ),
+                            )
+                        )
+
+                if tstype.dep is not None:
+                    program.import_(
+                        tstype.dep,
+                        True,
+                    )
+
+                observable_args.append(
+                    (
+                        expr.ExprIdent(propname),
+                        expr.ExprIdent("observable"),
+                    )
+                )
+                observable_args.append(
+                    (
+                        expr.ExprIdent(f"update{to_pascal_case(prop.name)}"),
+                        expr.ExprIdent("action"),
+                    )
+                )
+
+            prop_defs.append(stmt.DefClassVarStatement(propname, tstype.type))
+            prop_constructor_assigns.append(
+                stmt.AssignStatement(
+                    PredefinedFn.attr_getter(
+                        expr.ExprIdent("this"),
+                        expr.ExprIdent(propname),
+                    ),
+                    expr.ExprIdent("args." + propname),
+                )
+            )
+            create_args.append((expr.ExprIdent(propname), create_propvalue))
+            update_args.append(
+                (
+                    expr.ExprIdent(propname),
+                    # if this is mutable property, we need to copy to make it immutable.
+                    clone_prop(prop, update_propvalue),
+                )
+            )
+            update_field_funcs.append(
+                _update_field_func(prop, propname, tstype, draft_clsname)
+            )
+
+        prop_defs.append(stmt.DefClassVarStatement("stale", "boolean"))
+        prop_constructor_assigns.append(
+            stmt.AssignStatement(
+                PredefinedFn.attr_getter(
+                    expr.ExprIdent("this"), expr.ExprIdent("stale")
+                ),
+                expr.ExprIdent("args.stale"),
+            )
+        )
+        observable_args.append(
+            (
+                expr.ExprIdent("stale"),
+                expr.ExprIdent("observable"),
+            )
+        )
+        create_args.append(
+            (
+                expr.ExprIdent("stale"),
+                expr.ExprConstant(True),
+            ),
+        )
+        update_args.append(
+            (
+                expr.ExprIdent("stale"),
+                expr.ExprConstant(False),
+            ),
+        )
+        observable_args.sort(key=lambda x: {"observable": 0, "action": 1}[x[1].ident])
+
+        program.root(
+            lambda ast00: ast00.interface(
+                draft_clsname + "ConstructorArgs",
+            )(*prop_defs),
+            stmt.LineBreak(),
+            lambda ast10: ast10.class_(draft_clsname)(
+                *prop_defs,
+                stmt.LineBreak(),
+                lambda ast10: ast10.func(
+                    "constructor",
+                    [
+                        DeferredVar.simple(
+                            "args",
+                            expr.ExprIdent(draft_clsname + "ConstructorArgs"),
+                        ),
+                    ],
+                )(
+                    *prop_constructor_assigns,
+                    stmt.LineBreak(),
+                    stmt.SingleExprStatement(
+                        expr.ExprFuncCall(
+                            expr.ExprIdent("makeObservable"),
+                            [
+                                expr.ExprIdent("this"),
+                                PredefinedFn.dict(observable_args),
+                            ],
+                        )
+                    ),
+                ),
+                stmt.LineBreak(),
+                lambda ast11: (
+                    ast11.func(
+                        "is_new_record",
+                        [],
+                        expr.ExprIdent("boolean"),
+                        comment="Check if this draft is for creating a new record",
+                    )(
+                        stmt.ReturnStatement(
+                            expr.ExprEqual(
+                                PredefinedFn.attr_getter(
+                                    expr.ExprIdent("this"), cls_pk[0]
+                                ),
+                                cls_pk[1],
+                            )
+                        )
+                    )
+                    if cls_pk is not None
+                    else None
+                ),
+                stmt.LineBreak(),
+                lambda ast12: ast12.func(
+                    "create",
+                    [],
+                    expr.ExprIdent(draft_clsname),
+                    is_static=True,
+                    comment="Make a new draft for creating a new record",
+                )(
+                    stmt.ReturnStatement(
+                        expr.ExprNewInstance(
+                            expr.ExprIdent(draft_clsname),
+                            [PredefinedFn.dict(create_args)],
+                        )
+                    ),
+                ),
+                stmt.LineBreak(),
+                lambda ast13: ast13.func(
+                    "update",
+                    [DeferredVar.simple("record", expr.ExprIdent(cls.name))],
+                    expr.ExprIdent(draft_clsname),
+                    is_static=True,
+                    comment="Make a new draft for updating an existing record",
+                )(
+                    stmt.ReturnStatement(
+                        expr.ExprNewInstance(
+                            expr.ExprIdent(draft_clsname),
+                            [PredefinedFn.dict(update_args)],
+                        )
+                    ),
+                ),
+                *update_field_funcs,
+                stmt.LineBreak(),
+                lambda ast14: ast14.func(
+                    "ser",
+                    [],
+                    expr.ExprIdent("any"),
+                    comment="Serialize the draft to communicate with the server",
+                )(
+                    stmt.ReturnStatement(
+                        PredefinedFn.dict(ser_args),
+                    ),
+                ),
+            ),
+        )
+
+        pkg.module("Draft" + cls.name).write(program)
+
+    def make_table(cls: Class, pkg: Package):
+        if not cls.is_public or cls.db is None:
+            # skip classes that are not public and not stored in the database
+            return
+
+        outmod = pkg.module(cls.name + "Table")
+        if outmod.exists():
+            # skip if the module already exists
+            logger.info(f"Module {outmod.path} already exists, skip")
+            return
+
+        program = Program()
+        program.import_(f"@.models.{pkg.dir.name}.{cls.name}.{cls.name}", True)
+        program.import_(f"@.models.{pkg.dir.name}.{cls.name}.{cls.name}Id", True)
+        program.import_(f"@.models.{pkg.dir.name}.{cls.name}Query.query", True)
+        program.import_(
+            f"@.models.{pkg.dir.name}.Draft{cls.name}.Draft{cls.name}", True
+        )
+        program.import_("sera-db.Table", True)
+        program.import_("sera-db.DB", True)
+
+        program.root(
+            stmt.LineBreak(),
+            lambda ast00: ast00.class_(
+                f"{cls.name}Table",
+                [expr.ExprIdent(f"Table<{cls.name}Id, {cls.name}, Draft{cls.name}>")],
+            )(
+                lambda ast01: ast01.func(
+                    "constructor",
+                    [
+                        DeferredVar.simple(
+                            "db",
+                            expr.ExprIdent("DB"),
+                        )
+                    ],
+                )(
+                    stmt.SingleExprStatement(
+                        expr.ExprFuncCall(
+                            expr.ExprIdent("super"),
+                            [
+                                PredefinedFn.dict(
+                                    [
+                                        (
+                                            expr.ExprIdent("cls"),
+                                            expr.ExprIdent(cls.name),
+                                        ),
+                                        (
+                                            expr.ExprIdent("remoteURL"),
+                                            expr.ExprConstant(
+                                                f"/api/{to_snake_case(cls.name).replace('_', '-')}"
+                                            ),
+                                        ),
+                                        (
+                                            expr.ExprIdent("db"),
+                                            expr.ExprIdent("db"),
+                                        ),
+                                        (
+                                            expr.ExprIdent("queryProcessor"),
+                                            expr.ExprIdent("query"),
+                                        ),
+                                    ]
+                                )
+                            ],
+                        )
+                    )
+                ),
+            ),
+        )
+
+        outmod.write(program)
+
+    def make_query_processor(cls: Class, pkg: Package):
+        if not cls.is_public:
+            # skip classes that are not public
+            return
+
+        outmod = pkg.module(cls.name + "Query")
+
+        program = Program()
+        program.import_(f"@.models.{pkg.dir.name}.{cls.name}.{cls.name}", True)
+        program.import_(f"sera-db.QueryProcessor", True)
+
+        query_args = []
+        for prop in cls.properties.values():
+            propname = to_camel_case(prop.name)
+            if propname != prop.name:
+                query_args.append(
+                    (
+                        expr.ExprIdent(propname),
+                        expr.ExprConstant(prop.name),
+                    )
+                )
+
+        program.root(
+            stmt.LineBreak(),
+            stmt.TypescriptStatement(
+                f"export const query = "
+                + expr.ExprNewInstance(
+                    expr.ExprIdent(f"QueryProcessor<{cls.name}>"),
+                    [
+                        PredefinedFn.dict(query_args),
+                    ],
+                ).to_typescript()
+                + ";",
+            ),
+        )
+
+        outmod.write(program)
+
+    def make_definition(cls: Class, pkg: Package):
+        """Make schema definition for the class in frontend so that we can generate components"""
+        if not cls.is_public:
+            # skip classes that are not public
+            return
+
+        program = Program()
+        prop_defs: list[tuple[expr.Expr, expr.Expr]] = []
+
+        for prop in cls.properties.values():
+            if prop.data.is_private:
+                # skip private fields as this is for APIs exchange
+                continue
+            propname = to_camel_case(prop.name)
+            tsprop = {}
+
+            if isinstance(prop, DataProperty):
+                tstype = prop.get_data_model_datatype().get_typescript_type()
+                # for schema definition, we need to use the original type, not the type alias
+                # if prop.name == idprop.name:
+                #     # use id type alias
+                #     tstype = TsTypeWithDep(f"{cls.name}Id")
+                if tstype.dep is not None:
+                    program.import_(tstype.dep, True)
+                tsprop = [
+                    (expr.ExprIdent("datatype"), expr.ExprConstant(tstype.type)),
+                    (
+                        expr.ExprIdent("isList"),
+                        expr.ExprConstant(prop.datatype.is_list),
+                    ),
+                ]
+            else:
+                assert isinstance(prop, ObjectProperty)
+                if prop.target.db is not None:
+                    # this class is stored in the database, we store the id instead
+                    tstype = (
+                        assert_not_null(prop.target.get_id_property())
+                        .get_data_model_datatype()
+                        .get_typescript_type()
+                    )
+                else:
+                    # we are going to store the whole object
+                    tstype = TsTypeWithDep(
+                        prop.target.name,
+                        f"@.models.{prop.target.get_tsmodule_name()}.{prop.target.name}.{prop.target.name}",
+                    )
+
+                # we don't store the type itself, but just the name of the type
+                # so not need to import the dependency
+                # if tstype.dep is not None:
+                #     program.import_(
+                #         tstype.dep,
+                #         True,
+                #     )
+
+                tsprop = [
+                    (
+                        expr.ExprIdent("targetClass"),
+                        expr.ExprConstant(prop.target.name),
+                    ),
+                    (expr.ExprIdent("datatype"), expr.ExprConstant(tstype.type)),
+                    (
+                        expr.ExprIdent("isList"),
+                        expr.ExprConstant(prop.cardinality.is_star_to_many()),
+                    ),
+                    (
+                        expr.ExprIdent("isEmbedded"),
+                        expr.ExprConstant(prop.target.db is not None),
+                    ),
+                ]
+
+            prop_defs.append(
+                (
+                    expr.ExprIdent(propname),
+                    PredefinedFn.dict(
+                        [
+                            (expr.ExprIdent("name"), expr.ExprConstant(prop.name)),
+                            (
+                                expr.ExprIdent("label"),
+                                expr.ExprConstant(prop.label.to_dict()),
+                            ),
+                            (
+                                expr.ExprIdent("description"),
+                                expr.ExprConstant(prop.description.to_dict()),
+                            ),
+                        ]
+                        + tsprop
+                    ),
+                )
+            )
+
+        program.import_("sera-db.Schema", True)
+        program.import_(f"@.models.{pkg.dir.name}.{cls.name}.{cls.name}", True)
+        program.root(
+            stmt.LineBreak(),
+            stmt.TypescriptStatement(
+                f"export const {cls.name}Schema: Schema<{cls.name}> = "
+                + PredefinedFn.dict(
+                    [
+                        (expr.ExprIdent("properties"), PredefinedFn.dict(prop_defs)),
+                    ]
+                    + (
+                        [
+                            (
+                                expr.ExprIdent("primaryKey"),
+                                expr.ExprConstant(
+                                    assert_not_null(cls.get_id_property()).name
+                                ),
+                            )
+                        ]
+                        if cls.db is not None
+                        else []
+                    )
+                ).to_typescript()
+                + ";"
+            ),
+        )
+        pkg.module(cls.name + "Schema").write(program)
+
+    def make_index(pkg: Package):
+        outmod = pkg.module("index")
+        if outmod.exists():
+            # skip if the module already exists
+            logger.info(f"Module {outmod.path} already exists, skip")
+            return
+
+        program = Program()
+        program.import_(f"@.models.{pkg.dir.name}.{cls.name}.{cls.name}", True)
+        if cls.db is not None:
+            # only import the id if this class is stored in the database
+            program.import_(f"@.models.{pkg.dir.name}.{cls.name}.{cls.name}Id", True)
+        program.import_(
+            f"@.models.{pkg.dir.name}.{cls.name}Schema.{cls.name}Schema", True
+        )
+        program.import_(
+            f"@.models.{pkg.dir.name}.Draft{cls.name}.Draft{cls.name}", True
+        )
+        if cls.db is not None:
+            program.import_(
+                f"@.models.{pkg.dir.name}.{cls.name}Table.{cls.name}Table", True
+            )
+
+        program.root(
+            stmt.LineBreak(),
+            stmt.TypescriptStatement(
+                f"export {{ {cls.name}, Draft{cls.name}, {cls.name}Table, {cls.name}Schema }};"
+                if cls.db is not None
+                else f"export {{ {cls.name}, Draft{cls.name}, {cls.name}Schema }};"
+            ),
+            (
+                stmt.TypescriptStatement(f"export type {{ {cls.name}Id }};")
+                if cls.db
+                else None
+            ),
+        )
+
+        outmod.write(program)
+
+    for cls in schema.topological_sort():
+        pkg = target_pkg.pkg(cls.get_tsmodule_name())
+        make_normal(cls, pkg)
+        make_draft(cls, pkg)
+        make_query_processor(cls, pkg)
+        make_table(cls, pkg)
+        make_definition(cls, pkg)
+
+        make_index(pkg)