feat: discover models from subdirectories

Signed-off-by: warisniz02 <[email protected]> Signed-off-by: warisniz02 <[email protected]> Signed-off-by: warisniz02 <[email protected]>
loopbackio · Sep 22, 2024 · e329f85 · e329f85
1 parent ec7c6a0
commit e329f85
Show file tree

Hide file tree

Showing 3 changed files with 97 additions and 6 deletions.
diff --git a/packages/cli/generators/relation/index.js b/packages/cli/generators/relation/index.js
@@ -230,6 +230,26 @@ module.exports = class RelationGenerator extends ArtifactGenerator {
       /* istanbul ignore next */
       return this.exit(err);
     }
+    // Check if modelDir contains subdirectories
+    const subdirectories = await utils.getSubdirectories(
+      this.artifactInfo.modelDir,
+    );
+    // If subdirectories exist, retrieve models from them
+    if (subdirectories.length > 0) {
+      for (const subdirectory of subdirectories) {
+        try {
+          const subdirectoryModelList = await utils.getArtifactList(
+            subdirectory,
+            'model',
+          );
+          modelList = modelList.concat(subdirectoryModelList);
+        } catch (err) {
+          console.error(
+            `Error retrieving models from subdirectory ${subdirectory}: ${err}`,
+          );
+        }
+      }
+    }
     let repoList;
     try {
       debug(`repository list dir ${this.artifactInfo.repoDir}`);

diff --git a/packages/cli/generators/repository/index.js b/packages/cli/generators/repository/index.js
@@ -101,14 +101,48 @@ module.exports = class RepositoryGenerator extends ArtifactGenerator {
       this.artifactInfo.modelDir,
       utils.getModelFileName(modelName),
     );
-    try {
-      fileContent = this.fs.read(modelFile, {});
-    } catch (err) {
-      debug(`${ERROR_READING_FILE} ${modelFile}: ${err.message}`);
-      return this.exit(err);
+
+    // Check if the model file exists directly in the modelDir
+    if (fs.existsSync(modelFile)) {
+      try {
+        fileContent = this.fs.read(modelFile, {});
+      } catch (err) {
+        debug(`${ERROR_READING_FILE} ${modelFile}: ${err.message}`);
+        return this.exit(err);
+      }
+
+      return tsquery.getIdFromModel(fileContent);
+    }
+
+    // If the model file is not found directly, search in subdirectories
+    const subdirectories = await utils.getSubdirectories(
+      this.artifactInfo.modelDir,
+    );
+
+    for (const subdirectory of subdirectories) {
+      const subdirectoryModelFile = path.join(
+        subdirectory,
+        utils.getModelFileName(modelName),
+      );
+
+      if (fs.existsSync(subdirectoryModelFile)) {
+        try {
+          fileContent = this.fs.read(subdirectoryModelFile, {});
+        } catch (err) {
+          debug(
+            `${ERROR_READING_FILE} ${subdirectoryModelFile}: ${err.message}`,
+          );
+          return this.exit(err);
+        }
+
+        return tsquery.getIdFromModel(fileContent);
+      }
     }
 
-    return tsquery.getIdFromModel(fileContent);
+    // If the model file is not found in any subdirectory, return an error
+    return this.exit(
+      new Error(`Model ${modelName} not found in any subdirectory.`),
+    );
   }
 
   /**
@@ -330,6 +364,27 @@ module.exports = class RepositoryGenerator extends ArtifactGenerator {
       return this.exit(err);
     }
 
+    // Check if modelDir contains subdirectories
+    const subdirectories = await utils.getSubdirectories(
+      this.artifactInfo.modelDir,
+    );
+    // If subdirectories exist, retrieve models from them
+    if (subdirectories.length > 0) {
+      for (const subdirectory of subdirectories) {
+        try {
+          const subdirectoryModelList = await utils.getArtifactList(
+            subdirectory,
+            'model',
+          );
+          modelList = modelList.concat(subdirectoryModelList);
+        } catch (err) {
+          console.error(
+            `Error retrieving models from subdirectory ${subdirectory}: ${err}`,
+          );
+        }
+      }
+    }
+
     if (this.options.model) {
       debug(`Model name received from command line: ${this.options.model}`);
 

diff --git a/packages/cli/lib/utils.js b/packages/cli/lib/utils.js
@@ -6,6 +6,7 @@
 'use strict';
 
 const fs = require('node:fs');
+const fsp = require('fs/promises');
 const path = require('node:path');
 const util = require('node:util');
 const stream = require('node:stream');
@@ -383,6 +384,21 @@ exports.getArtifactList = async function (
   });
 };
 
+/**
+ * Retrieves a list of subdirectories within a given directory.
+ *
+ * @param {string} dir The directory path to search for subdirectories.
+ */
+exports.getSubdirectories = async function (dir) {
+  const entries = await fsp.readdir(dir, {withFileTypes: true});
+
+  const subdirectories = entries
+    .filter(entry => entry.isDirectory())
+    .map(entry => path.join(dir, entry.name));
+
+  return subdirectories;
+};
+
 /**
  * Check package.json and dependencies.json to find out versions for generated
  * dependencies