import { DENOM, KYVE_DECIMALS } from "@kyve/sdk-beta/dist/constants";
import BigNumber from "bignumber.js";

import { Node, standardizeJSON } from "../..";

/**
 * getBalances tries to retrieve the $KYVE balance of the staker account, the $KYVE
 * balance of the valaccount and the balance of the storage provider which
 * can be of any currency for metrics
 *
 * @method getBalances
 * @param {Node} this
 * @return {Promise<void>}
 */
export async function getBalances(this: Node): Promise<void> {
  try {
    this.logger.debug(
      `this.client.nativeClient.getBalance(${this.staker},${DENOM})`
    );

    const stakerBalanceRaw = await this.client.nativeClient.getBalance(
      this.staker,
      DENOM
    );

    this.logger.debug(JSON.stringify(stakerBalanceRaw));

    const stakerBalance = new BigNumber(stakerBalanceRaw.amount)
      .dividedBy(new BigNumber(10).exponentiatedBy(KYVE_DECIMALS))
      .toNumber();

    this.m.balance_staker.set(stakerBalance);
  } catch (err) {
    this.logger.error(`Failed to get $KYVE balance of staker`);
    this.logger.error(standardizeJSON(err));
  }

  try {
    this.logger.debug(
      `this.client.nativeClient.getBalance(${this.client.account.address},${DENOM})`
    );

    const valaccountBalanceRaw = await this.client.nativeClient.getBalance(
      this.client.account.address,
      DENOM
    );

    this.logger.debug(JSON.stringify(valaccountBalanceRaw));

    const valaccountBalance = new BigNumber(valaccountBalanceRaw.amount)
      .dividedBy(new BigNumber(10).exponentiatedBy(KYVE_DECIMALS))
      .toNumber();

    this.m.balance_valaccount.set(valaccountBalance);
  } catch (err) {
    this.logger.error(`Failed to get $KYVE balance of valaccount`);
    this.logger.error(standardizeJSON(err));
  }

  try {
    // get current storage provider defined on pool
    this.logger.debug(
      `storageProviderFactory(${
        this.pool.data?.current_storage_provider_id ?? 0
      }, $STORAGE_PRIV)`
    );
    const storageProvider = await this.storageProviderFactory(
      this.pool.data?.current_storage_provider_id ?? 0
    );

    this.logger.debug(`this.storageProvider.getBalance()`);

    const storageProviderBalanceRaw = await storageProvider.getBalance();

    this.logger.debug(JSON.stringify(storageProviderBalanceRaw));

    const storageProviderBalance = new BigNumber(storageProviderBalanceRaw)
      .dividedBy(new BigNumber(10).exponentiatedBy(storageProvider.decimals))
      .toNumber();

    this.m.balance_storage_provider.set(storageProviderBalance);
  } catch (err) {
    this.logger.error(`Failed to get balance of storage provider`);
    this.logger.error(standardizeJSON(err));
  }
}