Simplify @published implementation using Property

DeclarativeHub · Nov 21, 2024 · 5b0360c · 5b0360c
1 parent 57a902a
commit 5b0360c
Showing 1 changed file with 9 additions and 19 deletions.
diff --git a/Sources/Published.swift b/Sources/Published.swift
@@ -11,49 +11,39 @@
 @propertyWrapper
 public struct Published<Value> {
 
-    private var value: Value
-    private var publisher: Publisher?
+    private let publisher: Publisher
     private let willChangeSubject = PassthroughSubject<Void, Never>()
 
     public init(wrappedValue: Value) {
-        value = wrappedValue
+        publisher = Publisher(wrappedValue)
     }
 
     /// A publisher for properties used with the `@Published` attribute.
     public struct Publisher: SignalProtocol {
         public typealias Element = Value
         public typealias Error = Never
 
-        fileprivate let didChangeSubject: ReplayOneSubject<Value, Never>
+        fileprivate let property: Property<Value>
 
         public func observe(with observer: @escaping (Signal<Value, Never>.Event) -> Void) -> Disposable {
-            self.didChangeSubject.observe(with: observer)
+            self.property.observe(with: observer)
         }
 
         fileprivate init(_ output: Element) {
-            self.didChangeSubject = ReplayOneSubject()
-            self.didChangeSubject.send(output)
+            self.property = Property(output)
         }
     }
 
     public var wrappedValue: Value {
-        get { self.value }
-        set {
+        get { self.publisher.property.value }
+        nonmutating set {
             self.willChangeSubject.send()
-            self.value = newValue
-            self.publisher?.didChangeSubject.send(newValue)
+            self.publisher.property.value = newValue
         }
     }
 
     public var projectedValue: Publisher {
-        mutating get {
-            if let publisher = publisher {
-                return publisher
-            }
-            let publisher = Publisher(value)
-            self.publisher = publisher
-            return publisher
-        }
+        get { publisher }
     }
 }