Add support for TS enum type variables.

lit · Dec 8, 2022 · bb8ba31 · bb8ba31
1 parent 7fac64c
commit bb8ba31
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 3 deletions.
diff --git a/packages/labs/analyzer/src/lib/javascript/modules.ts b/packages/labs/analyzer/src/lib/javascript/modules.ts
@@ -26,6 +26,7 @@ import {getClassDeclarationInfo} from './classes.js';
 import {
   getExportAssignmentVariableDeclarationInfo,
   getVariableDeclarationInfo,
+  getEnumDeclarationInfo,
 } from './variables.js';
 import {AbsolutePath, PackagePath, absoluteToPackage} from '../paths.js';
 import {getPackageInfo} from './packages.js';
@@ -111,6 +112,8 @@ export const getModule = (
       addDeclaration(getClassDeclarationInfo(statement, analyzer));
     } else if (ts.isVariableStatement(statement)) {
       getVariableDeclarationInfo(statement, analyzer).forEach(addDeclaration);
+    } else if (ts.isEnumDeclaration(statement)) {
+      addDeclaration(getEnumDeclarationInfo(statement, analyzer));
     } else if (ts.isExportDeclaration(statement) && !statement.isTypeOnly) {
       const {exportClause, moduleSpecifier} = statement;
       if (exportClause === undefined) {

diff --git a/packages/labs/analyzer/src/lib/javascript/variables.ts b/packages/labs/analyzer/src/lib/javascript/variables.ts
@@ -30,7 +30,7 @@ type VariableName =
  * ts.Identifier within a potentially nested ts.VariableDeclaration.
  */
 const getVariableDeclaration = (
-  dec: ts.VariableDeclaration,
+  dec: ts.VariableDeclaration | ts.EnumDeclaration,
   name: ts.Identifier,
   analyzer: AnalyzerInterface
 ): VariableDeclaration => {
@@ -129,3 +129,14 @@ const getExportAssignmentVariableDeclaration = (
     type: getTypeForNode(exportAssignment.expression, analyzer),
   });
 };
+
+export const getEnumDeclarationInfo = (
+  statement: ts.EnumDeclaration,
+  analyzer: AnalyzerInterface
+) => {
+  return {
+    name: statement.name.text,
+    factory: () => getVariableDeclaration(statement, statement.name, analyzer),
+    isExport: hasExportKeyword(statement),
+  };
+};
diff --git a/packages/labs/analyzer/src/lib/model.ts b/packages/labs/analyzer/src/lib/model.ts
@@ -282,12 +282,15 @@ export abstract class Declaration {
 }
 
 export interface VariableDeclarationInit extends DeclarationInit {
-  node: ts.VariableDeclaration | ts.ExportAssignment;
+  node: ts.VariableDeclaration | ts.ExportAssignment | ts.EnumDeclaration;
   type: Type | undefined;
 }
 
 export class VariableDeclaration extends Declaration {
-  readonly node: ts.VariableDeclaration | ts.ExportAssignment;
+  readonly node:
+    | ts.VariableDeclaration
+    | ts.ExportAssignment
+    | ts.EnumDeclaration;
   readonly type: Type | undefined;
   constructor(init: VariableDeclarationInit) {
     super(init);