feat: implement Keystone Babylon integration with Cosmos signing support and Keplr compatibility

Author: slient-coder

package.json

@@ -43,6 +43,8 @@
     "@keystonehq/hw-app-base": "0.1.7",
     "@keystonehq/hw-transport-webusb": "0.4.0",
     "@metamask/obs-store": "^7.0.0",
+    "@noble/curves": "^1.0.0",
+    "@noble/hashes": "^1.7.1",
     "@pmmmwh/react-refresh-webpack-plugin": "^0.5.3",
     "@reduxjs/toolkit": "^1.9.2",
     "@sentry/browser": "^6.19.7",
@@ -59,7 +61,7 @@
     "@types/react": "^18.0.9",
     "@types/react-cache": "^2.0.1",
     "@types/react-dom": "^18.0.4",
-    "@unisat/wallet-sdk": "1.8.0",
+    "@unisat/wallet-sdk": "1.9.0",
     "antd": "^4.20.4",
     "antd-dayjs-webpack-plugin": "1.0.6",
     "assert": "^2.0.0",

src/background/controller/provider/controller.ts

@@ -1,20 +1,18 @@
-
 import { permissionService, sessionService } from '@/background/service';
 import { CHAINS, CHAINS_MAP, NETWORK_TYPES, VERSION } from '@/shared/constant';
-
 import { NetworkType } from '@/shared/types';
-import { getChainInfo } from '@/shared/utils';
+import { getChainInfo, objToUint8Array } from '@/shared/utils';
 import { amountToSatoshis } from '@/ui/utils';
+import * as encoding from '@cosmjs/encoding';
 import { bitcoin } from '@unisat/wallet-sdk/lib/bitcoin-core';
 import { verifyMessageOfBIP322Simple } from '@unisat/wallet-sdk/lib/message';
 import { toPsbtNetwork } from '@unisat/wallet-sdk/lib/network';
 import { ethErrors } from 'eth-rpc-errors';
 import BaseController from '../base';
 import wallet from '../wallet';
 
-import {
-  SignDoc
-} from "@keplr-wallet/proto-types/cosmos/tx/v1beta1/tx";
+import { encodeSecp256k1Signature, makeADR36AminoSignDoc, serializeSignDoc } from '@/background/service/keyring/CosmosKeyring';
+import { makeSignBytes } from '@cosmjs/proto-signing';
 
 function formatPsbtHex(psbtHex: string) {
   let formatData = '';
@@ -197,7 +195,7 @@ class ProviderController extends BaseController {
     return await wallet.pushTx(rawtx)
   }
 
-  @Reflect.metadata('APPROVAL', ['SignText', () => {
+  @Reflect.metadata('APPROVAL', ['SignText', (req) => {
     // todo check text
   }])
   signMessage = async ({ data: { params: { text, type } }, approvalRes }) => {
@@ -303,6 +301,13 @@ class ProviderController extends BaseController {
     return utxos;
   };
 
+
+  private _isKeystoneWallet = async () => {
+    const currentKeyring = await wallet.getCurrentKeyring();
+    return currentKeyring?.type === 'keystone';
+  }
+
+
   @Reflect.metadata('APPROVAL', ['CosmosConnect', (req) => {
     // todo check
   }])
@@ -312,6 +317,7 @@ class ProviderController extends BaseController {
     }
   };
 
+
   @Reflect.metadata('SAFE', true)
   cosmosExperimentalSuggestChain = async ( {data:{params:{chainData}}} ) => {
     // const chainInfo:CosmosChainInfo = chainData;
@@ -322,76 +328,71 @@ class ProviderController extends BaseController {
     throw new Error('not implemented')
   }
 
+
   @Reflect.metadata('SAFE', true)
   cosmosGetKey = async ({ data: { params: { chainId} } }) => {
     const cosmosKeyring = await wallet.getCosmosKeyring(chainId);
-    if(!cosmosKeyring){
-      return null;
-    }
-
-    const key = cosmosKeyring.getKey();
-    const _key = Object.assign({},key,{
-      address:key.address.toString(),
-      pubKey:key.pubKey.toString()
-    })
-    return _key
+      if(!cosmosKeyring){
+        return null;
+      }
+
+      const key = cosmosKeyring.getKey();
+      const _key = Object.assign({},key,{
+        address:key.address.toString(),
+        pubKey:key.pubKey.toString()
+      });
+      return _key;
   }
 
+
   @Reflect.metadata('APPROVAL', ['CosmosSign', (req) => {
-    // todo check
+    const signDoc = req.data.params.signDoc;
+    signDoc.bodyBytes = objToUint8Array(signDoc.bodyBytes);
+    signDoc.authInfoBytes = objToUint8Array(signDoc.authInfoBytes);
+    const signBytes = makeSignBytes(signDoc);
+    req.data.params.signBytesHex = encoding.toHex(signBytes);
+    
   }])
-  cosmosSignDirect = async ({ data: { params:msg } }) => {
-    const signDoc = SignDoc.fromPartial({
-      bodyBytes: msg.signDoc.bodyBytes,
-      authInfoBytes: msg.signDoc.authInfoBytes,
-      chainId: msg.signDoc.chainId,
-      accountNumber: msg.signDoc.accountNumber,
-    });
-
-    const cosmosKeyring = await wallet.getCosmosKeyring(msg.signDoc.chainId);
-    if(!cosmosKeyring){
-      throw new Error('no keyring')
+  cosmosSignDirect = async ({ data: { params: msg } ,approvalRes}) => {
+    if (!approvalRes) {
+      throw new Error('approvalRes is required')
     }
-
-    const response = await cosmosKeyring.signDirect(
-      msg.signDoc.chainId,
-      msg.signerAddress,
-      signDoc,
-    );
-
-    return response
-
-
+    const {bodyBytes,authInfoBytes,chainId,accountNumber} = msg.signDoc;
+    const signature = encodeSecp256k1Signature(encoding.fromHex(approvalRes.publicKey),  encoding.fromHex(approvalRes.signature));
+    const respone = {
+      signed: {
+        bodyBytes:objToUint8Array(bodyBytes),
+        authInfoBytes:objToUint8Array(authInfoBytes),
+        chainId,
+        accountNumber,
+      },
+      signature
+    } ;
+    return respone;
+    
   }
 
-  // signArbitrary
   @Reflect.metadata('APPROVAL', ['CosmosSign', (req) => {
-    // todo check
+    const signerAddress = req.data.params.signerAddress;
+    const data = req.data.params.data;
+    const signDoc = makeADR36AminoSignDoc(signerAddress, data);
+    const signBytes = serializeSignDoc(signDoc);
+    req.data.params.signBytesHex = encoding.toHex(signBytes);
   }])
-  cosmosSignArbitrary = async ({ data: { params:msg } }) => {
-
-    const cosmosKeyring = await wallet.getCosmosKeyring(msg.chainId);
-    if(!cosmosKeyring){
-      throw new Error('no keyring')
+  cosmosSignArbitrary = async ({ data: { params:msg } ,approvalRes}) => {
+    if (!approvalRes) {
+      throw new Error('approvalRes is required')
     }
 
-    const response = await cosmosKeyring.signAminoADR36(
-      msg.chainId,
-      msg.signer,
-      typeof msg.data === "string" ? Buffer.from(msg.data) : msg.data,
-    );
-
-    return response
+    const signature = encodeSecp256k1Signature(encoding.fromHex(approvalRes.publicKey),  encoding.fromHex(approvalRes.signature));
+    const respone =  signature
+    return respone;
+ 
+  }
 
 
-  }
 
 
-  @Reflect.metadata('SAFE', true)
-  cosmosSignAmino = async ({ data: { params: { signerAddress, signDoc } } }) => {
-    throw new Error('not implemented')
-    // return signerAddress
-  }
 }
 
 export default new ProviderController();