From d9db332af85582191d0f21cc4e67fd6c8f890275 Mon Sep 17 00:00:00 2001
From: Andrei Kuzmiankov <ender2016@gmail.com>
Date: Thu, 15 Oct 2020 20:07:13 -0700
Subject: [PATCH] feat: Added contract_info_query

---
 contract_info_query.go | 136 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 136 insertions(+)
 create mode 100644 contract_info_query.go

diff --git a/contract_info_query.go b/contract_info_query.go
new file mode 100644
index 000000000..c509b38bf
--- /dev/null
+++ b/contract_info_query.go
@@ -0,0 +1,136 @@
+package hedera
+
+import (
+	"time"
+
+	"github.com/hashgraph/hedera-sdk-go/proto"
+)
+
+// ContractInfoQuery retrieves information about a smart contract instance. This includes the account that it uses, the
+// file containing its bytecode, and the time when it will expire.
+type ContractInfoQuery struct {
+	Query
+	pb *proto.ContractGetInfoQuery
+}
+
+// ContractInfo is the information about the contract instance returned by a ContractInfoQuery
+type ContractInfo struct {
+	AccountID         AccountID
+	ContractID        ContractID
+	ContractAccountID string
+	AdminKey          PublicKey
+	ExpirationTime    time.Time
+	AutoRenewPeriod   time.Duration
+	Storage           uint64
+	ContractMemo      string
+}
+
+// NewContractInfoQuery creates a ContractInfoQuery query which can be used to construct and execute a
+// Contract Get Info Query.
+func NewContractInfoQuery() *ContractInfoQuery {
+	header := proto.QueryHeader{}
+	query := newQuery(true, &header)
+	pb := proto.ContractGetInfoQuery{Header: &header}
+	query.pb.Query = &proto.Query_ContractGetInfo{
+		ContractGetInfo: &pb,
+	}
+
+	return &ContractInfoQuery{
+		Query: query,
+		pb:    &pb,
+	}
+}
+
+// SetContractID sets the contract for which information is requested
+func (query *ContractInfoQuery) SetContractID(id ContractID) *ContractInfoQuery {
+	query.pb.ContractID = id.toProtobuf()
+	return query
+}
+
+func contractInfoQuery_mapResponseStatus(_ request, response response) Status {
+	return Status(response.query.GetContractGetInfo().Header.NodeTransactionPrecheckCode)
+}
+
+func contractInfoQuery_getMethod(_ request, channel *channel) method {
+	return method{
+		query: channel.getContract().GetContractInfo,
+	}
+}
+
+func (query *ContractInfoQuery) Execute(client *Client) (ContractInfo, error) {
+	if client == nil || client.operator == nil {
+		return ContractInfo{}, errNoClientProvided
+	}
+
+	query.queryPayment = NewHbar(2)
+	query.paymentTransactionID = TransactionIDGenerate(client.operator.accountID)
+
+	var cost Hbar
+	if query.queryPayment.tinybar != 0 {
+		cost = query.queryPayment
+	} else {
+		cost = client.maxQueryPayment
+
+		// actualCost := CostQuery()
+	}
+
+	err := query_generatePayments(&query.Query, client, cost)
+	if err != nil {
+		return ContractInfo{}, err
+	}
+
+
+	resp, err := execute(
+		client,
+		request{
+			query: &query.Query,
+		},
+		query_shouldRetry,
+		query_makeRequest,
+		query_advanceRequest,
+		query_getNodeId,
+		contractInfoQuery_getMethod,
+		contractInfoQuery_mapResponseStatus,
+		query_mapResponse,
+	)
+
+	if err != nil {
+		return ContractInfo{}, err
+	}
+
+	adminKey, err := publicKeyFromProtobuf(resp.query.GetContractGetInfo().GetContractInfo().GetAdminKey())
+
+	return ContractInfo{
+		AccountID:         accountIDFromProtobuf(resp.query.GetContractGetInfo().ContractInfo.AccountID),
+		ContractID:        contractIDFromProtobuf(resp.query.GetContractGetInfo().ContractInfo.ContractID),
+		ContractAccountID: resp.query.GetContractGetInfo().ContractInfo.ContractAccountID,
+		AdminKey:          PublicKey{
+			keyData: adminKey.toProtoKey().GetEd25519(),
+		},
+		ExpirationTime:    timeFromProtobuf(resp.query.GetContractGetInfo().ContractInfo.ExpirationTime),
+		AutoRenewPeriod:   durationFromProtobuf(resp.query.GetContractGetInfo().ContractInfo.AutoRenewPeriod),
+		Storage:           uint64(resp.query.GetContractGetInfo().ContractInfo.Storage),
+		ContractMemo:      resp.query.GetContractGetInfo().ContractInfo.Memo,
+	}, nil
+}
+
+// SetMaxQueryPayment sets the maximum payment allowed for this Query.
+func (query *ContractInfoQuery) SetMaxQueryPayment(maxPayment Hbar) *ContractInfoQuery {
+	query.Query.SetMaxQueryPayment(maxPayment)
+	return query
+}
+
+// SetQueryPayment sets the payment amount for this Query.
+func (query *ContractInfoQuery) SetQueryPayment(paymentAmount Hbar) *ContractInfoQuery {
+	query.Query.SetQueryPayment(paymentAmount)
+	return query
+}
+
+func (query *ContractInfoQuery) SetNodeAccountID(accountID AccountID) *ContractInfoQuery {
+	query.Query.SetNodeAccountID(accountID)
+	return query
+}
+
+func (query *ContractInfoQuery) GetNodeAccountId() AccountID {
+	return query.Query.GetNodeAccountId()
+}