feat(accountplus): Add query for accountplus account state (backport #2659)

indexer/packages/v4-protos/src/codegen/dydxprotocol/accountplus/query.lcd.ts

@@ -1,5 +1,5 @@
 import { LCDClient } from "@osmonauts/lcd";
-import { QueryParamsRequest, QueryParamsResponseSDKType, GetAuthenticatorRequest, GetAuthenticatorResponseSDKType, GetAuthenticatorsRequest, GetAuthenticatorsResponseSDKType } from "./query";
+import { QueryParamsRequest, QueryParamsResponseSDKType, GetAuthenticatorRequest, GetAuthenticatorResponseSDKType, GetAuthenticatorsRequest, GetAuthenticatorsResponseSDKType, AccountStateRequest, AccountStateResponseSDKType } from "./query";
 export class LCDQueryClient {
   req: LCDClient;
 
@@ -12,6 +12,7 @@ export class LCDQueryClient {
     this.params = this.params.bind(this);
     this.getAuthenticator = this.getAuthenticator.bind(this);
     this.getAuthenticators = this.getAuthenticators.bind(this);
+    this.accountState = this.accountState.bind(this);
   }
   /* Parameters queries the parameters of the module. */
 
@@ -34,5 +35,12 @@ export class LCDQueryClient {
     const endpoint = `dydxprotocol/accountplus/authenticators/${params.account}`;
     return await this.req.get<GetAuthenticatorsResponseSDKType>(endpoint);
   }
+  /* Queries for an account state (timestamp nonce). */
+
+
+  async accountState(params: AccountStateRequest): Promise<AccountStateResponseSDKType> {
+    const endpoint = `dydxprotocol/accountplus/account_state/${params.address}`;
+    return await this.req.get<AccountStateResponseSDKType>(endpoint);
+  }
 
 }

indexer/packages/v4-protos/src/codegen/dydxprotocol/accountplus/query.rpc.Query.ts

@@ -1,7 +1,7 @@
 import { Rpc } from "../../helpers";
 import * as _m0 from "protobufjs/minimal";
 import { QueryClient, createProtobufRpcClient } from "@cosmjs/stargate";
-import { QueryParamsRequest, QueryParamsResponse, GetAuthenticatorRequest, GetAuthenticatorResponse, GetAuthenticatorsRequest, GetAuthenticatorsResponse } from "./query";
+import { QueryParamsRequest, QueryParamsResponse, GetAuthenticatorRequest, GetAuthenticatorResponse, GetAuthenticatorsRequest, GetAuthenticatorsResponse, AccountStateRequest, AccountStateResponse } from "./query";
 /** Query defines the gRPC querier service. */
 
 export interface Query {
@@ -13,6 +13,9 @@ export interface Query {
   /** Queries all authenticators for a given account. */
 
   getAuthenticators(request: GetAuthenticatorsRequest): Promise<GetAuthenticatorsResponse>;
+  /** Queries for an account state (timestamp nonce). */
+
+  accountState(request: AccountStateRequest): Promise<AccountStateResponse>;
 }
 export class QueryClientImpl implements Query {
   private readonly rpc: Rpc;
@@ -22,6 +25,7 @@ export class QueryClientImpl implements Query {
     this.params = this.params.bind(this);
     this.getAuthenticator = this.getAuthenticator.bind(this);
     this.getAuthenticators = this.getAuthenticators.bind(this);
+    this.accountState = this.accountState.bind(this);
   }
 
   params(request: QueryParamsRequest = {}): Promise<QueryParamsResponse> {
@@ -42,6 +46,12 @@ export class QueryClientImpl implements Query {
     return promise.then(data => GetAuthenticatorsResponse.decode(new _m0.Reader(data)));
   }
 
+  accountState(request: AccountStateRequest): Promise<AccountStateResponse> {
+    const data = AccountStateRequest.encode(request).finish();
+    const promise = this.rpc.request("dydxprotocol.accountplus.Query", "AccountState", data);
+    return promise.then(data => AccountStateResponse.decode(new _m0.Reader(data)));
+  }
+
 }
 export const createRpcQueryExtension = (base: QueryClient) => {
   const rpc = createProtobufRpcClient(base);
@@ -57,6 +67,10 @@ export const createRpcQueryExtension = (base: QueryClient) => {
 
     getAuthenticators(request: GetAuthenticatorsRequest): Promise<GetAuthenticatorsResponse> {
       return queryService.getAuthenticators(request);
+    },
+
+    accountState(request: AccountStateRequest): Promise<AccountStateResponse> {
+      return queryService.accountState(request);
     }
 
   };

indexer/packages/v4-protos/src/codegen/dydxprotocol/accountplus/query.ts

@@ -1,7 +1,44 @@
+import { AccountState, AccountStateSDKType } from "./accountplus";
 import { Params, ParamsSDKType } from "./params";
 import { AccountAuthenticator, AccountAuthenticatorSDKType } from "./models";
 import * as _m0 from "protobufjs/minimal";
 import { DeepPartial, Long } from "../../helpers";
+/** AccountStateRequest is request type for the Query/AccountState RPC method. */
+
+export interface AccountStateRequest {
+  /** AccountStateRequest is request type for the Query/AccountState RPC method. */
+  address: string;
+}
+/** AccountStateRequest is request type for the Query/AccountState RPC method. */
+
+export interface AccountStateRequestSDKType {
+  /** AccountStateRequest is request type for the Query/AccountState RPC method. */
+  address: string;
+}
+/**
+ * AccountStateResponse is response type for the Query/GetAccountState RPC
+ * method.
+ */
+
+export interface AccountStateResponse {
+  /**
+   * AccountStateResponse is response type for the Query/GetAccountState RPC
+   * method.
+   */
+  accountState?: AccountState;
+}
+/**
+ * AccountStateResponse is response type for the Query/GetAccountState RPC
+ * method.
+ */
+
+export interface AccountStateResponseSDKType {
+  /**
+   * AccountStateResponse is response type for the Query/GetAccountState RPC
+   * method.
+   */
+  account_state?: AccountStateSDKType;
+}
 /** QueryParamsRequest is request type for the Query/Params RPC method. */
 
 export interface QueryParamsRequest {}
@@ -65,6 +102,96 @@ export interface GetAuthenticatorResponseSDKType {
   account_authenticator?: AccountAuthenticatorSDKType;
 }
 
+function createBaseAccountStateRequest(): AccountStateRequest {
+  return {
+    address: ""
+  };
+}
+
+export const AccountStateRequest = {
+  encode(message: AccountStateRequest, writer: _m0.Writer = _m0.Writer.create()): _m0.Writer {
+    if (message.address !== "") {
+      writer.uint32(10).string(message.address);
+    }
+
+    return writer;
+  },
+
+  decode(input: _m0.Reader | Uint8Array, length?: number): AccountStateRequest {
+    const reader = input instanceof _m0.Reader ? input : new _m0.Reader(input);
+    let end = length === undefined ? reader.len : reader.pos + length;
+    const message = createBaseAccountStateRequest();
+
+    while (reader.pos < end) {
+      const tag = reader.uint32();
+
+      switch (tag >>> 3) {
+        case 1:
+          message.address = reader.string();
+          break;
+
+        default:
+          reader.skipType(tag & 7);
+          break;
+      }
+    }
+
+    return message;
+  },
+
+  fromPartial(object: DeepPartial<AccountStateRequest>): AccountStateRequest {
+    const message = createBaseAccountStateRequest();
+    message.address = object.address ?? "";
+    return message;
+  }
+
+};
+
+function createBaseAccountStateResponse(): AccountStateResponse {
+  return {
+    accountState: undefined
+  };
+}
+
+export const AccountStateResponse = {
+  encode(message: AccountStateResponse, writer: _m0.Writer = _m0.Writer.create()): _m0.Writer {
+    if (message.accountState !== undefined) {
+      AccountState.encode(message.accountState, writer.uint32(10).fork()).ldelim();
+    }
+
+    return writer;
+  },
+
+  decode(input: _m0.Reader | Uint8Array, length?: number): AccountStateResponse {
+    const reader = input instanceof _m0.Reader ? input : new _m0.Reader(input);
+    let end = length === undefined ? reader.len : reader.pos + length;
+    const message = createBaseAccountStateResponse();
+
+    while (reader.pos < end) {
+      const tag = reader.uint32();
+
+      switch (tag >>> 3) {
+        case 1:
+          message.accountState = AccountState.decode(reader, reader.uint32());
+          break;
+
+        default:
+          reader.skipType(tag & 7);
+          break;
+      }
+    }
+
+    return message;
+  },
+
+  fromPartial(object: DeepPartial<AccountStateResponse>): AccountStateResponse {
+    const message = createBaseAccountStateResponse();
+    message.accountState = object.accountState !== undefined && object.accountState !== null ? AccountState.fromPartial(object.accountState) : undefined;
+    return message;
+  }
+
+};
+
 function createBaseQueryParamsRequest(): QueryParamsRequest {
   return {};
 }

proto/dydxprotocol/accountplus/query.proto

@@ -3,6 +3,7 @@ package dydxprotocol.accountplus;
 
 import "gogoproto/gogo.proto";
 import "google/api/annotations.proto";
+import "dydxprotocol/accountplus/accountplus.proto";
 import "dydxprotocol/accountplus/models.proto";
 import "dydxprotocol/accountplus/params.proto";
 
@@ -28,8 +29,21 @@ service Query {
     option (google.api.http).get =
         "/dydxprotocol/accountplus/authenticators/{account}";
   }
+
+  // Queries for an account state (timestamp nonce).
+  rpc AccountState(AccountStateRequest) returns (AccountStateResponse) {
+    option (google.api.http).get =
+        "/dydxprotocol/accountplus/account_state/{address}";
+  }
 }
 
+// AccountStateRequest is request type for the Query/AccountState RPC method.
+message AccountStateRequest { string address = 1; }
+
+// AccountStateResponse is response type for the Query/GetAccountState RPC
+// method.
+message AccountStateResponse { AccountState account_state = 1; }
+
 // QueryParamsRequest is request type for the Query/Params RPC method.
 message QueryParamsRequest {}
 

protocol/x/accountplus/client/cli/query.go

@@ -0,0 +1,144 @@
+package cli
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/spf13/cast"
+	"github.com/spf13/cobra"
+
+	"github.com/cosmos/cosmos-sdk/client"
+	"github.com/cosmos/cosmos-sdk/client/flags"
+	"github.com/dydxprotocol/v4-chain/protocol/x/accountplus/types"
+)
+
+// GetQueryCmd returns the cli query commands for this module.
+func GetQueryCmd() *cobra.Command {
+	// Group x/accountplus queries under a subcommand.
+	cmd := &cobra.Command{
+		Use:                        types.ModuleName,
+		Short:                      fmt.Sprintf("Querying commands for the %s module", types.ModuleName),
+		DisableFlagParsing:         true,
+		SuggestionsMinimumDistance: 2,
+		RunE:                       client.ValidateCmd,
+	}
+
+	cmd.AddCommand(
+		CmdQueryParam(),
+		CmdQueryGetAuthenticator(),
+		CmdQueryGetAllAuthenticators(),
+		CmdQueryAccountState(),
+	)
+	return cmd
+}
+
+func CmdQueryParam() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "param",
+		Short: "Get param",
+		Args:  cobra.ExactArgs(0),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			clientCtx, err := client.GetClientQueryContext(cmd)
+			if err != nil {
+				return err
+			}
+			queryClient := types.NewQueryClient(clientCtx)
+			res, err := queryClient.Params(
+				context.Background(),
+				&types.QueryParamsRequest{},
+			)
+			if err != nil {
+				return err
+			}
+			return clientCtx.PrintProto(res)
+		},
+	}
+	flags.AddQueryFlagsToCmd(cmd)
+	return cmd
+}
+
+func CmdQueryGetAuthenticator() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "get-authenticator [account] [authenticator_id]",
+		Short: "Get authenticator",
+		Args:  cobra.ExactArgs(2),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			clientCtx, err := client.GetClientQueryContext(cmd)
+			if err != nil {
+				return err
+			}
+			id, err := cast.ToUint64E(args[1])
+			if err != nil {
+				return err
+			}
+			queryClient := types.NewQueryClient(clientCtx)
+			res, err := queryClient.GetAuthenticator(
+				context.Background(),
+				&types.GetAuthenticatorRequest{
+					Account:         args[0],
+					AuthenticatorId: id,
+				},
+			)
+			if err != nil {
+				return err
+			}
+			return clientCtx.PrintProto(res)
+		},
+	}
+	flags.AddQueryFlagsToCmd(cmd)
+	return cmd
+}
+
+func CmdQueryGetAllAuthenticators() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "get-all-authenticators [account]",
+		Short: "Get all authenticators for an account",
+		Args:  cobra.ExactArgs(1),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			clientCtx, err := client.GetClientQueryContext(cmd)
+			if err != nil {
+				return err
+			}
+			queryClient := types.NewQueryClient(clientCtx)
+			res, err := queryClient.GetAuthenticators(
+				context.Background(),
+				&types.GetAuthenticatorsRequest{
+					Account: args[0],
+				},
+			)
+			if err != nil {
+				return err
+			}
+			return clientCtx.PrintProto(res)
+		},
+	}
+	flags.AddQueryFlagsToCmd(cmd)
+	return cmd
+}
+
+func CmdQueryAccountState() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "account-state [address]",
+		Short: "Get account state for an address",
+		Args:  cobra.ExactArgs(1),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			clientCtx, err := client.GetClientQueryContext(cmd)
+			if err != nil {
+				return err
+			}
+			queryClient := types.NewQueryClient(clientCtx)
+			res, err := queryClient.AccountState(
+				context.Background(),
+				&types.AccountStateRequest{
+					Address: args[0],
+				},
+			)
+			if err != nil {
+				return err
+			}
+			return clientCtx.PrintProto(res)
+		},
+	}
+	flags.AddQueryFlagsToCmd(cmd)
+	return cmd
+}