// Copyright © 2021 Encrypted Ink. All rights reserved. import Foundation import WalletConnect class WalletConnect { private let sessionStorage = SessionStorage.shared static let shared = WalletConnect() private init() {} private var interactors = [WCInteractor]() private var peers = [String: WCPeerMeta]() func sessionWithLink(_ link: String) -> WCSession? { return WCSession.from(string: link) } func connect(session: WCSession, address: String, uuid: UUID = UUID(), completion: @escaping ((Bool) -> Void)) { let clientMeta = WCPeerMeta(name: "Encrypted Ink", url: "https://encrypted.ink", description: "Ethereum agent for macOS", icons: ["https://encrypted.ink/icon.png"]) let interactor = WCInteractor(session: session, meta: clientMeta, uuid: uuid) configure(interactor: interactor, address: address) interactor.connect().done { connected in completion(connected) }.catch { [weak self, weak interactor] _ in completion(false) if let interactor = interactor { self?.didDisconnect(interactor: interactor) } } interactors.append(interactor) } func restartSessions() { let items = sessionStorage.loadAll() for item in items { guard let uuid = UUID(uuidString: item.clientId) else { continue } connect(session: item.session, address: item.address, uuid: uuid) { _ in } peers[item.clientId] = item.sessionDetails.peerMeta } } private func didDisconnect(interactor: WCInteractor) { if sessionStorage.shouldReconnect(interactor: interactor) { reconnectWhenPossible(interactor: interactor) } else { removeInteractor(id: interactor.clientId) } } private func removeInteractor(id: String) { interactors.removeAll(where: { $0.clientId == id }) peers.removeValue(forKey: id) sessionStorage.remove(clientId: id) } private func getPeerOfInteractor(_ interactor: WCInteractor?) -> WCPeerMeta? { guard let id = interactor?.clientId else { return nil } return peers[id] } private func configure(interactor: WCInteractor, address: String) { let accounts = [address] let chainId = 1 interactor.onError = { _ in } interactor.onSessionRequest = { [weak self, weak interactor] (id, peerParam) in guard let interactor = interactor else { return } self?.peers[interactor.clientId] = peerParam.peerMeta self?.sessionStorage.add(interactor: interactor, address: address, sessionDetails: peerParam) interactor.approveSession(accounts: accounts, chainId: chainId).cauterize() } interactor.onDisconnect = { [weak interactor, weak self] _ in if let interactor = interactor { self?.didDisconnect(interactor: interactor) } } interactor.eth.onSign = { [weak self, weak interactor] (id, payload) in self?.approveSign(id: id, payload: payload, address: address, interactor: interactor) } interactor.eth.onTransaction = { [weak self, weak interactor] (id, event, transaction) in self?.approveTransaction(id: id, wct: transaction, address: address, interactor: interactor) } } private func reconnectWhenPossible(interactor: WCInteractor) { DispatchQueue.main.asyncAfter(deadline: .now() + .seconds(10)) { [weak interactor] in interactor?.resume() // TODO: reconnect when appropriate. } } private func approveTransaction(id: Int64, wct: WCEthereumTransaction, address: String, interactor: WCInteractor?) { guard let to = wct.to else { rejectRequest(id: id, interactor: interactor, message: "Something went wrong.") return } let peer = getPeerOfInteractor(interactor) let transaction = Transaction(from: wct.from, to: to, nonce: wct.nonce, gasPrice: wct.gasPrice, gas: wct.gas, value: wct.value, data: wct.data) Agent.shared.showApprove(transaction: transaction, peerMeta: peer) { [weak self, weak interactor] transaction in if let transaction = transaction { self?.sendTransaction(transaction, address: address, requestId: id, interactor: interactor) } else { self?.rejectRequest(id: id, interactor: interactor, message: "Cancelled") } } } private func approveSign(id: Int64, payload: WCEthereumSignPayload, address: String, interactor: WCInteractor?) { var message: String? let title: String switch payload { case let .sign(data: data, raw: _): message = String(data: data, encoding: .utf8) title = "Sign Message" case let .personalSign(data: data, raw: _): message = String(data: data, encoding: .utf8) title = "Sign Personal Message" case let .signTypeData(id: _, data: _, raw: raw): title = "Sign Typed Data" if raw.count >= 2 { message = raw[1] } } let peer = getPeerOfInteractor(interactor) Agent.shared.showApprove(title: title, meta: message ?? "", peerMeta: peer) { [weak self, weak interactor] approved in if approved { self?.sign(id: id, message: message, payload: payload, address: address, interactor: interactor) } else { self?.rejectRequest(id: id, interactor: interactor, message: "Cancelled") } } } private func rejectRequest(id: Int64, interactor: WCInteractor?, message: String) { interactor?.rejectRequest(id: id, message: message).cauterize() } private func sendTransaction(_ transaction: Transaction, address: String, requestId: Int64, interactor: WCInteractor?) { guard let account = AccountsService.getAccountForAddress(address) else { rejectRequest(id: requestId, interactor: interactor, message: "Something went wrong.") return } guard let hash = try? Ethereum.send(transaction: transaction, account: account) else { rejectRequest(id: requestId, interactor: interactor, message: "Failed to send") return } interactor?.approveRequest(id: requestId, result: hash).cauterize() } private func sign(id: Int64, message: String?, payload: WCEthereumSignPayload, address: String, interactor: WCInteractor?) { guard let message = message, let account = AccountsService.getAccountForAddress(address) else { rejectRequest(id: id, interactor: interactor, message: "Something went wrong.") return } var signed: String? switch payload { case .personalSign: signed = try? Ethereum.signPersonal(message: message, account: account) case .signTypeData: signed = try? Ethereum.sign(typedData: message, account: account) case .sign: signed = try? Ethereum.sign(message: message, account: account) } guard let result = signed else { rejectRequest(id: id, interactor: interactor, message: "Something went wrong.") return } interactor?.approveRequest(id: id, result: result).cauterize() } }