self/contracts/test/utils/deployment.ts

import { ethers } from "hardhat";
import { Signer } from "ethers";
import { getSMTs } from "./generateProof";
import { PassportData } from "../../../common/src/utils/types";
import { genMockPassportData } from "../../../common/src/utils/passports/genMockPassportData";
import { RegisterVerifierId, DscVerifierId } from "../../../common/src/constants/constants";
import { getCscaTreeRoot } from "../../../common/src/utils/trees";
import serialized_csca_tree from "../../../common/pubkeys/serialized_csca_tree.json";
import {
    DeployedActors,
    VcAndDiscloseVerifier,
    RegisterVerifier,
    DscVerifier,
    IdentityVerificationHub,
    IdentityVerificationHubImplV1,
    IdentityRegistry,
    IdentityRegistryImplV1,
} from "./types";

// Verifier artifacts
import VcAndDiscloseVerifierArtifactLocal from "../../artifacts/contracts/verifiers/local/disclose/Verifier_vc_and_disclose.sol/Verifier_vc_and_disclose.json";
// import VcAndDiscloseVerifierArtifactProd from "../../artifacts/contracts/verifiers/disclose/Verifier_vc_and_disclose.sol/Verifier_vc_and_disclose.json";
import RegisterVerifierArtifactLocal from "../../artifacts/contracts/verifiers/local/register/Verifier_register_sha256_sha256_sha256_rsa_65537_4096.sol/Verifier_register_sha256_sha256_sha256_rsa_65537_4096.json";
// import RegisterVerifierArtifactProd from "../../artifacts/contracts/verifiers/register/Verifier_register_rsa_65537_sha256.sol/Verifier_register_rsa_65537_sha256.json";
import DscVerifierArtifactLocal from "../../artifacts/contracts/verifiers/local/dsc/Verifier_dsc_sha256_rsa_65537_4096.sol/Verifier_dsc_sha256_rsa_65537_4096.json";
// import DscVerifierArtifactProd from "../../artifacts/contracts/verifiers/dsc/Verifier_dsc_sha256_rsa_65537_4096.sol/Verifier_dsc_sha256_rsa_65537_4096.json";

export async function deploySystemFixtures(): Promise<DeployedActors> {
    let identityVerificationHubProxy: IdentityVerificationHub;
    let identityVerificationHubImpl: IdentityVerificationHubImplV1;
    let identityRegistryProxy: IdentityRegistry;
    let identityRegistryImpl: IdentityRegistryImplV1;
    let vcAndDiscloseVerifier: VcAndDiscloseVerifier;
    let registerVerifier: RegisterVerifier;
    let dscVerifier: DscVerifier;
    let owner: Signer;
    let user1: Signer;
    let user2: Signer;
    let mockPassport: PassportData;

    [owner, user1, user2] = await ethers.getSigners();

    const newBalance = "0x" + ethers.parseEther("10000").toString(16);

    await ethers.provider.send("hardhat_setBalance", [await owner.getAddress(), newBalance]);
    await ethers.provider.send("hardhat_setBalance", [await user1.getAddress(), newBalance]);
    await ethers.provider.send("hardhat_setBalance", [await user2.getAddress(), newBalance]);

    mockPassport = genMockPassportData(
        "sha256",
        "sha256",
        "rsa_sha256_65537_2048",
        "FRA",
        "940131",
        "401031"
    );

    // Deploy verifiers
    const vcAndDiscloseVerifierArtifact = process.env.TEST_ENV === "local"
        ? VcAndDiscloseVerifierArtifactLocal
        : VcAndDiscloseVerifierArtifactProd;
    const vcAndDiscloseVerifierFactory = await ethers.getContractFactory(
        vcAndDiscloseVerifierArtifact.abi,
        vcAndDiscloseVerifierArtifact.bytecode,
        owner
    );
    vcAndDiscloseVerifier = await vcAndDiscloseVerifierFactory.deploy();
    await vcAndDiscloseVerifier.waitForDeployment();

    // Deploy register verifier
    const registerVerifierArtifact = process.env.TEST_ENV === "local"
        ? RegisterVerifierArtifactLocal
        : RegisterVerifierArtifactProd;
    const registerVerifierFactory = await ethers.getContractFactory(
        registerVerifierArtifact.abi,
        registerVerifierArtifact.bytecode,
        owner
    );
    registerVerifier = await registerVerifierFactory.deploy();
    await registerVerifier.waitForDeployment();

    // Deploy dsc verifier
    const dscVerifierArtifact = process.env.TEST_ENV === "local"
        ? DscVerifierArtifactLocal
        : DscVerifierArtifactProd;
    const dscVerifierFactory = await ethers.getContractFactory(
        dscVerifierArtifact.abi,
        dscVerifierArtifact.bytecode,
        owner
    );
    dscVerifier = await dscVerifierFactory.deploy();
    await dscVerifier.waitForDeployment();

    // Deploy PoseidonT3
    const PoseidonT3Factory = await ethers.getContractFactory("PoseidonT3", owner);
    const poseidonT3 = await PoseidonT3Factory.deploy();
    await poseidonT3.waitForDeployment();

    // Deploy IdentityRegistryImplV1
    const IdentityRegistryImplFactory = await ethers.getContractFactory(
        "IdentityRegistryImplV1",
        {
            libraries: {
                PoseidonT3: poseidonT3.target
            }
        },
        owner
    );
    identityRegistryImpl = await IdentityRegistryImplFactory.deploy();
    await identityRegistryImpl.waitForDeployment();

    // Deploy IdentityVerificationHubImplV1
    const IdentityVerificationHubImplFactory = await ethers.getContractFactory("IdentityVerificationHubImplV1", owner);
    identityVerificationHubImpl = await IdentityVerificationHubImplFactory.deploy();
    await identityVerificationHubImpl.waitForDeployment();

    // Deploy registry with temporary hub address
    const temporaryHubAddress = "0x0000000000000000000000000000000000000000";
    const registryInitData = identityRegistryImpl.interface.encodeFunctionData("initialize", [
        temporaryHubAddress
    ]);
    const registryProxyFactory = await ethers.getContractFactory("IdentityRegistry", owner);
    identityRegistryProxy = await registryProxyFactory.deploy(identityRegistryImpl.target, registryInitData);
    await identityRegistryProxy.waitForDeployment();

    // Deploy hub with deployed registry and verifiers
    const initializeData = identityVerificationHubImpl.interface.encodeFunctionData("initialize", [
        identityRegistryProxy.target,
        vcAndDiscloseVerifier.target,
        [RegisterVerifierId.register_sha256_sha256_sha256_rsa_65537_4096],
        [registerVerifier.target],
        [DscVerifierId.dsc_sha256_rsa_65537_4096],
        [dscVerifier.target]
    ]);
    const hubFactory = await ethers.getContractFactory("IdentityVerificationHub", owner);
    identityVerificationHubProxy = await hubFactory.deploy(identityVerificationHubImpl.target, initializeData);
    await identityVerificationHubProxy.waitForDeployment();

    // Get contracts with implementation ABI and update hub address
    const registryContract = await ethers.getContractAt(
        "IdentityRegistryImplV1",
        identityRegistryProxy.target
    ) as IdentityRegistryImplV1;
    const updateHubTx = await registryContract.updateHub(identityVerificationHubProxy.target);
    await updateHubTx.wait();

    const hubContract = await ethers.getContractAt(
        "IdentityVerificationHubImplV1",
        identityVerificationHubProxy.target
    ) as IdentityVerificationHubImplV1;

    // Initialize roots
    const csca_root = getCscaTreeRoot(serialized_csca_tree);
    await registryContract.updateCscaRoot(csca_root, { from: owner });

    const {
        passportNo_smt,
        nameAndDob_smt,
        nameAndYob_smt
    } = getSMTs();

    await registryContract.updatePassportNoOfacRoot(passportNo_smt.root, { from: owner });
    await registryContract.updateNameAndDobOfacRoot(nameAndDob_smt.root, { from: owner });
    await registryContract.updateNameAndYobOfacRoot(nameAndYob_smt.root, { from: owner });

    return {
        hub: hubContract,
        hubImpl: identityVerificationHubImpl,
        registry: registryContract,
        registryImpl: identityRegistryImpl,
        vcAndDisclose: vcAndDiscloseVerifier,
        register: registerVerifier,
        dsc: dscVerifier,
        owner: owner,
        user1: user1,
        user2: user2,
        mockPassport: mockPassport
    };
}