feat(TCK): AddressBookQuery implementation (#2610)

Signed-off-by: Mustafa Uzun <mustafa.uzun@limechain.tech>

Author: mustafauzunn

tck/src/main/java/com/hedera/hashgraph/tck/methods/sdk/NodeService.java

@@ -6,15 +6,18 @@
 import com.hedera.hashgraph.tck.annotation.JSONRPC2Method;
 import com.hedera.hashgraph.tck.annotation.JSONRPC2Service;
 import com.hedera.hashgraph.tck.methods.AbstractJSONRPC2Service;
+import com.hedera.hashgraph.tck.methods.sdk.param.node.AddressBookQueryParams;
 import com.hedera.hashgraph.tck.methods.sdk.param.node.NodeCreateParams;
 import com.hedera.hashgraph.tck.methods.sdk.param.node.NodeDeleteParams;
 import com.hedera.hashgraph.tck.methods.sdk.param.node.NodeUpdateParams;
 import com.hedera.hashgraph.tck.methods.sdk.param.node.ServiceEndpointParams;
+import com.hedera.hashgraph.tck.methods.sdk.response.AddressBookResponse;
 import com.hedera.hashgraph.tck.methods.sdk.response.NodeResponse;
 import com.hedera.hashgraph.tck.util.KeyUtils;
 import java.time.Duration;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.stream.Collectors;
 import org.bouncycastle.util.encoders.Hex;
 
 @JSONRPC2Service
@@ -26,6 +29,35 @@ public NodeService(SdkService sdkService) {
         this.sdkService = sdkService;
     }
 
+    @JSONRPC2Method("getAddressBook")
+    public AddressBookResponse addressBookQuery(final AddressBookQueryParams params) {
+        AddressBookQuery query = new AddressBookQuery().setFileId(FileId.fromString(params.getFileId()));
+        Client client = sdkService.getClient(params.getSessionId());
+
+        NodeAddressBook addressBook = query.execute(client);
+        AddressBookResponse response = new AddressBookResponse();
+
+        addressBook.getNodeAddresses().forEach(address -> {
+            List<AddressBookResponse.Endpoint> mappedEndpoints = address.getAddresses().stream()
+                    .map(sdkEndpoint -> new AddressBookResponse.Endpoint(
+                            sdkEndpoint.getAddress() != null ? Hex.toHexString(sdkEndpoint.getAddress()) : null,
+                            sdkEndpoint.getPort(),
+                            sdkEndpoint.getDomainName() != null ? sdkEndpoint.getDomainName() : ""))
+                    .collect(Collectors.toList());
+
+            response.addNodeAddress(new AddressBookResponse.NodeAddress(
+                    address.getPublicKey() != null ? address.getPublicKey() : "",
+                    address.getAccountId() != null ? address.getAccountId().toString() : "",
+                    address.getNodeId(),
+                    address.getCertHash() != null ? address.getCertHash().toString() : "",
+                    mappedEndpoints,
+                    address.getDescription(),
+                    address.getStake()));
+        });
+
+        return response;
+    }
+
     @JSONRPC2Method("createNode")
     public NodeResponse createNode(final NodeCreateParams params) throws Exception {
         NodeCreateTransaction tx = new NodeCreateTransaction().setGrpcDeadline(DEFAULT_GRPC_DEADLINE);

tck/src/main/java/com/hedera/hashgraph/tck/methods/sdk/param/node/AddressBookQueryParams.java

@@ -0,0 +1,29 @@
+// SPDX-License-Identifier: Apache-2.0
+package com.hedera.hashgraph.tck.methods.sdk.param.node;
+
+import com.hedera.hashgraph.tck.methods.JSONRPC2Param;
+import com.hedera.hashgraph.tck.util.JSONRPCParamParser;
+import java.util.Map;
+import java.util.Objects;
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+
+@Getter
+@AllArgsConstructor
+@NoArgsConstructor
+public class AddressBookQueryParams extends JSONRPC2Param {
+    private String fileId;
+    private Long limit;
+    private String sessionId;
+
+    @Override
+    public JSONRPC2Param parse(Map<String, Object> jrpcParams) throws Exception {
+        Objects.requireNonNull(jrpcParams, "jrpcParams must not be null");
+
+        var parsedFileId = (String) jrpcParams.get("fileId");
+        var parsedLimit = (Long) jrpcParams.get("limit");
+
+        return new AddressBookQueryParams(parsedFileId, parsedLimit, JSONRPCParamParser.parseSessionId(jrpcParams));
+    }
+}

tck/src/main/java/com/hedera/hashgraph/tck/methods/sdk/response/AddressBookResponse.java

@@ -0,0 +1,68 @@
+// SPDX-License-Identifier: Apache-2.0
+package com.hedera.hashgraph.tck.methods.sdk.response;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Objects;
+import javax.annotation.Nullable;
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.Getter;
+
+@Data
+public class AddressBookResponse {
+    private final List<NodeAddress> nodeAddresses = new ArrayList<>();
+
+    @Getter
+    @AllArgsConstructor
+    public static class NodeAddress {
+        /**
+         * The RSA public key of the node.
+         */
+        @Nullable
+        String publicKey;
+        /**
+         * The account to be paid for queries and transactions sent to this node.
+         */
+        @Nullable
+        String accountId;
+        /**
+         * A non-sequential identifier for the node.
+         */
+        long nodeId;
+        /**
+         * A hash of the X509 cert used for gRPC traffic to this node.
+         */
+        @Nullable
+        String certHash;
+        /**
+         * A node's service IP addresses and ports.
+         */
+        List<Endpoint> serviceEndpoints;
+        /**
+         * A description of the node, with UTF-8 encoding up to 100 bytes.
+         */
+        @Nullable
+        String description;
+        /**
+         * The amount of tinybars staked to the node.
+         */
+        long stake;
+    }
+
+    @Getter
+    @AllArgsConstructor
+    public static class Endpoint {
+        @Nullable
+        String address;
+
+        int port;
+
+        String domainName;
+    }
+
+    public void addNodeAddress(NodeAddress nodeAddress) {
+        Objects.requireNonNull(nodeAddress, "nodeAddress must not be null");
+        nodeAddresses.add(nodeAddress);
+    }
+}