import {useSolanaStandardWallets, type SolanaStandardWallet} from '@privy-io/react-auth/solana';

function WalletComponent() {
  const {ready, wallets} = useSolanaStandardWallets();

  // Connect/Disconnect
  const connect = (wallet: SolanaStandardWallet) => wallet.features['standard:connect']!.connect();
  const disconnect = (wallet: SolanaStandardWallet) =>
    wallet.features['standard:disconnect']!.disconnect();

  // Sign Message
  const signMessage = async (
    wallet: SolanaStandardWallet,
    address: string,
    message: Uint8Array
  ) => {
    const account = wallet.accounts.find((a) => a.address === address)!;
    const [result] = await wallet.features['solana:signMessage']!.signMessage({
      account,
      message
    });
    return result;
  };

  // Sign Transaction
  const signTransaction = async (
    wallet: SolanaStandardWallet,
    address: string,
    transaction: Uint8Array
  ) => {
    const account = wallet.accounts.find((a) => a.address === address)!;
    const [result] = await wallet.features['solana:signTransaction']!.signTransaction({
      transaction,
      chain: 'solana:devnet',
      account
    });
    return result;
  };

  // Sign and Send Transaction
  const signAndSendTransaction = async (
    wallet: SolanaStandardWallet,
    address: string,
    transaction: Uint8Array
  ) => {
    const account = wallet.accounts.find((a) => a.address === address)!;
    return wallet.features['solana:signAndSendTransaction']!.signAndSendTransaction({
      transaction,
      chain: 'solana:devnet',
      account
    });
  };
}

// This is copied from @wallet-standard/wallet

import type {
  Wallet,
  WalletEventsWindow,
  WindowRegisterWalletEvent,
  WindowRegisterWalletEventCallback
} from '@wallet-standard/base';

class RegisterWalletEvent
  extends CustomEvent<WindowRegisterWalletEventCallback>
  implements WindowRegisterWalletEvent
{
  readonly #detail: WindowRegisterWalletEventCallback;

  get detail() {
    return this.#detail;
  }

  get type() {
    return 'wallet-standard:register-wallet' as const;
  }

  constructor(callback: WindowRegisterWalletEventCallback) {
    super('wallet-standard:register-wallet', {
      bubbles: false,
      cancelable: false,
      detail: callback
    });
    this.#detail = callback;
  }

  preventDefault(): never {
    throw new Error('preventDefault is not supported');
  }

  stopPropagation(): never {
    throw new Error('stopPropagation is not supported');
  }

  stopImmediatePropagation(): never {
    throw new Error('stopImmediatePropagation is not supported');
  }
}

export function registerWallet(wallet: Wallet): void {
  const callback: WindowRegisterWalletEventCallback = ({register}) => register(wallet);
  try {
    (window as WalletEventsWindow).dispatchEvent(new RegisterWalletEvent(callback));
  } catch (error) {
    console.error('wallet-standard:register-wallet event could not be dispatched\n', error);
  }
  try {
    (window as WalletEventsWindow).addEventListener('wallet-standard:app-ready', ({detail: api}) =>
      callback(api)
    );
  } catch (error) {
    console.error('wallet-standard:app-ready event listener could not be added\n', error);
  }
}

registerWallet(wallets.find((wallet) => wallet.name === 'Privy' && 'privy:' in wallet.features));