Add ObservedObject macro if store_prefix (#863)

* Add ObservedObject macro if store_prefix

* Add store_property_wrapper

Author: tinder-garricnahapetian

Sources/NodesGenerator/Config.swift

@@ -61,6 +61,7 @@ public struct Config: Codable, Equatable {
     public var isPeripheryCommentEnabled: Bool
 
     public var storePrefix: String { isObservableStoreEnabled ? "Observable" : "" }
+    public var storePropertyWrapper: String { isObservableStoreEnabled ? "@ObservedObject" : "" }
     public var isNimbleEnabled: Bool { baseTestImports.contains("Nimble") }
 
     public init(

Sources/NodesGenerator/PresetGenerator.swift

@@ -94,6 +94,7 @@ public final class PresetGenerator {
                 contextGenericTypes: config.contextGenericTypes,
                 workerGenericTypes: config.workerGenericTypes,
                 storePrefix: config.storePrefix,
+                storePropertyWrapper: config.storePropertyWrapper,
                 isPreviewProviderEnabled: config.isPreviewProviderEnabled,
                 isPeripheryCommentEnabled: config.isPeripheryCommentEnabled,
                 isNimbleEnabled: config.isNimbleEnabled

Sources/NodesGenerator/Resources/Stencils/ViewController-SwiftUI.stencil

@@ -52,7 +52,12 @@ internal struct {{ node_name }}View {
     internal weak var receiver: {{ node_name }}Receiver?
 
     /// The Store instance.
-    private let store: Any{{ store_prefix }}ViewStateStore<{{ node_name }}ViewState>
+    {% if store_property_wrapper %}
+    {{ store_property_wrapper }}
+    private var store: Any{{ store_prefix }}ViewStateStore<{{ node_name }}ViewState>
+    {% else %}
+    private let store: AnyViewStateStore<{{ node_name }}ViewState>
+    {% endif %}
 
     /// The initializer.
     ///

Sources/NodesGenerator/StencilContexts/NodeStencilContext.swift

@@ -2,6 +2,8 @@
 //  Copyright © 2021 Tinder (Match Group, LLC)
 //
 
+// swiftlint:disable file_length
+
 // swiftlint:disable:next type_body_length
 public struct NodeStencilContext: StencilContext {
 
@@ -47,6 +49,7 @@ public struct NodeStencilContext: StencilContext {
     private let contextGenericTypes: [String]
     private let workerGenericTypes: [String]
     private let storePrefix: String
+    private let storePropertyWrapper: String
     private let isPreviewProviderEnabled: Bool
     private let isPeripheryCommentEnabled: Bool
     private let isNimbleEnabled: Bool
@@ -96,6 +99,7 @@ public struct NodeStencilContext: StencilContext {
             "context_generic_types": contextGenericTypes,
             "worker_generic_types": workerGenericTypes,
             "store_prefix": storePrefix,
+            "store_property_wrapper": storePropertyWrapper,
             "is_preview_provider_enabled": isPreviewProviderEnabled,
             "is_periphery_comment_enabled": isPeripheryCommentEnabled,
             "is_nimble_enabled": isNimbleEnabled
@@ -146,6 +150,7 @@ public struct NodeStencilContext: StencilContext {
         contextGenericTypes: [String],
         workerGenericTypes: [String],
         storePrefix: String,
+        storePropertyWrapper: String,
         isPreviewProviderEnabled: Bool,
         isPeripheryCommentEnabled: Bool,
         isNimbleEnabled: Bool
@@ -194,12 +199,14 @@ public struct NodeStencilContext: StencilContext {
             contextGenericTypes: contextGenericTypes,
             workerGenericTypes: workerGenericTypes,
             storePrefix: storePrefix,
+            storePropertyWrapper: storePropertyWrapper,
             isPreviewProviderEnabled: isPreviewProviderEnabled,
             isPeripheryCommentEnabled: isPeripheryCommentEnabled,
             isNimbleEnabled: isNimbleEnabled
         )
     }
 
+    // swiftlint:disable:next function_body_length
     public init(
         preset: Preset,
         fileHeader: String,
@@ -238,6 +245,7 @@ public struct NodeStencilContext: StencilContext {
         contextGenericTypes: [String],
         workerGenericTypes: [String],
         storePrefix: String,
+        storePropertyWrapper: String,
         isPreviewProviderEnabled: Bool,
         isPeripheryCommentEnabled: Bool,
         isNimbleEnabled: Bool
@@ -288,6 +296,7 @@ public struct NodeStencilContext: StencilContext {
             contextGenericTypes: contextGenericTypes,
             workerGenericTypes: workerGenericTypes,
             storePrefix: storePrefix,
+            storePropertyWrapper: storePropertyWrapper,
             isPreviewProviderEnabled: isPreviewProviderEnabled,
             isPeripheryCommentEnabled: isPeripheryCommentEnabled,
             isNimbleEnabled: isNimbleEnabled
@@ -338,6 +347,7 @@ public struct NodeStencilContext: StencilContext {
         contextGenericTypes: [String],
         workerGenericTypes: [String],
         storePrefix: String,
+        storePropertyWrapper: String,
         isPreviewProviderEnabled: Bool,
         isPeripheryCommentEnabled: Bool,
         isNimbleEnabled: Bool
@@ -386,8 +396,11 @@ public struct NodeStencilContext: StencilContext {
         self.contextGenericTypes = contextGenericTypes
         self.workerGenericTypes = workerGenericTypes
         self.storePrefix = storePrefix
+        self.storePropertyWrapper = storePropertyWrapper
         self.isPreviewProviderEnabled = isPreviewProviderEnabled
         self.isPeripheryCommentEnabled = isPeripheryCommentEnabled
         self.isNimbleEnabled = isNimbleEnabled
     }
 }
+
+// swiftlint:enable file_length

Sources/NodesGenerator/XcodeTemplatePermutations/NodeXcodeTemplatePermutation.swift

@@ -58,6 +58,7 @@ internal struct NodeXcodeTemplatePermutation: XcodeTemplatePermutation {
             contextGenericTypes: config.contextGenericTypes,
             workerGenericTypes: config.workerGenericTypes,
             storePrefix: config.storePrefix,
+            storePropertyWrapper: config.storePropertyWrapper,
             isPreviewProviderEnabled: config.isPreviewProviderEnabled,
             isPeripheryCommentEnabled: config.isPeripheryCommentEnabled,
             isNimbleEnabled: config.isNimbleEnabled

Tests/NodesGeneratorTests/Support/TestFactories.swift

@@ -105,6 +105,7 @@ extension TestFactories {
             contextGenericTypes: .mock(with: "contextGenericType", count: mockCount),
             workerGenericTypes: .mock(with: "workerGenericType", count: mockCount),
             storePrefix: mockCount > 0 ? "<storePrefix>" : "",
+            storePropertyWrapper: mockCount > 0 ? "<storePropertyWrapper>" : "",
             isPreviewProviderEnabled: mockCount > 0,
             isPeripheryCommentEnabled: mockCount > 0,
             isNimbleEnabled: mockCount > 0
@@ -153,6 +154,7 @@ extension TestFactories {
             contextGenericTypes: .mock(with: "contextGenericType", count: mockCount),
             workerGenericTypes: .mock(with: "workerGenericType", count: mockCount),
             storePrefix: mockCount > 0 ? "<storePrefix>" : "",
+            storePropertyWrapper: mockCount > 0 ? "<storePropertyWrapper>" : "",
             isPreviewProviderEnabled: mockCount > 0,
             isPeripheryCommentEnabled: mockCount > 0,
             isNimbleEnabled: mockCount > 0

Tests/NodesGeneratorTests/__Snapshots__/StencilContextsTests/testNodePresetRootStencilContext.1.txt

@@ -1,4 +1,4 @@
-▿ 46 key/value pairs
+▿ 47 key/value pairs
   ▿ (2 elements)
     - key: "analytics_imports"
     ▿ value: 1 element
@@ -111,6 +111,9 @@
   ▿ (2 elements)
     - key: "store_prefix"
     - value: "<storePrefix>"
+  ▿ (2 elements)
+    - key: "store_property_wrapper"
+    - value: "<storePropertyWrapper>"
   ▿ (2 elements)
     - key: "view_controllable_flow_type"
     - value: "<viewControllableFlowType>"

Tests/NodesGeneratorTests/__Snapshots__/StencilContextsTests/testNodeStencilContext.1.txt

@@ -1,4 +1,4 @@
-▿ 46 key/value pairs
+▿ 47 key/value pairs
   ▿ (2 elements)
     - key: "analytics_imports"
     ▿ value: 1 element
@@ -111,6 +111,9 @@
   ▿ (2 elements)
     - key: "store_prefix"
     - value: "<storePrefix>"
+  ▿ (2 elements)
+    - key: "store_property_wrapper"
+    - value: "<storePropertyWrapper>"
   ▿ (2 elements)
     - key: "view_controllable_flow_type"
     - value: "<viewControllableFlowType>"

Tests/NodesGeneratorTests/__Snapshots__/StencilContextsTests/testNodeStencilContext_withPlugin.1.txt

@@ -1,4 +1,4 @@
-▿ 46 key/value pairs
+▿ 47 key/value pairs
   ▿ (2 elements)
     - key: "analytics_imports"
     ▿ value: 1 element
@@ -113,6 +113,9 @@
   ▿ (2 elements)
     - key: "store_prefix"
     - value: "<storePrefix>"
+  ▿ (2 elements)
+    - key: "store_property_wrapper"
+    - value: "<storePropertyWrapper>"
   ▿ (2 elements)
     - key: "view_controllable_flow_type"
     - value: "<viewControllableFlowType>"

Tests/NodesGeneratorTests/__Snapshots__/StencilRendererTests/testRenderNode.ViewController-AppKitSwiftUI-mockCount-1.txt

@@ -38,7 +38,8 @@ internal struct <nodeName>View {
     internal weak var receiver: <nodeName>Receiver?
 
     /// The Store instance.
-    private let store: Any<storePrefix>ViewStateStore<<nodeName>ViewState>
+    <storePropertyWrapper>
+    private var store: Any<storePrefix>ViewStateStore<<nodeName>ViewState>
 
     /// The initializer.
     ///