Tunnel creation: Compute public key from private key as the text is being edited

Signed-off-by: Roopesh Chander <roop@roopc.net>

1 files changed, 19 insertions, 0 deletions

diff --git a/WireGuard/WireGuard/UI/iOS/TunnelEditTableViewController.swift b/WireGuard/WireGuard/UI/iOS/TunnelEditTableViewController.swift
index 89446e9..f203e09 100644
--- a/WireGuard/WireGuard/UI/iOS/TunnelEditTableViewController.swift
+++ b/WireGuard/WireGuard/UI/iOS/TunnelEditTableViewController.swift
@@ -160,6 +160,17 @@ extension TunnelEditTableViewController {
                 cell.onValueChanged = { [weak interfaceData] value in
                     interfaceData?[field] = value
                 }
+                // Compute public key live
+                if (field == .privateKey) {
+                    cell.onValueBeingEdited = { [weak self, weak interfaceData] value in
+                        if let interfaceData = interfaceData, let s = self {
+                            interfaceData[.privateKey] = value
+                            if let row = s.interfaceFieldsBySection[section].firstIndex(of: .publicKey) {
+                                s.tableView.reloadRows(at: [IndexPath(row: row, section: section)], with: .none)
+                            }
+                        }
+                    }
+                }
                 return cell
             }
         } else if ((numberOfPeers > 0) && (section < (numberOfInterfaceSections + numberOfPeers * numberOfPeerSections))) {
@@ -289,6 +300,7 @@ class TunnelsEditTableViewKeyValueCell: UITableViewCell {
     }
 
     var onValueChanged: ((String) -> Void)? = nil
+    var onValueBeingEdited: ((String) -> Void)? = nil
 
     let keyLabel: UILabel
     let valueTextField: UITextField
@@ -352,6 +364,13 @@ extension TunnelsEditTableViewKeyValueCell: UITextFieldDelegate {
             onValueChanged(textField.text ?? "")
         }
     }
+    func textField(_ textField: UITextField, shouldChangeCharactersIn range: NSRange, replacementString string: String) -> Bool {
+        if let onValueBeingEdited = onValueBeingEdited {
+            let modifiedText = ((textField.text ?? "") as NSString).replacingCharacters(in: range, with: string)
+            onValueBeingEdited(modifiedText)
+        }
+        return true
+    }
 }
 
 class TunnelsEditTableViewButtonCell: UITableViewCell {