/*

    Copyright 2020 DODO ZOO.
    SPDX-License-Identifier: Apache-2.0

*/

import { EVM, getDefaultWeb3 } from "./EVM";
import Web3 from "web3";
import { Contract } from "web3-eth-contract";
import BigNumber from "bignumber.js";
import * as contracts from "./Contracts";
import { decimalStr, gweiStr, MAX_UINT256 } from "./Converter";
import * as log from "./Log";

BigNumber.config({
  EXPONENTIAL_AT: 1000,
  DECIMAL_PLACES: 80,
});

export interface DODOContextInitConfig {
  lpFeeRate: string,
  mtFeeRate: string,
  k: string,
  gasPriceLimit: string,
}

/*
  price curve when k=0.1
  +──────────────────────+───────────────+
  | purchase percentage  | avg slippage  |
  +──────────────────────+───────────────+
  | 1%                   | 0.1%         |
  | 5%                   | 0.5%         |
  | 10%                  | 1.1%         |
  | 20%                  | 2.5%         |
  | 50%                  | 10%          |
  | 70%                  | 23.3%        |
  +──────────────────────+───────────────+
*/
export let DefaultDODOContextInitConfig = {
  lpFeeRate: decimalStr("0.002"),
  mtFeeRate: decimalStr("0.001"),
  k: decimalStr("0.1"),
  gasPriceLimit: gweiStr("100"),
}

export class DODOContext {
  EVM: EVM
  Web3: Web3
  DODO: Contract
  DODOZoo: Contract
  BASE: Contract
  BaseCapital: Contract
  QUOTE: Contract
  QuoteCapital: Contract
  ORACLE: Contract
  Deployer: string
  Supervisor: string
  Maintainer: string
  spareAccounts: string[]

  constructor() { }

  async init(config: DODOContextInitConfig) {
    this.EVM = new EVM
    this.Web3 = getDefaultWeb3()
    this.DODOZoo = await contracts.newContract(contracts.DODO_ZOO_CONTRACT_NAME)
    this.BASE = await contracts.newContract(contracts.TEST_ERC20_CONTRACT_NAME, ["TestBase", 18])
    this.QUOTE = await contracts.newContract(contracts.TEST_ERC20_CONTRACT_NAME, ["TestQuote", 18])
    this.ORACLE = await contracts.newContract(contracts.NAIVE_ORACLE_CONTRACT_NAME)

    const allAccounts = await this.Web3.eth.getAccounts();
    this.Deployer = allAccounts[0]
    this.Supervisor = allAccounts[1]
    this.Maintainer = allAccounts[2]
    this.spareAccounts = allAccounts.slice(3, 10)

    await this.DODOZoo.methods.breedDODO(
      this.Supervisor,
      this.Maintainer,
      this.BASE.options.address,
      this.QUOTE.options.address,
      this.ORACLE.options.address,
      config.lpFeeRate,
      config.mtFeeRate,
      config.k,
      config.gasPriceLimit
    ).send(this.sendParam(this.Deployer))

    this.DODO = contracts.getContractWithAddress(contracts.DODO_CONTRACT_NAME, await this.DODOZoo.methods.getDODO(this.BASE.options.address, this.QUOTE.options.address).call())

    this.BaseCapital = contracts.getContractWithAddress(contracts.DODO_LP_TOKEN_CONTRACT_NAME, await this.DODO.methods._BASE_CAPITAL_TOKEN_().call())
    this.QuoteCapital = contracts.getContractWithAddress(contracts.DODO_LP_TOKEN_CONTRACT_NAME, await this.DODO.methods._QUOTE_CAPITAL_TOKEN_().call())

    this.DODO.methods.claimOwnership().send(this.sendParam(this.Deployer))

    console.log(log.blueText("[Init dodo context]"))
  }

  sendParam(sender, value = "0") {
    return {
      from: sender,
      gas: process.env["COVERAGE"] ? 10000000000 : 7000000,
      gasPrice: process.env.GAS_PRICE,
      value: decimalStr(value)
    }
  }

  async setOraclePrice(price: string) {
    await this.ORACLE.methods.setPrice(price).send(this.sendParam(this.Deployer))
  }

  async mintTestToken(to: string, base: string, quote: string) {
    await this.BASE.methods.mint(to, base).send(this.sendParam(this.Deployer))
    await this.QUOTE.methods.mint(to, quote).send(this.sendParam(this.Deployer))
  }

  async approveDODO(account: string) {
    await this.BASE.methods.approve(this.DODO.options.address, MAX_UINT256).send(this.sendParam(account))
    await this.QUOTE.methods.approve(this.DODO.options.address, MAX_UINT256).send(this.sendParam(account))
  }
}

export async function getDODOContext(config: DODOContextInitConfig = DefaultDODOContextInitConfig): Promise<DODOContext> {
  var context = new DODOContext()
  await context.init(config)
  return context
}