Merge pull request #12 from timocov/enums

Author: timocov

src/transformer.ts

@@ -113,6 +113,11 @@ function createTransformerFactory(program: ts.Program, options?: Partial<RenameO
 					return handleClassMember(node);
 				}
 
+				// enum Enum { node }
+				if (ts.isEnumMember(node.parent) && node.parent.name === node) {
+					return handleEnumMember(node);
+				}
+
 				// const a = { node: 123 }
 				if (ts.isPropertyAssignment(node.parent) && node.parent.name === node) {
 					return handlePropertyAssignment(node);
@@ -181,6 +186,11 @@ function createTransformerFactory(program: ts.Program, options?: Partial<RenameO
 			return createNewIdentifier(node);
 		}
 
+		// enum Enum { node }
+		function handleEnumMember(node: ts.Identifier): ts.Identifier {
+			return createNewIdentifier(node);
+		}
+
 		// const { node: localName } = obj;
 		function handleBindingElement(node: ts.Identifier): ts.Identifier {
 			return createNewIdentifier(node);

tests/test-cases/enums/input.ts

@@ -0,0 +1,12 @@
+enum TestEnum {
+	Test1,
+	Test2,
+	Test3,
+}
+
+console.log(TestEnum.Test1);
+console.log(TestEnum['Test2']);
+
+export enum ExportedEnum {
+	Foo,
+}

tests/test-cases/enums/output.js

@@ -0,0 +1,15 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ExportedEnum = void 0;
+var TestEnum;
+(function (TestEnum) {
+    TestEnum[TestEnum["_internal_Test1"] = 0] = "_internal_Test1";
+    TestEnum[TestEnum["_internal_Test2"] = 1] = "_internal_Test2";
+    TestEnum[TestEnum["_internal_Test3"] = 2] = "_internal_Test3";
+})(TestEnum || (TestEnum = {}));
+console.log(TestEnum._internal_Test1);
+console.log(TestEnum["_internal_Test2"]);
+var ExportedEnum;
+(function (ExportedEnum) {
+    ExportedEnum[ExportedEnum["Foo"] = 0] = "Foo";
+})(ExportedEnum = exports.ExportedEnum || (exports.ExportedEnum = {}));