Skip to content
Snippets Groups Projects
polkadotExtensionContext.tsx 2.56 KiB
Newer Older
Niklas P's avatar
Niklas P committed
import { ReactNode, createContext, useEffect, useState } from "react";
Niklas P's avatar
Niklas P committed
import type { InjectedAccountWithMeta, InjectedExtension, InjectedWindow } from "@polkadot/extension-inject/types";
Niklas P's avatar
Niklas P committed
import { isDeepStrictEqual } from "util";
// import { web3Accounts, web3Enable, web3FromAddress, web3AccountsSubscribe, web3EnablePromise, isWeb3Injected } from "@polkadot/extension-dapp";
Niklas P's avatar
Niklas P committed
import { WalletAccount, getWallets, isWalletInstalled } from "@talismn/connect-wallets";
Niklas P's avatar
Niklas P committed

type PolkadotExtensionContextType = {
Niklas P's avatar
Niklas P committed
    accounts: WalletAccount[] | undefined;
    actingAccountIdx: number | undefined;
Niklas P's avatar
Niklas P committed
    setActingAccountIdx: (idx: number) => void;
    setActingAccountByAddress: (address: string) => void;
Niklas P's avatar
Niklas P committed
    injector: InjectedExtension | undefined;
Niklas P's avatar
Niklas P committed
    isWeb3Injected: boolean;
Niklas P's avatar
Niklas P committed
}

export const PolkadotExtensionContext = 
    createContext<PolkadotExtensionContextType>({} as PolkadotExtensionContextType);

Niklas P's avatar
Niklas P committed
export const PolkadotExtensionProvider = ( { children } : { children : ReactNode }) => {
  const [ initialized, setInitialized ] = useState<boolean>( false )
Niklas P's avatar
Niklas P committed
  const [accounts, setAccounts] = useState<WalletAccount[] | undefined>([]);
  const [actingAccountIdx, setActingAccountIdx] = useState<number | undefined>( undefined );
  const [isWeb3Injected, setIsWeb3Injected] = useState<boolean>(false);
  const [injector, setInjector] = useState<InjectedExtension>();
Niklas P's avatar
Niklas P committed

Niklas P's avatar
Niklas P committed
  const setActingAccountByAddress = (address: string) => {
    setActingAccountIdx( accounts?.findIndex( account => account.address === address ) )
  }
Niklas P's avatar
Niklas P committed
  const Accounts = async () => {      
    if (typeof window !== "undefined" ) {
      const installedWallets = getWallets().filter(wallet => wallet.installed)
      const firstWallet = installedWallets[0]
Niklas P's avatar
Niklas P committed
      
Niklas P's avatar
Niklas P committed
      // enable the wallet
      if (firstWallet) {
        try {
          await firstWallet.enable( "nextjs connect" )
          if ( ! isWeb3Injected ) {
            const unsubscribe = await firstWallet.subscribeAccounts((allAccounts: WalletAccount[] | undefined) => {
                console.log("got accounts via talisman connect", allAccounts)
                setAccounts( allAccounts )
                setActingAccountIdx( 0 )
            });
          }
          setIsWeb3Injected( true )
        } catch (error) {
          console.log( error )
Niklas P's avatar
Niklas P committed
      } 
Niklas P's avatar
Niklas P committed
    }
Niklas P's avatar
Niklas P committed
  }
Niklas P's avatar
Niklas P committed
  Accounts()
Niklas P's avatar
Niklas P committed

Niklas P's avatar
Niklas P committed
  return (
      <PolkadotExtensionContext.Provider value={ { 
        accounts,
        actingAccountIdx,
        setActingAccountIdx,
        setActingAccountByAddress,
        injector,
        isWeb3Injected,
      } }>
          {children}
      </PolkadotExtensionContext.Provider>
  )
Niklas P's avatar
Niklas P committed
}