Pass INetworkProvider where needed

multiversx · Dec 9, 2024 · 17177ac · 17177ac
1 parent fa2b75a
commit 17177ac
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 9 deletions.
diff --git a/src/entrypoints/entrypoints.ts b/src/entrypoints/entrypoints.ts
@@ -6,6 +6,7 @@ import { DelegationController } from "../delegation";
 import { ErrInvalidNetworkProviderKind } from "../errors";
 import { Message, MessageComputer } from "../message";
 import { ApiNetworkProvider, ProxyNetworkProvider } from "../networkProviders";
+import { INetworkProvider } from "../networkProviders/interface";
 import { RelayedController } from "../relayed/relayedController";
 import { SmartContractController } from "../smartContracts/smartContractController";
 import { TokenManagementController } from "../tokenManagement";
@@ -18,7 +19,7 @@ import { UserVerifier } from "../wallet";
 import { DevnetEntrypointConfig, MainnetEntrypointConfig, TestnetEntrypointConfig } from "./config";
 
 class NetworkEntrypoint {
-    private networkProvider: ApiNetworkProvider | ProxyNetworkProvider;
+    private networkProvider: INetworkProvider;
     private chainId: string;
 
     constructor(options: { networkProviderUrl: string; networkProviderKind: string; chainId: string }) {
@@ -80,7 +81,7 @@ class NetworkEntrypoint {
         return transactionAwaiter.awaitCompleted(txHash);
     }
 
-    createNetworkProvider(): ApiNetworkProvider | ProxyNetworkProvider {
+    createNetworkProvider(): INetworkProvider {
         return this.networkProvider;
     }
 

diff --git a/src/smartContracts/smartContractController.ts b/src/smartContracts/smartContractController.ts
@@ -1,7 +1,6 @@
 import { AbiRegistry, ArgSerializer, isTyped, NativeSerializer } from "../abi";
 import { IAccount } from "../accounts/interfaces";
 import { Err, ErrSmartContractQuery } from "../errors";
-import { ApiNetworkProvider, ProxyNetworkProvider } from "../networkProviders";
 import { INetworkProvider } from "../networkProviders/interface";
 import { SmartContractQuery, SmartContractQueryInput, SmartContractQueryResponse } from "../smartContractQuery";
 import { Transaction } from "../transaction";
@@ -18,14 +17,10 @@ export class SmartContractController {
     private parser: SmartContractTransactionsOutcomeParser;
     private transactionWatcher: TransactionWatcher;
     private txComputer: TransactionComputer;
-    private networkProvider: ApiNetworkProvider | ProxyNetworkProvider | INetworkProvider;
+    private networkProvider: INetworkProvider;
     private abi?: AbiRegistry;
 
-    constructor(options: {
-        chainID: string;
-        networkProvider: ApiNetworkProvider | ProxyNetworkProvider | INetworkProvider;
-        abi?: AbiRegistry;
-    }) {
+    constructor(options: { chainID: string; networkProvider: INetworkProvider; abi?: AbiRegistry }) {
         this.factory = new SmartContractTransactionsFactory({
             config: new TransactionsFactoryConfig({ chainID: options.chainID }),
             abi: options.abi,