Expose _ViewControllerKey (#40)

eure · Apr 16, 2024 · d5897bd · d5897bd
1 parent fcb306f
commit d5897bd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/Sources/StorybookKit/Internals/TargetViewControllerKey.swift b/Sources/StorybookKit/Internals/TargetViewControllerKey.swift
@@ -1,7 +1,7 @@
 import SwiftUI
 
-enum _ViewControllerKey: EnvironmentKey {
-  static var defaultValue: UIViewController?
+public enum _ViewControllerKey: EnvironmentKey {
+  public static var defaultValue: UIViewController?
 }
 
 enum TestKey: EnvironmentKey {

diff --git a/Sources/StorybookKit/StorybookDisplayRootView.swift b/Sources/StorybookKit/StorybookDisplayRootView.swift
@@ -26,8 +26,8 @@ public struct BookActionHosting<Content: View>: View {
 
   private let content: Content
 
-  public init(_ content: Content) {
-    self.content = content
+  public init(@ViewBuilder content: () -> Content) {
+    self.content = content()
   }
 
   public var body: some View {
@@ -186,7 +186,7 @@ final class _ViewController<Content: View>: UIViewController {
 #if DEBUG
 
 #Preview {
-  BookActionHosting(BookAction(title: "Hello", action: { vc in }))
+  BookActionHosting { BookAction(title: "Hello", action: { vc in }) }
 }
 
 #endif