✨ Implement useful authz amino types

likecoin · Jun 15, 2023 · 1ac7fb6 · 1ac7fb6
1 parent 63b5bdf
commit 1ac7fb6
Show file tree

Hide file tree

Showing 2 changed files with 141 additions and 11 deletions.
diff --git a/src/messages/amino.ts b/src/messages/amino.ts
@@ -1,9 +1,11 @@
 import { AminoTypes, createDefaultAminoConverters } from '@cosmjs/stargate';
 import { LikeNftAminoTypes } from '@likecoin/iscn-message-types/dist/likechain/likenft/v1/amino';
 import { IscnAminoTypes } from '@likecoin/iscn-message-types/dist/likechain/iscn/amino';
+import { createAuthzAminoConverters } from './authz';
 
 export const aminoTypes = new AminoTypes({
   ...createDefaultAminoConverters(),
+  ...createAuthzAminoConverters(),
   ...LikeNftAminoTypes,
   ...IscnAminoTypes,
 });

diff --git a/src/messages/authz.ts b/src/messages/authz.ts
@@ -1,5 +1,8 @@
 // eslint-disable-next-line import/no-extraneous-dependencies
 import { EncodeObject } from '@cosmjs/proto-signing';
+import { AminoConverter } from '@cosmjs/stargate';
+import { GenericAuthorization } from 'cosmjs-types/cosmos/authz/v1beta1/authz';
+import { MsgGrant, MsgRevoke } from 'cosmjs-types/cosmos/authz/v1beta1/tx';
 import { SendAuthorization } from 'cosmjs-types/cosmos/bank/v1beta1/authz';
 import { MsgSend } from 'cosmjs-types/cosmos/bank/v1beta1/tx';
 import { Coin } from 'cosmjs-types/cosmos/base/v1beta1/coin';
@@ -41,9 +44,11 @@ export function formatMsgGrantSendAuthorization(
     senderAddress,
     granteeAddress,
     '/cosmos.bank.v1beta1.SendAuthorization',
-    SendAuthorization.encode(SendAuthorization.fromPartial({
-      spendLimit,
-    })).finish(),
+    SendAuthorization.encode(
+      SendAuthorization.fromPartial({
+        spendLimit,
+      }),
+    ).finish(),
     expirationDateInMs,
   );
 }
@@ -58,14 +63,18 @@ export function formatMsgExecSendAuthorization(
     typeUrl: '/cosmos.authz.v1beta1.MsgExec',
     value: {
       grantee: execAddress,
-      msgs: [{
-        typeUrl: '/cosmos.bank.v1beta1.MsgSend',
-        value: MsgSend.encode(MsgSend.fromPartial({
-          fromAddress: granterAddress,
-          toAddress,
-          amount: amounts,
-        })).finish(),
-      }],
+      msgs: [
+        {
+          typeUrl: '/cosmos.bank.v1beta1.MsgSend',
+          value: MsgSend.encode(
+            MsgSend.fromPartial({
+              fromAddress: granterAddress,
+              toAddress,
+              amount: amounts,
+            }),
+          ).finish(),
+        },
+      ],
     },
   };
   return message;
@@ -85,3 +94,122 @@ export function formatMsgRevokeSendAuthorization(
   };
   return message;
 }
+
+export function createAuthzAminoConverters(): Record<string, AminoConverter> {
+  return {
+    '/cosmos.authz.v1beta1.MsgGrant': {
+      aminoType: 'cosmos-sdk/MsgGrant',
+      toAmino: ({ granter, grantee, grant }: MsgGrant) => {
+        if (!grant || !grant.authorization) {
+          throw new Error(
+            `Unsupported grant type: '${grant?.authorization?.typeUrl}'`,
+          );
+        }
+        let authorizationValue;
+        switch (grant?.authorization?.typeUrl) {
+          case '/cosmos.authz.v1beta1.GenericAuthorization': {
+            const generic = GenericAuthorization.decode(
+              grant.authorization.value,
+            );
+            authorizationValue = {
+              type: 'cosmos-sdk/GenericAuthorization',
+              value: {
+                msg: generic.msg,
+              },
+            };
+            break;
+          }
+          case '/cosmos.bank.v1beta1.SendAuthorization': {
+            const spend = SendAuthorization.decode(grant.authorization.value);
+            authorizationValue = {
+              type: 'cosmos-sdk/SendAuthorization',
+              value: {
+                spend_limit: spend.spendLimit,
+              },
+            };
+            break;
+          }
+          default:
+            throw new Error(
+              `Unsupported grant type: '${grant.authorization.typeUrl}'`,
+            );
+        }
+        const expiration = grant.expiration?.seconds;
+        return {
+          granter,
+          grantee,
+          grant: {
+            authorization: authorizationValue,
+            expiration: expiration
+              ? new Date(expiration.toNumber() * 1000).toISOString().replace(/\.000Z$/, 'Z')
+              : undefined,
+          },
+        };
+      },
+      fromAmino: ({ granter, grantee, grant }: {
+        granter: string; grantee: string; grant: any;
+      }): MsgGrant => {
+        const authorizationType = grant?.authorization?.type;
+        let authorizationValue;
+        switch (authorizationType) {
+          case 'cosmos-sdk/GenericAuthorization': {
+            authorizationValue = {
+              typeUrl: '/cosmos.authz.v1beta1.GenericAuthorization',
+              value: GenericAuthorization
+                .encode({ msg: grant.authorization.value.msg })
+                .finish(),
+            };
+            break;
+          }
+          case 'cosmos-sdk/SendAuthorization': {
+            authorizationValue = {
+              typeUrl: '/cosmos.bank.v1beta1.SendAuthorization',
+              value: SendAuthorization
+                .encode(SendAuthorization
+                  .fromPartial({ spendLimit: grant.authorization.value.spend_limit }))
+                .finish(),
+            };
+            break;
+          }
+          default:
+            throw new Error(
+              `Unsupported grant type: '${grant?.authorization?.type}'`,
+            );
+        }
+        const expiration = grant.expiration
+          ? Date.parse(grant.expiration)
+          : undefined;
+        return MsgGrant.fromPartial({
+          granter,
+          grantee,
+          grant: {
+            authorization: authorizationValue,
+            expiration: expiration
+              ? Timestamp.fromPartial({
+                seconds: expiration / 1000,
+                nanos: (expiration % 1000) * 1e6,
+              })
+              : undefined,
+          },
+        });
+      },
+    },
+    '/cosmos.authz.v1beta1.MsgRevoke': {
+      aminoType: 'cosmos-sdk/MsgRevoke',
+      toAmino: ({ granter, grantee, msgTypeUrl }: MsgRevoke) => ({
+        granter,
+        grantee,
+        msg_type_url: msgTypeUrl,
+      }),
+      /* eslint-disable camelcase */
+      fromAmino: ({ granter, grantee, msg_type_url }: {
+        granter: string, grantee: string, msg_type_url: string
+      }): MsgRevoke => MsgRevoke.fromPartial({
+        granter,
+        grantee,
+        msgTypeUrl: msg_type_url,
+      }),
+      /* eslint-enable camelcase */
+    },
+  };
+}