fix generic struct-like enums options

AnyDSL · Jul 9, 2024 · 5d1e2b6 · 5d1e2b6
1 parent 2263837
commit 5d1e2b6
Show file tree

Hide file tree

Showing 5 changed files with 24 additions and 3 deletions.
diff --git a/include/artic/types.h b/include/artic/types.h
@@ -594,9 +594,7 @@ struct TypeApp : public Type {
     }
 
     /// Returns the type of the given member of the applied type, if it is a complex type.
-    const Type* member_type(size_t i) const {
-        return applied->as<ComplexType>()->member_type(i)->replace(replace_map());
-    }
+    const Type* member_type(size_t i) const;
 
     void print(Printer&) const override;
     bool equals(const Type*) const override;

diff --git a/src/check.cpp b/src/check.cpp
@@ -703,6 +703,8 @@ const artic::Type* Path::infer(TypeChecker& checker, bool value_expected, Ptr<Ex
                 if (enum_type->decl.options[*index]->struct_type) {
                     // If the enumeration option uses the record syntax, we use the corresponding structure type
                     type = enum_type->decl.options[*index]->struct_type;
+                    if (type_app)
+                        type = checker.type_table.type_app(type->as<StructType>(), type_app->type_args);
                     is_value = false;
                     is_ctor = true;
                 } else {

diff --git a/src/types.cpp b/src/types.cpp
@@ -476,6 +476,12 @@ const ModType::Members& ModType::members() const {
     return *members_;
 }
 
+const Type* TypeApp::member_type(size_t i) const {
+    if (auto enum_t = applied->isa<EnumType>(); enum_t && enum_t->decl.options[i]->struct_type)
+        return type_table.type_app(enum_t->decl.options[i]->struct_type->as<StructType>(), type_args);
+    return applied->as<ComplexType>()->member_type(i)->replace(replace_map());
+}
+
 // Misc. ---------------------------------------------------------------------------
 
 bool Type::subtype(const Type* other) const {

diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt
@@ -47,6 +47,7 @@ add_test(NAME simple_enums1      COMMAND artic --print-ast ${CMAKE_CURRENT_SOURC
 add_test(NAME simple_enums2      COMMAND artic --print-ast ${CMAKE_CURRENT_SOURCE_DIR}/simple/enums2.art)
 add_test(NAME simple_enums3      COMMAND artic --print-ast ${CMAKE_CURRENT_SOURCE_DIR}/simple/enums3.art)
 add_test(NAME simple_enums4      COMMAND artic --print-ast ${CMAKE_CURRENT_SOURCE_DIR}/simple/enums4.art)
+add_test(NAME simple_enums5      COMMAND artic --print-ast ${CMAKE_CURRENT_SOURCE_DIR}/simple/enums5.art)
 add_test(NAME simple_types       COMMAND artic --print-ast ${CMAKE_CURRENT_SOURCE_DIR}/simple/types.art)
 add_test(NAME simple_loops       COMMAND artic --print-ast ${CMAKE_CURRENT_SOURCE_DIR}/simple/loops.art)
 add_test(NAME simple_return      COMMAND artic --print-ast ${CMAKE_CURRENT_SOURCE_DIR}/simple/return.art)

diff --git a/test/simple/enums5.art b/test/simple/enums5.art
@@ -0,0 +1,14 @@
+enum G[T] { A, B(bool), C { x: bool, y: T }, D {} }
+#[export]
+fn woah(k: G[i32]) {
+    match k {
+        G[i32]::A => 1,
+        G[i32]::B(true) => 3,
+        G[i32]::B(false) => 4,
+        G[i32]::C { x = true, y = 1 }  => 5,
+        G[i32]::C { x = true, y = _ } => 6,
+        G[i32]::C { x = false, y = 2 } => 7,
+        G[i32]::C { x = false, y = _ } => 8,
+        G[i32]::D {} => 9
+    }
+}