feat: add better reusability for the CLI (#2010)

asyncapi · May 21, 2024 · 0cd15c0 · 0cd15c0
1 parent faa6bfd
commit 0cd15c0
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 13 deletions.
diff --git a/modelina-cli/src/commands/generate.ts b/modelina-cli/src/commands/generate.ts
@@ -32,6 +32,20 @@ export default class Models extends ModelinaCommand {
       },
     }
 
-    await generateModels(flags, document, logger, language as Languages);
+    const generatedModels = await generateModels(flags, document, logger, language as Languages);
+
+    if (flags.output) {
+      const generatedModelsString = generatedModels.map((model) => { return model.modelName; });
+      logger.info(`Successfully generated the following models: ${generatedModelsString.join(', ')}`);
+    } else {
+      const generatedModelsString = generatedModels.map((model) => {
+        return `## Model name: ${model.modelName}
+${model.result}
+`;
+      });
+      logger.info(`Successfully generated the following models: ${generatedModelsString.join('\n')}`);
+    }
+
+
   }
 }
diff --git a/modelina-cli/src/helpers/generate.ts b/modelina-cli/src/helpers/generate.ts
@@ -110,23 +110,13 @@ export async function generateModels(flags: any, document: any, logger: any, lan
   const { fileGenerator, fileOptions } = mapper(flags);
 
   if (output) {
-    const models = await fileGenerator.generateToFiles(
+    return fileGenerator.generateToFiles(
       document,
       output,
       { ...fileOptions, });
-    const generatedModels = models.map((model) => { return model.modelName; });
-    logger.info(`Successfully generated the following models: ${generatedModels.join(', ')}`);
-    return;
   }
 
-  const models = await fileGenerator.generateCompleteModels(
+  return fileGenerator.generateCompleteModels(
     document,
     { ...fileOptions });
-  const generatedModels = models.map((model) => {
-    return `
-  ## Model name: ${model.modelName}
-  ${model.result}
-  `;
-  });
-  logger.info(`Successfully generated the following models: ${generatedModels.join('\n')}`);
 }