import {
  Secp256k1Keypair,
  RoochAddress,
  BitcoinAddress,
} from '@roochnetwork/rooch-sdk';
import { getKeyPair } from './keypair';
import { generateMnemonic } from './crypto';
import type { AccountMangerParams, DerivePathParams } from '../../types';
import RoochSDK from '@roochnetwork/rooch-sdk';

export class RoochAccountManager {
  private mnemonics: string;
  private secretKey: string;
  public currentKeyPair: Secp256k1Keypair;
  public currentAddress: RoochAddress;
  public currentBitcoinAddress: BitcoinAddress;

  /**
   * Support the following ways to init the SuiToolkit:
   * 1. mnemonics
   * 2. secretKey (base64 or hex)
   * If none of them is provided, will generate a random mnemonics with 24 words.
   *
   * @param mnemonics, 12 or 24 mnemonics words, separated by space
   * @param secretKey, base64 or hex string, when mnemonics is provided, secretKey will be ignored
   */
  constructor({ mnemonics, secretKey }: AccountMangerParams = {}) {
    // If the mnemonics or secretKey is provided, use it
    // Otherwise, generate a random mnemonics with 24 words
    this.mnemonics = mnemonics || '';
    this.secretKey = secretKey || '';
    if (!this.mnemonics && !this.secretKey) {
      this.mnemonics = generateMnemonic(24);
    }

    this.currentKeyPair = this.secretKey
      ? Secp256k1Keypair.fromSecretKey(secretKey!)
      : getKeyPair(this.mnemonics);

    this.currentAddress = this.currentKeyPair.getRoochAddress();
    this.currentBitcoinAddress = this.currentKeyPair.getBitcoinAddress();
  }

  /**
   * if derivePathParams is not provided or mnemonics is empty, it will return the currentKeyPair.
   * else:
   * it will generate keyPair from the mnemonic with the given derivePathParams.
   */
  getKeyPair(derivePathParams?: DerivePathParams) {
    if (!derivePathParams || !this.mnemonics) return this.currentKeyPair;
    return getKeyPair(this.mnemonics, derivePathParams);
  }

  /**
   * if derivePathParams is not provided or mnemonics is empty, it will return the currentAddress.
   * else:
   * it will generate address from the mnemonic with the given derivePathParams.
   */
  getAddress(derivePathParams?: DerivePathParams): RoochAddress {
    if (!derivePathParams || !this.mnemonics) return this.currentAddress;
    return getKeyPair(this.mnemonics, derivePathParams).getRoochAddress();
  }

  getBitcoinAddress(derivePathParams?: DerivePathParams): BitcoinAddress {
    if (!derivePathParams || !this.mnemonics) return this.currentBitcoinAddress;
    return getKeyPair(this.mnemonics, derivePathParams).getBitcoinAddress();
  }

  /**
   * Switch the current account with the given derivePathParams.
   * This is only useful when the mnemonics is provided. For secretKey mode, it will always use the same account.
   */
  switchAccount(derivePathParams: DerivePathParams) {
    if (this.mnemonics) {
      this.currentKeyPair = getKeyPair(this.mnemonics, derivePathParams);
      this.currentAddress = this.currentKeyPair.getRoochAddress();
      this.currentBitcoinAddress = this.currentKeyPair.getBitcoinAddress();
    }
  }
}