import { Query as BankQuery } from '@routerprotocol/chain-api/cosmos/bank/v1beta1/query_pb_service';
import {
  QueryAllBalancesRequest,
  QueryAllBalancesResponse,
  QueryBalanceRequest,
  QueryBalanceResponse,
  QueryTotalSupplyRequest,
  QueryTotalSupplyResponse,
  QueryParamsRequest as QueryBankParamsRequest,
  QueryParamsResponse as QueryBankParamsResponse,
} from '@routerprotocol/chain-api/cosmos/bank/v1beta1/query_pb';
import BaseConsumer from '../../BaseGrpcConsumer';
import { PaginationOption } from '../../../types/pagination';
import { paginationRequestFromPagination } from '../../../utils/pagination';
import { ChainGrpcBankTransformer } from '../transformers';

/**
 * The bank module is responsible for handling multi-asset coin transfers between accounts.
 * 
 * @group gRPC API
 * 
 * @example
 * To use Bank methods, initialise a {@link ChainGrpcBankApi} object to with a gRPC endpoint. An endpoint can be retrieved by using {@link networkEndpoints}.
 * ```ts
 * const endpoint =  getEndpointsForNetwork(Network.Devnet).grpcEndpoint;
 * const bankClient = new ChainGrpcBankApi(endpoint);
 * const response = await bankClient.fetchBalances("router12avkt8c0lk76atge8f3pe2t6fj4jsc8038d742");
 * ```
 */

export class ChainGrpcBankApi extends BaseConsumer {
  async fetchModuleParams() {
    const request = new QueryBankParamsRequest();

    try {
      const response = await this.request<
        QueryBankParamsRequest,
        QueryBankParamsResponse,
        typeof BankQuery.Params
      >(request, BankQuery.Params);

      return ChainGrpcBankTransformer.moduleParamsResponseToModuleParams(
        response
      );
    } catch (e) {
      //@ts-ignore
      throw new Error(e.message);
    }
  }

  /**
   * Get balance of a coin.
   * 
   * @param accountAddress account address.
   * @param denom coin denomination.
   * @returns balance.
   */
  async fetchBalance({
    accountAddress,
    denom,
  }: {
    accountAddress: string;
    denom: string;
  }) {
    const request = new QueryBalanceRequest();
    request.setAddress(accountAddress);
    request.setDenom(denom);

    try {
      const response = await this.request<
        QueryBalanceRequest,
        QueryBalanceResponse,
        typeof BankQuery.Balance
      >(request, BankQuery.Balance);

      return ChainGrpcBankTransformer.balanceResponseToBalance(response);
    } catch (e) {
      //@ts-ignore
      throw new Error(e.message);
    }
  }

  /**
   * Get balance of all coins
   * 
   * @param address account address.
   * @returns list of balances.
   */
  async fetchBalances(address: string) {
    const request = new QueryAllBalancesRequest();
    request.setAddress(address);

    try {
      const response = await this.request<
        QueryAllBalancesRequest,
        QueryAllBalancesResponse,
        typeof BankQuery.AllBalances
      >(request, BankQuery.AllBalances);

      return ChainGrpcBankTransformer.balancesResponseToBalances(response);
    } catch (e) {
      //@ts-ignore
      throw new Error(e.message);
    }
  }

  /**
   * Get total supply of coins.
   * 
   * @param pagination pagination options (optional param).
   * @returns total supply.
   */
  async fetchTotalSupply(pagination?: PaginationOption) {
    const request = new QueryTotalSupplyRequest();
    const paginationForRequest = paginationRequestFromPagination(pagination);

    if (paginationForRequest) {
      request.setPagination(paginationForRequest);
    }

    try {
      const response = await this.request<
        QueryTotalSupplyRequest,
        QueryTotalSupplyResponse,
        typeof BankQuery.TotalSupply
      >(request, BankQuery.TotalSupply);

      return ChainGrpcBankTransformer.totalSupplyResponseToTotalSupply(
        response
      );
    } catch (e) {
      //@ts-ignore
      throw new Error(e.message);
    }
  }
}