From 158b7063ca242002312f870f8deea219b6b7f051 Mon Sep 17 00:00:00 2001
From: Joseph Duffy <commits@josephduffy.co.uk>
Date: Thu, 16 May 2019 17:00:11 +0100
Subject: [PATCH] Move update listeners to PartialBuilder

---
 Source/Partial.swift            |  5 ---
 Source/PartialBuilder.swift     | 77 ++++++++++++++++++++++++++++++++-
 Source/PartialConvertible.swift |  2 -
 3 files changed, 75 insertions(+), 9 deletions(-)

diff --git a/Source/Partial.swift b/Source/Partial.swift
index f1d96ed3..c9545da1 100644
--- a/Source/Partial.swift
+++ b/Source/Partial.swift
@@ -11,8 +11,6 @@ public struct Partial<Wrapped>: CustomStringConvertible, CustomDebugStringConver
 
     internal var backingValue: Wrapped? = nil
 
-    internal var updateHandlers: [PartialKeyPath<Wrapped>: [(Any?) -> Void]] = [:]
-
     public init(backingValue: Wrapped? = nil) {
         self.backingValue = backingValue
     }
@@ -47,7 +45,6 @@ public struct Partial<Wrapped>: CustomStringConvertible, CustomDebugStringConver
 
     public mutating func set<Value>(value: Value, for key: KeyPath<Wrapped, Value>) {
         values[key] = value
-        updateHandlers[key]?.forEach { $0(value as Any) }
     }
 
     public mutating func set<Value>(value: Value?, for key: KeyPath<Wrapped, Value?>) {
@@ -58,12 +55,10 @@ public struct Partial<Wrapped>: CustomStringConvertible, CustomDebugStringConver
          a `backingValue` is set and a key is explicitly set to `nil`
          */
         values.updateValue(value, forKey: key)
-        updateHandlers[key]?.forEach { $0(value as Any) }
     }
 
     public mutating func removeValue<Value>(for key: KeyPath<Wrapped, Value>) {
         values.removeValue(forKey: key)
-        updateHandlers[key]?.forEach { $0(nil) }
     }
 
     public subscript<Value>(key: KeyPath<Wrapped, Value>) -> Value? {
diff --git a/Source/PartialBuilder.swift b/Source/PartialBuilder.swift
index 95bcd365..3aefa376 100644
--- a/Source/PartialBuilder.swift
+++ b/Source/PartialBuilder.swift
@@ -1,6 +1,8 @@
-public class PartialBuilder<Wrapped> {
+public final class PartialBuilder<Wrapped> {
     
-    public var partial: Partial<Wrapped>
+    public private(set) var partial: Partial<Wrapped>
+    
+    private var updateHandlers: [PartialKeyPath<Wrapped>: [(Any?) -> Void]] = [:]
     
     public init(partial: Partial<Wrapped> = Partial<Wrapped>()) {
         self.partial = partial
@@ -10,6 +12,77 @@ public class PartialBuilder<Wrapped> {
         partial = Partial(backingValue: backingValue)
     }
     
+    public func value<Value>(for key: KeyPath<Wrapped, Value>) throws -> Value {
+        return try partial.value(for: key)
+    }
+    
+    public func value<Value>(for key: KeyPath<Wrapped, Value?>) throws -> Value {
+        return try partial.value(for: key)
+    }
+    
+    public func value<Value>(for key: KeyPath<Wrapped, Value>) throws -> Value where Value: PartialConvertible {
+        return try partial.value(for: key)
+    }
+    
+    public func value<Value>(for key: KeyPath<Wrapped, Value?>) throws -> Value where Value: PartialConvertible {
+        return try partial.value(for: key)
+    }
+    
+    public func partialValue<Value>(for key: KeyPath<Wrapped, Value>) throws -> Partial<Value> where Value: PartialConvertible {
+        return try partial.partialValue(for: key)
+    }
+    
+    public func partialValue<Value>(for key: KeyPath<Wrapped, Value?>) throws -> Partial<Value> where Value: PartialConvertible {
+        return try partial.partialValue(for: key)
+    }
+    
+    public func set<Value>(value: Value, for key: KeyPath<Wrapped, Value>) {
+        partial.set(value: value, for: key)
+        updateHandlers[key]?.forEach { $0(value as Any) }
+    }
+    
+    public func set<Value>(value: Value?, for key: KeyPath<Wrapped, Value?>) {
+        partial.set(value: value, for: key)
+        updateHandlers[key]?.forEach { $0(value as Any) }
+    }
+    
+    public func set<Value>(value: Partial<Value>, for key: KeyPath<Wrapped, Value>) where Value: PartialConvertible {
+        partial.set(value: value, for: key)
+        updateHandlers[key]?.forEach { $0(value as Any) }
+    }
+    
+    public func removeValue<Value>(for key: KeyPath<Wrapped, Value>) {
+        partial.removeValue(for: key)
+        updateHandlers[key]?.forEach({ $0(nil) })
+    }
+    
+    public subscript<Value>(key: KeyPath<Wrapped, Value>) -> Value? {
+        get {
+            return partial[key]
+        }
+        set {
+            partial.removeValue(for: key)
+        }
+    }
+    
+    public subscript<Value>(key: KeyPath<Wrapped, Value?>) -> Value? {
+        get {
+            return partial[key]
+        }
+        set {
+            set(value: newValue, for: key)
+        }
+    }
+    
+    public subscript<Value>(key: KeyPath<Wrapped, Value>) -> Partial<Value> where Value: PartialConvertible {
+        get {
+            return partial[key]
+        }
+        set {
+            set(value: newValue, for: key)
+        }
+    }
+    
 }
 
 extension PartialBuilder where Wrapped: PartialConvertible {
diff --git a/Source/PartialConvertible.swift b/Source/PartialConvertible.swift
index 3351e09c..a2500dcf 100644
--- a/Source/PartialConvertible.swift
+++ b/Source/PartialConvertible.swift
@@ -80,12 +80,10 @@ extension Partial {
     
     public mutating func set<Value>(value: Partial<Value>, for key: KeyPath<Wrapped, Value>) where Value: PartialConvertible {
         values[key] = value
-        updateHandlers[key]?.forEach { $0(value as Any) }
     }
     
     public mutating func set<Value>(value: Partial<Value>, for key: KeyPath<Wrapped, Value?>) where Value: PartialConvertible {
         values[key] = value
-        updateHandlers[key]?.forEach { $0(value as Any) }
     }
     
     public subscript<Value>(key: KeyPath<Wrapped, Value>) -> Partial<Value> where Value: PartialConvertible {