Enum and enum literal

Arthur Ozga · Arthur Ozga · commit 2958649a9594 · 2017-03-22T14:34:14.000-07:00
diff --git a/src/compiler/checker.ts b/src/compiler/checker.ts
@@ -2266,7 +2266,8 @@ namespace ts {
                     return createKeywordTypeNode(SyntaxKind.BooleanKeyword);
                 }
                 if (type.flags & TypeFlags.Enum) {
-                    throw new Error ("enums not implemented")
+                    const name = createNameFromSymbol(type.symbol);
+                    return createTypeReferenceNode(name, /*typeArguments*/ undefined);
                 }
                 if (type.flags & (TypeFlags.StringLiteral)) {
                     return createLiteralTypeNode((createLiteral((<LiteralType>type).text)));
@@ -2278,7 +2279,8 @@ namespace ts {
                     return (<IntrinsicType>type).intrinsicName === "true" ? createTrue() : createFalse();
                 }
                 if (type.flags & TypeFlags.EnumLiteral) {
-                    throw new Error("enum literal not implemented");
+                    const name = createNameFromSymbol(type.symbol);
+                    return createTypeReferenceNode(name, /*typeArguments*/ undefined);
                 }
                 if (type.flags & TypeFlags.Void) {
                     return createKeywordTypeNode(SyntaxKind.VoidKeyword);
diff --git a/tests/cases/fourslash/codeFixClassImplementInterfacePropertyEnum.ts b/tests/cases/fourslash/codeFixClassImplementInterfacePropertyEnum.ts
@@ -0,0 +1,13 @@
+/// <reference path='fourslash.ts' />
+
+// @lib: es2017
+
+//// enum E { a,b,c }
+//// interface I {
+////     a: E;
+//// }
+//// class C implements I {[| |]}
+
+verify.rangeAfterCodeFix(`
+    a: E;
+`);
diff --git a/tests/cases/fourslash/codeFixClassImplementInterfacePropertyEnumLiteral.ts b/tests/cases/fourslash/codeFixClassImplementInterfacePropertyEnumLiteral.ts
@@ -0,0 +1,13 @@
+/// <reference path='fourslash.ts' />
+
+// @lib: es2017
+
+//// enum E { a,b,c }
+//// interface I {
+////     a: E.a
+//// }
+//// class C implements I {[| |]}
+
+verify.rangeAfterCodeFix(`
+    a: E.a;
+`);

Original file line number	Diff line number	Diff line change
`@@ -2266,7 +2266,8 @@ namespace ts {`
`2266`	`2266`	`return createKeywordTypeNode(SyntaxKind.BooleanKeyword);`
`2267`	`2267`	`}`
`2268`	`2268`	`if (type.flags & TypeFlags.Enum) {`
`2269`		`- throw new Error ("enums not implemented")`
	`2269`	`+ const name = createNameFromSymbol(type.symbol);`
	`2270`	`+ return createTypeReferenceNode(name, /typeArguments/ undefined);`
`2270`	`2271`	`}`
`2271`	`2272`	`if (type.flags & (TypeFlags.StringLiteral)) {`
`2272`	`2273`	`return createLiteralTypeNode((createLiteral((<LiteralType>type).text)));`
`@@ -2278,7 +2279,8 @@ namespace ts {`
`2278`	`2279`	`return (<IntrinsicType>type).intrinsicName === "true" ? createTrue() : createFalse();`
`2279`	`2280`	`}`
`2280`	`2281`	`if (type.flags & TypeFlags.EnumLiteral) {`
`2281`		`- throw new Error("enum literal not implemented");`
	`2282`	`+ const name = createNameFromSymbol(type.symbol);`
	`2283`	`+ return createTypeReferenceNode(name, /typeArguments/ undefined);`
`2282`	`2284`	`}`
`2283`	`2285`	`if (type.flags & TypeFlags.Void) {`
`2284`	`2286`	`return createKeywordTypeNode(SyntaxKind.VoidKeyword);`