Add config for interface imports (#908)

Author: tinder-cfuller

Sources/NodesGenerator/Config.swift

@@ -38,6 +38,7 @@ public struct Config: Codable, Equatable {
     public var dependencyInjectionImports: Set<String>
     public var builderImports: Set<String>
     public var flowImports: Set<String>
+    public var interfaceImports: Set<String>
     public var pluginListImports: Set<String>
     public var viewControllerImports: Set<String>
     public var dependencies: [Variable]
@@ -94,6 +95,7 @@ extension Config {
         dependencyInjectionImports = ["NeedleFoundation"]
         builderImports = []
         flowImports = []
+        interfaceImports = []
         pluginListImports = []
         viewControllerImports = []
         dependencies = []
@@ -168,6 +170,9 @@ extension Config {
         flowImports =
             (try? decoder.decode(CodingKeys.flowImports))
             ?? defaults.flowImports
+        interfaceImports =
+            (try? decoder.decode(CodingKeys.interfaceImports))
+            ?? defaults.interfaceImports
         pluginListImports =
             (try? decoder.decode(CodingKeys.pluginListImports))
             ?? defaults.pluginListImports

Sources/NodesGenerator/StencilTemplate.swift

@@ -287,6 +287,7 @@ public enum StencilTemplate: CustomStringConvertible, Equatable, Sendable {
         case .interface:
             config.baseImports
                 .union(["Nodes"])
+                .union(config.interfaceImports)
         case .plugin:
             config.baseImports
                 .union(["Nodes"])

Tests/NodesGeneratorTests/ConfigTests.swift

@@ -157,6 +157,9 @@ final class ConfigTests: XCTestCase, TestFactories {
         flowImports:
           - <flowImports-1>
           - <flowImports-2>
+        interfaceImports:
+          - <interfaceImports-1>
+          - <interfaceImports-2>
         pluginListImports:
           - <pluginListImports-1>
           - <pluginListImports-2>

Tests/NodesGeneratorTests/StencilTemplateTests.swift

@@ -341,6 +341,7 @@ final class StencilTemplateTests: XCTestCase, TestFactories {
                 case .interface:
                     expect(imports) == [
                         "<baseImport>",
+                        "<interfaceImport>",
                         "Nodes"
                     ]
                 case .plugin:
@@ -466,6 +467,7 @@ final class StencilTemplateTests: XCTestCase, TestFactories {
             case .interface:
                 expect(imports) == [
                     "<baseImport>",
+                    "<interfaceImport>",
                     "Nodes"
                 ]
             case .plugin:

Tests/NodesGeneratorTests/Support/TestFactories.swift

@@ -34,6 +34,7 @@ extension TestFactories {
         config.dependencyInjectionImports = ["<dependencyInjectionImport>"]
         config.builderImports = ["<builderImport>"]
         config.flowImports = ["<flowImport>"]
+        config.interfaceImports = ["<interfaceImport>"]
         config.pluginListImports = ["<pluginListImport>"]
         config.viewControllerImports = ["<viewControllerImport>"]
         config.dependencies = [Variable(name: "<dependencyName>", type: "<dependencyType>")]

Tests/NodesGeneratorTests/__Snapshots__/ConfigTests/testConfig.1.txt

@@ -38,6 +38,9 @@
     ▿ Variable
       - name: "<flowProperties-name-2>"
       - type: "<flowProperties-type-2>"
+  ▿ interfaceImports: 2 members
+    - "<interfaceImports-1>"
+    - "<interfaceImports-2>"
   - isObservableStoreEnabled: false
   - isPeripheryCommentEnabled: true
   - isPreviewProviderEnabled: true

Tests/NodesGeneratorTests/__Snapshots__/ConfigTests/testConfigWithEmptyFileContents.1.txt

@@ -12,6 +12,7 @@
     - "NeedleFoundation"
   - flowImports: 0 members
   - flowProperties: 0 elements
+  - interfaceImports: 0 members
   - isObservableStoreEnabled: false
   - isPeripheryCommentEnabled: false
   - isPreviewProviderEnabled: false

Tests/NodesGeneratorTests/__Snapshots__/ConfigTests/testDecodingFromEmptyString.1.txt

@@ -12,6 +12,7 @@
     - "NeedleFoundation"
   - flowImports: 0 members
   - flowProperties: 0 elements
+  - interfaceImports: 0 members
   - isObservableStoreEnabled: false
   - isPeripheryCommentEnabled: false
   - isPreviewProviderEnabled: false

Tests/NodesGeneratorTests/__Snapshots__/ConfigTests/testDecodingWithEmptyArray.1.txt

@@ -12,6 +12,7 @@
     - "NeedleFoundation"
   - flowImports: 0 members
   - flowProperties: 0 elements
+  - interfaceImports: 0 members
   - isObservableStoreEnabled: false
   - isPeripheryCommentEnabled: false
   - isPreviewProviderEnabled: false

Tests/NodesGeneratorTests/__Snapshots__/XcodeTemplatePermutationTests/testNodeViewInjectedXcodeTemplatePermutation.1.txt

@@ -55,8 +55,9 @@
           - value: "<flowPropertyType>"
     ▿ flowTestsImports: 1 element
       - "<baseTestImport>"
-    ▿ interfaceImports: 2 elements
+    ▿ interfaceImports: 3 elements
       - "<baseImport>"
+      - "<interfaceImport>"
       - "Nodes"
     - isNimbleEnabled: false
     - isPeripheryCommentEnabled: true