Fixed incorrect keywords for non-exported enums (both simple and constant)

Fixes #205

Author: timocov

src/generate-output.ts

@@ -141,7 +141,7 @@ function compareStatementText(a: StatementText, b: StatementText): number {
 }
 
 function needAddDeclareKeyword(statement: ts.Statement, nodeText: string): boolean {
-	// for some reason TypeScript allows to do not write `declare` keyword for ClassDeclaration, FunctionDeclaration and VariableDeclaration
+	// for some reason TypeScript allows to not write `declare` keyword for ClassDeclaration, FunctionDeclaration and VariableDeclaration
 	// if it already has `export` keyword - so we need to add it
 	// to avoid TS1046: Top-level declarations in .d.ts files must start with either a 'declare' or 'export' modifier.
 	if (ts.isClassDeclaration(statement) && (/^class\b/.test(nodeText) || /^abstract\b/.test(nodeText))) {
@@ -156,6 +156,10 @@ function needAddDeclareKeyword(statement: ts.Statement, nodeText: string): boole
 		return true;
 	}
 
+	if (ts.isEnumDeclaration(statement) && (/^(const)\b/.test(nodeText) || /^(enum)\b/.test(nodeText))) {
+		return true;
+	}
+
 	return false;
 }
 

tests/e2e/test-cases/disable-non-direct-exports/enums.d.ts

@@ -0,0 +1,5 @@
+// note it doesn't have `declare` keyword
+export const enum ConstEnum2 {}
+
+// note it doesn't have `declare` keyword
+export enum Enum2 {}

tests/e2e/test-cases/disable-non-direct-exports/input.ts

@@ -7,11 +7,14 @@ import {
 	Type,
 	variable,
 } from './module';
+import { Enum2, ConstEnum2 } from './enums';
 
 export interface ExportedInterface {
 	class: Class;
 	constEnum: ConstEnum;
+	constEnum2: ConstEnum2;
 	enum: Enum;
+	enum2: Enum2;
 	func: typeof func;
 	interface: Interface;
 	type: Type;

tests/e2e/test-cases/disable-non-direct-exports/output.d.ts

@@ -9,10 +9,16 @@ declare function func(): void;
 interface Interface {
 }
 declare type Type = string;
+declare const enum ConstEnum2 {
+}
+declare enum Enum2 {
+}
 export interface ExportedInterface {
 	class: Class;
 	constEnum: ConstEnum;
+	constEnum2: ConstEnum2;
 	enum: Enum;
+	enum2: Enum2;
 	func: typeof func;
 	interface: Interface;
 	type: Type;