Make it possible to manually construct a ThresholdKey

hiero-ledger · Nov 6, 2019 · 62a272e · 62a272e
1 parent ebb7386
commit 62a272e
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/Sources/Hedera/crypto/ThresholdKey.swift b/Sources/Hedera/crypto/ThresholdKey.swift
@@ -2,6 +2,12 @@ public final class ThresholdKey: PublicKey {
     let threshold: UInt32
     var keys: [PublicKey]
 
+    init(threshold: UInt32, keys: [PublicKey] = []) {
+        self.threshold = threshold
+        self.keys = keys
+        super.init()
+    }
+
     required init?(_ proto: Proto_Key) {
         guard proto.thresholdKey.hasKeys else { return nil }
         threshold = proto.thresholdKey.threshold
@@ -27,4 +33,16 @@ public final class ThresholdKey: PublicKey {
 
         return proto
     }
+
+    @discardableResult
+    public func addKey(_ key: PublicKey) -> Self {
+        keys.append(key)
+        return self
+    }
+
+    @discardableResult
+    public func addKeys(_ keys: [PublicKey]) -> Self {
+        self.keys.append(contentsOf: keys)
+        return self
+    }
 }