import {
  APPROVAL_AMOUNT_LENDING_POOL_CORE,
  MOCK_ETH_ADDRESS,
  AAVE_REFERRAL,
  MAX_UINT_AMOUNT,
  ZERO_ADDRESS,
} from "../helpers/constants";
import {AToken} from "../types/AToken";
import {MintableErc20} from "../types/MintableErc20";
import {LendingPool} from "../types/LendingPool";
import {LendingPoolCore} from "../types/LendingPoolCore";
import {
  getAaveProtocolTestHelpers,
  getMintableErc20,
  getAToken,
  convertToCurrencyDecimals,
  getEthersSigners,
  getLendingPoolCore,
  getLendingPool,
} from "../helpers/contracts-helpers";
import {expect} from "chai";
import {Signer, ethers} from "ethers";
import {RateMode} from "../helpers/types";
import { makeSuite } from './helpers/make-suite';



makeSuite("AToken: Transfer", () => {
  let deployer: Signer;
  let users: Signer[];
  let _aDai: AToken;
  let _dai: MintableErc20;
  let _lendingPool: LendingPool;
  let _lendingPoolCore: LendingPoolCore;

  before(async () => {
    const [_deployer, ..._users] = await getEthersSigners();
    deployer = _deployer;
    users = _users;

    _lendingPool = await getLendingPool();
    _lendingPoolCore = await getLendingPoolCore();

    const testHelpers = await getAaveProtocolTestHelpers();

    const aDaiAddress = (await testHelpers.getAllATokens()).find(
      (aToken) => aToken.symbol === "aDAI"
    )?.tokenAddress;

    const daiAddress = (await testHelpers.getAllReservesTokens()).find(
      (token) => token.symbol === "DAI"
    )?.tokenAddress;
    if (!aDaiAddress) {
      console.log(`atoken-modifiers.spec: aDAI not correctly initialized`);
      process.exit(1);
    }
    if (!daiAddress) {
      console.log(`atoken-modifiers.spec: DAI not correctly initialized`);
      process.exit(1);
    }

    _aDai = await getAToken(aDaiAddress);
    _dai = await getMintableErc20(daiAddress);
  });

  it("User 0 deposits 1000 DAI, transfers to user 1", async () => {
    await _dai
      .connect(users[0])
      .mint(await convertToCurrencyDecimals(_dai.address, "1000"));

    await _dai
      .connect(users[0])
      .approve(_lendingPoolCore.address, APPROVAL_AMOUNT_LENDING_POOL_CORE);

    //user 1 deposits 1000 DAI
    const amountDAItoDeposit = await convertToCurrencyDecimals(
      _dai.address,
      "1000"
    );

    await _lendingPool
      .connect(users[0])
      .deposit(_dai.address, amountDAItoDeposit, "0");

    await _aDai
      .connect(users[0])
      .transfer(await users[1].getAddress(), amountDAItoDeposit);

    const fromBalance = await _aDai.balanceOf(await users[0].getAddress());
    const toBalance = await _aDai.balanceOf(await users[1].getAddress());

    expect(fromBalance.toString()).to.be.equal(
      "0",
      "Invalid from balance after transfer"
    );
    expect(toBalance.toString()).to.be.equal(
      amountDAItoDeposit.toString(),
      "Invalid to balance after transfer"
    );
  });

  it("User 1 redirects interest to user 2, transfers 500 DAI back to user 0", async () => {
    await _aDai
      .connect(users[1])
      .redirectInterestStream(await users[2].getAddress());

    const aDAIRedirected = await convertToCurrencyDecimals(
      _dai.address,
      "1000"
    );

    const aDAItoTransfer = await convertToCurrencyDecimals(_dai.address, "500");

    const user2RedirectedBalanceBefore = await _aDai.getRedirectedBalance(
      await users[2].getAddress()
    );
    expect(user2RedirectedBalanceBefore.toString()).to.be.equal(
      aDAIRedirected,
      "Invalid redirected balance for user 2 before transfer"
    );

    await _aDai
      .connect(users[1])
      .transfer(await users[0].getAddress(), aDAItoTransfer);

    const user2RedirectedBalanceAfter = await _aDai.getRedirectedBalance(
      await users[2].getAddress()
    );
    const user1RedirectionAddress = await _aDai.getInterestRedirectionAddress(
      await users[1].getAddress()
    );

    expect(user2RedirectedBalanceAfter.toString()).to.be.equal(
      aDAItoTransfer,
      "Invalid redirected balance for user 2 after transfer"
    );
    expect(user1RedirectionAddress.toString()).to.be.equal(
      await users[2].getAddress(),
      "Invalid redirection address for user 1"
    );
  });

  it("User 0 transfers back to user 1", async () => {
    const aDAItoTransfer = await convertToCurrencyDecimals(_dai.address, "500");

    await _aDai
      .connect(users[0])
      .transfer(await users[1].getAddress(), aDAItoTransfer);

    const user2RedirectedBalanceAfter = await _aDai.getRedirectedBalance(
      await users[2].getAddress()
    );

    const user1BalanceAfter = await _aDai.balanceOf(
      await users[1].getAddress()
    );

    expect(user2RedirectedBalanceAfter.toString()).to.be.equal(
      user1BalanceAfter.toString(),
      "Invalid redirected balance for user 2 after transfer"
    );
  });

  it("User 0 deposits 1 ETH and user tries to borrow, but the aTokens received as a transfer are not available as collateral (revert expected)", async () => {
    await _lendingPool
      .connect(users[0])
      .deposit(MOCK_ETH_ADDRESS, ethers.utils.parseEther("1.0"), "0", {
        value: ethers.utils.parseEther("1.0"),
      });
    await expect(
      _lendingPool
        .connect(users[1])
        .borrow(
          MOCK_ETH_ADDRESS,
          ethers.utils.parseEther("0.1"),
          RateMode.Stable,
          AAVE_REFERRAL
        ),
      "The collateral balance is 0"
    ).to.be.revertedWith("The collateral balance is 0");
  });

  it("User 1 sets the DAI as collateral and borrows, tries to transfer everything back to user 0 (revert expected)", async () => {
    await _lendingPool
      .connect(users[1])
      .setUserUseReserveAsCollateral(_dai.address, true);

    const aDAItoTransfer = await convertToCurrencyDecimals(
      _dai.address,
      "1000"
    );

    await _lendingPool
      .connect(users[1])
      .borrow(
        MOCK_ETH_ADDRESS,
        ethers.utils.parseEther("0.1"),
        RateMode.Stable,
        AAVE_REFERRAL
      );

    await expect(
      _aDai
        .connect(users[1])
        .transfer(await users[0].getAddress(), aDAItoTransfer),
      "Transfer cannot be allowed."
    ).to.be.revertedWith("Transfer cannot be allowed.");
  });

  it("User 0 tries to transfer 0 balance (revert expected)", async () => {
    await expect(
      _aDai.connect(users[0]).transfer(await users[1].getAddress(), "0"),
      "Transferred amount needs to be greater than zero"
    ).to.be.revertedWith("Transferred amount needs to be greater than zero");
  });

  it("User 1 repays the borrow, transfers aDAI back to user 0", async () => {
    await _lendingPool
      .connect(users[1])
      .repay(MOCK_ETH_ADDRESS, MAX_UINT_AMOUNT, await users[1].getAddress(), {
        value: ethers.utils.parseEther("1"),
      });

    const aDAItoTransfer = await convertToCurrencyDecimals(
      _aDai.address,
      "1000"
    );

    await _aDai
      .connect(users[1])
      .transfer(await users[0].getAddress(), aDAItoTransfer);

    const user2RedirectedBalanceAfter = await _aDai.getRedirectedBalance(
      await users[2].getAddress()
    );

    const user1RedirectionAddress = await _aDai.getInterestRedirectionAddress(
      await users[1].getAddress()
    );

    expect(user2RedirectedBalanceAfter.toString()).to.be.equal(
      "0",
      "Invalid redirected balance for user 2 after transfer"
    );

    expect(user1RedirectionAddress.toString()).to.be.equal(
      ZERO_ADDRESS,
      "Invalid redirected address for user 1"
    );
  });

  it("User 0 redirects interest to user 2, transfers 500 aDAI to user 1. User 1 redirects to user 3. User 0 transfers another 100 aDAI", async () => {
    let aDAItoTransfer = await convertToCurrencyDecimals(_aDai.address, "500");

    await _aDai
      .connect(users[0])
      .redirectInterestStream(await users[2].getAddress());

    await _aDai
      .connect(users[0])
      .transfer(await users[1].getAddress(), aDAItoTransfer);

    await _aDai
      .connect(users[1])
      .redirectInterestStream(await users[3].getAddress());

    aDAItoTransfer = await convertToCurrencyDecimals(_aDai.address, "100");

    await _aDai
      .connect(users[0])
      .transfer(await users[1].getAddress(), aDAItoTransfer);

    const user2RedirectedBalanceAfter = await _aDai.getRedirectedBalance(
      await users[2].getAddress()
    );
    const user3RedirectedBalanceAfter = await _aDai.getRedirectedBalance(
      await users[3].getAddress()
    );

    const expectedUser2Redirected = await convertToCurrencyDecimals(
      _aDai.address,
      "400"
    );
    const expectedUser3Redirected = await convertToCurrencyDecimals(
      _aDai.address,
      "600"
    );

    expect(user2RedirectedBalanceAfter.toString()).to.be.equal(
      expectedUser2Redirected,
      "Invalid redirected balance for user 2 after transfer"
    );
    expect(user3RedirectedBalanceAfter.toString()).to.be.equal(
      expectedUser3Redirected,
      "Invalid redirected balance for user 3 after transfer"
    );
  });
});