Refactor

atomex-protocol · Sep 5, 2022 · 81ecba8 · 81ecba8
1 parent 574acd9
commit 81ecba8
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/src/tezos/atomexProtocol/taquitoAtomexProtocolMultiChain.ts b/src/tezos/atomexProtocol/taquitoAtomexProtocolMultiChain.ts
@@ -1,4 +1,4 @@
-import type { TezosToolkit, TransactionWalletOperation } from '@taquito/taquito';
+import type { ContractAbstraction, TezosToolkit, TransactionWalletOperation, Wallet } from '@taquito/taquito';
 import BigNumber from 'bignumber.js';
 
 import type {
@@ -78,6 +78,13 @@ export abstract class TaquitoAtomexProtocolMultiChain implements AtomexProtocolM
     return taquitoWallet;
   }
 
+  protected async getSwapContractCore<T extends ContractAbstraction<Wallet>>(address: string): Promise<T> {
+    const wallet = await this.getWallet(address);
+    const contract = await wallet.toolkit.wallet.at<T>(this.swapContractAddress);
+
+    return contract;
+  }
+
   protected async getTransaction(operation: TransactionWalletOperation): Promise<Transaction> {
     const status = await operation.status();
     const confirmation = await operation.confirmation();

diff --git a/src/tezos/atomexProtocol/tezosTaquitoAtomexProtocolMultiChain.ts b/src/tezos/atomexProtocol/tezosTaquitoAtomexProtocolMultiChain.ts
@@ -77,9 +77,6 @@ export class TezosTaquitoAtomexProtocolMultiChain extends TaquitoAtomexProtocolM
   }
 
   protected async getSwapContract(address: string): Promise<TezosMultiChainSmartContract<Wallet>> {
-    const wallet = await this.getWallet(address);
-    const contract = await wallet.toolkit.wallet.at<TezosMultiChainSmartContract<Wallet>>(this.swapContractAddress);
-
-    return contract;
+    return this.getSwapContractCore<TezosMultiChainSmartContract<Wallet>>(address);
   }
 }