Add more chains to Blockchain

trustwallet · Jul 13, 2018 · b5385c8 · b5385c8
1 parent f82f561
commit b5385c8
Show file tree

Hide file tree

Showing 3 changed files with 36 additions and 6 deletions.
diff --git a/Sources/Blockchain.swift b/Sources/Blockchain.swift
@@ -10,4 +10,15 @@ import Foundation
 public enum Blockchain {
     case bitcoin
     case ethereum
+    case poa
+    case ethereumClassic
+    case callisto
+    case gochain
+
+    // test networks
+
+    case ethereumRopsten
+    case ethereumKovan
+    case ethereumRinkeby
+    case ethereumSokol
 }
diff --git a/Sources/HDWallet/HDWallet.swift b/Sources/HDWallet/HDWallet.swift
@@ -55,18 +55,29 @@ public class HDWallet {
 extension Blockchain {
     public var coinType: Int {
         switch self {
-        case .bitcoin:
-            return 0
-        case .ethereum:
-            return 60
+        case .bitcoin: return 0
+        case .ethereum: return 60
+        case .ethereumKovan, .ethereumRinkeby, .ethereumRopsten, .ethereumSokol: return 60
+        case .poa: return 178
+        case .ethereumClassic: return 61
+        case .callisto: return 820
+        case .gochain: return 6060
         }
     }
 
     public func derivationPath(at index: Int) -> DerivationPath {
         switch self {
         case .bitcoin:
             return DerivationPath(purpose: 44, coinType: coinType, account: 0, change: 0, address: index)
-        case .ethereum:
+        case .ethereum,
+             .poa,
+             .ethereumClassic,
+             .callisto,
+             .gochain,
+             .ethereumKovan,
+             .ethereumRinkeby,
+             .ethereumRopsten,
+             .ethereumSokol:
             return DerivationPath(purpose: 44, coinType: coinType, account: 0, change: 0, address: index)
         }
     }

diff --git a/Sources/PrivateKey.swift b/Sources/PrivateKey.swift
@@ -66,7 +66,15 @@ public final class PrivateKey: Hashable, CustomStringConvertible {
         switch blockchain {
         case .bitcoin:
             return BitcoinPublicKey(data: Crypto.getPublicKey(from: data))!
-        case .ethereum:
+        case .ethereum,
+             .poa,
+             .ethereumClassic,
+             .callisto,
+             .gochain,
+             .ethereumKovan,
+             .ethereumRinkeby,
+             .ethereumRopsten,
+             .ethereumSokol:
             return EthereumPublicKey(data: Crypto.getPublicKey(from: data))!
         }
     }