FuncSelectors.h.sol

// SPDX-License-Identifier: UNLICENSED
pragma solidity 0.8.18;

import {IDiamondCut} from "diamond/contracts/interfaces/IDiamondCut.sol";
import {IDiamondLoupe} from "diamond/contracts/interfaces/IDiamondLoupe.sol";
import {IERC721} from "@openzeppelin/contracts/token/ERC721/IERC721.sol";
import {IERC165} from "diamond/contracts/interfaces/IERC165.sol";
import {OwnershipFacet} from "diamond/contracts/facets/OwnershipFacet.sol";
import {IERC721Receiver} from "@openzeppelin/contracts/token/ERC721/IERC721Receiver.sol";

import {IAuctionFacet} from "../interface/IAuctionFacet.sol";
import {IBorrowFacet} from "../interface/IBorrowFacet.sol";
import {IClaimFacet} from "../interface/IClaimFacet.sol";
import {IProtocolFacet} from "../interface/IProtocolFacet.sol";
import {IRepayFacet} from "../interface/IRepayFacet.sol";
import {IAdminFacet} from "../interface/IAdminFacet.sol";
import {ISignature} from "../interface/ISignature.sol";

import {SupplyPositionFacet} from "../SupplyPositionFacet.sol";
import {DiamondERC721} from "../SupplyPositionLogic/DiamondERC721.sol";

/* solhint-disable func-visibility */

/// @notice This file is for function selectors getters of facets
/// @dev create a new function for each new facet and update them
///     according to their interface

function loupeFS() pure returns (bytes4[] memory) {
    bytes4[] memory functionSelectors = new bytes4[](5);

    functionSelectors[0] = IDiamondLoupe.facets.selector;
    functionSelectors[1] = IDiamondLoupe.facetFunctionSelectors.selector;
    functionSelectors[2] = IDiamondLoupe.facetAddresses.selector;
    functionSelectors[3] = IDiamondLoupe.facetAddress.selector;
    functionSelectors[4] = IERC165.supportsInterface.selector;

    return functionSelectors;
}

function ownershipFS() pure returns (bytes4[] memory) {
    bytes4[] memory functionSelectors = new bytes4[](2);

    functionSelectors[0] = OwnershipFacet.transferOwnership.selector;
    functionSelectors[1] = OwnershipFacet.owner.selector;

    return functionSelectors;
}

function cutFS() pure returns (bytes4[] memory) {
    bytes4[] memory functionSelectors = new bytes4[](1);

    functionSelectors[0] = IDiamondCut.diamondCut.selector;

    return functionSelectors;
}

function borrowFS() pure returns (bytes4[] memory) {
    bytes4[] memory functionSelectors = new bytes4[](3);

    functionSelectors[0] = IERC721Receiver.onERC721Received.selector;
    functionSelectors[1] = ISignature.offerDigest.selector;
    functionSelectors[2] = IBorrowFacet.borrow.selector;

    return functionSelectors;
}

function supplyPositionFS() pure returns (bytes4[] memory) {
    bytes4[] memory functionSelectors = new bytes4[](13);

    functionSelectors[0] = IERC721.balanceOf.selector;
    functionSelectors[1] = IERC721.ownerOf.selector;
    functionSelectors[2] = DiamondERC721.name.selector;
    functionSelectors[3] = DiamondERC721.symbol.selector;
    functionSelectors[4] = IERC721.approve.selector;
    functionSelectors[5] = IERC721.getApproved.selector;
    functionSelectors[6] = IERC721.setApprovalForAll.selector;
    functionSelectors[7] = IERC721.isApprovedForAll.selector;
    functionSelectors[8] = IERC721.transferFrom.selector;
    functionSelectors[9] = getSelector("safeTransferFrom(address,address,uint256)");
    functionSelectors[10] = getSelector("safeTransferFrom(address,address,uint256,bytes)");
    functionSelectors[11] = SupplyPositionFacet.position.selector;
    functionSelectors[12] = SupplyPositionFacet.totalSupply.selector;

    return functionSelectors;
}

/// @notice protocol facet function selectors
function protoFS() pure returns (bytes4[] memory) {
    bytes4[] memory functionSelectors = new bytes4[](4);

    functionSelectors[0] = IProtocolFacet.getRateOfTranche.selector;
    functionSelectors[1] = IProtocolFacet.getParameters.selector;
    functionSelectors[2] = IProtocolFacet.getLoan.selector;
    functionSelectors[3] = IProtocolFacet.getMinOfferCostAndBorrowableAmount.selector;

    return functionSelectors;
}

function repayFS() pure returns (bytes4[] memory) {
    bytes4[] memory functionSelectors = new bytes4[](1);

    functionSelectors[0] = IRepayFacet.repay.selector;

    return functionSelectors;
}

function auctionFS() pure returns (bytes4[] memory) {
    bytes4[] memory functionSelectors = new bytes4[](2);

    functionSelectors[0] = IAuctionFacet.buy.selector;
    functionSelectors[1] = IAuctionFacet.price.selector;

    return functionSelectors;
}

function claimFS() pure returns (bytes4[] memory) {
    bytes4[] memory functionSelectors = new bytes4[](2);

    functionSelectors[0] = IClaimFacet.claim.selector;
    functionSelectors[1] = IClaimFacet.claimAsBorrower.selector;

    return functionSelectors;
}

function adminFS() pure returns (bytes4[] memory) {
    bytes4[] memory functionSelectors = new bytes4[](5);

    functionSelectors[0] = IAdminFacet.setAuctionDuration.selector;
    functionSelectors[1] = IAdminFacet.setAuctionPriceFactor.selector;
    functionSelectors[2] = IAdminFacet.createTranche.selector;
    functionSelectors[3] = IAdminFacet.setMinOfferCost.selector;
    functionSelectors[4] = IAdminFacet.setBorrowAmountPerOfferLowerBound.selector;

    return functionSelectors;
}

function getSelector(string memory _func) pure returns (bytes4) {
    return bytes4(keccak256(bytes(_func)));
}