contracts.ts

import { Addressable, BaseContract } from "ethers";
import { BasePlugin, RecoveryWithDelayPlugin, RelayPlugin, SafeProtocolRegistryAttestation, WhitelistPlugin, WhitelistHook } from "../../typechain-types";
import { HardhatRuntimeEnvironment } from "hardhat/types";
import { getProtocolManagerAddress, getProtocolRegistryAddress } from "./protocol";

export const getInstance = async <T extends BaseContract>(hre: HardhatRuntimeEnvironment, name: string, address: string | Addressable): Promise<T> => {
    // TODO: this typecasting should be refactored
    return (await hre.ethers.getContractAt(name, address)) as unknown as T;
};

export const getSingleton = async <T extends BaseContract>(hre: HardhatRuntimeEnvironment, name: string): Promise<T> => {
    const deployment = await hre.deployments.get(name);
    return getInstance<T>(hre, name, deployment.address);
};

export const getPlugin = (hre: HardhatRuntimeEnvironment, address: string) => getInstance<BasePlugin>(hre, "BasePlugin", address);
export const getRelayPlugin = (hre: HardhatRuntimeEnvironment) => getSingleton<RelayPlugin>(hre, "RelayPlugin");
export const getRegistry = async (hre: HardhatRuntimeEnvironment) => getInstance<SafeProtocolRegistryAttestation>(hre, "SafeProtocolRegistryAttestation", await getProtocolRegistryAddress(hre));
export const getWhiteListPlugin = async (hre: HardhatRuntimeEnvironment) => getSingleton<WhitelistPlugin>(hre, "WhitelistPlugin");
export const getRecoveryWithDelayPlugin= async(hre: HardhatRuntimeEnvironment) => getSingleton<RecoveryWithDelayPlugin>(hre, "TestPlugin");
export const getWhiteListHook= async(hre: HardhatRuntimeEnvironment) => getSingleton<WhitelistHook>(hre, "WhitelistHook");
// export const getTestPlugin= async(hre: HardhatRuntimeEnvironment) => getSingleton<TestPlugin>(hre, "TestPlugin");