mango-ui-v3/utils/wallet-adapters/bitpie.ts

import EventEmitter from 'eventemitter3'
import { PublicKey, Transaction } from '@solana/web3.js'
import { notify } from '../../utils/notifications'
import { WalletAdapter } from '../../@types/types'

interface BitpieWallet {
  getAccount(): Promise<string>
  signTransaction(transaction: Transaction): Promise<Transaction>
  signAllTransactions(transactions: Transaction[]): Promise<Transaction[]>
}

interface BitpieWalletWindow extends Window {
  bitpie?: BitpieWallet
}

declare const window: BitpieWalletWindow

export class BitpieWalletAdapter extends EventEmitter implements WalletAdapter {
  private _connecting: boolean
  private _wallet: BitpieWallet | null
  private _publicKey: PublicKey | null

  constructor() {
    super()
    this._connecting = false
    this._wallet = null
    this._publicKey = null
  }

  get publicKey(): PublicKey | null {
    return this._publicKey
  }

  get ready(): boolean {
    return typeof window !== 'undefined' && !!window.bitpie
  }

  get connecting(): boolean {
    return this._connecting
  }

  get connected(): boolean {
    return !!this._wallet
  }

  get autoApprove() {
    return true
  }

  async connect(): Promise<void> {
    try {
      if (this.connected || this.connecting) return
      this._connecting = true

      const wallet = typeof window !== 'undefined' && window.bitpie
      if (!wallet) return

      let account: string
      try {
        account = await wallet.getAccount()
      } catch (error: any) {
        notify({
          title: 'Connection Error',
          type: 'error',
          description:
            'Please install Bitpie wallet and then reload this page.',
        })
      }

      this._wallet = wallet
      this._publicKey = new PublicKey(account)

      this.emit('connect')
    } catch (error: any) {
      this.emit('error', error)
      throw error
    } finally {
      this._connecting = false
    }
  }

  async disconnect(): Promise<void> {
    if (this._wallet) {
      this._wallet = null
      this._publicKey = null
    }

    this.emit('disconnect')
  }

  async signTransaction(transaction: Transaction): Promise<Transaction> {
    try {
      const wallet = this._wallet
      if (!wallet) return

      return (await wallet.signTransaction(transaction)) || transaction
    } catch (error: any) {
      this.emit('error', error)
      throw error
    }
  }

  async signAllTransactions(
    transactions: Transaction[]
  ): Promise<Transaction[]> {
    try {
      const wallet = this._wallet
      if (!wallet) return

      return (await wallet.signAllTransactions(transactions)) || transactions
    } catch (error: any) {
      this.emit('error', error)
      throw error
    }
  }
}