chore: use generator to export all api model enums (#117)

package.json

@@ -63,7 +63,6 @@
     "@types/jest": "26.0.23",
     "@types/lodash": "4.14.169",
     "@types/node": "13.13.48",
-    "@types/swagger-parser": "7.0.1",
     "danger": "10.6.4",
     "dotenv": "9.0.2",
     "env-var": "7.0.1",
@@ -72,7 +71,6 @@
     "jest": "26.6.3",
     "lodash": "4.17.21",
     "rimraf": "3.0.2",
-    "swagger-parser": "10.0.2",
     "ts-jest": "26.5.4",
     "ts-morph": "10.1.0",
     "ts-node": "9.1.1",

utils/generator/api-model-generator.ts

@@ -1,8 +1,9 @@
-import SwaggerParser from '@apidevtools/swagger-parser'
 import log from 'fancy-log'
-import { camelCase, has, isEmpty, upperFirst } from 'lodash'
+import { camelCase, upperFirst } from 'lodash'
+import { Project } from 'ts-morph'
 
 import { APIModel } from './api-model'
+import { API_MODEL_FILE_NAME, TS_CONFIG_FILE_PATH, TS_LIB_FOLDER_PATH } from './constants'
 // eslint-disable-next-line @typescript-eslint/no-var-requires
 const fs = require('fs')
 
@@ -13,7 +14,6 @@ const REDUNDANTS: string[] = [
   'git_push.sh',
   '.openapi-generator',
 ]
-const EXCLUDE_EXPORTED_OBJECTS = new Set(['ErrorList', 'Error'])
 
 export async function removeRedundantObjects(model: APIModel): Promise<APIModel> {
   /**
@@ -37,58 +37,23 @@ export async function removeRedundantObjects(model: APIModel): Promise<APIModel>
   return model
 }
 
-/**
- * Verify Data Models (Schemas) in OpenAPI definitions.
- * OpenAPI definitions have various data models. Such as: array, boolean, object.
- * We don't need to export all of them. Only export: concrete objects and enums.
- *
- * @param definitions Record<string, any>
- * @param key string
- * @returns boolean
- */
-function verifyObjectDefinition(definitions: Record<string, any>, key: string): boolean {
-  const definitionType = definitions[key].type
-
-  return (
-    (definitionType === 'object' &&
-      !EXCLUDE_EXPORTED_OBJECTS.has(key) &&
-      /**
-       * The 'properties' keyword is used to define the object properties.
-       * Docs: https://swagger.io/docs/specification/data-models/data-types/#ctxM:~:text=The%20properties%20keyword%20is%20used%20to%20define%20the%20object%20properties
-       */
-      has(definitions[key], 'properties') &&
-      /**
-       * 'additionalProperties' is used to define a free form object.
-       * Docs: https://swagger.io/docs/specification/data-models/dictionaries/#free-form:~:text=Free%2DForm%20Objects
-       */
-
-      isEmpty(definitions[key].additionalProperties)) ||
-    /**
-     * Use to combine schemas
-     * Docs: https://swagger.io/docs/specification/data-models/oneof-anyof-allof-not/
-     */
-    has(definitions[key], 'oneOf') ||
-    has(definitions[key], 'allOf') ||
-    has(definitions[key], 'oneOf') ||
-    /**
-     *
-     */
-    (definitionType === 'string' && has(definitions[key], 'enum'))
-  )
-}
-
 export async function generateExportStatement(model: APIModel): Promise<string> {
-  return SwaggerParser.parse(model.modelPath).then(({ definitions }) => {
-    const exportings: string[] = []
+  // Export all enums and interfaces
+  const sourceFile = new Project({
+    tsConfigFilePath: TS_CONFIG_FILE_PATH,
+    libFolderPath: TS_LIB_FOLDER_PATH,
+  }).getSourceFileOrThrow(`${model.outputPath}/${API_MODEL_FILE_NAME}`)
 
-    for (const key in definitions) {
-      if (has(definitions, key) && verifyObjectDefinition(definitions, key)) {
-        exportings.push(`${key} as ${upperFirst(camelCase(model.dirname))}${key}`)
-      }
-    }
+  const exports = [...sourceFile.getEnums(), ...sourceFile.getInterfaces()]
+    .map(
+      (declaration) =>
+        `${declaration.getName()} as ${upperFirst(
+          camelCase(model.dirname),
+        )}${declaration.getName()}`,
+    )
+    .join(', ')
 
-    return `export { ${exportings.join(', ')} } from './${model.dirname}'`
-  })
+  return `export { ${exports} } from './${model.dirname}'`
 }
 
 export function writeStatementsToFile(statements: string[]): void {