Cover (fix) referenced imports

Author: reggie_7

lib/generator.js

@@ -44,20 +44,28 @@ function makeInitial(srcData, generator) {
     return template
 }
 
-function genImports(properties) {
-    var result = new Set()
+function genImportsSet(properties, target) {
     for(var propName in properties) {
         var props = properties[propName]
         var annotate = props['x-annotate']
-        if (propName !== 'children' && props['x-source'] == 'inject' && annotate) {
-            forEachSizeAnnotation(annotate, _ => {
-                result.add('com.peregrine.model.api.ImageInfo')
-                result.add('java.awt.Dimension')
-            })
+        if (propName !== 'children') {
+            if(props['x-source'] == 'inject' && annotate) {
+                forEachSizeAnnotation(annotate, _ => {console.log(`${propName}: ${JSON.stringify(props)}`)
+                    target.add('com.peregrine.model.api.ImageInfo')
+                    target.add('java.awt.Dimension')
+                })
+            } else if(props['x-form-type'] === 'reference') {
+                genImportsSet(props.properties, target)
+            }
         }
     }
 
-    return Array.from(result).reduce((all, next) => `${all}import ${next};\n`, '')
+    return target
+}
+
+function genImports(properties) {
+    return Array.from(genImportsSet(properties, new Set()))
+        .reduce((all, next) => `${all}import ${next};\n`, '')
 }
 
 function forEachSizeAnnotation(crudeValues, callback) {