chore: make follow and community nfts non-upgradeable

horuslabsio · Feb 2, 2025 · 5c15abb · 5c15abb
1 parent 17a985e
commit 5c15abb
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 24 deletions.
diff --git a/src/community/communitynft.cairo b/src/community/communitynft.cairo
@@ -3,7 +3,7 @@ pub mod CommunityNFT {
     // *************************************************************************
     //                             IMPORTS
     // *************************************************************************
-    use starknet::{ContractAddress, ClassHash, get_block_timestamp};
+    use starknet::{ContractAddress, get_block_timestamp};
     use core::num::traits::zero::Zero;
     use openzeppelin_introspection::src5::SRC5Component;
     use openzeppelin_token::erc721::{ERC721Component, ERC721HooksEmptyImpl};
@@ -117,15 +117,6 @@ pub mod CommunityNFT {
             self.user_token_id.write(user_address, 0);
         }
 
-        /// @notice upgrades the nft contract
-        /// @param new_class_hash classhash to upgrade to
-        fn upgrade(ref self: ContractState, new_class_hash: ClassHash) {
-            self.ownable.assert_only_owner();
-
-            // Replace the class hash upgrading the contract
-            self.upgradeable.upgrade(new_class_hash);
-        }
-
         // *************************************************************************
         //                            GETTERS
         // *************************************************************************

diff --git a/src/follownft/follownft.cairo b/src/follownft/follownft.cairo
@@ -4,7 +4,7 @@ pub mod Follow {
     //                            IMPORT
     // *************************************************************************
     use starknet::{
-        ContractAddress, ClassHash, get_block_timestamp,
+        ContractAddress, get_block_timestamp,
         storage::{
             StoragePointerWriteAccess, StoragePointerReadAccess, Map, StorageMapReadAccess,
             StorageMapWriteAccess
@@ -244,15 +244,6 @@ pub mod Follow {
             return true;
         }
 
-        /// @notice upgrades the nft contract
-        /// @param new_class_hash classhash to upgrade to
-        fn upgrade(ref self: ContractState, new_class_hash: ClassHash) {
-            self.ownable.assert_only_owner();
-
-            // Replace the class hash upgrading the contract
-            self.upgradeable.upgrade(new_class_hash);
-        }
-
         // *************************************************************************
         //                            GETTERS
         // *************************************************************************

diff --git a/src/interfaces/ICustomNFT.cairo b/src/interfaces/ICustomNFT.cairo
@@ -1,4 +1,4 @@
-use starknet::{ContractAddress, ClassHash};
+use starknet::ContractAddress;
 
 // *************************************************************************
 //                              INTERFACE of ICommunity NFT
@@ -11,7 +11,6 @@ pub trait ICustomNFT<TState> {
 
     fn mint_nft(ref self: TState, user_address: ContractAddress) -> u256;
     fn burn_nft(ref self: TState, user_address: ContractAddress, token_id: u256);
-    fn upgrade(ref self: TState, new_class_hash: ClassHash);
 
     // *************************************************************************
     //                            GETTERS

diff --git a/src/interfaces/IFollowNFT.cairo b/src/interfaces/IFollowNFT.cairo
@@ -1,4 +1,4 @@
-use starknet::{ContractAddress, ClassHash};
+use starknet::ContractAddress;
 use coloniz::base::constants::types::FollowData;
 
 // *************************************************************************
@@ -13,7 +13,6 @@ pub trait IFollowNFT<TState> {
     fn unfollow(ref self: TState, unfollower_profile_address: ContractAddress);
     fn process_block(ref self: TState, follower_profile_address: ContractAddress) -> bool;
     fn process_unblock(ref self: TState, follower_profile_address: ContractAddress) -> bool;
-    fn upgrade(ref self: TState, new_class_hash: ClassHash);
     // *************************************************************************
     //                            GETTERS
     // *************************************************************************