const { expect } = require("chai");
const hre = require("hardhat");
const { web3, deployments, waffle, ethers } = hre;
const { provider, deployContract } = waffle

const deployAndEnableConnector = require("../../scripts/deployAndEnableConnector.js")
const buildDSAv2 = require("../../scripts/buildDSAv2")
const encodeSpells = require("../../scripts/encodeSpells.js")
const getMasterSigner = require("../../scripts/getMasterSigner")

const addresses = require("../../scripts/constant/addresses");
const abis = require("../../scripts/constant/abis");
const constants = require("../../scripts/constant/constant");
const tokens = require("../../scripts/constant/tokens");

const connectV2CompoundArtifacts = require("../../artifacts/contracts/mainnet/connectors/compound/main.sol/ConnectV2Compound.json")
const connectV2PoolTogetherArtifacts = require("../../artifacts/contracts/mainnet/connectors/pooltogether/main.sol/ConnectV2PoolTogether.json")

const token = tokens.dai.address // DAI Token
const prizePool = "0xEBfb47A7ad0FD6e57323C8A42B2E5A6a4F68fc1a" // DAI Prize Pool
const controlledToken = "0x334cBb5858417Aee161B53Ee0D5349cCF54514CF" // PT DAI Ticket
const daiPoolFaucet = "0xF362ce295F2A4eaE4348fFC8cDBCe8d729ccb8Eb"  // DAI POOL Faucet
const poolTokenAddress = "0x0cEC1A9154Ff802e7934Fc916Ed7Ca50bDE6844e"
const tokenFaucetProxyFactory = "0xE4E9cDB3E139D7E8a41172C20b6Ed17b6750f117" // TokenFaucetProxyFactory for claimAll

describe("PoolTogether", function () {
    const connectorName = "COMPOUND-TEST-A"
    const ptConnectorName = "POOLTOGETHER-TEST-A"
    
    let dsaWallet0
    let masterSigner;
    let instaConnectorsV2;
    let connector;
    let ptConnector;
    
    const wallets = provider.getWallets()
    const [wallet0, wallet1, wallet2, wallet3] = wallets
    before(async () => {
        masterSigner = await getMasterSigner(wallet3)
        instaConnectorsV2 = await ethers.getContractAt(abis.core.connectorsV2, addresses.core.connectorsV2);

        // Deploy and enable Compound Connector
        connector = await deployAndEnableConnector({
            connectorName,
            contractArtifact: connectV2CompoundArtifacts,
            signer: masterSigner,
            connectors: instaConnectorsV2
        })

        // Deploy and enable Pool Together Connector
        ptConnector = await deployAndEnableConnector({
            connectorName: ptConnectorName,
            contractArtifact: connectV2PoolTogetherArtifacts,
            signer: masterSigner,
            connectors: instaConnectorsV2
        })
  })

  it("Should have contracts deployed.", async function () {
    expect(!!instaConnectorsV2.address).to.be.true;
    expect(!!connector.address).to.be.true;
    expect(!!ptConnector.address).to.be.true;
    expect(!!masterSigner.address).to.be.true;
  });

  describe("DSA wallet setup", function () {
    it("Should build DSA v2", async function () {
        dsaWallet0 = await buildDSAv2(wallet0.address)
        expect(!!dsaWallet0.address).to.be.true;
    });

    it("Deposit ETH into DSA wallet", async function () {
        await wallet0.sendTransaction({
            to: dsaWallet0.address,
            value: ethers.utils.parseEther("10")
        });
        expect(await ethers.provider.getBalance(dsaWallet0.address)).to.be.gte(ethers.utils.parseEther("10"));
    });
  });

  describe("Main - DAI Prize Pool Test", function () {

    it("Should deposit ETH in Compound", async function () {
        const amount = ethers.utils.parseEther("1") // 1 ETH
        const spells = [
            {
                connector: connectorName,
                method: "deposit",
                args: ["ETH-A", amount, 0, 0]
            }
        ]

        const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(spells), wallet1.address)
        const receipt = await tx.wait()
        expect(await ethers.provider.getBalance(dsaWallet0.address)).to.be.lte(ethers.utils.parseEther("9"));
    });

    it("Should borrow DAI from Compound and deposit DAI into DAI Prize Pool", async function () {
        const amount = ethers.utils.parseEther("100") // 100 DAI
        const setId = "83478237"
        const spells = [
            {
                connector: connectorName,
                method: "borrow",
                args: ["DAI-A", amount, 0, setId]
            },
            {
                connector: ptConnectorName,
                method: "depositTo",
                args: [prizePool, dsaWallet0.address, amount, controlledToken, constants.address_zero, setId, 0]
            }
        ]
        // Before Spell
        // DAI balance 0
        let daiToken = await ethers.getContractAt(abis.basic.erc20, token)
        let daiBalance = await daiToken.balanceOf(dsaWallet0.address);
        console.log("Before spell:");
        console.log("\tBalance before: ", daiBalance.toString(), tokens.dai.symbol);

        // PT DAI Ticket balance is 0
        let cToken = await ethers.getContractAt(abis.basic.erc20, controlledToken)
        const balance = await cToken.balanceOf(dsaWallet0.address)
        const tokenName = await cToken.name()
        console.log("\tBalance before: ", balance.toString(), tokenName)

        // Run spell transaction
        const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(spells), wallet1.address)
        const receipt = await tx.wait()

        // After spell
        // Expect DAI balance to equal 0
        daiBalance = await daiToken.balanceOf(dsaWallet0.address);
        console.log("After spell:");
        console.log("\tBalance after: ", daiBalance.toString(), tokens.dai.symbol);
        expect(daiBalance).to.be.eq(ethers.utils.parseEther("0"));

        // Expect PT DAI Ticket to equal 100
        const balanceAfter = await cToken.balanceOf(dsaWallet0.address)
        console.log("\tBalance after: ", balanceAfter.toString(), tokenName)
        expect(balanceAfter.toString()).to.be.eq(ethers.utils.parseEther("100"));

        expect(await ethers.provider.getBalance(dsaWallet0.address)).to.be.lte(ethers.utils.parseEther("9"));
    });

    it("Should wait 11 days, withdraw all PrizePool, get back 100 DAI, and claim POOL", async function () {
        const amount = ethers.utils.parseEther("100") // 100 DAI
        const spells = [
            {
                connector: ptConnectorName,
                method: "withdrawInstantlyFrom",
                args: [prizePool, dsaWallet0.address, amount, controlledToken, amount, 0, 0]
            },
            {
                connector: ptConnectorName,
                method: "claim",
                args: [daiPoolFaucet, dsaWallet0.address]
            }
        ]
        
        // Before spell
        // DAI balance is 0
        let daiToken = await ethers.getContractAt(abis.basic.erc20, token)
        let daiBalance = await daiToken.balanceOf(dsaWallet0.address);
        console.log("Before Spell:")
        console.log("\tBalance before: ", daiBalance.toString(), tokens.dai.symbol);

        // PT Dai Ticket is 100
        let cToken = await ethers.getContractAt(abis.basic.erc20, controlledToken)
        const balance = await cToken.balanceOf(dsaWallet0.address)
        const tokenName = await cToken.name()
        console.log("\tBalance before: ", balance.toString(), tokenName)

        // PoolToken is 0
        let poolToken = await ethers.getContractAt(abis.basic.erc20, poolTokenAddress)
        const poolBalance = await poolToken.balanceOf(dsaWallet0.address)
        const poolTokenName = await poolToken.name()
        console.log("\tBalance before: ", poolBalance.toString(), poolTokenName)

        // Increase time by 11 days so we get back all DAI without early withdrawal fee
        await ethers.provider.send("evm_increaseTime", [11*24*60*60]);

        // Run spell transaction
        const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(spells), wallet1.address)
        const receipt = await tx.wait()

        // After spell
        // Expect DAI balance to be equal to 100, because of no early withdrawal fee
        daiBalance = await daiToken.balanceOf(dsaWallet0.address);
        console.log("After spell: ");
        console.log("\tBalance after: ", daiBalance.toString(), tokens.dai.symbol);
        expect(daiBalance).to.be.eq(ethers.utils.parseEther("100"));

        // Expect PT Dai Ticket to equal 0
        const balanceAfter = await cToken.balanceOf(dsaWallet0.address)
        console.log("\tBalance after: ", balanceAfter.toString(), tokenName)
        expect(balanceAfter.toNumber()).to.be.eq(0);

        // Expect Pool Token Balance to be greater than 0
        const poolBalanceAfter = await poolToken.balanceOf(dsaWallet0.address)
        console.log("\tBalance after: ", poolBalanceAfter.toString(), poolTokenName)
        expect(poolBalanceAfter).to.be.gt(ethers.utils.parseEther("0"));
    });

    it("Should deposit and withdraw all PrizePool, get back less than 100 DAI", async function() {
        const amount = ethers.utils.parseEther("100") // 100 DAI
        const spells = [
            {
                connector: ptConnectorName,
                method: "depositTo",
                args: [prizePool, dsaWallet0.address, amount, controlledToken, constants.address_zero, 0, 0]
            },
            {
                connector: ptConnectorName,
                method: "withdrawInstantlyFrom",
                args: [prizePool, dsaWallet0.address, amount, controlledToken, amount, 0, 0]
            }
        ]
        
        // Before spell
        // DAI balance is 0
        let daiToken = await ethers.getContractAt(abis.basic.erc20, token)
        let daiBalance = await daiToken.balanceOf(dsaWallet0.address);
        console.log("Before Spell:")
        console.log("\tBalance before: ", daiBalance.toString(), tokens.dai.symbol);

        // PT Dai Ticket is 100
        let cToken = await ethers.getContractAt(abis.basic.erc20, controlledToken)
        const balance = await cToken.balanceOf(dsaWallet0.address)
        const tokenName = await cToken.name()
        console.log("\tBalance before: ", balance.toString(), tokenName)

        // PoolToken is 0 
        let poolToken = await ethers.getContractAt(abis.basic.erc20, poolTokenAddress)
        const poolBalance = await poolToken.balanceOf(dsaWallet0.address)
        const poolTokenName = await poolToken.name()
        console.log("\tBalance before: ", poolBalance.toString(), poolTokenName)

        // Run spell transaction
        const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(spells), wallet1.address)
        const receipt = await tx.wait()

        // After spell
        // Expect DAI balance to be less than 100, because of early withdrawal fee
        daiBalance = await daiToken.balanceOf(dsaWallet0.address);
        console.log("After spell: ");
        console.log("\tBalance after: ", daiBalance.toString(), tokens.dai.symbol);
        expect(daiBalance).to.be.lt(ethers.utils.parseEther("100"));

        // Expect PT Dai Ticket to equal 0
        const balanceAfter = await cToken.balanceOf(dsaWallet0.address)
        console.log("\tBalance after: ", balanceAfter.toString(), tokenName)
        expect(balanceAfter.toNumber()).to.be.eq(0);

        // Expect Pool Token Balance to greater than 0
        const poolBalanceAfter = await poolToken.balanceOf(dsaWallet0.address)
        console.log("\tBalance after: ", poolBalanceAfter.toString(), poolTokenName)
        expect(poolBalanceAfter).to.be.gt(ethers.utils.parseEther("0"));

    });

    it("Should deposit, wait 11 days, and withdraw all PrizePool, get 99 DAI, and claim all POOL using claimAll", async function() {
        const amount = ethers.utils.parseEther("99") // 99 DAI
        const depositSpells = [
            {
                connector: ptConnectorName,
                method: "depositTo",
                args: [prizePool, dsaWallet0.address, amount, controlledToken, constants.address_zero, 0, 0]
            }
        ]

        const withdrawSpells = [
            {
                connector: ptConnectorName,
                method: "withdrawInstantlyFrom",
                args: [prizePool, dsaWallet0.address, amount, controlledToken, amount, 0, 0]
            },
            {
                connector: ptConnectorName,
                method: "claimAll",
                args: [tokenFaucetProxyFactory, dsaWallet0.address, [daiPoolFaucet]]
            }
        ]

        // Before spell
        // DAI balance is 0
        let daiToken = await ethers.getContractAt(abis.basic.erc20, token)
        let daiBalance = await daiToken.balanceOf(dsaWallet0.address);
        console.log("Before Spell:")
        console.log("\tBalance before: ", daiBalance.toString(), tokens.dai.symbol);

        // PT Dai Ticket is 0
        let cToken = await ethers.getContractAt(abis.basic.erc20, controlledToken)
        const balance = await cToken.balanceOf(dsaWallet0.address)
        const tokenName = await cToken.name()
        console.log("\tBalance before: ", balance.toString(), tokenName)

        // PoolToken is 0
        let poolToken = await ethers.getContractAt(abis.basic.erc20, poolTokenAddress)
        const poolBalance = await poolToken.balanceOf(dsaWallet0.address)
        const poolTokenName = await poolToken.name()
        console.log("\tBalance before: ", poolBalance.toString(), poolTokenName)

        // Run spell transaction
        const tx = await dsaWallet0.connect(wallet0).cast(...encodeSpells(depositSpells), wallet1.address)
        const receipt = await tx.wait()

        // Increase time by 11 days so we get back all DAI without early withdrawal fee
        await ethers.provider.send("evm_increaseTime", [11*24*60*60]);

        // Run spell transaction
        const tx2 = await dsaWallet0.connect(wallet0).cast(...encodeSpells(withdrawSpells), wallet1.address)
        const receipt2 = await tx2.wait()

        // After spell
        // Expect DAI balance to be 99
        daiBalance = await daiToken.balanceOf(dsaWallet0.address);
        console.log("After spell: ");
        console.log("\tBalance after: ", daiBalance.toString(), tokens.dai.symbol);
        expect(daiBalance).to.be.eq(ethers.utils.parseEther("99"));

        // Expect PT Dai Ticket to equal 0
        const balanceAfter = await cToken.balanceOf(dsaWallet0.address)
        console.log("\tBalance after: ", balanceAfter.toString(), tokenName)
        expect(balanceAfter.toNumber()).to.be.eq(0);

        // Expect Pool Token Balance to be greateir than 0
        const poolBalanceAfter = await poolToken.balanceOf(dsaWallet0.address)
        console.log("\tBalance after: ", poolBalanceAfter.toString(), poolTokenName)
        expect(poolBalanceAfter).to.be.gt(ethers.utils.parseEther("0"));
    });
  })
})