Fix sequencing issue with typedefs for structs/enums

3 files changed, 25 insertions, 5 deletions

diff --git a/compile.c b/compile.c
index d2554ed2..154a9e07 100644
--- a/compile.c
+++ b/compile.c
@@ -4389,7 +4389,6 @@ CORD compile_statement_type_header(env_t *env, ast_t *ast)
         auto def = Match(ast, LangDef);
         CORD full_name = CORD_cat(namespace_prefix(env, env->namespace), def->name);
         return CORD_all(
-            "typedef Text_t ", namespace_prefix(env, env->namespace), def->name, "_t;\n"
             // Constructor macro:
             "#define ", namespace_prefix(env, env->namespace), def->name,
                 "(text) ((", namespace_prefix(env, env->namespace), def->name, "_t){.length=sizeof(text)-1, .tag=TEXT_ASCII, .ascii=\"\" text})\n"
@@ -4494,7 +4493,28 @@ typedef struct {
     CORD *header;
 } compile_typedef_info_t;
 
-static void _visit_statement(compile_typedef_info_t *info, ast_t *ast)
+static void _make_typedefs(compile_typedef_info_t *info, ast_t *ast)
+{
+    if (ast->tag == StructDef) {
+        auto def = Match(ast, StructDef);
+        CORD full_name = CORD_cat(namespace_prefix(info->env, info->env->namespace), def->name);
+        *info->header = CORD_all(*info->header, "typedef struct ", full_name, "_s ", full_name, "_t;\n");
+    } else if (ast->tag == EnumDef) {
+        auto def = Match(ast, EnumDef);
+        CORD full_name = CORD_cat(namespace_prefix(info->env, info->env->namespace), def->name);
+        *info->header = CORD_all(*info->header, "typedef struct ", full_name, "_s ", full_name, "_t;\n");
+
+        for (tag_ast_t *tag = def->tags; tag; tag = tag->next) {
+            if (!tag->fields) continue;
+            *info->header = CORD_all(*info->header, "typedef struct ", full_name, "$", tag->name, "_s ", full_name, "$", tag->name, "_t;\n");
+        }
+    } else if (ast->tag == LangDef) {
+        auto def = Match(ast, LangDef);
+        *info->header = CORD_all(*info->header, "typedef Text_t ", namespace_prefix(info->env, info->env->namespace), def->name, "_t;\n");
+    }
+}
+
+static void _define_types(compile_typedef_info_t *info, ast_t *ast)
 {
     *info->header = CORD_all(*info->header,
                              compile_statement_type_header(info->env, ast),
@@ -4509,7 +4529,8 @@ CORD compile_file_header(env_t *env, ast_t *ast)
         "#include <tomo/tomo.h>\n");
 
     compile_typedef_info_t info = {.env=env, .header=&header};
-    visit_topologically(Match(ast, Block)->statements, (Closure_t){.fn=(void*)_visit_statement, &info});
+    visit_topologically(Match(ast, Block)->statements, (Closure_t){.fn=(void*)_make_typedefs, &info});
+    visit_topologically(Match(ast, Block)->statements, (Closure_t){.fn=(void*)_define_types, &info});
 
     header = CORD_all(header, "void _$", env->namespace->name, "$$initialize(void);\n");
     return header;
diff --git a/enums.c b/enums.c
index 354d9752..25e55336 100644
--- a/enums.c
+++ b/enums.c
@@ -68,7 +68,7 @@ CORD compile_enum_header(env_t *env, ast_t *ast)
 {
     auto def = Match(ast, EnumDef);
     CORD full_name = CORD_all(namespace_prefix(env, env->namespace), def->name);
-    CORD all_defs = CORD_all("typedef struct ", full_name, "_s ", full_name, "_t;\n");
+    CORD all_defs = CORD_EMPTY;
     CORD enum_def = CORD_all("struct ", full_name, "_s {\n"
                              "\tenum { ", full_name, "$null=0, ");
 
diff --git a/structs.c b/structs.c
index 94a04de9..7c652678 100644
--- a/structs.c
+++ b/structs.c
@@ -62,7 +62,6 @@ CORD compile_struct_header(env_t *env, ast_t *ast)
     struct_code = CORD_all(struct_code, "};\n");
     type_t *t = Table$str_get(*env->types, def->name);
     return CORD_all(
-        "typedef struct ", full_name, "_s ", full_name, "_t;\n",
         "struct ", full_name, "_s {\n",
         fields,
         "};\n",