From 064a23c22a6d9b1edf7cd98d0192eba38d5459f5 Mon Sep 17 00:00:00 2001
From: Nimish Agrawal <realnimish@gmail.com>
Date: Fri, 19 Jul 2024 00:57:44 +0530
Subject: [PATCH] feat: retain old `register()` signature for backward
 compatibility

---
 src/azns_registry/lib.rs | 21 ++++++++++++++++-----
 1 file changed, 16 insertions(+), 5 deletions(-)

diff --git a/src/azns_registry/lib.rs b/src/azns_registry/lib.rs
index acd16b0..78124c5 100644
--- a/src/azns_registry/lib.rs
+++ b/src/azns_registry/lib.rs
@@ -386,10 +386,8 @@ mod azns_registry {
         }
 
         /// Register specific name with caller as owner.
-        ///
-        /// NOTE: Whitelisted addresses can buy one name during the whitelist phase by submitting its proof
         #[ink(message, payable)]
-        pub fn register(
+        pub fn register_v2(
             &mut self,
             name: String,
             years_to_register: u8,
@@ -410,6 +408,19 @@ mod azns_registry {
             Ok(())
         }
 
+        /// register_v1
+        #[ink(message, payable)]
+        pub fn register(
+            &mut self,
+            name: String,
+            years_to_register: u8,
+            referrer: Option<String>,
+            _merkle_proof: Option<Vec<[u8; 32]>>,
+            set_as_primary_name: bool,
+        ) -> Result<()> {
+            self.register_v2(name, years_to_register, referrer, None, set_as_primary_name)
+        }
+
         #[ink(message, payable)]
         pub fn renew(
             &mut self,
@@ -3302,7 +3313,7 @@ mod tests {
         set_next_caller(default_accounts().alice);
         transfer_in::<DefaultEnvironment>(1000);
         contract
-            .register(name1.clone(), 1, None, Some(bonus_name.clone()), false)
+            .register_v2(name1.clone(), 1, None, Some(bonus_name.clone()), false)
             .unwrap();
 
         assert_eq!(
@@ -3311,7 +3322,7 @@ mod tests {
         );
 
         contract
-            .register(name2, 10, None, Some(bonus_name.clone()), false)
+            .register_v2(name2, 10, None, Some(bonus_name.clone()), false)
             .unwrap();
         assert_eq!(contract.get_owner(bonus_name), Ok(accounts.alice))
     }