feat: add x/authz message parser (#97)

forbole · Jul 4, 2023 · 2aa8d96 · 2aa8d96
1 parent 007d6e4
commit 2aa8d96
Show file tree

Hide file tree

Showing 3 changed files with 94 additions and 5 deletions.
diff --git a/.github/workflows/lint.yml b/.github/workflows/lint.yml
@@ -18,7 +18,7 @@ jobs:
         uses: actions/checkout@v3
 
       - name: Compute diff 📜
-        uses: technote-space/get-diff-action@v6.1.0
+        uses: technote-space/get-diff-action@v6.1.2
         with:
           SUFFIX_FILTER: |
             .go
@@ -27,9 +27,9 @@ jobs:
 
       - name: Setup Go 🧰
         if: "env.GIT_DIFF != ''"
-        uses: actions/setup-go@v3
+        uses: actions/setup-go@v4
         with:
-          go-version: 1.18
+          go-version: "1.20"
 
       - name: Run lint ✅
         if: "env.GIT_DIFF != ''"

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -24,9 +24,9 @@ jobs:
         uses: actions/checkout@v3
 
       - name: Setup Go 🧰
-        uses: actions/setup-go@v3
+        uses: actions/setup-go@v4
         with:
-          go-version: 1.18
+          go-version: "1.20"
 
       - name: Compute diff 📜
         uses: technote-space/get-diff-action@v6.1.0

diff --git a/modules/messages/account_parser.go b/modules/messages/account_parser.go
@@ -2,6 +2,7 @@ package messages
 
 import (
 	"fmt"
+	"strings"
 
 	"github.com/gogo/protobuf/proto"
 
@@ -14,6 +15,7 @@ import (
 
 	"github.com/cosmos/cosmos-sdk/codec"
 	sdk "github.com/cosmos/cosmos-sdk/types"
+	authztypes "github.com/cosmos/cosmos-sdk/x/authz"
 	banktypes "github.com/cosmos/cosmos-sdk/x/bank/types"
 	distrtypes "github.com/cosmos/cosmos-sdk/x/distribution/types"
 	stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
@@ -55,6 +57,7 @@ var CosmosMessageAddressesParser = JoinMessageParsers(
 	IBCTransferMessagesParser,
 	SlashingMessagesParser,
 	StakingMessagesParser,
+	AuthzMessageParser,
 	DefaultMessagesParser,
 )
 
@@ -230,3 +233,89 @@ func StakingMessagesParser(_ codec.Codec, cosmosMsg sdk.Msg) ([]string, error) {
 
 	return nil, MessageNotSupported(cosmosMsg)
 }
+
+// AuthzMessageParser returns the list of all the accounts involved in the given
+// message if it's related to the x/authz module
+func AuthzMessageParser(c codec.Codec, cosmosMsg sdk.Msg) ([]string, error) {
+	switch msg := cosmosMsg.(type) {
+	case *authztypes.MsgGrant:
+		return []string{msg.Grantee, msg.Granter}, nil
+	case *authztypes.MsgRevoke:
+		return []string{msg.Grantee, msg.Granter}, nil
+	case *authztypes.MsgExec:
+		for _, index := range msg.Msgs {
+			var executedMsg sdk.Msg
+			if err := c.UnpackAny(index, &executedMsg); err != nil {
+				return nil, fmt.Errorf("error while unpacking message from authz: %s", err)
+			}
+
+			switch {
+			case strings.Contains(index.TypeUrl, "bank"):
+				addresses, err := BankMessagesParser(c, executedMsg)
+				if err != nil {
+					return nil, MessageNotSupported(executedMsg)
+				}
+				addresses = append(addresses, msg.Grantee)
+				return addresses, nil
+			case strings.Contains(index.TypeUrl, "crisis"):
+				addresses, err := CrisisMessagesParser(c, executedMsg)
+				if err != nil {
+					return nil, MessageNotSupported(executedMsg)
+				}
+				addresses = append(addresses, msg.Grantee)
+				return addresses, nil
+			case strings.Contains(index.TypeUrl, "distribution"):
+				addresses, err := DistributionMessagesParser(c, executedMsg)
+				if err != nil {
+					return nil, MessageNotSupported(executedMsg)
+				}
+				addresses = append(addresses, msg.Grantee)
+				return addresses, nil
+			case strings.Contains(index.TypeUrl, "evidence"):
+				addresses, err := EvidenceMessagesParser(c, executedMsg)
+				if err != nil {
+					return nil, MessageNotSupported(executedMsg)
+				}
+				addresses = append(addresses, msg.Grantee)
+				return addresses, nil
+			case strings.Contains(index.TypeUrl, "gov"):
+				addresses, err := GovMessagesParser(c, executedMsg)
+				if err != nil {
+					return nil, MessageNotSupported(executedMsg)
+				}
+				addresses = append(addresses, msg.Grantee)
+				return addresses, nil
+			case strings.Contains(index.TypeUrl, "ibc"):
+				addresses, err := IBCTransferMessagesParser(c, executedMsg)
+				if err != nil {
+					return nil, MessageNotSupported(executedMsg)
+				}
+				addresses = append(addresses, msg.Grantee)
+				return addresses, nil
+			case strings.Contains(index.TypeUrl, "slashing"):
+				addresses, err := SlashingMessagesParser(c, executedMsg)
+				if err != nil {
+					return nil, MessageNotSupported(executedMsg)
+				}
+				addresses = append(addresses, msg.Grantee)
+				return addresses, nil
+			case strings.Contains(index.TypeUrl, "staking"):
+				addresses, err := StakingMessagesParser(c, executedMsg)
+				if err != nil {
+					return nil, MessageNotSupported(executedMsg)
+				}
+				addresses = append(addresses, msg.Grantee)
+				return addresses, nil
+			default:
+				addresses, err := DefaultMessagesParser(c, executedMsg)
+				if err != nil {
+					return nil, MessageNotSupported(executedMsg)
+				}
+				addresses = append(addresses, msg.Grantee)
+				return addresses, nil
+			}
+		}
+	}
+
+	return nil, MessageNotSupported(cosmosMsg)
+}