import { TransactionBlock } from '@mysten/sui.js/transactions';

import { ContractConfig } from '@/common/env';
import { Globals } from '@/common/globals';
import { BaseContract } from '@/transaction/contracts/BaseContract';
import { MoveNumber } from '@/transaction/contracts/common';

export class FeeContract extends BaseContract {
  static ModuleName = 'fee_module';

  static MethodName = {
    set_streaming_fee: 'set_streaming_fee',
    set_claim_fee: 'set_claim_fee',
    set_streaming_flat_fee: 'set_streaming_flat_fee',
    streaming_flat_fee: 'streaming_flat_fee',
    streaming_fee: 'streaming_fee',
    claim_fee: 'claim_fee',
    fee_denominator: 'fee_denominator',
  } as const;

  constructor(
    public readonly config: ContractConfig,
    public readonly globals: Globals,
  ) {
    super(FeeContract.ModuleName, config, globals);
  }

  setStreamingFee(txb: TransactionBlock, createFeeNumerator: MoveNumber) {
    const roleObject = this.roleObject();
    const feeObject = this.feeObject();
    return this.addContractCall(txb, {
      method: FeeContract.MethodName.set_streaming_fee,
      arguments: [roleObject, feeObject, createFeeNumerator],
      typeArgs: [],
    });
  }

  setStreamingFlatFee(txb: TransactionBlock, flatFee: MoveNumber) {
    const roleObject = this.roleObject();
    const feeObject = this.feeObject();
    return this.addContractCall(txb, {
      method: FeeContract.MethodName.set_streaming_flat_fee,
      arguments: [roleObject, feeObject, flatFee],
      typeArgs: [],
    });
  }

  setClaimFee(txb: TransactionBlock, claimFee: MoveNumber) {
    const roleObject = this.roleObject();
    const feeObject = this.feeObject();
    return this.addContractCall(txb, {
      method: FeeContract.MethodName.set_claim_fee,
      arguments: [roleObject, feeObject, claimFee],
      typeArgs: [],
    });
  }

  streamingFee(txb: TransactionBlock, amount: MoveNumber) {
    const feeObject = this.feeObject();
    return this.addContractCall(txb, {
      method: FeeContract.MethodName.streaming_fee,
      arguments: [feeObject, amount],
      typeArgs: [],
    });
  }

  claimFee(txb: TransactionBlock, amount: MoveNumber) {
    return this.addContractCall(txb, {
      method: FeeContract.MethodName.claim_fee,
      arguments: [this.feeObject(), amount],
      typeArgs: [],
    });
  }

  feeDenominator(txb: TransactionBlock) {
    return this.addContractCall(txb, {
      method: FeeContract.MethodName.fee_denominator,
      arguments: [this.feeObject()],
      typeArgs: [],
    });
  }
}