std::comptime: add enum support for the Elem method of comptimeTypeInfo

julelang · Jun 26, 2024 · 58a6325 · 58a6325
1 parent 27f9a10
commit 58a6325
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/std/comptime/type.jule b/std/comptime/type.jule
@@ -59,7 +59,7 @@ impl comptimeTypeInfo {
     fn Bits(self): int
 
     // Returns comptimeTypeInfo for element type.
-    // Supports only pointers (except unsafe pointer), smart pointers, arrays, and slices.
+    // Supports only pointers (except unsafe pointer), smart pointers, arrays, slices, and enums.
     fn Elem(self): comptimeTypeInfo
 
     // Returns size of array.

diff --git a/std/jule/sema/comptime.jule b/std/jule/sema/comptime.jule
@@ -281,6 +281,10 @@ impl comptimeType {
         if array != nil {
             ret buildComptimeTypeInfoData(e.s, array.Elem)
         }
+        let mut enm = self.base.Enum()
+        if enm != nil {
+            ret buildComptimeTypeInfoData(e.s, enm.Kind.Kind)
+        }
         e.pushErr(fc.Token, LogMsg.InvalidTypeForFn, self.base.Str(), "Elem")
         ret nil
     }