import { ChainGetter, requireCosmosInfo } from "../../../chain";
import { CoinPretty, Int } from "@keplr-wallet/unit";
import { ObservableCosmwasmContractChainQuery } from "../contract-query";
import { QuerySharedContext } from "../../../common";
import { AppCurrency } from "@keplr-wallet/types";
import { ObservableChainQueryMap } from "../../chain-query";
import { computed } from "mobx";

interface NeutronStakingRewardsResponse {
  pending_rewards?: {
    denom?: string;
    amount?: string;
  };
}

class ObservableQueryNeutronStakingRewardsInner extends ObservableCosmwasmContractChainQuery<NeutronStakingRewardsResponse> {
  public static readonly NEUTRON_REWARDS_CONTRACT_ADDRESS =
    "neutron1gqq3c735pj6ese3yru5xr6ud0fvxgltxesygvyyzpsrt74v6yg4sgkrgwq";

  constructor(
    sharedContext: QuerySharedContext,
    chainId: string,
    chainGetter: ChainGetter,
    protected readonly bech32Address: string
  ) {
    super(
      sharedContext,
      chainId,
      chainGetter,
      ObservableQueryNeutronStakingRewardsInner.NEUTRON_REWARDS_CONTRACT_ADDRESS,
      {
        rewards: { user: bech32Address },
      }
    );
  }

  protected override canFetch(): boolean {
    return super.canFetch() && this.bech32Address !== "";
  }

  @computed
  get pendingReward(): CoinPretty {
    if (!this.response?.data?.pending_rewards) {
      const mcInfo2 = this.chainGetter.getModularChain(this.chainId);
      const cosmosInfo = requireCosmosInfo(mcInfo2);
      const defaultCurrency =
        cosmosInfo.stakeCurrency || cosmosInfo.currencies[0];

      return new CoinPretty(defaultCurrency, new Int(0)).ready(false);
    }

    const reward = this.response.data.pending_rewards;
    if (!reward.denom || !reward.amount) {
      const mcInfo2 = this.chainGetter.getModularChain(this.chainId);
      const cosmosInfo = requireCosmosInfo(mcInfo2);
      const defaultCurrency =
        cosmosInfo.stakeCurrency || cosmosInfo.currencies[0];

      return new CoinPretty(defaultCurrency, new Int(0)).ready(false);
    }
    const mcInfo2 = this.chainGetter.getModularChain(this.chainId);
    const currency = mcInfo2.forceFindCurrency(reward.denom);

    return new CoinPretty(currency, new Int(reward.amount)).ready(
      !this.isFetching
    );
  }

  @computed
  get rewardCurrency(): AppCurrency | undefined {
    if (!this.response?.data?.pending_rewards) {
      return undefined;
    }
    const reward = this.response.data.pending_rewards;
    if (!reward.denom || !reward.amount) {
      return undefined;
    }
    const mcInfo2 = this.chainGetter.getModularChain(this.chainId);
    return mcInfo2.findCurrency(reward.denom);
  }
}

export class ObservableQueryNeutronStakingRewards extends ObservableChainQueryMap<NeutronStakingRewardsResponse> {
  constructor(
    sharedContext: QuerySharedContext,
    chainId: string,
    chainGetter: ChainGetter
  ) {
    super(sharedContext, chainId, chainGetter, (contractAddress: string) => {
      return new ObservableQueryNeutronStakingRewardsInner(
        this.sharedContext,
        this.chainId,
        this.chainGetter,
        contractAddress
      );
    });
  }

  getRewardFor(
    bech32Address: string
  ): ObservableQueryNeutronStakingRewardsInner {
    return this.get(bech32Address) as ObservableQueryNeutronStakingRewardsInner;
  }
}