// SPDX-License-Identifier: MIT
pragma solidity ^0.8.29;

// interfaces
import {IDiamondInitHelper} from "./IDiamondInitHelper.sol";

// libraries
import {DeployDiamondCut} from "@towns-protocol/diamond/scripts/deployments/facets/DeployDiamondCut.sol";
import {DeployDiamondLoupe} from "@towns-protocol/diamond/scripts/deployments/facets/DeployDiamondLoupe.sol";
import {DeployIntrospection} from "@towns-protocol/diamond/scripts/deployments/facets/DeployIntrospection.sol";
import {DeployOwnable} from "@towns-protocol/diamond/scripts/deployments/facets/DeployOwnable.sol";
import {DeployMetadata} from "../facets/DeployMetadata.s.sol";
import {DeployL2RegistrarFacet} from "../facets/DeployL2RegistrarFacet.s.sol";
import {LibString} from "solady/utils/LibString.sol";

// contracts
import {Diamond} from "@towns-protocol/diamond/src/Diamond.sol";
import {MultiInit} from "@towns-protocol/diamond/src/initializers/MultiInit.sol";
import {DiamondHelper} from "@towns-protocol/diamond/scripts/common/helpers/DiamondHelper.s.sol";

// deployers
import {DeployFacet} from "../../common/DeployFacet.s.sol";
import {Deployer} from "../../common/Deployer.s.sol";

contract DeployL2Registrar is IDiamondInitHelper, DiamondHelper, Deployer {
    using LibString for string;

    bytes32 internal constant METADATA_NAME = bytes32("L2Registrar");

    DeployFacet private facetHelper = new DeployFacet();

    address private REGISTRY;
    address private SPACE_FACTORY;
    address private USDC;

    function setRegistry(address registry) external {
        REGISTRY = registry;
    }

    function setSpaceFactory(address spaceFactory) external {
        SPACE_FACTORY = spaceFactory;
    }

    function setUSDC(address usdc) external {
        USDC = usdc;
    }

    function diamondInitHelper(
        address deployer,
        string[] memory facetNames
    ) external override returns (FacetCut[] memory) {
        for (uint256 i; i < facetNames.length; ++i) {
            facetHelper.add(facetNames[i]);
        }

        facetHelper.deployBatch(deployer);

        for (uint256 i; i < facetNames.length; ++i) {
            string memory facetName = facetNames[i];
            address facet = facetHelper.getDeployedAddress(facetName);

            if (facetName.eq("L2RegistrarFacet")) {
                addCut(makeCut(facet, FacetCutAction.Add, DeployL2RegistrarFacet.selectors()));
            }
        }

        return baseFacets();
    }

    function diamondInitParams(address deployer) public returns (Diamond.InitParams memory) {
        // Queue up feature facets for batch deployment
        facetHelper.add("MultiInit");
        facetHelper.add("MetadataFacet");
        facetHelper.add("L2RegistrarFacet");

        // Deploy the batch of facets
        facetHelper.deployBatch(deployer);

        // Add MetadataFacet
        address facet = facetHelper.getDeployedAddress("MetadataFacet");
        addFacet(
            makeCut(facet, FacetCutAction.Add, DeployMetadata.selectors()),
            facet,
            DeployMetadata.makeInitData(METADATA_NAME, "")
        );

        // Add L2RegistrarFacet
        facet = facetHelper.getDeployedAddress("L2RegistrarFacet");
        addFacet(
            makeCut(facet, FacetCutAction.Add, DeployL2RegistrarFacet.selectors()),
            facet,
            DeployL2RegistrarFacet.makeInitData(getRegistry(), getSpaceFactory(), getUSDC())
        );

        address multiInit = facetHelper.getDeployedAddress("MultiInit");

        return
            Diamond.InitParams({
                baseFacets: baseFacets(),
                init: multiInit,
                initData: abi.encodeCall(MultiInit.multiInit, (_initAddresses, _initDatas))
            });
    }

    function versionName() public pure override returns (string memory) {
        return "l2Registrar";
    }

    function getRegistry() internal returns (address) {
        if (REGISTRY == address(0)) return getDeployment("l2Resolver");
        else return REGISTRY;
    }

    function getSpaceFactory() internal returns (address) {
        if (SPACE_FACTORY == address(0)) return getDeployment("spaceFactory");
        else return SPACE_FACTORY;
    }

    function getUSDC() internal returns (address) {
        if (USDC == address(0)) return getDeployment("usdc");
        else return USDC;
    }

    function addImmutableCuts(address deployer) internal {
        // Queue up all core facets for batch deployment
        facetHelper.add("DiamondCutFacet");
        facetHelper.add("DiamondLoupeFacet");
        facetHelper.add("IntrospectionFacet");
        facetHelper.add("OwnableFacet");

        // Get predicted addresses and add facets
        address facet = facetHelper.predictAddress("DiamondCutFacet");
        addFacet(
            makeCut(facet, FacetCutAction.Add, DeployDiamondCut.selectors()),
            facet,
            DeployDiamondCut.makeInitData()
        );

        facet = facetHelper.predictAddress("DiamondLoupeFacet");
        addFacet(
            makeCut(facet, FacetCutAction.Add, DeployDiamondLoupe.selectors()),
            facet,
            DeployDiamondLoupe.makeInitData()
        );

        facet = facetHelper.predictAddress("IntrospectionFacet");
        addFacet(
            makeCut(facet, FacetCutAction.Add, DeployIntrospection.selectors()),
            facet,
            DeployIntrospection.makeInitData()
        );

        facet = facetHelper.predictAddress("OwnableFacet");
        addFacet(
            makeCut(facet, FacetCutAction.Add, DeployOwnable.selectors()),
            facet,
            DeployOwnable.makeInitData(deployer)
        );
    }

    function __deploy(address deployer) internal override returns (address) {
        addImmutableCuts(deployer);

        Diamond.InitParams memory initDiamondCut = diamondInitParams(deployer);

        vm.broadcast(deployer);
        Diamond diamond = new Diamond(initDiamondCut);

        return address(diamond);
    }
}