feat: adds getter functions for address constants

codegen/payments.ts

@@ -37,7 +37,7 @@ export function generateFHEPayment(priceData: PriceData): string {
       uint256 private constant MAJOR_VERSION = 0;
       uint256 private constant MINOR_VERSION = 1;
       uint256 private constant PATCH_VERSION = 0;
-      address public constant tfheExecutorAddress = tfheExecutorAdd;
+      address private constant tfheExecutorAddress = tfheExecutorAdd;
 
       uint256 private constant FHE_GAS_BLOCKLIMIT = 10_000_000;
       uint256 private constant MIN_FHE_GASPRICE = 0; // eg: 10_000_000 means a minimum of 0.01 Gwei

lib/ACL.sol

@@ -34,6 +34,7 @@ contract ACL is UUPSUpgradeable, Ownable2StepUpgradeable {
         }
     }
 
+    /// @notice Getter function for the TFHEExecutor contract address
     function getTFHEExecutorAddress() public view virtual returns (address) {
         return tfheExecutorAddress;
     }

lib/FHEPayment.sol

@@ -27,7 +27,7 @@ contract FHEPayment is UUPSUpgradeable, Ownable2StepUpgradeable {
     uint256 private constant MAJOR_VERSION = 0;
     uint256 private constant MINOR_VERSION = 1;
     uint256 private constant PATCH_VERSION = 0;
-    address public constant tfheExecutorAddress = tfheExecutorAdd;
+    address private constant tfheExecutorAddress = tfheExecutorAdd;
 
     uint256 private constant FHE_GAS_BLOCKLIMIT = 10_000_000;
     uint256 private constant MIN_FHE_GASPRICE = 0; // eg: 10_000_000 means a minimum of 0.01 Gwei
@@ -53,6 +53,7 @@ contract FHEPayment is UUPSUpgradeable, Ownable2StepUpgradeable {
         }
     }
 
+    /// @notice Getter function for the TFHEExecutor contract address
     function getTFHEExecutorAddress() public view virtual returns (address) {
         return tfheExecutorAddress;
     }

lib/InputVerifier.coprocessor.sol

@@ -29,7 +29,7 @@ contract InputVerifier is UUPSUpgradeable, Ownable2StepUpgradeable, EIP712Upgrad
     /// @notice Handle version
     uint8 public constant HANDLE_VERSION = 0;
 
-    address constant coprocessorAddress = coprocessorAdd;
+    address private constant coprocessorAddress = coprocessorAdd;
 
     KMSVerifier public constant kmsVerifier = KMSVerifier(kmsVerifierAdd);
     string public constant CIPHERTEXTVERIFICATION_COPRO_TYPE =
@@ -51,6 +51,16 @@ contract InputVerifier is UUPSUpgradeable, Ownable2StepUpgradeable, EIP712Upgrad
         return CIPHERTEXTVERIFICATION_COPRO_TYPE;
     }
 
+    /// @notice Getter function for the Coprocessor account address
+    function getCoprocessorAddress() public view virtual returns (address) {
+        return coprocessorAddress;
+    }
+
+    /// @notice Getter function for the KMSVerifier contract address
+    function getKMSVerifierAddress() public view virtual returns (address) {
+        return address(kmsVerifier);
+    }
+
     /// @custom:oz-upgrades-unsafe-allow constructor
     constructor() {
         _disableInitializers();

lib/InputVerifier.native.sol

@@ -32,6 +32,11 @@ contract InputVerifier is UUPSUpgradeable, Ownable2StepUpgradeable {
 
     function _authorizeUpgrade(address _newImplementation) internal virtual override onlyOwner {}
 
+    /// @notice Getter function for the KMSVerifier contract address
+    function getKMSVerifierAddress() public view virtual returns (address) {
+        return address(kmsVerifier);
+    }
+
     /// @custom:oz-upgrades-unsafe-allow constructor
     constructor() {
         _disableInitializers();

lib/TFHEExecutor.sol

@@ -58,6 +58,21 @@ contract TFHEExecutor is UUPSUpgradeable, Ownable2StepUpgradeable {
 
     function _authorizeUpgrade(address _newImplementation) internal virtual override onlyOwner {}
 
+    /// @notice Getter function for the ACL contract address
+    function getACLAddress() public view virtual returns (address) {
+        return address(acl);
+    }
+
+    /// @notice Getter function for the FHEPayment contract address
+    function getFHEPaymentAddress() public view virtual returns (address) {
+        return address(fhePayment);
+    }
+
+    /// @notice Getter function for the InputVerifier contract address
+    function getInputVerifierAddress() public view virtual returns (address) {
+        return address(inputVerifier);
+    }
+
     /// @custom:oz-upgrades-unsafe-allow constructor
     constructor() {
         _disableInitializers();