feat: improve swift api

Author: okwasniewski

apps/example/swift/App.swift

@@ -11,7 +11,6 @@ let initialState = BrownfieldStore(
 
 @main
 struct MyApp: App {
-
   init() {
     ReactNativeBrownfield.shared.startReactNative {
       print("loaded")
@@ -41,30 +40,25 @@ struct MyApp: App {
   }
 
   struct NativeView: View {
-    @UseStore<BrownfieldStore, User>(\.user, key: BrownfieldStore.storeName) var user
-    @UseStore<BrownfieldStore, Double>(\.counter, key: BrownfieldStore.storeName) var counter
+    @UseStore<BrownfieldStore> var store
 
     var body: some View {
       VStack {
         Text("Native Side")
           .font(.headline)
           .padding(.top)
 
-        Text("User: \(user.name)")
-        Text("Count: \(Int(counter))")
+        Text("User: \(store.state.user.name)")
+        Text("Count: \(Int(store.state.counter))")
 
-        TextField("Name", text: Binding(get: { user.name }, set: { data in
-          $user.set { state in
-            state.user.name = data
-          }
+        TextField("Name", text: Binding(get: { store.state.user.name }, set: { data in
+          store.set { $0.user.name = data }
         }))
         .textFieldStyle(.roundedBorder)
         .padding(.horizontal)
 
         Button("Increment") {
-          $counter.set { state in
-            state.counter += 1
-          }
+          store.set { $0.counter += 1 }
         }
         .buttonStyle(.borderedProminent)
 

packages/brownie/ios/BrownieStore.swift

@@ -6,6 +6,10 @@ extension Notification.Name {
   public static let BrownieStoreUpdated = Notification.Name("BrownieStoreUpdated")
 }
 
+public protocol BrownieStoreProtocol: Codable {
+  static var storeName: String { get }
+}
+
 public struct StoreKey<State: Codable>: EnvironmentKey {
   public static var defaultValue: Store<State> { fatalError("Store not provided") }
 }
@@ -18,22 +22,17 @@ public extension EnvironmentValues {
 
 @MainActor
 @propertyWrapper
-public struct UseStore<State: Codable, Value>: DynamicProperty {
-  private let keyPath: KeyPath<State, Value>
+public struct UseStore<State: BrownieStoreProtocol>: DynamicProperty {
   @StateObject private var store: Store<State>
 
-  public init(_ keyPath: KeyPath<State, Value>, key: String) {
-    self.keyPath = keyPath
+  public init() {
+    let key = State.storeName
     let foundStore = StoreManager.shared.store(key: key, as: State.self)
     guard let foundStore else { fatalError("Store not found for key: \(key)") }
     self._store = StateObject(wrappedValue: foundStore)
   }
 
-  public var wrappedValue: Value {
-    store.state[keyPath: keyPath]
-  }
-
-  public var projectedValue: Store<State> {
+  public var wrappedValue: Store<State> {
     store
   }
 }
@@ -146,4 +145,9 @@ public class Store<State: Codable>: ObservableObject {
   public func get<Value>(_ keyPath: KeyPath<State, Value>) -> Value {
     state[keyPath: keyPath]
   }
+
+  /// Convenience subscript for property access
+  public subscript<Value>(_ keyPath: KeyPath<State, Value>) -> Value {
+    state[keyPath: keyPath]
+  }
 }

packages/brownie/scripts/generators/swift.ts

@@ -67,12 +67,13 @@ export async function generateSwift(
   });
 
   const storeNameExtension = `
-extension ${typeName} {
-  static let storeName = "${name}"
+extension ${typeName}: BrownieStoreProtocol {
+  public static let storeName = "${name}"
 }
 `;
 
-  const swiftOutput = lines.join('\n') + storeNameExtension;
+  const swiftOutput =
+    'import Brownie\n\n' + lines.join('\n') + storeNameExtension;
   const absoluteOutputPath = path.resolve(process.cwd(), outputPath);
   const outputDir = path.dirname(absoluteOutputPath);