fix: remove non-object generated classes (#62)

asyncapi · Jun 17, 2020 · 53de017 · 53de017
1 parent d98713c
commit 53de017
Show file tree

Hide file tree

Showing 5 changed files with 18 additions and 1 deletion.
diff --git a/hooks/inverseOperations.js → hooks/01_inverseOperations.js b/hooks/inverseOperations.js → hooks/01_inverseOperations.js
diff --git a/hooks/removeNotRelevantParts.js → hooks/02_removeNotRelevantParts.js b/hooks/removeNotRelevantParts.js → hooks/02_removeNotRelevantParts.js
diff --git a/hooks/03_removeNonObjectSchemas.js b/hooks/03_removeNonObjectSchemas.js
@@ -0,0 +1,17 @@
+const fs = require('fs');
+const path = require('path');
+
+module.exports = {
+    'generate:after': generator => {
+        const asyncapi = generator.asyncapi;
+        const schemas = asyncapi.allSchemas();
+
+        for (let [key, value] of schemas) {
+            if (value.type() !== 'object') {
+                try {
+                    fs.unlinkSync(path.resolve(generator.targetDir, `src/main/java/com/asyncapi/model/${key}.java`));
+                } catch (e) {}
+            }
+        }
+    }
+};
diff --git a/hooks/fixModelFileNames.js → hooks/04_fixModelFileNames.js b/hooks/fixModelFileNames.js → hooks/04_fixModelFileNames.js
@@ -15,7 +15,7 @@ module.exports = {
             }
         }
         for (let [key, value] of schemas) {
-            if (_.upperFirst(key) !== key) {
+            if (_.upperFirst(key) !== key && value.type() === 'object') {
                 fs.renameSync(path.resolve(generator.targetDir, `src/main/java/com/asyncapi/model/${key}.java`),
                     path.resolve(generator.targetDir, `src/main/java/com/asyncapi/model/${_.upperFirst(key)}.java`));
             }

diff --git a/hooks/packageProcessor.js → hooks/05_packageProcessor.js b/hooks/packageProcessor.js → hooks/05_packageProcessor.js