import { Cw20ContractTokenInfo } from "./types";
import { ObservableChainQueryMap } from "../chain-query";
import { ChainGetter } from "../../chain";
import { computed } from "mobx";
import { ObservableCosmwasmContractChainQuery } from "./contract-query";
import { QuerySharedContext } from "../../common";

export class ObservableQueryCw20ContactInfoInner extends ObservableCosmwasmContractChainQuery<Cw20ContractTokenInfo> {
  constructor(
    sharedContext: QuerySharedContext,
    chainId: string,
    chainGetter: ChainGetter,
    contractAddress: string
  ) {
    super(sharedContext, chainId, chainGetter, contractAddress, {
      token_info: {},
    });
  }

  @computed
  get tokenInfo(): Cw20ContractTokenInfo | undefined {
    if (!this.response || !this.response.data) {
      return undefined;
    }

    return this.response.data;
  }
}

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

  getQueryContract(
    contractAddress: string
  ): ObservableQueryCw20ContactInfoInner {
    return this.get(contractAddress) as ObservableQueryCw20ContactInfoInner;
  }
}