feat: query mirror node for AccountBalances, AccountInfo, ContractInfo

Package.swift

@@ -97,6 +97,7 @@ let package = Package(
         .package(url: "https://github.com/krzyzanowskim/CryptoSwift.git", from: "1.0.0"),
         .package(url: "https://github.com/apple/swift-docc-plugin", from: "1.0.0"),
         .package(url: "https://github.com/pointfreeco/swift-snapshot-testing.git", from: "1.0.0"),
+        .package(url: "https://github.com/swift-server/async-http-client.git", from: "1.21.1"),
     ],
     targets: [
         .target(
@@ -123,6 +124,7 @@ let package = Package(
                 .product(name: "Atomics", package: "swift-atomics"),
                 .product(name: "secp256k1", package: "secp256k1.swift"),
                 "CryptoSwift",
+                .product(name: "AsyncHTTPClient", package: "async-http-client"),
             ]
             // todo: find some way to enable these locally.
             // swiftSettings: [

Sources/Hedera/Account/AccountBalance.swift

@@ -21,7 +21,7 @@
 import Foundation
 import HederaProtobufs
 
-private struct TokenBalance {
+internal struct TokenBalance {
     fileprivate let id: TokenId
     fileprivate let balance: UInt64
     fileprivate let decimals: UInt32
@@ -59,6 +59,16 @@ public struct AccountBalance: Sendable {
 
     private let tokensInner: [TokenBalance]
 
+    internal init(
+        accountId: AccountId,
+        hbars: Hbar,
+        tokensInner: [TokenBalance]
+    ) {
+        self.accountId = accountId
+        self.hbars = hbars
+        self.tokensInner = tokensInner
+    }
+
     // hack to work around deprecated warning
     private var tokenBalancesInner: [TokenId: UInt64] {
         Dictionary(uniqueKeysWithValues: tokensInner.map { ($0.id, $0.balance) })

Sources/Hedera/Account/AccountBalanceQuery.swift

@@ -85,12 +85,22 @@ public final class AccountBalanceQuery: Query<AccountBalance> {
         try await Proto_CryptoServiceAsyncClient(channel: channel).cryptoGetBalance(request)
     }
 
-    internal override func makeQueryResponse(_ response: Proto_Response.OneOf_Response) throws -> Response {
+    internal override func makeQueryResponse(_ context: Context, _ response: Proto_Response.OneOf_Response) async throws
+        -> Response
+    {
+        let mirrorNodeGateway = try MirrorNodeGateway.forNetwork(context.mirrorNetworkNodes, context.ledgerId)
+        let mirrorNodeService = MirrorNodeService.init(mirrorNodeGateway)
+
         guard case .cryptogetAccountBalance(let proto) = response else {
             throw HError.fromProtobuf("unexpected \(response) received, expected `cryptogetAccountBalance`")
         }
 
-        return try .fromProtobuf(proto)
+        let accountId = try AccountId.fromProtobuf(proto.accountID)
+        let tokenBalanceProto = try await mirrorNodeService.getTokenBalancesForAccount(String(accountId.num))
+
+        return AccountBalance(
+            accountId: accountId, hbars: .fromTinybars(Int64(proto.balance)),
+            tokensInner: .fromProtobuf(tokenBalanceProto))
     }
 
     public override func validateChecksums(on ledgerId: LedgerId) throws {

Sources/Hedera/Account/AccountId.swift

@@ -135,6 +135,15 @@ public struct AccountId: Sendable, EntityId, ValidateChecksums {
     public func toBytes() -> Data {
         toProtobufBytes()
     }
+
+    public func populateAccountIdNum(_ client: Client) async throws -> Self {
+        let mirrorNodeGateway = try MirrorNodeGateway.forClient(client)
+        let mirrorNodeService = MirrorNodeService(mirrorNodeGateway)
+
+        let accountNumFromMirror = try await mirrorNodeService.getAccountNum(self.evmAddress!.toString())
+
+        return Self(shard: shard, realm: realm, num: accountNumFromMirror)
+    }
 }
 
 extension AccountId: TryProtobufCodable {

Sources/Hedera/Account/AccountInfo.swift

@@ -47,6 +47,7 @@ public struct AccountInfo: Sendable {
         maxAutomaticTokenAssociations: UInt32,
         aliasKey: PublicKey?,
         ethereumNonce: UInt64,
+        tokenRelationships: [TokenId: TokenRelationship],
         ledgerId: LedgerId,
         staking: StakingInfo?
     ) {
@@ -66,6 +67,7 @@ public struct AccountInfo: Sendable {
         self.ethereumNonce = ethereumNonce
         self.ledgerId = ledgerId
         self.staking = staking
+        self.tokenRelationships = tokenRelationships
         self.guts = DeprecatedGuts(
             proxyAccountId: proxyAccountId,
             sendRecordThreshold: sendRecordThreshold,
@@ -154,6 +156,9 @@ public struct AccountInfo: Sendable {
     /// Staking metadata for this account.
     public let staking: StakingInfo?
 
+    /// Staking metadata for this account.
+    public let tokenRelationships: [TokenId: TokenRelationship]
+
     /// Decode `Self` from protobuf-encoded `bytes`.
     ///
     /// - Throws: ``HError/ErrorKind/fromProtobuf`` if:
@@ -178,6 +183,11 @@ extension AccountInfo: TryProtobufCodable {
         let staking = proto.hasStakingInfo ? proto.stakingInfo : nil
         let proxyAccountId = proto.hasProxyAccountID ? proto.proxyAccountID : nil
 
+        var tokenRelationships: [TokenId: TokenRelationship] = [:]
+        for relationship in proto.tokenRelationships {
+            tokenRelationships[.fromProtobuf(relationship.tokenID)] = try TokenRelationship.fromProtobuf(relationship)
+        }
+
         self.init(
             accountId: try .fromProtobuf(proto.accountID),
             contractAccountId: proto.contractAccountID,
@@ -196,6 +206,7 @@ extension AccountInfo: TryProtobufCodable {
             maxAutomaticTokenAssociations: UInt32(proto.maxAutomaticTokenAssociations),
             aliasKey: try .fromAliasBytes(proto.alias),
             ethereumNonce: UInt64(proto.ethereumNonce),
+            tokenRelationships: tokenRelationships,
             ledgerId: .fromBytes(proto.ledgerID),
             staking: try .fromProtobuf(staking)
         )

Sources/Hedera/Account/AccountInfoQuery.swift

@@ -59,12 +59,48 @@ public final class AccountInfoQuery: Query<AccountInfo> {
         try await Proto_CryptoServiceAsyncClient(channel: channel).getAccountInfo(request)
     }
 
-    internal override func makeQueryResponse(_ response: Proto_Response.OneOf_Response) throws -> Response {
+    internal override func makeQueryResponse(_ context: Context, _ response: Proto_Response.OneOf_Response) async throws
+        -> Response
+    {
+        let mirrorNodeGateway = try MirrorNodeGateway.forNetwork(context.mirrorNetworkNodes, context.ledgerId)
+        let mirrorNodeService = MirrorNodeService.init(mirrorNodeGateway)
+
         guard case .cryptoGetInfo(let proto) = response else {
             throw HError.fromProtobuf("unexpected \(response) received, expected `cryptoGetInfo`")
         }
 
-        return try .fromProtobuf(proto.accountInfo)
+        let accountInfo = try AccountInfo.fromProtobuf(proto.accountInfo)
+        let tokenRelationshipsProto = try await mirrorNodeService.getTokenRelationshipsForAccount(
+            String(describing: accountInfo.accountId.num))
+
+        var tokenRelationships: [TokenId: TokenRelationship] = [:]
+
+        for relationship in tokenRelationshipsProto {
+            tokenRelationships[.fromProtobuf(relationship.tokenID)] = try TokenRelationship.fromProtobuf(relationship)
+        }
+
+        return AccountInfo(
+            accountId: accountInfo.accountId,
+            contractAccountId: accountInfo.contractAccountId,
+            isDeleted: accountInfo.isDeleted,
+            proxyAccountId: accountInfo.proxyAccountId,
+            proxyReceived: accountInfo.proxyReceived,
+            key: accountInfo.key,
+            balance: accountInfo.balance,
+            sendRecordThreshold: accountInfo.sendRecordThreshold,
+            receiveRecordThreshold: accountInfo.receiveRecordThreshold,
+            isReceiverSignatureRequired: accountInfo.isReceiverSignatureRequired,
+            expirationTime: accountInfo.expirationTime,
+            autoRenewPeriod: accountInfo.autoRenewPeriod,
+            accountMemo: accountInfo.accountMemo,
+            ownedNfts: accountInfo.ownedNfts,
+            maxAutomaticTokenAssociations: accountInfo.maxAutomaticTokenAssociations,
+            aliasKey: accountInfo.aliasKey,
+            ethereumNonce: accountInfo.ethereumNonce,
+            tokenRelationships: tokenRelationships,
+            ledgerId: accountInfo.ledgerId,
+            staking: accountInfo.staking
+        )
     }
 
     internal override func validateChecksums(on ledgerId: LedgerId) throws {

Sources/Hedera/Account/AccountRecordsQuery.swift

@@ -57,7 +57,9 @@ public final class AccountRecordsQuery: Query<[TransactionRecord]> {
         try await Proto_CryptoServiceAsyncClient(channel: channel).getAccountRecords(request)
     }
 
-    internal override func makeQueryResponse(_ response: Proto_Response.OneOf_Response) throws -> Response {
+    internal override func makeQueryResponse(_ context: Context, _ response: Proto_Response.OneOf_Response) async throws
+        -> Response
+    {
         guard case .cryptoGetAccountRecords(let proto) = response else {
             throw HError.fromProtobuf("unexpected \(response) received, expected `cryptoGetAccountRecords`")
         }

Sources/Hedera/Account/AccountStakersQuery.swift

@@ -57,7 +57,9 @@ public final class AccountStakersQuery: Query<[ProxyStaker]> {
         try await Proto_CryptoServiceAsyncClient(channel: channel).getStakersByAccountID(request)
     }
 
-    internal override func makeQueryResponse(_ response: Proto_Response.OneOf_Response) throws -> Response {
+    internal override func makeQueryResponse(_ context: Context, _ response: Proto_Response.OneOf_Response) async throws
+        -> Response
+    {
         guard case .cryptoGetProxyStakers(let proto) = response else {
             throw HError.fromProtobuf("unexpected \(response) received, expected `cryptoGetProxyStakers`")
         }

Sources/Hedera/ChunkedTransaction.swift

@@ -206,7 +206,7 @@ extension ChunkedTransaction.FirstChunkView: Execute {
         self.transaction.regenerateTransactionId
     }
 
-    internal func makeRequest(_ transactionId: TransactionId?, _ nodeAccountId: AccountId) throws -> (
+    internal func makeRequest(_ client: Client, _ transactionId: TransactionId?, _ nodeAccountId: AccountId) throws -> (
         GrpcRequest, Context
     ) {
         assert(transaction.isFrozen)
@@ -265,7 +265,7 @@ extension ChunkedTransaction.ChunkView: Execute {
         self.transaction.regenerateTransactionId
     }
 
-    internal func makeRequest(_ transactionId: TransactionId?, _ nodeAccountId: AccountId) throws -> (
+    internal func makeRequest(_ client: Client, _ transactionId: TransactionId?, _ nodeAccountId: AccountId) throws -> (
         GrpcRequest, Context
     ) {
         assert(transaction.isFrozen)

Sources/Hedera/Contract/ContractBytecodeQuery.swift

@@ -55,7 +55,9 @@ public final class ContractBytecodeQuery: Query<Data> {
         try await Proto_SmartContractServiceAsyncClient(channel: channel).contractGetBytecode(request)
     }
 
-    internal override func makeQueryResponse(_ response: Proto_Response.OneOf_Response) throws -> Response {
+    internal override func makeQueryResponse(_ context: Context, _ response: Proto_Response.OneOf_Response) async throws
+        -> Response
+    {
         guard case .contractGetBytecodeResponse(let proto) = response else {
             throw HError.fromProtobuf("unexpected \(response) received, expected `contractGetBytecodeResponse`")
         }

Sources/Hedera/Contract/ContractCallQuery.swift

@@ -133,7 +133,9 @@ public final class ContractCallQuery: Query<ContractFunctionResult> {
         try await Proto_SmartContractServiceAsyncClient(channel: channel).contractCallLocalMethod(request)
     }
 
-    internal override func makeQueryResponse(_ response: Proto_Response.OneOf_Response) throws -> Response {
+    internal override func makeQueryResponse(_ context: Context, _ response: Proto_Response.OneOf_Response) async throws
+        -> Response
+    {
         guard case .contractCallLocal(let proto) = response else {
             throw HError.fromProtobuf("unexpected \(response) received, expected `contractCallLocal`")
         }

Sources/Hedera/Contract/ContractId.swift

@@ -119,6 +119,17 @@ public struct ContractId: EntityId {
     public func toBytes() -> Data {
         toProtobufBytes()
     }
+
+    public func populateContractNum(_ client: Client) async throws -> Self {
+        let address = try EvmAddress.fromBytes(self.evmAddress!)
+
+        let mirrorNodeGateway = try MirrorNodeGateway.forClient(client)
+        let mirrorNodeService = MirrorNodeService(mirrorNodeGateway)
+
+        let contractNum = try await mirrorNodeService.getContractNum(address.toString())
+
+        return Self(shard: shard, realm: realm, num: contractNum)
+    }
 }
 
 #if compiler(>=5.7)

Sources/Hedera/Contract/ContractInfo.swift

@@ -61,6 +61,11 @@ public struct ContractInfo {
     /// The maximum number of tokens that a contract can be implicitly associated with.
     public let maxAutomaticTokenAssociations: UInt32
 
+    /// The tokens associated to the contract
+    ///
+    /// Query mirror node
+    public let tokenRelationships: [TokenId: TokenRelationship]
+
     /// Ledger ID for the network the response was returned from.
     public let ledgerId: LedgerId
 
@@ -91,9 +96,15 @@ extension ContractInfo: TryProtobufCodable {
         let autoRenewPeriod = proto.hasAutoRenewPeriod ? proto.autoRenewPeriod : nil
         let autoRenewAccountId = proto.hasAutoRenewAccountID ? proto.autoRenewAccountID : nil
 
+        var tokenRelationships: [TokenId: TokenRelationship] = [:]
+
+        for relationship in proto.tokenRelationships {
+            tokenRelationships[.fromProtobuf(relationship.tokenID)] = try TokenRelationship.fromProtobuf(relationship)
+        }
+
         self.init(
             contractId: try .fromProtobuf(proto.contractID),
-            accountId: try .fromProtobuf(proto.accountID),
+            accountId: try AccountId.fromProtobuf(proto.accountID),
             contractAccountId: proto.contractAccountID,
             adminKey: try .fromProtobuf(adminKey),
             expirationTime: .fromProtobuf(expirationTime),
@@ -104,6 +115,7 @@ extension ContractInfo: TryProtobufCodable {
             isDeleted: proto.deleted,
             autoRenewAccountId: try .fromProtobuf(autoRenewAccountId),
             maxAutomaticTokenAssociations: UInt32(proto.maxAutomaticTokenAssociations),
+            tokenRelationships: tokenRelationships,
             ledgerId: .fromBytes(proto.ledgerID),
             stakingInfo: try .fromProtobuf(proto.stakingInfo)
         )