hook up imports from plugins in generator

hackthehackerman · Feb 17, 2019 · 0ed78d2 · 0ed78d2
1 parent af8dc02
commit 0ed78d2
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/Sources/ImmutableSwift/generating/Generator.swift b/Sources/ImmutableSwift/generating/Generator.swift
@@ -8,6 +8,13 @@ class Generator {
             output = output + Generator.ImportsToString(datamodel.imports!) + "\n"
         }
 
+        if pluginList != nil {
+            let importsFromPlugins = Generator.GenerateImportsWithPluginList(pluginList!)
+            if importsFromPlugins.count > 0{
+                output = output  + importsFromPlugins + "\n"
+            }
+        }
+
         // optional accesscontrol
         if datamodel.accessControlLevel != nil {
             output = output + Generator.AccessControlLevelToString(datamodel.accessControlLevel!) + " "
@@ -76,6 +83,18 @@ class Generator {
         return output
     }
 
+    static func GenerateImportsWithPluginList(_ pluginList: PluginList) -> String {
+        var imports: Set<String> = []
+        for pluginName in pluginList.plugins {
+            if Plugins.PLUGIN_MAP[pluginName] != nil {
+                for importFromPlugin in Plugins.PLUGIN_MAP[pluginName]!.imports() {
+                    imports.insert("import "+importFromPlugin)
+                }
+            }
+        }
+        return imports.joined(separator: "\n") + "\n"
+    }
+
     static func AccessControlLevelToString(_ accessControl: AccessControl) -> String {
         switch accessControl {
         case AccessControl.levelPublic: