import { expect } from "chai";
import hre, { ethers, deployments } from "hardhat";
import { AddressZero } from "@ethersproject/constants";
import { getSafeWithSingleton, getSafeSingletonAt, getMock } from "../utils/setup";
import deploymentData from "../json/safeDeployment.json";
import {
    buildSafeTransaction,
    executeContractCallWithSigners,
    executeTx,
    executeTxWithSigners,
    safeApproveHash,
} from "../../src/utils/execution";

const FALLBACK_HANDLER_STORAGE_SLOT = "0x6c9a6c4a39284e37ed1cf53d337577d14212a4870fb976a4366c693b939918d5";

const GUARD_STORAGE_SLOT = "0x4a204f620c8c5ccdca3fd54d003badd85ba500436a431f0cbda4f558c93c34c8";

describe("SafeToL2Migration library", () => {
    const migratedInterface = new ethers.utils.Interface(["function masterCopy() view returns(address)"]);

    const setupTests = deployments.createFixture(async ({ deployments }) => {
        await deployments.fixture();

        const LATEST_SAFE_SINGLETON_ADDRESS = (await deployments.get("Safe"))?.address;
        const LATEST_SAFE_SINGLETON_L2_ADDRESS = (await deployments.get("SafeL2"))?.address;
        const LASTEST_COMPATIBILITY_FALLBACK_HANDLER = (await deployments.get("CompatibilityFallbackHandler"))?.address;
        if (!LATEST_SAFE_SINGLETON_ADDRESS || !LATEST_SAFE_SINGLETON_L2_ADDRESS || !LASTEST_COMPATIBILITY_FALLBACK_HANDLER) {
            throw new Error("Could not get Safe or SafeL2 addresses");
        }

        const signers = await ethers.getSigners();
        const [user1] = signers;
        const singleton111Address = (await (await user1.sendTransaction({ data: deploymentData.safe111 })).wait())?.contractAddress;
        const singleton130Address = (await (await user1.sendTransaction({ data: deploymentData.safe130 })).wait())?.contractAddress;
        const singleton130L2Address = (await (await user1.sendTransaction({ data: deploymentData.safe130l2 })).wait())?.contractAddress;

        if (!singleton111Address || !singleton130Address || !singleton130L2Address) {
            throw new Error("Could not deploy Safe111, Safe130 or Safe130L2");
        }
        const singleton111 = await getSafeSingletonAt(singleton111Address);
        const singleton130 = await getSafeSingletonAt(singleton130Address);
        const singletonLatest = await getSafeSingletonAt(LATEST_SAFE_SINGLETON_ADDRESS);

        const guardContract = await hre.ethers.getContractAt("Guard", AddressZero);
        const guardEip165Calldata = guardContract.interface.encodeFunctionData("supportsInterface", ["0x945b8148"]);
        const validGuardMock = await getMock();
        await validGuardMock.givenCalldataReturnBool(guardEip165Calldata, true);

        const invalidGuardMock = await getMock();
        await invalidGuardMock.givenCalldataReturnBool(guardEip165Calldata, false);

        const safeWith1967Proxy = await getSafeSingletonAt(
            await hre.ethers
                .getContractFactory("UpgradeableProxy")
                .then((factory) =>
                    factory.deploy(
                        singleton130Address,
                        singleton130.interface.encodeFunctionData("setup", [
                            [user1.address],
                            1,
                            AddressZero,
                            "0x",
                            AddressZero,
                            AddressZero,
                            0,
                            AddressZero,
                        ]),
                    ),
                )
                .then((proxy) => proxy.address),
        );
        const safeToL2MigrationContract = await hre.ethers.getContractFactory("SafeToL2Migration");
        const migration = await safeToL2MigrationContract.deploy();
        return {
            safe111: await getSafeWithSingleton(singleton111, [user1.address]),
            safe130: await getSafeWithSingleton(singleton130, [user1.address]),
            safeLatest: await getSafeWithSingleton(singletonLatest, [user1.address]),
            safeWith1967Proxy,
            migration,
            signers,
            validGuardMock,
            invalidGuardMock,
            singleton130Address,
            singleton130L2Address,
            LATEST_SAFE_SINGLETON_ADDRESS,
            LATEST_SAFE_SINGLETON_L2_ADDRESS,
            LASTEST_COMPATIBILITY_FALLBACK_HANDLER,
        };
    });

    describe("migrateToL2", () => {
        it("reverts if the singleton is not set", async () => {
            const {
                migration,
                safeWith1967Proxy,
                signers: [user1],
                singleton130L2Address,
            } = await setupTests();

            await expect(
                executeContractCallWithSigners(safeWith1967Proxy, migration, "migrateToL2", [singleton130L2Address], [user1], true),
            ).to.be.revertedWith("GS013");
        });

        it("reverts if new singleton is the same as the old one", async () => {
            const {
                safe130,
                migration,
                signers: [user1],
                singleton130Address,
            } = await setupTests();
            console.log("fixtures done");
            await expect(
                executeContractCallWithSigners(safe130, migration, "migrateToL2", [singleton130Address], [user1], true),
            ).to.be.revertedWith("GS013");
        });

        it("reverts if the new singleton is not the same version as the old one", async () => {
            const {
                safe130,
                migration,
                LATEST_SAFE_SINGLETON_L2_ADDRESS,
                signers: [user1],
            } = await setupTests();
            await expect(
                executeContractCallWithSigners(safe130, migration, "migrateToL2", [LATEST_SAFE_SINGLETON_L2_ADDRESS], [user1], true),
            ).to.be.revertedWith("GS013");
        });

        it("reverts if nonce > 0", async () => {
            const {
                safe130,
                migration,
                signers: [user1],
                singleton130Address,
                singleton130L2Address,
            } = await setupTests();
            const safeAddress = safe130.address;
            expect(await safe130.nonce()).to.be.eq(0);

            // Increase nonce by sending eth
            await user1.sendTransaction({ to: safeAddress, value: ethers.utils.parseEther("1") });
            const nonce = 0;
            const safeTx = buildSafeTransaction({ to: user1.address, value: ethers.utils.parseEther("1"), nonce });
            await executeTxWithSigners(safe130, safeTx, [user1]);

            expect(await safe130.nonce()).to.be.eq(1);
            await expect(
                executeContractCallWithSigners(safe130, migration, "migrateToL2", [singleton130L2Address], [user1], true),
            ).to.be.revertedWith("GS013");

            const singletonResp = await user1.call({ to: safeAddress, data: migratedInterface.encodeFunctionData("masterCopy") });
            expect(migratedInterface.decodeFunctionResult("masterCopy", singletonResp)[0]).to.eq(singleton130Address);
        });

        it("migrates from singleton 1.3.0 to 1.3.0L2", async () => {
            const {
                safe130,
                migration,
                signers: [user1],
                singleton130L2Address,
            } = await setupTests();
            const safeAddress = safe130.address;
            // The emit matcher checks the address, which is the Safe as delegatecall is used
            const migrationSafe = migration.attach(safeAddress);
            const migrationAddress = migration.address;

            const functionName = "migrateToL2";
            const expectedData = migration.interface.encodeFunctionData(functionName, [singleton130L2Address]);
            const safeThreshold = await safe130.getThreshold();
            const additionalInfo = hre.ethers.utils.defaultAbiCoder.encode(
                ["uint256", "address", "uint256"],
                [0, user1.address, safeThreshold],
            );
            await expect(executeContractCallWithSigners(safe130, migration, functionName, [singleton130L2Address], [user1], true))
                .to.emit(migrationSafe, "ChangedMasterCopy")
                .withArgs(singleton130L2Address)
                .to.emit(migrationSafe, "SafeMultiSigTransaction")
                .withArgs(
                    migrationAddress,
                    0,
                    expectedData,
                    1,
                    0,
                    0,
                    0,
                    AddressZero,
                    AddressZero,
                    "0x", // We cannot detect signatures
                    additionalInfo,
                );

            const singletonResp = await user1.call({ to: safeAddress, data: migratedInterface.encodeFunctionData("masterCopy") });
            expect(migratedInterface.decodeFunctionResult("masterCopy", singletonResp)[0]).to.eq(singleton130L2Address);
            expect(await safe130.nonce()).to.be.eq(1);
        });

        it("migrates from singleton 1.4.1 to 1.4.1L2", async () => {
            const {
                safeLatest,
                migration,
                signers: [user1],
                LATEST_SAFE_SINGLETON_L2_ADDRESS,
            } = await setupTests();
            const safeAddress = safeLatest.address;
            // The emit matcher checks the address, which is the Safe as delegatecall is used
            const migrationSafe = migration.attach(safeAddress);
            const migrationAddress = migration.address;

            const functionName = "migrateToL2";
            const expectedData = migration.interface.encodeFunctionData(functionName, [LATEST_SAFE_SINGLETON_L2_ADDRESS]);
            const safeThreshold = await safeLatest.getThreshold();
            const additionalInfo = hre.ethers.utils.defaultAbiCoder.encode(
                ["uint256", "address", "uint256"],
                [0, user1.address, safeThreshold],
            );
            await expect(
                executeContractCallWithSigners(safeLatest, migration, functionName, [LATEST_SAFE_SINGLETON_L2_ADDRESS], [user1], true),
            )
                .to.emit(migrationSafe, "ChangedMasterCopy")
                .withArgs(LATEST_SAFE_SINGLETON_L2_ADDRESS)
                .to.emit(migrationSafe, "SafeMultiSigTransaction")
                .withArgs(
                    migrationAddress,
                    0,
                    expectedData,
                    1,
                    0,
                    0,
                    0,
                    AddressZero,
                    AddressZero,
                    "0x", // We cannot detect signatures
                    additionalInfo,
                );

            const singletonResp = await user1.call({ to: safeAddress, data: migratedInterface.encodeFunctionData("masterCopy") });
            expect(migratedInterface.decodeFunctionResult("masterCopy", singletonResp)[0]).to.eq(LATEST_SAFE_SINGLETON_L2_ADDRESS);
            expect(await safeLatest.nonce()).to.be.eq(1);
        });

        it("migrates from singleton 1.1.1 to 1.4.1L2", async () => {
            const {
                safe111,
                migration,
                signers: [user1],
                LATEST_SAFE_SINGLETON_L2_ADDRESS,
                LASTEST_COMPATIBILITY_FALLBACK_HANDLER,
            } = await setupTests();
            const safeAddress = safe111.address;
            expect(await safe111.VERSION()).eq("1.1.1");
            expect("0x" + (await hre.ethers.provider.getStorageAt(safeAddress, FALLBACK_HANDLER_STORAGE_SLOT)).slice(26)).to.be.eq(
                AddressZero,
            );

            // The emit matcher checks the address, which is the Safe as delegatecall is used
            const migrationSafe = migration.attach(safeAddress);
            const migrationAddress = migration.address;

            const functionName = "migrateFromV111";
            const data = migration.interface.encodeFunctionData(functionName, [
                LATEST_SAFE_SINGLETON_L2_ADDRESS,
                LASTEST_COMPATIBILITY_FALLBACK_HANDLER,
            ]);
            const nonce = await safe111.nonce();
            expect(nonce).to.be.eq(0);
            const safeThreshold = await safe111.getThreshold();
            const additionalInfo = hre.ethers.utils.defaultAbiCoder.encode(
                ["uint256", "address", "uint256"],
                [0, user1.address, safeThreshold],
            );

            const tx = buildSafeTransaction({ to: migrationAddress, data, operation: 1, nonce });

            await expect(executeTx(safe111, tx, [await safeApproveHash(user1, safe111, tx, true)]))
                .to.emit(migrationSafe, "ChangedMasterCopy")
                .withArgs(LATEST_SAFE_SINGLETON_L2_ADDRESS)
                .to.emit(migrationSafe, "SafeMultiSigTransaction")
                .withArgs(
                    migrationAddress,
                    0,
                    data,
                    1,
                    0,
                    0,
                    0,
                    AddressZero,
                    AddressZero,
                    "0x", // We cannot detect signatures
                    additionalInfo,
                )
                .to.emit(migrationSafe, "SafeSetup")
                .withArgs(migrationAddress, await safe111.getOwners(), safeThreshold, AddressZero, LASTEST_COMPATIBILITY_FALLBACK_HANDLER);

            expect(await safe111.nonce()).to.be.eq(1);
            expect(await safe111.VERSION()).to.be.eq("1.4.1");
            const singletonResp = await user1.call({ to: safeAddress, data: migratedInterface.encodeFunctionData("masterCopy") });
            expect(migratedInterface.decodeFunctionResult("masterCopy", singletonResp)[0]).to.eq(LATEST_SAFE_SINGLETON_L2_ADDRESS);
            expect("0x" + (await hre.ethers.provider.getStorageAt(safeAddress, FALLBACK_HANDLER_STORAGE_SLOT)).slice(26)).to.be.eq(
                LASTEST_COMPATIBILITY_FALLBACK_HANDLER.toLowerCase(),
            );
        });

        it("doesn't touch important storage slots", async () => {
            const {
                safe130,
                migration,
                signers: [user1],
                singleton130L2Address,
            } = await setupTests();
            const safeAddress = safe130.address;

            const ownerCountBeforeMigration = await hre.ethers.provider.getStorageAt(safeAddress, 3);
            const thresholdBeforeMigration = await hre.ethers.provider.getStorageAt(safeAddress, 4);
            const nonceBeforeMigration = await hre.ethers.provider.getStorageAt(safeAddress, 5);
            const guardBeforeMigration = await hre.ethers.provider.getStorageAt(safeAddress, GUARD_STORAGE_SLOT);
            const fallbackHandlerBeforeMigration = await hre.ethers.provider.getStorageAt(safeAddress, FALLBACK_HANDLER_STORAGE_SLOT);

            await expect(executeContractCallWithSigners(safe130, migration, "migrateToL2", [singleton130L2Address], [user1], true));

            expect(await hre.ethers.provider.getStorageAt(safeAddress, 3)).to.be.eq(ownerCountBeforeMigration);
            expect(await hre.ethers.provider.getStorageAt(safeAddress, 4)).to.be.eq(thresholdBeforeMigration);
            expect(await hre.ethers.provider.getStorageAt(safeAddress, 5)).to.be.eq(nonceBeforeMigration);
            expect(await hre.ethers.provider.getStorageAt(safeAddress, GUARD_STORAGE_SLOT)).to.be.eq(guardBeforeMigration);
            expect(await hre.ethers.provider.getStorageAt(safeAddress, FALLBACK_HANDLER_STORAGE_SLOT)).to.be.eq(
                fallbackHandlerBeforeMigration,
            );
        });
    });
});