Feat/cosmos enigma utils OK-50764 OK-50786 OK-50962

@types/miscreant.d.ts

@@ -0,0 +1,60 @@
+declare module 'miscreant' {
+  interface ICryptoProvider {
+    importBlockCipherKey(keyData: Uint8Array): Promise<IBlockCipher>;
+  }
+
+  interface IBlockCipher {
+    clear(): this;
+    encryptBlock(block: Uint8Array): Promise<Uint8Array>;
+  }
+
+  export class PolyfillCryptoProvider implements ICryptoProvider {
+    importBlockCipherKey(keyData: Uint8Array): Promise<IBlockCipher>;
+  }
+
+  export class WebCryptoProvider implements ICryptoProvider {
+    importBlockCipherKey(keyData: Uint8Array): Promise<IBlockCipher>;
+  }
+
+  export class SIV {
+    static importKey(
+      keyData: Uint8Array,
+      algorithm: string,
+      provider: ICryptoProvider,
+    ): Promise<SIV>;
+
+    seal(
+      plaintext: Uint8Array,
+      associatedData: Uint8Array[],
+    ): Promise<Uint8Array>;
+
+    open(
+      ciphertext: Uint8Array,
+      associatedData: Uint8Array[],
+    ): Promise<Uint8Array>;
+
+    clear(): this;
+  }
+
+  export class AEAD {
+    static importKey(
+      keyData: Uint8Array,
+      algorithm: string,
+      provider: ICryptoProvider,
+    ): Promise<AEAD>;
+
+    seal(
+      plaintext: Uint8Array,
+      nonce: Uint8Array,
+      associatedData?: Uint8Array,
+    ): Promise<Uint8Array>;
+
+    open(
+      ciphertext: Uint8Array,
+      nonce: Uint8Array,
+      associatedData?: Uint8Array,
+    ): Promise<Uint8Array>;
+
+    clear(): this;
+  }
+}

package.json

@@ -101,19 +101,19 @@
     "@mysten/sui": "1.28.0",
     "@ngraveio/bc-ur": "^1.1.13",
     "@nktkas/hyperliquid": "0.29.1",
-    "@onekeyfe/cross-inpage-provider-core": "2.2.62",
-    "@onekeyfe/cross-inpage-provider-errors": "2.2.62",
-    "@onekeyfe/cross-inpage-provider-injected": "2.2.62",
-    "@onekeyfe/cross-inpage-provider-types": "2.2.62",
-    "@onekeyfe/extension-bridge-hosted": "2.2.62",
+    "@onekeyfe/cross-inpage-provider-core": "2.2.63",
+    "@onekeyfe/cross-inpage-provider-errors": "2.2.63",
+    "@onekeyfe/cross-inpage-provider-injected": "2.2.63",
+    "@onekeyfe/cross-inpage-provider-types": "2.2.63",
+    "@onekeyfe/extension-bridge-hosted": "2.2.63",
     "@onekeyfe/hd-ble-sdk": "1.1.24-alpha.2",
     "@onekeyfe/hd-common-connect-sdk": "1.1.24-alpha.2",
     "@onekeyfe/hd-core": "1.1.24-alpha.2",
     "@onekeyfe/hd-shared": "1.1.24-alpha.2",
     "@onekeyfe/hd-transport": "1.1.24-alpha.2",
     "@onekeyfe/hd-transport-electron": "1.1.24-alpha.2",
     "@onekeyfe/hd-web-sdk": "1.1.24-alpha.2",
-    "@onekeyfe/onekey-cross-webview": "2.2.62",
+    "@onekeyfe/onekey-cross-webview": "2.2.63",
     "@polkadot/extension-inject": "0.54.1",
     "@polkadot/types": "14.3.1",
     "@polkadot/util-crypto": "13.2.3",

packages/kit-bg/package.json

@@ -5,7 +5,8 @@
   "main": "src/index.tsx",
   "dependencies": {
     "eth-url-parser": "^1.0.4",
-    "idb": "^7.1.1"
+    "idb": "^7.1.1",
+    "miscreant": "^0.3.2"
   },
   "scripts": {
     "_folderslint": "yarn folderslint",

packages/kit-bg/src/providers/ProviderApiBtc.ts

@@ -354,6 +354,58 @@ class ProviderApiBtc extends ProviderApiBase {
     };
   }
 
+  @providerApiMethod()
+  public async getBalanceV2(request: IJsBridgeMessagePayload) {
+    const { accountInfo: { networkId, accountId } = {} } = (
+      await this.getAccountsInfo(request)
+    )[0];
+
+    const { balance } =
+      await this.backgroundApi.serviceAccountProfile.fetchAccountDetails({
+        networkId: networkId ?? '',
+        accountId: accountId ?? '',
+      });
+    return {
+      available: balance,
+      unavailable: 0,
+      total: balance,
+    };
+  }
+
+  @providerApiMethod()
+  public async getBitcoinUtxos(
+    request: IJsBridgeMessagePayload,
+    params?: { cursor?: number; size?: number },
+  ) {
+    const { accountInfo: { networkId, accountId } = {} } = (
+      await this.getAccountsInfo(request)
+    )[0];
+
+    const { utxoList } =
+      await this.backgroundApi.serviceAccountProfile.fetchAccountDetails({
+        networkId: networkId ?? '',
+        accountId: accountId ?? '',
+        withUTXOList: true,
+      });
+
+    const mappedUtxos =
+      utxoList?.map((it) => {
+        const bn = new BigNumber(it.value ?? 0);
+        const satoshis = bn.isNaN() ? 0 : bn.toNumber();
+
+        return {
+          txid: it.txid,
+          vout: it.vout,
+          pubkey: it.txPubkey,
+          satoshis,
+          scriptPk: it.scriptPublicKey,
+        };
+      }) ?? [];
+
+    const { cursor = 0, size = mappedUtxos.length } = params ?? {};
+    return mappedUtxos.slice(cursor, cursor + size);
+  }
+
   @providerApiMethod()
   public async getInscriptions() {
     throw web3Errors.rpc.methodNotSupported();

packages/kit-bg/src/providers/ProviderApiCosmos.ts

@@ -24,6 +24,10 @@ import {
   IMPL_COSMOS,
 } from '@onekeyhq/shared/src/engine/engineConsts';
 import { OneKeyLocalError } from '@onekeyhq/shared/src/errors';
+import {
+  EDAppConnectionModal,
+  EModalRoutes,
+} from '@onekeyhq/shared/src/routes';
 import { defaultLogger } from '@onekeyhq/shared/src/logger/logger';
 import accountUtils from '@onekeyhq/shared/src/utils/accountUtils';
 import hexUtils from '@onekeyhq/shared/src/utils/hexUtils';
@@ -33,8 +37,11 @@ import type { INetworkAccount } from '@onekeyhq/shared/types/account';
 import type { IConnectionAccountInfo } from '@onekeyhq/shared/types/dappConnection';
 import { EMessageTypesCommon } from '@onekeyhq/shared/types/message';
 
+import type { SecretNetworkEncryption } from '../vaults/impls/cosmos/sdkCosmos/SecretNetworkEncryption';
+import { vaultFactory } from '../vaults/factory';
 import ProviderApiBase from './ProviderApiBase';
 
+import type VaultCosmos from '../vaults/impls/cosmos/Vault';
 import type { IProviderBaseBackgroundNotifyInfo } from './ProviderApiBase';
 import type { IJsBridgeMessagePayload } from '@onekeyfe/cross-inpage-provider-types';
 
@@ -165,7 +172,7 @@ class ProviderApiCosmos extends ProviderApiBase {
       networkId,
     });
     if (!network) {
-      throw new OneKeyLocalError('Invalid chainId');
+      return undefined;
     }
 
     try {
@@ -722,6 +729,90 @@ class ProviderApiCosmos extends ProviderApiBase {
     });
   }
 
+  // Enigma (Secret Network) support
+  private async _getOrCreateEnigmaUtils(
+    request: IJsBridgeMessagePayload,
+    chainId: string,
+  ): Promise<SecretNetworkEncryption> {
+    const networkId = this.convertCosmosChainId(chainId);
+    if (!networkId) throw new OneKeyLocalError('Invalid chainId');
+
+    const account = await this._getAccount(request, networkId);
+
+    const { accountInfo } = account;
+    const walletId = accountInfo?.walletId ?? '';
+    const accountId = accountInfo?.accountId ?? account.account.id;
+
+    let password = await this.backgroundApi.servicePassword.getCachedPassword();
+
+    if (!password) {
+      const result = (await this.backgroundApi.serviceDApp.openModal({
+        request,
+        screens: [
+          EModalRoutes.DAppConnectionModal,
+          EDAppConnectionModal.CosmosEnigmaUnlockModal,
+        ],
+        params: {
+          walletId,
+          accountId,
+          networkId: accountInfo?.networkId ?? networkId,
+        },
+        fullScreen: true,
+      })) as { password: string };
+      password = result.password;
+    }
+
+    const vault = (await vaultFactory.getVault({
+      networkId: accountInfo?.networkId ?? networkId,
+      accountId,
+    })) as VaultCosmos;
+
+    return vault.getOrCreateEnigmaUtils({ password });
+  }
+
+  @providerApiMethod()
+  public async getEnigmaPubKey(
+    request: IJsBridgeMessagePayload,
+    params: { chainId: string },
+  ): Promise<string> {
+    const utils = await this._getOrCreateEnigmaUtils(request, params.chainId);
+    const pubkey = await utils.getPubkey();
+    return bytesToHex(pubkey);
+  }
+
+  @providerApiMethod()
+  public async enigmaEncrypt(
+    request: IJsBridgeMessagePayload,
+    params: { chainId: string; contractCodeHash: string; msg: object },
+  ): Promise<string> {
+    const utils = await this._getOrCreateEnigmaUtils(request, params.chainId);
+    const encrypted = await utils.encrypt(params.contractCodeHash, params.msg);
+    return bytesToHex(encrypted);
+  }
+
+  @providerApiMethod()
+  public async enigmaDecrypt(
+    request: IJsBridgeMessagePayload,
+    params: { chainId: string; ciphertext: string; nonce: string },
+  ): Promise<string> {
+    const utils = await this._getOrCreateEnigmaUtils(request, params.chainId);
+    const decrypted = await utils.decrypt(
+      hexToBytes(params.ciphertext),
+      hexToBytes(params.nonce),
+    );
+    return bytesToHex(decrypted);
+  }
+
+  @providerApiMethod()
+  public async enigmaGetTxEncryptionKey(
+    request: IJsBridgeMessagePayload,
+    params: { chainId: string; nonce: string },
+  ): Promise<string> {
+    const utils = await this._getOrCreateEnigmaUtils(request, params.chainId);
+    const key = await utils.getTxEncryptionKey(hexToBytes(params.nonce));
+    return bytesToHex(key);
+  }
+
   @providerApiMethod()
   public async cosmos_signDirect(
     request: IJsBridgeMessagePayload,

packages/kit-bg/src/vaults/impls/cosmos/KeyringHd.ts

@@ -2,6 +2,7 @@ import coreChainApi from '@onekeyhq/core/src/instance/coreChainApi';
 import type { ISignedTxPro } from '@onekeyhq/core/src/types';
 
 import { KeyringHdBase } from '../../base/KeyringHdBase';
+import { SecretNetworkEncryption } from './sdkCosmos/SecretNetworkEncryption';
 
 import type { IDBAccount } from '../../../dbs/local/types';
 import type {
@@ -44,4 +45,15 @@ export class KeyringHd extends KeyringHdBase {
   override async signMessage(params: ISignMessageParams): Promise<string[]> {
     return this.baseSignMessage(params);
   }
+
+  async getEnigmaSeed(params: { password: string }): Promise<Uint8Array> {
+    const { password } = params;
+    const privateKeys = await this.baseGetPrivateKeys({ password });
+    const account = await this.vault.getAccount();
+    const encryptedPrivateKeyHex = privateKeys[account.path];
+    return SecretNetworkEncryption.deriveEnigmaSeed({
+      encryptedPrivateKeyHex,
+      password,
+    });
+  }
 }

packages/kit-bg/src/vaults/impls/cosmos/KeyringImported.ts

@@ -2,6 +2,7 @@ import coreChainApi from '@onekeyhq/core/src/instance/coreChainApi';
 import type { ISignedMessagePro, ISignedTxPro } from '@onekeyhq/core/src/types';
 
 import { KeyringImportedBase } from '../../base/KeyringImportedBase';
+import { SecretNetworkEncryption } from './sdkCosmos/SecretNetworkEncryption';
 
 import type { IDBAccount } from '../../../dbs/local/types';
 import type {
@@ -48,4 +49,16 @@ export class KeyringImported extends KeyringImportedBase {
   ): Promise<ISignedMessagePro> {
     return this.baseSignMessage(params);
   }
+
+  async getEnigmaSeed(params: { password: string }): Promise<Uint8Array> {
+    const { password } = params;
+    const privateKeys = await this.baseGetPrivateKeys({ password });
+    const account = await this.vault.getAccount();
+    const encryptedPrivateKeyHex =
+      privateKeys[account.path] || Object.values(privateKeys)[0];
+    return SecretNetworkEncryption.deriveEnigmaSeed({
+      encryptedPrivateKeyHex,
+      password,
+    });
+  }
 }