Fix the CaseStudies (UIKit)

Author: Czajnikowski

Examples/CaseStudies/UIKitCaseStudies/RootViewController.swift

@@ -12,58 +12,62 @@ struct CaseStudy {
   }
 }
 
-let dataSource: [CaseStudy] = [
-  CaseStudy(
-    title: "Basics",
-    viewController: CounterViewController(
-      store: Store(
-        initialState: CounterState(),
-        reducer: counterReducer,
-        environment: CounterEnvironment()
-      )
-    )
-  ),
-  CaseStudy(
-    title: "Lists",
-    viewController: CountersTableViewController(
-      store: Store(
-        initialState: CounterListState(
-          counters: [
-            CounterState(),
-            CounterState(),
-            CounterState(),
-          ]
-        ),
-        reducer: counterListReducer,
-        environment: CounterListEnvironment()
-      )
-    )
-  ),
-  CaseStudy(
-    title: "Navigate and load",
-    viewController: EagerNavigationViewController(
-      store: Store(
-        initialState: EagerNavigationState(),
-        reducer: eagerNavigationReducer,
-        environment: EagerNavigationEnvironment(
-          mainQueue: .main
+extension RootViewController {
+  var dataSource: [CaseStudy] {
+    [
+      CaseStudy(
+        title: "Basics",
+        viewController: CounterViewController(
+          store: Store(
+            initialState: CounterState(),
+            reducer: counterReducer,
+            environment: CounterEnvironment()
+          )
         )
-      )
-    )
-  ),
-  CaseStudy(
-    title: "Load then navigate",
-    viewController: LazyNavigationViewController(
-      store: Store(
-        initialState: LazyNavigationState(),
-        reducer: lazyNavigationReducer,
-        environment: LazyNavigationEnvironment(
-          mainQueue: .main
+      ),
+      CaseStudy(
+        title: "Lists",
+        viewController: CountersTableViewController(
+          store: Store(
+            initialState: CounterListState(
+              counters: [
+                CounterState(),
+                CounterState(),
+                CounterState(),
+              ]
+            ),
+            reducer: counterListReducer,
+            environment: CounterListEnvironment()
+          )
         )
-      )
-    )
-  ),
-]
+      ),
+      CaseStudy(
+        title: "Navigate and load",
+        viewController: EagerNavigationViewController(
+          store: Store(
+            initialState: EagerNavigationState(),
+            reducer: eagerNavigationReducer,
+            environment: EagerNavigationEnvironment(
+              mainQueue: .main
+            )
+          )
+        )
+      ),
+      CaseStudy(
+        title: "Load then navigate",
+        viewController: LazyNavigationViewController(
+          store: Store(
+            initialState: LazyNavigationState(),
+            reducer: lazyNavigationReducer,
+            environment: LazyNavigationEnvironment(
+              mainQueue: .main
+            )
+          )
+        )
+      ),
+    ]
+  }
+}
 
 final class RootViewController: UITableViewController {
   override func viewDidLoad() {