Add imports for uiFramework

Sources/XcodeTemplateGeneratorLibrary/StencilTemplate.swift

@@ -130,4 +130,27 @@ public enum StencilTemplate: Equatable, CaseIterable, CustomStringConvertible {
             return description.appending(variation.rawValue)
         }
     }
+
+    internal func imports(for uiFramework: UIFramework, config: XcodeTemplates.Config) -> Set<String> {
+        switch self {
+        case .analytics, .builder, .context, .flow, .plugin, .pluginList, .state, .viewState, .worker:
+            return imports(config: config)
+        case .viewController:
+            return imports(config: config).union([uiFramework.import])
+        }
+    }
+
+    internal func imports(config: XcodeTemplates.Config) -> Set<String> {
+        let imports: Set<String> = config.baseImports.union(["Nodes"])
+        switch self {
+        case .analytics, .flow, .state, .viewState:
+            return imports
+        case .builder:
+            return imports.union(config.reactiveImports).union(config.dependencyInjectionImports)
+        case .context, .viewController, .worker:
+            return imports.union(config.reactiveImports)
+        case .plugin, .pluginList:
+            return imports.union(config.dependencyInjectionImports)
+        }
+    }
 }

Tests/XcodeTemplateGeneratorLibraryTests/StencilTemplateTests.swift

@@ -9,7 +9,7 @@ import Nimble
 @testable import XcodeTemplateGeneratorLibrary
 import XCTest
 
-final class StencilTemplateTests: XCTestCase {
+final class StencilTemplateTests: XCTestCase, TestFactories {
 
     func testVariationRawValue() {
         StencilTemplate.Variation.allCases.forEach { variation in
@@ -57,6 +57,7 @@ final class StencilTemplateTests: XCTestCase {
         }
     }
 
+    // swiftlint:disable:next cyclomatic_complexity
     func testName() {
         for stencilTemplate in StencilTemplate.allCases {
             let name: String = stencilTemplate.name
@@ -85,6 +86,7 @@ final class StencilTemplateTests: XCTestCase {
         }
     }
 
+    // swiftlint:disable:next cyclomatic_complexity
     func testFilename() {
         for stencilTemplate in StencilTemplate.allCases {
             let filename: String = stencilTemplate.filename
@@ -135,4 +137,37 @@ final class StencilTemplateTests: XCTestCase {
             .worker
         ]
     }
+
+    // swiftlint:disable:next cyclomatic_complexity
+    func testImports() {
+        let config: Config = givenConfig()
+        for stencilTemplate in StencilTemplate.allCases {
+            for uiFramework in config.uiFrameworks {
+                let imports: Set<String> = stencilTemplate.imports(for: uiFramework, config: config)
+                let uiFrameworkImport: String
+                switch uiFramework.kind {
+                case .appKit:
+                    uiFrameworkImport = "AppKit"
+                case .uiKit:
+                    uiFrameworkImport = "UIKit"
+                case .swiftUI:
+                    uiFrameworkImport = "SwiftUI"
+                case .custom:
+                    uiFrameworkImport = "<uiFrameworkImport>"
+                }
+                switch stencilTemplate {
+                case .analytics, .flow, .state, .viewState:
+                    expect(imports) == ["Nodes", "<baseImports>"]
+                case .builder:
+                    expect(imports) == ["Nodes", "<baseImports>", "<reactiveImports>", "<dependencyInjectionImports>"]
+                case .context, .worker:
+                    expect(imports) == ["Nodes", "<baseImports>", "<reactiveImports>"]
+                case .viewController:
+                    expect(imports) == ["Nodes", "<baseImports>", "<reactiveImports>", uiFrameworkImport]
+                case .plugin, .pluginList:
+                    expect(imports) == ["Nodes", "<baseImports>", "<dependencyInjectionImports>"]
+                }
+            }
+        }
+    }
 }