@rango-dev/provider-okx 0.63.1-next.1 → 0.63.1-next.3

package/src/signers/ton.ts

@@ -0,0 +1,70 @@
+import type {
+  TonConnectWalletResponseError,
+  TonProviderApi,
+} from '../namespaces/ton/types.js';
+import type { GenericSigner, TonTransaction } from 'rango-types';
+
+import { dynamicImportWithRefinedError } from '@rango-dev/wallets-shared';
+import { SignerError, SignerErrorCode, TonChainID } from 'rango-types';
+
+import { TON_CONNECT_USER_REJECTED_CODE } from '../constants.js';
+import { nextTonRequestId } from '../namespaces/ton/utils.js';
+
+function toSignerError(response: TonConnectWalletResponseError): SignerError {
+  const code =
+    response.error.code === TON_CONNECT_USER_REJECTED_CODE
+      ? SignerErrorCode.REJECTED_BY_USER
+      : SignerErrorCode.SEND_TX_ERROR;
+  return new SignerError(code, response.error.message, response.error);
+}
+
+export class OKXTonSigner implements GenericSigner<TonTransaction> {
+  private provider: TonProviderApi;
+
+  constructor(provider: TonProviderApi) {
+    this.provider = provider;
+  }
+
+  async signMessage(): Promise<string> {
+    throw SignerError.UnimplementedError('signMessage');
+  }
+
+  async signAndSendTx(tx: TonTransaction): Promise<{ hash: string }> {
+    const { validUntil, network, from, messages } = tx;
+
+    const { Address, Cell } = await dynamicImportWithRefinedError(
+      async () => await import('@ton/core')
+    );
+
+    const transactionPayload = {
+      valid_until: validUntil,
+      network: network ?? TonChainID.MAINNET,
+      ...(from && { from: Address.parse(from).toRawString() }),
+      messages: messages.map(({ stateInit, payload, ...message }) => ({
+        ...message,
+        ...(stateInit != null && { stateInit }),
+        ...(payload != null && { payload }),
+      })),
+    };
+
+    try {
+      const response = await this.provider.send({
+        method: 'sendTransaction',
+        params: [JSON.stringify(transactionPayload)],
+        id: nextTonRequestId(),
+      });
+
+      if ('error' in response) {
+        throw toSignerError(response);
+      }
+
+      const hash = Cell.fromBase64(response.result).hash().toString('hex');
+      return { hash };
+    } catch (error) {
+      if (error instanceof SignerError) {
+        throw error;
+      }
+      throw new SignerError(SignerErrorCode.SEND_TX_ERROR, undefined, error);
+    }
+  }
+}

package/src/types.ts

@@ -1,10 +1,14 @@
+import type { TonProviderApi } from './namespaces/ton/types.js';
 import type { ProviderAPI as EvmProviderApi } from '@hub3js/evm';
 import type {
   EVM_NAMESPACE,
   SOLANA_NAMESPACE,
   UTXO_NAMESPACE,
+  TON_NAMESPACE,
+  TRON_NAMESPACE,
 } from '@hub3js/namespaces';
 import type { ProviderAPI as SolanaProviderApi } from '@hub3js/solana';
+import type { ProviderAPI as TronProviderApi } from '@rango-dev/wallets-core/namespaces/tron';
 import type { ProviderAPI as UtxoProviderApi } from '@rango-dev/wallets-core/namespaces/utxo';
 
 export type OkxBtcAddress = {
@@ -12,12 +16,28 @@ export type OkxBtcAddress = {
   publicKey: string;
   compressedPublicKey: string;
 };
+
 export type ProviderObject = {
   [EVM_NAMESPACE]: EvmProviderApi;
   [SOLANA_NAMESPACE]: SolanaProviderApi;
   [UTXO_NAMESPACE]: UtxoProviderApi;
+  [TON_NAMESPACE]: TonProviderApi;
+  [TRON_NAMESPACE]: TronProviderApi;
 };
 export type Provider = Map<
   keyof ProviderObject,
   ProviderObject[keyof ProviderObject]
 >;
+
+/*
+ * On `accountsChanged`, `data.address` is the new base58 address, or `false`
+ * when disconnected; on `disconnect`, `data` is absent.
+ */
+export type OkxTronMessageEvent = {
+  message: {
+    action: string;
+    data?: {
+      address: string | false;
+    };
+  };
+};

package/src/utils.ts

@@ -1,17 +1,21 @@
-import type { OkxBtcAddress, Provider } from './types.js';
+import type { TonProviderApi } from './namespaces/ton/types.js';
+import type { OkxBtcAddress, OkxTronMessageEvent, Provider } from './types.js';
 import type { ProviderAPI as EvmProviderApi } from '@hub3js/evm';
 import type { ProviderAPI as SolanaProviderApi } from '@hub3js/solana';
+import type { ProviderAPI as TronProviderApi } from '@rango-dev/wallets-core/namespaces/tron';
 import type { ProviderAPI as UtxoProviderApi } from '@rango-dev/wallets-core/namespaces/utxo';
 
 import {
   EVM_NAMESPACE,
   SOLANA_NAMESPACE,
   UTXO_NAMESPACE,
+  TON_NAMESPACE,
+  TRON_NAMESPACE,
 } from '@hub3js/namespaces';
 
 export function okx(): Provider | null {
-  const { okxwallet } = window;
-  if (!okxwallet) {
+  const { okxwallet, okxTonWallet } = window;
+  if (!okxwallet && !okxTonWallet) {
     return null;
   }
   const instances: Provider = new Map();
@@ -24,6 +28,12 @@ export function okx(): Provider | null {
   if (okxwallet.bitcoin) {
     instances.set(UTXO_NAMESPACE, okxwallet.bitcoin);
   }
+  if (okxTonWallet?.tonconnect) {
+    instances.set(TON_NAMESPACE, okxTonWallet.tonconnect);
+  }
+  if (okxwallet.tronLink) {
+    instances.set(TRON_NAMESPACE, okxwallet.tronLink);
+  }
   return instances;
 }
 
@@ -75,6 +85,32 @@ export function bitcoinOKX(): UtxoProviderApi {
 
   return bitcoinInstance;
 }
+
+export function tonOKX(): TonProviderApi {
+  const instance = okx();
+  const tonInstance = instance?.get(TON_NAMESPACE);
+
+  if (!tonInstance) {
+    throw new Error(
+      'OKX Wallet not injected or TON not enabled. Please check your wallet.'
+    );
+  }
+
+  return tonInstance as TonProviderApi;
+}
+
+export function tronOKX(): TronProviderApi {
+  const instance = okx();
+  const tronInstance = instance?.get(TRON_NAMESPACE);
+
+  if (!tronInstance) {
+    throw new Error(
+      'OKX Wallet not injected or Tron not enabled. Please check your wallet.'
+    );
+  }
+
+  return tronInstance;
+}
 export async function getBitcoinAccounts(): Promise<OkxBtcAddress> {
   const instance = bitcoinOKX();
   const requestResult = await instance.connect();
@@ -85,3 +121,17 @@ export async function getBitcoinAccounts(): Promise<OkxBtcAddress> {
 
   return requestResult;
 }
+
+export function isOkxTronMessageEvent(
+  value: unknown
+): value is OkxTronMessageEvent {
+  if (typeof value !== 'object' || value === null) {
+    return false;
+  }
+  const event = value as { message?: { action?: unknown } };
+  return (
+    typeof event.message === 'object' &&
+    event.message !== null &&
+    typeof event.message.action === 'string'
+  );
+}