Fix enum nested in namespace (#1682)

felixhirt · web-flow · commit e1f5d4f911e2 · 2026-02-23T21:45:24.000+01:00
* fix enum nested in namespace

* use snapshot to test
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/__snapshots__/enum.spec.ts.snap b/test/unit/__snapshots__/enum.spec.ts.snap
@@ -0,0 +1,12 @@
+// Jest Snapshot v1, https://goo.gl/fbAQLP
+
+exports[`enum nested in namespace 1`] = `
+"A = A or ({})
+do
+    A.TestEnum = A.TestEnum or ({})
+    A.TestEnum.B = 0
+    A.TestEnum[A.TestEnum.B] = "B"
+    A.TestEnum.C = 1
+    A.TestEnum[A.TestEnum.C] = "C"
+end"
+`;
diff --git a/test/unit/enum.spec.ts b/test/unit/enum.spec.ts
@@ -203,3 +203,14 @@ test("enum merging multiple files", () => {
         )
         .expectToMatchJsResult();
 });
+
+test("enum nested in namespace", () => {
+    util.testModule`
+        namespace A {
+            export enum TestEnum {
+                B,
+                C
+            }
+        }
+    `.expectLuaToMatchSnapshot();
+});
