@sphereon/ssi-sdk-web3.headless-provider/src/ethers-headless-provider.ts

import { TransactionRequest } from '@ethersproject/abstract-provider'
import { toUtf8String } from '@ethersproject/strings'
import { signTypedData, SignTypedDataVersion } from '@metamask/eth-sig-util'
import assert from 'assert/strict'
import { ethers, Signer, Wallet } from 'ethers'
// import {hashMessage} from "ethers/lib/utils";
import { BehaviorSubject, filter, first, firstValueFrom, from, switchMap, tap } from 'rxjs'

import { Deny, Disconnected, ErrorWithCode, Unauthorized, UnrecognizedChainID } from './errors'
import { EthersKMSSigner } from './ethers-kms-signer'
import { EventEmitter } from './event-emitter'
import { ChainConnection, IWeb3Provider, PendingRequest, Web3Method, Web3ProviderConfig, without } from './types'

export class EthersHeadlessProvider extends EventEmitter implements IWeb3Provider {
  private _pendingRequests = new BehaviorSubject<PendingRequest[]>([])
  private _signers: Signer[] = []
  private _activeChainId: number
  private _rpc: Record<number, ethers.providers.JsonRpcProvider> = {}
  private _config: { debug: boolean; logger: typeof console.log }
  private _authorizedRequests: { [K in Web3Method | string]?: boolean } = {}

  constructor(
    signers: Signer[],
    private readonly chains: ChainConnection[],
    // @ts-ignore
    config: Web3ProviderConfig = {},
  ) {
    super()
    this._signers = signers
    this._activeChainId = chains[0].chainId
    this._config = Object.assign({ debug: true, logger: console.log }, config)
  }

  request(args: { method: 'eth_call'; params: any[] }): Promise<any>
  request(args: { method: 'eth_getBalance'; params: string[] }): Promise<string>
  request(args: { method: 'eth_accounts'; params: [] }): Promise<string[]>
  request(args: { method: 'eth_requestAccounts'; params: string[] }): Promise<string[]>

  request(args: { method: 'net_version'; params: [] }): Promise<number>
  request(args: { method: 'eth_chainId'; params: [] }): Promise<string>
  request(args: { method: 'personal_sign'; params: string[] }): Promise<string>
  request(args: { method: 'eth_signTypedData' | 'eth_signTypedData_v1'; params: [object[], string] }): Promise<string>

  request(args: { method: 'eth_signTypedData_v3' | 'eth_signTypedData_v4'; params: string[] }): Promise<string>

  request(args: { method: 'eth_sendTransaction'; params: TransactionRequest[] }): Promise<string>

  async request({ method, params }: { method: string; params: any[] }): Promise<any> {
    if (this._config.debug) {
      this._config.logger(JSON.stringify({ method, params }))
    }

    switch (method) {
      case 'eth_call':
      case 'eth_getBalance':
      case 'eth_estimateGas':
      case 'eth_blockNumber':
      case 'eth_getBlockByNumber':
      case 'eth_getTransactionByHash':
      case 'eth_getTransactionReceipt':
      case 'eth_feeHistory':
        return this.getRpc().send(method, params)

      case 'eth_requestAccounts':
      case 'eth_accounts':
        return this.waitAuthorization(
          { method, params },
          async () => {
            const { chainId } = this.getCurrentChain()
            this.emit('connect', { chainId })
            return Promise.all(this._signers.map((wallet) => wallet.getAddress()))
          },
          true,
          'eth_requestAccounts',
        )

      case 'eth_chainId': {
        const { chainId } = this.getCurrentChain()
        return '0x' + chainId.toString(16)
      }

      case 'net_version': {
        const { chainId } = this.getCurrentChain()
        return '' + chainId
      }

      case 'eth_sendTransaction': {
        return this.waitAuthorization({ method, params }, async () => {
          const wallet = this.getCurrentWallet()
          const rpc = this.getRpc()
          const { gas, from, ...txRequest } = params[0]
          const tx = await wallet.connect(rpc).sendTransaction(txRequest)
          return tx.hash
        })
      }

      case 'eth_sign': {
        return this.waitAuthorization({ method, params }, async () => {
          const wallet = this.getCurrentWallet()
          const rpc = this.getRpc()
          const message = params[1]
          return await (wallet.connect(rpc) as EthersKMSSigner).signMessage(message)
        })
      }

      case 'wallet_addEthereumChain': {
        return this.waitAuthorization({ method, params }, async () => {
          const chainId = Number(params[0].chainId)
          const { rpcUrl } = params[0]
          this.addNetwork(chainId, rpcUrl)
          return null
        })
      }

      case 'wallet_switchEthereumChain': {
        if (this._activeChainId === Number(params[0].chainId)) {
          return null
        }
        return this.waitAuthorization({ method, params }, async () => {
          const chainId = Number(params[0].chainId)
          this.switchNetwork(chainId)
          return null
        })
      }

      case 'personal_sign': {
        return this.waitAuthorization({ method, params }, async () => {
          const wallet = this.getCurrentWallet()
          const address = await wallet.getAddress()
          assert.equal(address, ethers.utils.getAddress(params[1]))
          const message = toUtf8String(params[0])

          const signature = await wallet.signMessage(message)
          if (this._config.debug) {
            this._config.logger('personal_sign', {
              message,
              signature,
            })
          }

          return signature
        })
      }

      case 'eth_signTypedData':
      case 'eth_signTypedData_v1': {
        return this.waitAuthorization({ method, params }, async () => {
          const wallet = this.getCurrentWallet() as Wallet
          const address = await wallet.getAddress()
          assert.equal(address, ethers.utils.getAddress(params[1]))

          const msgParams = params[0]

          return signTypedData({
            privateKey: Buffer.from(wallet.privateKey.slice(2), 'hex'),
            data: msgParams,
            version: SignTypedDataVersion.V1,
          })
        })
      }

      case 'eth_signTypedData_v3':
      case 'eth_signTypedData_v4': {
        return this.waitAuthorization({ method, params }, async () => {
          const wallet = this.getCurrentWallet() as Wallet
          const address = await wallet.getAddress()
          assert.equal(address, ethers.utils.getAddress(params[0]))

          const msgParams = JSON.parse(params[1])

          return signTypedData({
            privateKey: Buffer.from(wallet.privateKey.slice(2), 'hex'),
            data: msgParams,
            version: method === 'eth_signTypedData_v4' ? SignTypedDataVersion.V4 : SignTypedDataVersion.V3,
          })
        })
      }

      default:
        return this.getRpc().send(method, params)
      // throw UnsupportedMethod()
    }
  }

  getCurrentWallet(): Signer {
    const wallet = this._signers[0]

    if (!wallet) {
      throw Unauthorized()
    }

    return wallet
  }

  waitAuthorization<T>(requestInfo: PendingRequest['requestInfo'], task: () => Promise<T>, permanentPermission = false, methodOverride?: string) {
    const method = methodOverride ?? requestInfo.method

    if (this._authorizedRequests[method]) {
      return task()
    }

    return new Promise((resolve, reject) => {
      const pendingRequest: PendingRequest = {
        requestInfo,
        authorize: async () => {
          if (permanentPermission) {
            this._authorizedRequests[method] = true
          }

          resolve(await task())
        },
        reject(err) {
          reject(err)
        },
      }

      this._pendingRequests.next(this._pendingRequests.getValue().concat(pendingRequest))
    })
  }

  private consumeRequest(requestKind: Web3Method) {
    return firstValueFrom(
      this._pendingRequests.pipe(
        switchMap((a) => from(a)),
        filter((request) => {
          return request.requestInfo.method === requestKind
        }),
        first(),
        tap((item) => {
          this._pendingRequests.next(without(this._pendingRequests.getValue(), item))
        }),
      ),
    )
  }

  private consumeAllRequests() {
    const pendingRequests = this._pendingRequests.getValue()
    this._pendingRequests.next([])
    return pendingRequests
  }

  getPendingRequests(): PendingRequest['requestInfo'][] {
    return this._pendingRequests.getValue().map((pendingRequest) => pendingRequest.requestInfo)
  }

  getPendingRequestCount(requestKind?: Web3Method): number {
    const pendingRequests = this._pendingRequests.getValue()
    if (!requestKind) {
      return pendingRequests.length
    }

    return pendingRequests.filter((pendingRequest) => pendingRequest.requestInfo.method === requestKind).length
  }

  async authorize(requestKind: Web3Method): Promise<void> {
    const pendingRequest = await this.consumeRequest(requestKind)
    return pendingRequest.authorize()
  }

  async reject(requestKind: Web3Method, reason: ErrorWithCode = Deny()): Promise<void> {
    const pendingRequest = await this.consumeRequest(requestKind)
    return pendingRequest.reject(reason)
  }

  authorizeAll(): void {
    this.consumeAllRequests().forEach((request) => request.authorize())
  }

  rejectAll(reason: ErrorWithCode = Deny()): void {
    this.consumeAllRequests().forEach((request) => request.reject(reason))
  }

  async changeAccounts(signers: Signer[]): Promise<void> {
    this._signers = signers
    this.emit('accountsChanged', await Promise.all(this._signers.map((signer) => signer.getAddress())))
  }

  private getCurrentChain(): ChainConnection {
    const chainConn = this.chains.find((chainConn) => chainConn.chainId === this._activeChainId)
    if (!chainConn) {
      throw Disconnected()
    }
    return chainConn
  }

  public getRpc(): ethers.providers.JsonRpcProvider {
    const chainConn = this.getCurrentChain()
    let rpc = this._rpc[chainConn.chainId]

    if (!rpc) {
      rpc = new ethers.providers.JsonRpcProvider(chainConn.rpcUrl, chainConn.chainId)
      this._rpc[chainConn.chainId] = rpc
    }

    return rpc
  }

  getNetwork(): ChainConnection {
    return this.getCurrentChain()
  }

  getNetworks(): ChainConnection[] {
    return this.chains
  }

  addNetwork(chainId: number, rpcUrl: string): void {
    this.chains.push({ chainId, rpcUrl })
  }

  switchNetwork(chainId: number): void {
    const idx = this.chains.findIndex((connection) => connection.chainId === chainId)
    if (idx < 0) {
      throw UnrecognizedChainID()
    }
    if (chainId !== this._activeChainId) {
      this._activeChainId = chainId
      this.emit('chainChanged', chainId)
    }
  }
}