import { expect } from "chai";
import hre from "hardhat";
const { web3, deployments, waffle, ethers } = hre;
const { provider, deployContract } = waffle

import { deployAndEnableConnector } from "../../../scripts/tests/deployAndEnableConnector"
import { buildDSAv2 } from "../../../scripts/tests/buildDSAv2"
import { encodeSpells } from "../../../scripts/tests/encodeSpells"
import { getMasterSigner } from "../../../scripts/tests/getMasterSigner"
import { addresses } from "../../../scripts/tests/mainnet/addresses";
import { abis } from "../../../scripts/constant/abis";
import { tokens } from "../../../scripts/tests/mainnet/tokens";
import { ConnectV2BMakerDAO__factory } from "../../../typechain";
import type { Signer, Contract } from "ethers";

describe("B.Maker", function () {
    const connectorName = "B.MAKER-TEST-A"

    let dsaWallet0: any;
    let dsaWallet1: any;
    let masterSigner: Signer;
    let instaConnectorsV2: Contract;
    let connector: any;
    let manager: any;
    let vat: any;
    let dai: any;

    const wallets = provider.getWallets()
    const [wallet0, wallet1, wallet2, wallet3] = wallets
    before(async () => {
        await hre.network.provider.request({
            method: "hardhat_reset",
            params: [
                {
                    forking: {
                        // @ts-ignore
                        jsonRpcUrl: hre.config.networks.hardhat.forking.url,
                        blockNumber: 12696000,
                    },
                },
            ],
        });
        masterSigner = await getMasterSigner()
        instaConnectorsV2 = await ethers.getContractAt(abis.core.connectorsV2, addresses.core.connectorsV2);
        connector = await deployAndEnableConnector({
            connectorName,
            contractArtifact: ConnectV2BMakerDAO__factory,
            signer: masterSigner,
            connectors: instaConnectorsV2
        })

        manager = await ethers.getContractAt("BManagerLike", "0x3f30c2381CD8B917Dd96EB2f1A4F96D91324BBed")
        vat = await ethers.getContractAt("../artifacts/contracts/mainnet/connectors/b.protocol/makerdao/interface.sol:VatLike", await manager.vat())
        dai = await ethers.getContractAt("../artifacts/contracts/mainnet/common/interfaces.sol:TokenInterface", tokens.dai.address)

        console.log("Connector address", connector.address)
    })

    it("test veryClose.", async function () {
        expect(veryClose(1000001, 1000000)).to.be.true
        expect(veryClose(1000000, 1000001)).to.be.true
        expect(veryClose(1003000, 1000001)).to.be.false
        expect(veryClose(1000001, 1000300)).to.be.false
    });

    it("Should have contracts deployed.", async function () {
        expect(!!instaConnectorsV2.address).to.be.true;
        expect(!!connector.address).to.be.true;
        expect(!!(await masterSigner.getAddress())).to.be.true;
        expect(await connector.name()).to.be.equal("B.MakerDAO-v1.0");
    });

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

            dsaWallet1 = await buildDSAv2(wallet1.address)
            expect(!!dsaWallet1.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"));

            await wallet1.sendTransaction({
                to: dsaWallet1.address,
                value: ethers.utils.parseEther("10")
            });
            expect(await ethers.provider.getBalance(dsaWallet1.address)).to.be.gte(ethers.utils.parseEther("10"));
        });
    });

    describe("Main", function () {
        let vault: any;
        let ilk: any;
        let urn: any;

        it("Should open ETH-A vault Maker", async function () {
            vault = Number(await manager.cdpi()) + 1
            const spells = [
                {
                    connector: connectorName,
                    method: "open",
                    args: ["ETH-A"]
                }
            ]

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

            expect(await manager.owns(vault)).to.be.equal(dsaWallet0.address)

            ilk = await manager.ilks(vault)
            expect(ilk).to.be.equal("0x4554482d41000000000000000000000000000000000000000000000000000000")

            urn = await manager.urns(vault)
        });

        it("Should deposit", async function () {
            const amount = ethers.utils.parseEther("7") // 7 ETH
            const setId = "83478237"

            const spells = [
                {
                    connector: connectorName,
                    method: "deposit",
                    args: [vault, amount, 0, setId]
                }
            ]

            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.gte(ethers.utils.parseEther("3"))

            const urnData = await vat.urns(ilk, urn)
            expect(urnData[0]).to.be.equal(amount) // ink
            expect(urnData[1]).to.be.equal("0") // art        

        });

        it("Should withdraw", async function () {
            const amount = ethers.utils.parseEther("1") // 1 ETH
            const setId = "83478237"

            const spells = [
                {
                    connector: connectorName,
                    method: "withdraw",
                    args: [vault, amount, 0, setId]
                }
            ]

            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.gte(ethers.utils.parseEther("4"))

            const urnData = await vat.urns(ilk, urn)
            expect(urnData[0]).to.be.equal(ethers.utils.parseEther("6")) // ink
            expect(urnData[1]).to.be.equal("0") // art        

        });

        it("Should borrow", async function () {
            const amount = ethers.utils.parseEther("6000") // 6000 dai
            const setId = "83478237"

            const spells = [
                {
                    connector: connectorName,
                    method: "borrow",
                    args: [vault, amount, 0, setId]
                }
            ]

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

            const urnData = await vat.urns(ilk, urn)
            expect(urnData[0]).to.be.equal(ethers.utils.parseEther("6")) // ink
            expect(urnData[1]).to.be.equal(await daiToArt(vat, ilk, amount)) // art

            expect(await dai.balanceOf(dsaWallet0.address)).to.be.equal(amount)
        });

        it("Should repay", async function () {
            const amount = ethers.utils.parseEther("500") // 500 dai
            const setId = "83478237"

            const spells = [
                {
                    connector: connectorName,
                    method: "payback",
                    args: [vault, amount, 0, setId]
                }
            ]

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

            const urnData = await vat.urns(ilk, urn)
            expect(urnData[0]).to.be.equal(ethers.utils.parseEther("6")) // ink
            expect(urnData[1]).to.be.equal(await daiToArt(vat, ilk, ethers.utils.parseEther("5500"))) // art        
            expect(await dai.balanceOf(dsaWallet0.address)).to.be.equal(ethers.utils.parseEther("5500"))
        });

        it("Should depositAndBorrow", async function () {
            const borrowAmount = ethers.utils.parseEther("1000") // 1000 dai
            const depositAmount = ethers.utils.parseEther("1") // 1 dai

            const setId = "83478237"

            const spells = [
                {
                    connector: connectorName,
                    method: "depositAndBorrow",
                    args: [vault, depositAmount, borrowAmount, 0, 0, 0, 0]
                }
            ]

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

            const urnData = await vat.urns(ilk, urn)
            expect(urnData[0]).to.be.equal(ethers.utils.parseEther("7")) // ink
            expect(await dai.balanceOf(dsaWallet0.address)).to.be.equal(ethers.utils.parseEther("6500"))
            // calculation is not precise as the jug was dripped
            expect(veryClose(urnData[1], await daiToArt(vat, ilk, ethers.utils.parseEther("6500")))).to.be.true
            expect(await ethers.provider.getBalance(dsaWallet0.address)).to.be.gte(ethers.utils.parseEther("1"))
        });

        it("Should close", async function () {
            // open a new vault
            const newVault = vault + 1
            let spells = [
                {
                    connector: connectorName,
                    method: "open",
                    args: ["ETH-A"]
                }
            ]

            let tx = await dsaWallet1.connect(wallet1).cast(...encodeSpells(spells), wallet1.address)
            let receipt = await tx.wait()

            expect(await manager.owns(newVault)).to.be.equal(dsaWallet1.address)

            ilk = await manager.ilks(newVault)
            expect(ilk).to.be.equal("0x4554482d41000000000000000000000000000000000000000000000000000000")

            urn = await manager.urns(newVault)

            // deposit and borrow
            const borrowAmount = ethers.utils.parseEther("6000") // 6000 dai
            const depositAmount = ethers.utils.parseEther("5") // 5 ETH

            spells = [
                {
                    connector: connectorName,
                    method: "depositAndBorrow",
                    args: [newVault, depositAmount, borrowAmount, 0, 0, 0, 0]
                }
            ]

            tx = await dsaWallet1.connect(wallet1).cast(...encodeSpells(spells), wallet1.address)
            receipt = await tx.wait()

            const setId = 0

            // repay borrow
            spells = [
                {
                    connector: connectorName,
                    method: "payback",
                    args: [newVault, borrowAmount, 0, setId]
                }
            ]

            tx = await dsaWallet1.connect(wallet1).cast(...encodeSpells(spells), wallet1.address)
            receipt = await tx.wait()

            // withdraw deposit
            spells = [
                {
                    connector: connectorName,
                    method: "withdraw",
                    args: [newVault, depositAmount, 0, setId]
                }
            ]

            tx = await dsaWallet1.connect(wallet1).cast(...encodeSpells(spells), wallet1.address)
            receipt = await tx.wait()

            // close
            spells = [
                {
                    connector: connectorName,
                    method: "close",
                    args: [newVault]
                }
            ]

            tx = await dsaWallet1.connect(wallet1).cast(...encodeSpells(spells), wallet1.address)
            receipt = await tx.wait()

            expect(await manager.owns(newVault)).not.to.be.equal(dsaWallet1.address)
        });
    })
})

async function daiToArt(vat: any, ilk: any, dai: any) {
    const ilks = await vat.ilks(ilk)
    const rate = ilks[1] // second parameter
    const _1e27 = ethers.utils.parseEther("1000000000") // 1e9 * 1e18
    const art = dai.mul(_1e27).div(rate)

    return art.add(1)
}

function veryClose(n1: any, n2: any) {
    n1 = web3.utils.toBN(n1)
    n2 = web3.utils.toBN(n2)

    let _10000 = web3.utils.toBN(10000)
    let _9999 = web3.utils.toBN(9999)

    if (n1.mul(_10000).lt(n2.mul(_9999))) return false
    if (n2.mul(_10000).lt(n1.mul(_9999))) return false

    return true
}