cgen: fix enum with comptime const value (fix #22386) (#22388)

vlang · Oct 2, 2024 · c01186c · c01186c
1 parent cf2ccbc
commit c01186c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/vlib/v/gen/c/cgen.v b/vlib/v/gen/c/cgen.v
@@ -4406,6 +4406,8 @@ fn (mut g Gen) enum_decl(node ast.EnumDecl) {
 				const_def := g.global_const_defs[util.no_dots(field.expr.name)]
 				if const_def.def.starts_with('#define') {
 					g.enum_typedefs.write_string(const_def.def.all_after_last(' '))
+				} else if const_def.def.contains('const') {
+					g.enum_typedefs.write_string(const_def.def.all_after_last('=').all_before_last(';'))
 				} else {
 					g.enum_typedefs.write_string(expr_str)
 				}

diff --git a/vlib/v/tests/enums/enum_with_comptime_const_test.v b/vlib/v/tests/enums/enum_with_comptime_const_test.v
@@ -0,0 +1,10 @@
+const enum_value = $if linux { 1 } $else { 2 }
+
+pub enum Test {
+	a = enum_value
+}
+
+fn test_enum_with_comptime_const() {
+	println(Test.a)
+	assert true
+}