fix enum nested in namespace

felixhirt · felixhirt · commit ba317ff7f6c9 · 2026-01-26T08:47:58.000+01:00
diff --git a/src/transformation/visitors/enum.ts b/src/transformation/visitors/enum.ts
@@ -2,7 +2,7 @@ import * as ts from "typescript";
 import * as lua from "../../LuaAST";
 import { FunctionVisitor, TransformationContext } from "../context";
 import { AnnotationKind, getTypeAnnotations } from "../utils/annotations";
-import { getSymbolExportScope } from "../utils/export";
+import { addExportToIdentifier, getSymbolExportScope } from "../utils/export";
 import { createLocalOrExportedOrGlobalDeclaration } from "../utils/lua-ast";
 import { isFirstDeclaration } from "../utils/typescript";
 import { transformIdentifier } from "./identifier";
@@ -32,7 +32,7 @@ export const transformEnumDeclaration: FunctionVisitor<ts.EnumDeclaration> = (no
     if (!membersOnly && isFirstDeclaration(context, node)) {
         const name = transformIdentifier(context, node.name);
         const table = lua.createBinaryExpression(
-            lua.cloneIdentifier(name),
+            addExportToIdentifier(context, name),
             lua.createTableExpression(),
             lua.SyntaxKind.OrOperator
         );
diff --git a/test/unit/enum.spec.ts b/test/unit/enum.spec.ts
@@ -203,3 +203,19 @@ test("enum merging multiple files", () => {
         )
         .expectToMatchJsResult();
 });
+
+test("enum nested in namespace", () => {
+    util.testModule`
+        namespace A {
+            export enum TestEnum {
+                C,
+                D
+            }
+        }
+    `.tap(builder => {
+        const lua = builder.getMainLuaCodeChunk();
+        expect(lua).toMatch(
+            'A = A or ({})\ndo\n    A.TestEnum = A.TestEnum or ({})\n    A.TestEnum.C = 0\n    A.TestEnum[A.TestEnum.C] = "C"\n    A.TestEnum.D = 1\n    A.TestEnum[A.TestEnum.D] = "D"\nend'
+        );
+    });
+});
