From d2990da05c9fe06c96ef97707406d887bbd8bf12 Mon Sep 17 00:00:00 2001
From: rrr523 <gaoyibobo@foxmail.com>
Date: Thu, 14 Sep 2023 16:34:38 +0800
Subject: [PATCH] feat: Custom Tx

---
 .changeset/warm-clouds-decide.md              |  5 ++
 .../nextjs/src/components/customtx/index.tsx  | 63 +++++++--------
 packages/chain-sdk/src/clients/txClient.ts    | 80 +++++++++++++++++++
 packages/chain-sdk/src/types/tx.ts            | 19 +++++
 4 files changed, 135 insertions(+), 32 deletions(-)
 create mode 100644 .changeset/warm-clouds-decide.md

diff --git a/.changeset/warm-clouds-decide.md b/.changeset/warm-clouds-decide.md
new file mode 100644
index 00000000..028aaef4
--- /dev/null
+++ b/.changeset/warm-clouds-decide.md
@@ -0,0 +1,5 @@
+---
+'@bnb-chain/greenfield-js-sdk': patch
+---
+
+feat: Custom any tx by raw info
diff --git a/examples/nextjs/src/components/customtx/index.tsx b/examples/nextjs/src/components/customtx/index.tsx
index 6417a914..f6b0ee0f 100644
--- a/examples/nextjs/src/components/customtx/index.tsx
+++ b/examples/nextjs/src/components/customtx/index.tsx
@@ -1,7 +1,6 @@
 import { client } from '@/client';
-import { useAccount } from 'wagmi';
 import { MsgSend } from '@bnb-chain/greenfield-cosmos-types/cosmos/bank/v1beta1/tx';
-import { parseEther } from 'viem';
+import { useAccount } from 'wagmi';
 
 export const CustomTx = () => {
   const { address } = useAccount();
@@ -14,61 +13,63 @@ export const CustomTx = () => {
       <button
         onClick={async () => {
           if (!address) return;
-          const transferParams: MsgSend = {
-            amount: [
-              {
-                denom: 'BNB',
-                amount: '100000000',
-              },
-            ],
-            fromAddress: address,
-            toAddress: '0x0000000000000000000000000000000000000001',
-          };
-
-          const transferTx = await client.txClient.tx(
-            '/cosmos.bank.v1beta1.MsgSend',
+          const tx = await client.txClient.txRaw({
             address,
-            {
+            eip712MsgType: {
               Msg1: [
                 {
-                  name: 'type',
+                  name: 'expiration_time',
                   type: 'string',
                 },
                 {
-                  name: 'from_address',
+                  name: 'operator',
                   type: 'string',
                 },
                 {
-                  name: 'to_address',
+                  name: 'principal',
+                  type: 'TypeMsg1Principal',
+                },
+                {
+                  name: 'resource',
                   type: 'string',
                 },
                 {
-                  name: 'amount',
-                  type: 'TypeMsg1Amount[]',
+                  name: 'type',
+                  type: 'string',
                 },
               ],
-              TypeMsg1Amount: [
+              TypeMsg1Principal: [
                 {
-                  name: 'denom',
+                  name: 'type',
                   type: 'string',
                 },
                 {
-                  name: 'amount',
+                  name: 'value',
                   type: 'string',
                 },
               ],
             },
-            MsgSend.toSDK(transferParams),
-            MsgSend.encode(transferParams).finish(),
-          );
+            msgData: {
+              expiration_time: '',
+              operator: '0x1C893441AB6c1A75E01887087ea508bE8e07AAae',
+              principal: {
+                type: 'PRINCIPAL_TYPE_GNFD_ACCOUNT',
+                value: '0x91D7deA99716Cbb247E81F1cfB692009164a967E',
+              },
+              resource: 'grn:o::foo/vvv',
+              type: '/greenfield.storage.MsgPutPolicy',
+            },
+            txRawHex:
+              '0x0a93010a90010a202f677265656e6669656c642e73746f726167652e4d7367507574506f6c696379126c0a2a307831433839333434314142366331413735453031383837303837656135303862453865303741416165122e0801122a3078393144376465413939373136436262323437453831463163664236393230303931363461393637451a0e67726e3a6f3a3a666f6f2f76767612021200',
+          });
 
-          const simulateInfo = await transferTx.simulate({
+          const simulateInfo = await tx.simulate({
             denom: 'BNB',
           });
 
-          console.log('simulateInfo', simulateInfo);
+          console.log(simulateInfo);
 
-          const res = await transferTx.broadcast({
+          const res = await tx.broadcast({
             denom: 'BNB',
             gasLimit: Number(simulateInfo?.gasLimit),
             gasPrice: simulateInfo?.gasPrice || '5000000000',
@@ -76,8 +77,6 @@ export const CustomTx = () => {
             granter: '',
           });
 
-          console.log('res', res);
-
           if (res.code === 0) {
             alert('success');
           }
diff --git a/packages/chain-sdk/src/clients/txClient.ts b/packages/chain-sdk/src/clients/txClient.ts
index 859a16ca..1fa4ad21 100644
--- a/packages/chain-sdk/src/clients/txClient.ts
+++ b/packages/chain-sdk/src/clients/txClient.ts
@@ -22,6 +22,7 @@ import { signTypedData, SignTypedDataVersion } from '@metamask/eth-sig-util';
 import { container, inject, injectable } from 'tsyringe';
 import {
   BroadcastOptions,
+  CustomTx,
   ISimulateGasFee,
   MetaTxInfo,
   SignOptions,
@@ -66,6 +67,13 @@ export interface ITxClient {
     msgBytes: MetaTxInfo['msgBytes'],
   ): Promise<TxResponse>;
 
+  txRaw({
+    address,
+    txRawHex,
+    eip712MsgType,
+    msgData,
+  }: CustomTx): Promise<Omit<TxResponse, 'metaTxInfo'>>;
+
   /**
    *
    */
@@ -125,6 +133,78 @@ export class TxClient implements ITxClient {
     };
   }
 
+  public async txRaw({
+    address,
+    txRawHex,
+    eip712MsgType,
+    msgData,
+  }: CustomTx): Promise<Omit<TxResponse, 'metaTxInfo'>> {
+    const accountInfo = await this.account.getAccount(address);
+    const txRawBytes = arrayify(txRawHex);
+    const txRawP = TxRaw.decode(txRawBytes);
+
+    return {
+      simulate: async (opts: SimulateOptions) => {
+        return await this.simulateRawTx(txRawP.bodyBytes, accountInfo, opts);
+      },
+      broadcast: async (opts: BroadcastOptions) => {
+        const {
+          denom,
+          gasLimit,
+          gasPrice,
+          payer,
+          granter,
+          privateKey,
+          signTypedDataCallback = defaultSignTypedData,
+        } = opts;
+
+        const fee = generateFee(
+          String(BigInt(gasLimit) * BigInt(gasPrice)),
+          denom,
+          String(gasLimit),
+          payer,
+          granter,
+        );
+        const wrapperTypes = generateTypes(eip712MsgType);
+        const messages = generateMessage(
+          accountInfo.accountNumber.toString(),
+          accountInfo.sequence.toString(),
+          this.chainId,
+          '',
+          fee,
+          msgData,
+          '0',
+        );
+
+        const eip712 = createEIP712(wrapperTypes, this.chainId, messages);
+        // console.log('eip712', eip712);
+        const { pubKey, signature } = privateKey
+          ? this.getSignByPriKey(eip712, privateKey)
+          : await this.getSignByWallet(eip712, accountInfo.address, signTypedDataCallback);
+
+        const authInfoBytes = this.getAuthInfoBytes({
+          denom,
+          sequence: accountInfo.sequence + '',
+          gasLimit,
+          gasPrice,
+          pubKey,
+          granter,
+          payer,
+        });
+
+        const txRaw = TxRaw.fromPartial({
+          bodyBytes: txRawP.bodyBytes,
+          authInfoBytes,
+          signatures: [arrayify(signature)],
+        });
+        const txBytes = TxRaw.encode(txRaw).finish();
+
+        // console.log('txBytes', hexlify(txBytes));
+        return await this.broadcastRawTx(txBytes);
+      },
+    };
+  }
+
   private getBodyBytes(params: { typeUrl: string; msgBytes: Uint8Array }[]) {
     const multiMsgBytes = params.map((tx) => {
       return generateMsg(tx.typeUrl, tx.msgBytes);
diff --git a/packages/chain-sdk/src/types/tx.ts b/packages/chain-sdk/src/types/tx.ts
index a056c6b6..17336079 100644
--- a/packages/chain-sdk/src/types/tx.ts
+++ b/packages/chain-sdk/src/types/tx.ts
@@ -44,3 +44,22 @@ export type TxResponse = {
   broadcast: (opts: Readonly<BroadcastOptions>) => Promise<DeliverTxResponse>;
   metaTxInfo: MetaTxInfo;
 };
+
+export type CustomTx = {
+  /**
+   * account
+   */
+  address: string;
+  /**
+   * txRaw infomation (hex string)
+   */
+  txRawHex: string;
+  /**
+   * EIP712 structure
+   */
+  eip712MsgType: object;
+  /**
+   * tx message
+   */
+  msgData: object;
+};