[WIP] [Migration ZChain] Domain Migration: Backend Registration/Transfer

.gitignore

@@ -20,3 +20,11 @@ docker*.tgz
 
 # We don't ever use the generated manifests
 .openzeppelin
+
+# Ignore the output files of migration
+valid-root-domains.json
+valid-subdomains.json
+invalid-domains.json
+
+# Deploy logs
+deploy-*.log

contracts/price/ZNSCurvePricer.sol

@@ -134,15 +134,15 @@ contract ZNSCurvePricer is AAccessControlled, ARegistryWired, UUPSUpgradeable, I
         bytes32 domainHash,
         CurvePriceConfig calldata priceConfig
     ) public override {
-        setPrecisionMultiplier(domainHash, priceConfig.precisionMultiplier);
-        priceConfigs[domainHash].baseLength = priceConfig.baseLength;
-        priceConfigs[domainHash].maxPrice = priceConfig.maxPrice;
-        priceConfigs[domainHash].minPrice = priceConfig.minPrice;
-        priceConfigs[domainHash].maxLength = priceConfig.maxLength;
-        setFeePercentage(domainHash, priceConfig.feePercentage);
-        priceConfigs[domainHash].isSet = true;
+        // Validate a new or existing config
+        _setPrecisionMultiplier(domainHash, priceConfig.precisionMultiplier);
+        _validateCurveMultiplier(domainHash, priceConfig.curveMultiplier, priceConfig);
+        // _validateBaseLength
+        // _validateMaxPrice
+        // _validateMaxLength
+        // _validateFeePercentage
 
-        _validateConfig(domainHash);
+        priceConfigs[domainHash].isSet = true;
 
         emit PriceConfigSet(
             domainHash,
@@ -186,11 +186,27 @@ contract ZNSCurvePricer is AAccessControlled, ARegistryWired, UUPSUpgradeable, I
         bytes32 domainHash,
         uint256 minPrice
     ) external override onlyOwnerOrOperator(domainHash) {
-        priceConfigs[domainHash].minPrice = minPrice;
+        // Grab an existing config
+        CurvePriceConfig memory config = priceConfigs[domainHash];
 
-        _validateConfig(domainHash);
+        _validateCurveMultiplier(domainHash, curveMultiplier, config);
+
+        emit CurveMultiplierSet(domainHash, curveMultiplier);
+    }
 
-        emit MinPriceSet(domainHash, minPrice);
+    function _validateCurveMultiplier(
+        bytes32 domainHash,
+        uint256 curveMultiplier,
+        CurvePriceConfig memory config
+    ) internal {
+        if (
+            (curveMultiplier == 0 && config.baseLength == 0) || 
+            (curveMultiplier == 0 && config.maxLength == 0)
+        ) {
+            revert DivisionByZero(0x0);
+        }
+
+        priceConfigs[domainHash].curveMultiplier = curveMultiplier;
     }
 
     /**
@@ -287,6 +303,44 @@ contract ZNSCurvePricer is AAccessControlled, ARegistryWired, UUPSUpgradeable, I
         _setRegistry(registry_);
     }
 
+    function validateConfig (
+        bytes32 domainHash
+    ) internal view {
+        CurvePriceConfig memory config = priceConfigs[domainHash];
+
+        if (config.maxLength < config.baseLength)
+            revert MaxLengthSmallerThanBaseLength(
+                domainHash
+            );
+
+        if (config.maxPrice == 0 && config.maxLength == 0)
+            revert InvalidMaxPrice(domainHash);
+
+        // curveMultiplier = 0 && baseLength = 0
+        // curveMultiplier = 0 && maxLength = 0
+        // baseLength = 0 && curveMultiplier == 0
+        // baseLength = 0 && maxLength = 0
+
+        if (
+            (config.curveMultiplier == 0 && config.baseLength == 0) || 
+            (config.curveMultiplier == 0 && config.maxLength == 0) ||
+            (config.baseLength == 0 && config.maxLength == 0)
+        )
+            revert DivisionByZero(domainHash);
+
+        // if (config.baseLength == 0 && (config.curveMultiplier == 0 || config.maxLength == 0))
+        //     revert DivisionByZero(domainHash);
+
+        if (config.feePercentage > PERCENTAGE_BASIS)
+            revert FeePercentageValueTooLarge(
+                config.feePercentage,
+                PERCENTAGE_BASIS
+            );
+
+        if (config.precisionMultiplier == 0 || config.precisionMultiplier > 10**18
+            ) revert InvalidMultiplierPassed(config.precisionMultiplier);
+    }
+
     /**
      * @notice Internal function to calculate price based on the config set,
      * and the length of the domain label.

contracts/registrar/IZNSRootRegistrar.sol

@@ -22,6 +22,15 @@ struct CoreRegisterArgs {
     PaymentConfig paymentConfig;
 }
 
+// Args for bulk tx
+struct DomainData {
+    string[] names;
+    address[] domainAddresses;
+    string [] tokenURIs;
+    IDistributionConfig.DistributionConfig[] distributionConfigs;
+    PaymentConfig[] paymentConfigs;
+}
+
 /**
  * @title IZNSRootRegistrar.sol - Interface for the ZNSRootRegistrar contract resposible for registering root domains.
  * @notice Below are docs for the types in this file:
@@ -42,6 +51,16 @@ struct CoreRegisterArgs {
  *      + `isStakePayment`: A flag for whether the payment is a stake payment or not
  */
 interface IZNSRootRegistrar is IDistributionConfig {
+    // TODO TEMP DELETE after migration
+    struct RootBulkMigrationArgs {
+        address[] tokenOwners;
+        address[] recordOwners;
+        string[] names;
+        address[] domainAddresses;
+        string[] tokenURIs;
+        DistributionConfig[] distributionConfigs;
+        PaymentConfig[] paymentConfigs;
+    }
     error NotTheOwnerOf(
         OwnerOf ownerOf,
         address candidate,

contracts/registrar/IZNSSubRegistrar.sol

@@ -5,13 +5,37 @@ import { IDistributionConfig } from "../types/IDistributionConfig.sol";
 import { PaymentConfig } from "../treasury/IZNSTreasury.sol";
 import { IZNSPricer } from "../types/IZNSPricer.sol";
 
-
 /**
  * @title IZNSSubRegistrar.sol - Interface for the ZNSSubRegistrar contract responsible for registering subdomains.
-*/
+ */
 interface IZNSSubRegistrar is IDistributionConfig {
+
+    // TODO TEMP DELETE after migration
+    struct SubBulkMigrationArgs {
+        address domainToken; // DomainToken contract for transfers
+        address[] tokenOwners;
+        address[] recordOwners;
+        bytes32[] parentHashes;
+        string[] labels;
+        address[] domainAddresses;
+        string[] tokenURIs;
+        DistributionConfig[] distributionConfigs;
+        PaymentConfig[] paymentConfigs;
+    }
+
+    struct RegisterSubdomainArgs {
+        bytes32 parentHash;
+        string label;
+        address domainAddress;
+        address recordOwner;
+        string tokenURI;
+        DistributionConfig distrConfig;
+        PaymentConfig paymentConfig;
+    }
+
     /**
-     * @notice Reverted when someone other than parent owner is trying to buy a subdomain under the parent that is locked\
+     * @notice Reverted when someone other than parent owner is trying to buy 
+     a subdomain under the parent that is locked\
      * or when the parent provided does not exist.
      */
     error ParentLockedOrDoesntExist(bytes32 parentHash);
@@ -23,22 +47,25 @@ interface IZNSSubRegistrar is IDistributionConfig {
 
     /**
      * @notice Emitted when a new `DistributionConfig.pricerContract` is set for a domain.
-    */
-    event PricerContractSet(bytes32 indexed domainHash, address indexed pricerContract);
+     */
+    event PricerContractSet(
+        bytes32 indexed domainHash,
+        address indexed pricerContract
+    );
 
     /**
      * @notice Emitted when a new `DistributionConfig.paymentType` is set for a domain.
-    */
+     */
     event PaymentTypeSet(bytes32 indexed domainHash, PaymentType paymentType);
 
     /**
      * @notice Emitted when a new `DistributionConfig.accessType` is set for a domain.
-    */
+     */
     event AccessTypeSet(bytes32 indexed domainHash, AccessType accessType);
 
     /**
      * @notice Emitted when a new full `DistributionConfig` is set for a domain at once.
-    */
+     */
     event DistributionConfigSet(
         bytes32 indexed domainHash,
         IZNSPricer pricerContract,
@@ -48,7 +75,7 @@ interface IZNSSubRegistrar is IDistributionConfig {
 
     /**
      * @notice Emitted when a `mintlist` is updated for a domain.
-    */
+     */
     event MintlistUpdated(
         bytes32 indexed domainHash,
         uint256 indexed ownerIndex,
@@ -64,16 +91,17 @@ interface IZNSSubRegistrar is IDistributionConfig {
 
     /**
      * @notice Emitted when the ZNSRootRegistrar address is set in state.
-    */
+     */
     event RootRegistrarSet(address registrar);
 
     function distrConfigs(
         bytes32 domainHash
-    ) external view returns (
-        IZNSPricer pricerContract,
-        PaymentType paymentType,
-        AccessType accessType
-    );
+    )
+        external view returns (
+            IZNSPricer pricerContract,
+            PaymentType paymentType,
+            AccessType accessType
+        );
 
     function isMintlistedForDomain(
         bytes32 domainHash,
@@ -87,12 +115,7 @@ interface IZNSSubRegistrar is IDistributionConfig {
     ) external;
 
     function registerSubdomain(
-        bytes32 parentHash,
-        string calldata label,
-        address domainAddress,
-        string calldata tokenURI,
-        DistributionConfig calldata configForSubdomains,
-        PaymentConfig calldata paymentConfig
+        RegisterSubdomainArgs calldata args
     ) external returns (bytes32);
 
     function hashWithParent(

contracts/registrar/ZNSRootRegistrar.sol

@@ -13,6 +13,10 @@ import { UUPSUpgradeable } from "@openzeppelin/contracts-upgradeable/proxy/utils
 import { StringUtils } from "../utils/StringUtils.sol";
 import { ZeroAddressPassed, DomainAlreadyExists } from "../utils/CommonErrors.sol";
 
+// import { BulkMigrationArgs } from "./IZNSRootRegistrar.sol"; 
+import { IERC20 } from "@openzeppelin/contracts/token/ERC20/IERC20.sol";
+
+import { console } from "hardhat/console.sol";
 
 /**
  * @title Main entry point for the three main flows of ZNS - Register Root Domain, Reclaim and Revoke any domain.
@@ -69,6 +73,31 @@ contract ZNSRootRegistrar is
         setDomainToken(domainToken_);
     }
 
+    function registerRootDomainBulk(
+        RootBulkMigrationArgs calldata args
+    ) public onlyAdmin {
+        // Expect all arrays are the same length
+        for (uint256 i = 0; i < args.names.length;) {
+            bytes32 domainHash = registerRootDomain(
+                args.names[i],
+                args.domainAddresses[i],
+                args.tokenURIs[i],
+                args.distributionConfigs[i],
+                args.paymentConfigs[i]
+            );
+
+            // Transfer ERC721 to domain token owner
+            domainToken.transferFrom(msg.sender, args.tokenOwners[i], uint256(domainHash));
+
+            // Update the domain record in the registry with the proper record owner
+            registry.updateDomainOwnerForMigration(domainHash, args.recordOwners[i]);
+
+            unchecked {
+                ++i;
+            }
+        }
+    }
+
     /**
      * @notice This function is the main entry point for the Register Root Domain flow.
      * Registers a new root domain such as `0://wilder`.
@@ -94,7 +123,7 @@ contract ZNSRootRegistrar is
         string calldata tokenURI,
         DistributionConfig calldata distributionConfig,
         PaymentConfig calldata paymentConfig
-    ) external override returns (bytes32) {
+    ) public override onlyAdmin returns (bytes32) {
         // Confirms string values are only [a-z0-9-]
         name.validate();
 
@@ -164,6 +193,7 @@ contract ZNSRootRegistrar is
         CoreRegisterArgs memory args
     ) internal {
         // payment part of the logic
+        // TODO uncomment after migration
         if (args.price > 0) {
             _processPayment(args);
         }