Merge pull request #47 from argentlabs/fix/add-custom-provider

fix: allow custom provider for injected connectors

src/connectors/argentMobile/index.ts

@@ -2,7 +2,7 @@ import type {
   AccountChangeEventHandler,
   StarknetWindowObject,
 } from "get-starknet-core"
-import type { AccountInterface } from "starknet"
+import type { AccountInterface, ProviderInterface } from "starknet"
 import { constants } from "starknet"
 import { DEFAULT_ARGENT_MOBILE_ICON, DEFAULT_PROJECT_ID } from "./constants"
 import {
@@ -26,6 +26,7 @@ export interface ArgentMobileConnectorOptions {
   description?: string
   url?: string
   icons?: string[]
+  provider?: ProviderInterface
 }
 
 export class ArgentMobileConnector extends Connector {
@@ -153,6 +154,10 @@ export class ArgentMobileConnector extends Connector {
       description,
       url,
       icons,
+      rpcUrl:
+        chainId === constants.NetworkName.SN_MAIN
+          ? "https://cloud.argent-api.com/v1/starknet/mainnet/rpc/v0.5"
+          : "https://api.hydrogen.argent47.net/v1/starknet/goerli/rpc/v0.5",
     }
 
     if (projectId === DEFAULT_PROJECT_ID) {
@@ -171,6 +176,14 @@ export class ArgentMobileConnector extends Connector {
     }
 
     const _wallet = await getStarknetWindowObject(options)
+
+    const { provider } = this._options
+    if (provider) {
+      Object.assign(_wallet, {
+        provider,
+      })
+    }
+
     this._wallet = _wallet
 
     // wallet connect rpc enable

src/connectors/argentMobile/modal/starknet/adapter.ts

@@ -11,7 +11,7 @@ import type {
   ProviderInterface,
   SignerInterface,
 } from "starknet"
-import { Provider, constants } from "starknet"
+import { RpcProvider, constants } from "starknet"
 
 import type { NamespaceAdapterOptions } from "../adapter"
 import { NamespaceAdapter } from "../adapter"
@@ -84,12 +84,7 @@ export class StarknetAdapter
 
     this.remoteSigner = new StarknetRemoteSigner(this.walletRpc)
 
-    if (rpcUrl) {
-      this.provider = new Provider({ rpc: { nodeUrl: rpcUrl } })
-    } else {
-      const network = this.getNetworkName(this.chainId)
-      this.provider = new Provider({ sequencer: { network } })
-    }
+    this.provider = new RpcProvider({ nodeUrl: rpcUrl })
     this.account = new StarknetRemoteAccount(
       this.provider,
       "",

src/connectors/injected/index.ts

@@ -1,5 +1,5 @@
 import type { StarknetWindowObject } from "get-starknet-core"
-import { AccountInterface, constants } from "starknet"
+import { AccountInterface, ProviderInterface, constants } from "starknet"
 import {
   ConnectorNotConnectedError,
   ConnectorNotFoundError,
@@ -23,6 +23,8 @@ export interface InjectedConnectorOptions {
   name?: string
   /** Wallet icons. */
   icon?: ConnectorIcons
+  /** Provider */
+  provider?: ProviderInterface
 }
 
 export class InjectedConnector extends Connector {
@@ -211,6 +213,13 @@ export class InjectedConnector extends Connector {
     const installed = getAvailableWallets(globalThis)
     const wallet = installed.filter((w) => w.id === this._options.id)[0]
     if (wallet) {
+      const { provider } = this._options
+      if (provider) {
+        Object.assign(wallet, {
+          provider,
+        })
+      }
+
       this._wallet = wallet
     }
   }

src/connectors/webwallet/constants.ts

@@ -19,4 +19,4 @@ export const RPC_NODE_URL_TESTNET =
   "https://api.hydrogen.argent47.net/v1/starknet/goerli/rpc/v0.5"
 
 export const RPC_NODE_URL_MAINNET =
-  "https://cloud.argent-api.com/v1/starknet/goerli/rpc/v0.5"
+  "https://cloud.argent-api.com/v1/starknet/mainnet/rpc/v0.5"

src/connectors/webwallet/index.ts

@@ -2,7 +2,7 @@ import type {
   AccountChangeEventHandler,
   StarknetWindowObject,
 } from "get-starknet-core"
-import type { AccountInterface } from "starknet"
+import type { AccountInterface, ProviderInterface } from "starknet"
 import {
   Connector,
   type ConnectorData,
@@ -23,6 +23,7 @@ let _wallet: StarknetWindowObject | null = null
 
 interface WebWalletConnectorOptions {
   url?: string
+  provider?: ProviderInterface
 }
 
 export class WebWalletConnector extends Connector {
@@ -163,11 +164,16 @@ export class WebWalletConnector extends Connector {
 
   private async ensureWallet(): Promise<void> {
     const origin = this._options.url || DEFAULT_WEBWALLET_URL
+    const provider = this._options.provider
     setPopupOptions({
       origin,
       location: "/interstitialLogin",
     })
-    const wallet = await getWebWalletStarknetObject(origin, trpcProxyClient({}))
+    const wallet = await getWebWalletStarknetObject(
+      origin,
+      trpcProxyClient({}),
+      provider,
+    )
 
     _wallet = wallet ?? null
     this._wallet = _wallet

src/connectors/webwallet/starknetWindowObject/getWebWalletStarknetObject.ts

@@ -1,5 +1,5 @@
 import type { CreateTRPCProxyClient } from "@trpc/client"
-import { RpcProvider } from "starknet"
+import { ProviderInterface, RpcProvider } from "starknet"
 
 import { mapTargetUrlToNodeUrl } from "../helpers/mapTargetUrlToNodeUrl"
 import type { AppRouter } from "../helpers/trpc"
@@ -9,14 +9,15 @@ import { getArgentStarknetWindowObject } from "./argentStarknetWindowObject"
 export const getWebWalletStarknetObject = async (
   target: string,
   proxyLink: CreateTRPCProxyClient<AppRouter>,
+  provider?: ProviderInterface,
 ): Promise<WebWalletStarknetWindowObject> => {
   const globalWindow = typeof window !== "undefined" ? window : undefined
   if (!globalWindow) {
     throw new Error("window is not defined")
   }
 
   const nodeUrl = mapTargetUrlToNodeUrl(target)
-  const defaultProvider = new RpcProvider({ nodeUrl })
+  const defaultProvider = provider ?? new RpcProvider({ nodeUrl })
   const starknetWindowObject = getArgentStarknetWindowObject(
     {
       host: globalWindow.location.origin,

src/helpers/defaultConnectors.ts

@@ -1,3 +1,4 @@
+import { ProviderInterface } from "starknet"
 import { type Connector } from "../connectors"
 import {
   ArgentMobileConnector,
@@ -9,25 +10,31 @@ import { WebWalletConnector } from "../connectors/webwallet"
 export const defaultConnectors = ({
   argentMobileOptions,
   webWalletUrl,
+  provider,
 }: {
   argentMobileOptions?: ArgentMobileConnectorOptions
   webWalletUrl?: string
+  provider?: ProviderInterface
 }): Connector[] => {
   const isSafari = /^((?!chrome|android).)*safari/i.test(navigator.userAgent)
 
   const defaultConnectors: Connector[] = []
 
   if (!isSafari) {
     defaultConnectors.push(
-      new InjectedConnector({ options: { id: "argentX" } }),
+      new InjectedConnector({ options: { id: "argentX", provider } }),
     )
     defaultConnectors.push(
-      new InjectedConnector({ options: { id: "braavos" } }),
+      new InjectedConnector({ options: { id: "braavos", provider } }),
     )
   }
 
-  defaultConnectors.push(new ArgentMobileConnector(argentMobileOptions))
-  defaultConnectors.push(new WebWalletConnector({ url: webWalletUrl }))
+  defaultConnectors.push(
+    new ArgentMobileConnector({ ...argentMobileOptions, provider }),
+  )
+  defaultConnectors.push(
+    new WebWalletConnector({ url: webWalletUrl, provider }),
+  )
 
   return defaultConnectors
 }

src/main.ts

@@ -31,6 +31,7 @@ export const connect = async ({
   webWalletUrl = DEFAULT_WEBWALLET_URL,
   argentMobileOptions,
   connectors = [],
+  provider,
   ...restOptions
 }: ConnectOptions = {}): Promise<StarknetWindowObject | null> => {
   // force null in case it was disconnected from mobile app
@@ -40,6 +41,7 @@ export const connect = async ({
       ? defaultConnectors({
           argentMobileOptions,
           webWalletUrl,
+          provider,
         })
       : connectors
 

src/types/modal.ts

@@ -1,6 +1,7 @@
 import type { GetWalletOptions } from "get-starknet-core"
 import type { Connector, ConnectorIcons } from "../connectors/connector"
 import type { ArgentMobileConnectorOptions } from "../connectors/argentMobile"
+import { ProviderInterface } from "starknet"
 
 export type StoreVersion = "chrome" | "firefox" | "edge"
 
@@ -12,6 +13,7 @@ export interface ConnectOptions extends GetWalletOptions {
   modalTheme?: "light" | "dark" | "system"
   storeVersion?: StoreVersion | null
   webWalletUrl?: string
+  provider?: ProviderInterface
 }
 
 export type ModalWallet = {