/*
Copyright 2018 - 2022 The Alephium Authors
This file is part of the alephium project.

The library is free software: you can redistribute it and/or modify
it under the terms of the GNU Lesser General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

The library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Lesser General Public License for more details.

You should have received a copy of the GNU Lesser General Public License
along with the library. If not, see <http://www.gnu.org/licenses/>.
*/

import {
  web3,
  Account,
  SignerProviderWithMultipleAccounts,
  NodeProvider,
  groupOfAddress,
  ExplorerProvider,
  Address
} from '@alephium/web3'

export class NodeWallet extends SignerProviderWithMultipleAccounts {
  walletName: string
  readonly nodeProvider: NodeProvider
  readonly explorerProvider: ExplorerProvider | undefined

  constructor(walletName: string, nodeProvider?: NodeProvider, explorerProvider?: ExplorerProvider) {
    super()
    this.walletName = walletName
    this.nodeProvider = nodeProvider ?? web3.getCurrentNodeProvider()
    this.explorerProvider = explorerProvider ?? web3.getCurrentExplorerProvider()
  }

  async setSelectedAccount(address: Address): Promise<void> {
    await this.nodeProvider.wallets.postWalletsWalletNameChangeActiveAddress(this.walletName, { address: address })
  }

  async getAccounts(): Promise<Account[]> {
    const walletAddresses = await this.nodeProvider.wallets.getWalletsWalletNameAddresses(this.walletName)
    const accounts: Account[] = walletAddresses.addresses.map<Account>((acc) => ({
      keyType: 'default',
      publicKey: acc.publicKey,
      address: acc.address,
      group: acc.group
    }))
    return accounts
  }

  protected async unsafeGetSelectedAccount(): Promise<Account> {
    const response = await this.nodeProvider.wallets.getWalletsWalletNameAddresses(this.walletName)
    const selectedAddressInfo = response.addresses.find((info) => info.address === response.activeAddress)!
    return {
      keyType: 'default',
      address: selectedAddressInfo.address,
      group: groupOfAddress(selectedAddressInfo.address),
      publicKey: selectedAddressInfo.publicKey
    }
  }

  async signRaw(signerAddress: string, hexString: string): Promise<string> {
    const provider = this.nodeProvider
    const currentActiveAddressResponse = await provider.wallets.getWalletsWalletNameAddresses(this.walletName)
    const { activeAddress } = currentActiveAddressResponse
    await provider.wallets.postWalletsWalletNameChangeActiveAddress(this.walletName, { address: signerAddress })
    const { signature } = await provider.wallets.postWalletsWalletNameSign(this.walletName, { data: hexString })

    await provider.wallets.postWalletsWalletNameChangeActiveAddress(this.walletName, { address: activeAddress }) // set the address that's active back to previous state

    return signature
  }

  async unlock(password: string): Promise<void> {
    return await this.nodeProvider.wallets.postWalletsWalletNameUnlock(this.walletName, { password })
  }

  async lock(): Promise<void> {
    return await this.nodeProvider.wallets.postWalletsWalletNameLock(this.walletName)
  }
}