* clear

* feat: Add everything

* fix: Commenht
This commit is contained in:
Prad Nukala
2025-10-03 14:45:52 -04:00
committed by GitHub
parent 43b4a11c06
commit 13e6c3e84d
1935 changed files with 655061 additions and 40058 deletions
+286
View File
@@ -0,0 +1,286 @@
package cli
import (
"context"
"fmt"
"github.com/cosmos/cosmos-sdk/client"
"github.com/cosmos/cosmos-sdk/client/flags"
"github.com/spf13/cobra"
"github.com/sonr-io/sonr/x/dex/types"
)
// NewQueryCmd creates and returns the query command
func NewQueryCmd() *cobra.Command {
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(
CmdQueryParams(),
CmdQueryAccount(),
CmdQueryAccounts(),
CmdQueryBalance(),
CmdQueryPool(),
CmdQueryOrders(),
CmdQueryHistory(),
)
return cmd
}
// CmdQueryParams queries the module parameters
func CmdQueryParams() *cobra.Command {
cmd := &cobra.Command{
Use: "params",
Short: "Query the current DEX module parameters",
Args: cobra.NoArgs,
RunE: func(cmd *cobra.Command, _ []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
}
// CmdQueryAccount queries a DEX account
func CmdQueryAccount() *cobra.Command {
cmd := &cobra.Command{
Use: "account [did] [connection-id]",
Short: "Query a DEX account by DID and connection",
Args: cobra.ExactArgs(2),
RunE: func(cmd *cobra.Command, args []string) error {
clientCtx, err := client.GetClientQueryContext(cmd)
if err != nil {
return err
}
did := args[0]
connectionID := args[1]
queryClient := types.NewQueryClient(clientCtx)
res, err := queryClient.Account(context.Background(), &types.QueryAccountRequest{
Did: did,
ConnectionId: connectionID,
})
if err != nil {
return err
}
return clientCtx.PrintProto(res)
},
}
flags.AddQueryFlagsToCmd(cmd)
return cmd
}
// CmdQueryAccounts queries all DEX accounts
func CmdQueryAccounts() *cobra.Command {
cmd := &cobra.Command{
Use: "accounts",
Short: "Query all DEX accounts",
Args: cobra.NoArgs,
RunE: func(cmd *cobra.Command, _ []string) error {
clientCtx, err := client.GetClientQueryContext(cmd)
if err != nil {
return err
}
did, _ := cmd.Flags().GetString("did")
queryClient := types.NewQueryClient(clientCtx)
pageReq, err := client.ReadPageRequest(cmd.Flags())
if err != nil {
return err
}
res, err := queryClient.Accounts(context.Background(), &types.QueryAccountsRequest{
Did: did,
Pagination: pageReq,
})
if err != nil {
return err
}
return clientCtx.PrintProto(res)
},
}
cmd.Flags().String("did", "", "Filter by DID")
flags.AddQueryFlagsToCmd(cmd)
flags.AddPaginationFlagsToCmd(cmd, "accounts")
return cmd
}
// CmdQueryBalance queries remote chain balances
func CmdQueryBalance() *cobra.Command {
cmd := &cobra.Command{
Use: "balance [did] [connection-id]",
Short: "Query remote chain balances for a DID",
Args: cobra.ExactArgs(2),
RunE: func(cmd *cobra.Command, args []string) error {
clientCtx, err := client.GetClientQueryContext(cmd)
if err != nil {
return err
}
did := args[0]
connectionID := args[1]
denom, _ := cmd.Flags().GetString("denom")
queryClient := types.NewQueryClient(clientCtx)
res, err := queryClient.Balance(context.Background(), &types.QueryBalanceRequest{
Did: did,
ConnectionId: connectionID,
Denom: denom,
})
if err != nil {
return err
}
return clientCtx.PrintProto(res)
},
}
cmd.Flags().String("denom", "", "Filter by specific denomination")
flags.AddQueryFlagsToCmd(cmd)
return cmd
}
// CmdQueryPool queries pool information
func CmdQueryPool() *cobra.Command {
cmd := &cobra.Command{
Use: "pool [connection-id] [pool-id]",
Short: "Query pool information on a remote chain",
Args: cobra.ExactArgs(2),
RunE: func(cmd *cobra.Command, args []string) error {
clientCtx, err := client.GetClientQueryContext(cmd)
if err != nil {
return err
}
connectionID := args[0]
poolID := args[1]
queryClient := types.NewQueryClient(clientCtx)
res, err := queryClient.Pool(context.Background(), &types.QueryPoolRequest{
ConnectionId: connectionID,
PoolId: poolID,
})
if err != nil {
return err
}
return clientCtx.PrintProto(res)
},
}
flags.AddQueryFlagsToCmd(cmd)
return cmd
}
// CmdQueryOrders queries orders for a DID
func CmdQueryOrders() *cobra.Command {
cmd := &cobra.Command{
Use: "orders [did] [connection-id]",
Short: "Query orders for a DID on a specific connection",
Args: cobra.ExactArgs(2),
RunE: func(cmd *cobra.Command, args []string) error {
clientCtx, err := client.GetClientQueryContext(cmd)
if err != nil {
return err
}
did := args[0]
connectionID := args[1]
status, _ := cmd.Flags().GetString("status")
queryClient := types.NewQueryClient(clientCtx)
pageReq, err := client.ReadPageRequest(cmd.Flags())
if err != nil {
return err
}
res, err := queryClient.Orders(context.Background(), &types.QueryOrdersRequest{
Did: did,
ConnectionId: connectionID,
Status: status,
Pagination: pageReq,
})
if err != nil {
return err
}
return clientCtx.PrintProto(res)
},
}
cmd.Flags().String("status", "", "Filter by order status (pending|open|filled|cancelled)")
flags.AddQueryFlagsToCmd(cmd)
flags.AddPaginationFlagsToCmd(cmd, "orders")
return cmd
}
// CmdQueryHistory queries transaction history
func CmdQueryHistory() *cobra.Command {
cmd := &cobra.Command{
Use: "history [did]",
Short: "Query transaction history for a DID",
Args: cobra.ExactArgs(1),
RunE: func(cmd *cobra.Command, args []string) error {
clientCtx, err := client.GetClientQueryContext(cmd)
if err != nil {
return err
}
did := args[0]
connectionID, _ := cmd.Flags().GetString("connection")
operationType, _ := cmd.Flags().GetString("type")
queryClient := types.NewQueryClient(clientCtx)
pageReq, err := client.ReadPageRequest(cmd.Flags())
if err != nil {
return err
}
res, err := queryClient.History(context.Background(), &types.QueryHistoryRequest{
Did: did,
ConnectionId: connectionID,
OperationType: operationType,
Pagination: pageReq,
})
if err != nil {
return err
}
return clientCtx.PrintProto(res)
},
}
cmd.Flags().String("connection", "", "Filter by connection ID")
cmd.Flags().String("type", "", "Filter by transaction type (swap|liquidity|order)")
flags.AddQueryFlagsToCmd(cmd)
flags.AddPaginationFlagsToCmd(cmd, "history")
return cmd
}
+325
View File
@@ -0,0 +1,325 @@
package cli
import (
"fmt"
"strconv"
"strings"
"time"
"cosmossdk.io/math"
"github.com/cosmos/cosmos-sdk/client"
"github.com/cosmos/cosmos-sdk/client/flags"
"github.com/cosmos/cosmos-sdk/client/tx"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/spf13/cobra"
"github.com/sonr-io/sonr/x/dex/types"
)
// NewTxCmd creates and returns the tx command
func NewTxCmd() *cobra.Command {
cmd := &cobra.Command{
Use: types.ModuleName,
Short: fmt.Sprintf("%s transactions subcommands", types.ModuleName),
DisableFlagParsing: true,
SuggestionsMinimumDistance: 2,
RunE: client.ValidateCmd,
}
cmd.AddCommand(
CmdRegisterDEXAccount(),
CmdExecuteSwap(),
CmdProvideLiquidity(),
CmdRemoveLiquidity(),
CmdCreateLimitOrder(),
CmdCancelOrder(),
)
return cmd
}
// CmdRegisterDEXAccount returns a command to register a DEX account
func CmdRegisterDEXAccount() *cobra.Command {
cmd := &cobra.Command{
Use: "register-account [did] [connection-id] [features]",
Short: "Register a new ICA account for DEX operations",
Args: cobra.ExactArgs(3),
RunE: func(cmd *cobra.Command, args []string) error {
clientCtx, err := client.GetClientTxContext(cmd)
if err != nil {
return err
}
did := args[0]
connectionID := args[1]
features := strings.Split(args[2], ",")
msg := &types.MsgRegisterDEXAccount{
Did: did,
ConnectionId: connectionID,
Features: features,
}
if err := msg.ValidateBasic(); err != nil {
return err
}
return tx.GenerateOrBroadcastTxCLI(clientCtx, cmd.Flags(), msg)
},
}
flags.AddTxFlagsToCmd(cmd)
return cmd
}
// CmdExecuteSwap returns a command to execute a swap
func CmdExecuteSwap() *cobra.Command {
cmd := &cobra.Command{
Use: "swap [did] [connection-id] [token-in] [token-out-denom] [min-amount-out] [pool-id]",
Short: "Execute a token swap through ICA",
Args: cobra.ExactArgs(6),
RunE: func(cmd *cobra.Command, args []string) error {
clientCtx, err := client.GetClientTxContext(cmd)
if err != nil {
return err
}
did := args[0]
connectionID := args[1]
tokenIn, err := sdk.ParseCoinNormalized(args[2])
if err != nil {
return fmt.Errorf("invalid token-in: %w", err)
}
tokenOutDenom := args[3]
minAmountOut, ok := math.NewIntFromString(args[4])
if !ok {
return fmt.Errorf("invalid min-amount-out: %s", args[4])
}
poolID, err := strconv.ParseUint(args[5], 10, 64)
if err != nil {
return fmt.Errorf("invalid pool-id: %w", err)
}
msg := &types.MsgExecuteSwap{
Did: did,
ConnectionId: connectionID,
SourceDenom: tokenIn.Denom,
TargetDenom: tokenOutDenom,
Amount: tokenIn.Amount,
MinAmountOut: minAmountOut,
Route: fmt.Sprintf("pool:%d", poolID),
}
if err := msg.ValidateBasic(); err != nil {
return err
}
return tx.GenerateOrBroadcastTxCLI(clientCtx, cmd.Flags(), msg)
},
}
flags.AddTxFlagsToCmd(cmd)
return cmd
}
// CmdProvideLiquidity returns a command to provide liquidity
func CmdProvideLiquidity() *cobra.Command {
cmd := &cobra.Command{
Use: "provide-liquidity [did] [connection-id] [pool-id] [token-a] [token-b] [min-shares]",
Short: "Provide liquidity to a pool through ICA",
Args: cobra.ExactArgs(6),
RunE: func(cmd *cobra.Command, args []string) error {
clientCtx, err := client.GetClientTxContext(cmd)
if err != nil {
return err
}
did := args[0]
connectionID := args[1]
poolID, err := strconv.ParseUint(args[2], 10, 64)
if err != nil {
return fmt.Errorf("invalid pool-id: %w", err)
}
tokenA, err := sdk.ParseCoinNormalized(args[3])
if err != nil {
return fmt.Errorf("invalid token-a: %w", err)
}
tokenB, err := sdk.ParseCoinNormalized(args[4])
if err != nil {
return fmt.Errorf("invalid token-b: %w", err)
}
minShares, ok := math.NewIntFromString(args[5])
if !ok {
return fmt.Errorf("invalid min-shares: %s", args[5])
}
msg := &types.MsgProvideLiquidity{
Did: did,
ConnectionId: connectionID,
PoolId: fmt.Sprintf("%d", poolID),
Assets: sdk.NewCoins(tokenA, tokenB),
MinShares: minShares,
Timeout: time.Now().Add(5 * time.Minute),
}
if err := msg.ValidateBasic(); err != nil {
return err
}
return tx.GenerateOrBroadcastTxCLI(clientCtx, cmd.Flags(), msg)
},
}
flags.AddTxFlagsToCmd(cmd)
return cmd
}
// CmdRemoveLiquidity returns a command to remove liquidity
func CmdRemoveLiquidity() *cobra.Command {
cmd := &cobra.Command{
Use: "remove-liquidity [did] [connection-id] [pool-id] [shares] [min-amount-a] [min-amount-b]",
Short: "Remove liquidity from a pool through ICA",
Args: cobra.ExactArgs(6),
RunE: func(cmd *cobra.Command, args []string) error {
clientCtx, err := client.GetClientTxContext(cmd)
if err != nil {
return err
}
did := args[0]
connectionID := args[1]
poolID, err := strconv.ParseUint(args[2], 10, 64)
if err != nil {
return fmt.Errorf("invalid pool-id: %w", err)
}
shares, ok := math.NewIntFromString(args[3])
if !ok {
return fmt.Errorf("invalid shares: %s", args[3])
}
minAmountA, ok := math.NewIntFromString(args[4])
if !ok {
return fmt.Errorf("invalid min-amount-a: %s", args[4])
}
minAmountB, ok := math.NewIntFromString(args[5])
if !ok {
return fmt.Errorf("invalid min-amount-b: %s", args[5])
}
msg := &types.MsgRemoveLiquidity{
Did: did,
ConnectionId: connectionID,
PoolId: fmt.Sprintf("%d", poolID),
Shares: shares,
MinAmounts: sdk.NewCoins(
sdk.NewCoin("token", minAmountA),
sdk.NewCoin("token", minAmountB),
),
Timeout: time.Now().Add(5 * time.Minute),
}
if err := msg.ValidateBasic(); err != nil {
return err
}
return tx.GenerateOrBroadcastTxCLI(clientCtx, cmd.Flags(), msg)
},
}
flags.AddTxFlagsToCmd(cmd)
return cmd
}
// CmdCreateLimitOrder returns a command to create a limit order
func CmdCreateLimitOrder() *cobra.Command {
cmd := &cobra.Command{
Use: "create-order [did] [connection-id] [token-in] [token-out-denom] [price]",
Short: "Create a limit order through ICA",
Args: cobra.ExactArgs(5),
RunE: func(cmd *cobra.Command, args []string) error {
clientCtx, err := client.GetClientTxContext(cmd)
if err != nil {
return err
}
did := args[0]
connectionID := args[1]
tokenIn, err := sdk.ParseCoinNormalized(args[2])
if err != nil {
return fmt.Errorf("invalid token-in: %w", err)
}
tokenOutDenom := args[3]
price, err := math.LegacyNewDecFromStr(args[4])
if err != nil {
return fmt.Errorf("invalid price: %w", err)
}
msg := &types.MsgCreateLimitOrder{
Did: did,
ConnectionId: connectionID,
SellDenom: tokenIn.Denom,
BuyDenom: tokenOutDenom,
Amount: tokenIn.Amount,
Price: price,
Expiration: time.Now().Add(24 * time.Hour),
}
if err := msg.ValidateBasic(); err != nil {
return err
}
return tx.GenerateOrBroadcastTxCLI(clientCtx, cmd.Flags(), msg)
},
}
flags.AddTxFlagsToCmd(cmd)
return cmd
}
// CmdCancelOrder returns a command to cancel an order
func CmdCancelOrder() *cobra.Command {
cmd := &cobra.Command{
Use: "cancel-order [did] [connection-id] [order-id]",
Short: "Cancel an existing order through ICA",
Args: cobra.ExactArgs(3),
RunE: func(cmd *cobra.Command, args []string) error {
clientCtx, err := client.GetClientTxContext(cmd)
if err != nil {
return err
}
did := args[0]
connectionID := args[1]
orderID := args[2]
msg := &types.MsgCancelOrder{
Did: did,
ConnectionId: connectionID,
OrderId: orderID,
}
if err := msg.ValidateBasic(); err != nil {
return err
}
return tx.GenerateOrBroadcastTxCLI(clientCtx, cmd.Flags(), msg)
},
}
flags.AddTxFlagsToCmd(cmd)
return cmd
}