reputation contract init

Author: Bosun-Josh121

contracts/reputation-scoring-contract/src/contract.rs

@@ -1,6 +1,7 @@
 use crate::error::ReputationError;
+use crate::events;
 use crate::storage;
-use soroban_sdk::{Address, Env};
+use soroban_sdk::{Address, BytesN, Env};
 
 pub fn initialize(
     env: &Env,
@@ -14,3 +15,37 @@ pub fn initialize(
     storage::set_vault_address(env, vault_address);
     Ok(())
 }
+
+pub fn pause(env: &Env) -> Result<(), ReputationError> {
+    let admin = storage::get_admin(env).ok_or(ReputationError::NotInitialized)?;
+    admin.require_auth();
+    storage::set_paused(env, true);
+    events::contract_paused(env, true);
+    Ok(())
+}
+
+pub fn unpause(env: &Env) -> Result<(), ReputationError> {
+    let admin = storage::get_admin(env).ok_or(ReputationError::NotInitialized)?;
+    admin.require_auth();
+    storage::set_paused(env, false);
+    events::contract_paused(env, false);
+    Ok(())
+}
+
+pub fn transfer_admin(env: &Env, new_admin: &Address) -> Result<(), ReputationError> {
+    let admin = storage::get_admin(env).ok_or(ReputationError::NotInitialized)?;
+    admin.require_auth();
+    if storage::is_paused(env) {
+        return Err(ReputationError::ContractPaused);
+    }
+    storage::set_admin(env, new_admin);
+    events::admin_transferred(env, &admin, new_admin);
+    Ok(())
+}
+
+pub fn upgrade_contract(env: &Env, new_wasm_hash: BytesN<32>) -> Result<(), ReputationError> {
+    let admin = storage::get_admin(env).ok_or(ReputationError::NotInitialized)?;
+    admin.require_auth();
+    env.deployer().update_current_contract_wasm(new_wasm_hash);
+    Ok(())
+}

contracts/reputation-scoring-contract/src/error.rs

@@ -4,5 +4,7 @@ use soroban_sdk::contracterror;
 #[derive(Copy, Clone, Debug, Eq, PartialEq, PartialOrd, Ord)]
 #[repr(u32)]
 pub enum ReputationError {
-    AlreadyInitialized = 1,
+    NotInitialized = 1,
+    AlreadyInitialized = 2,
+    ContractPaused = 3,
 }

contracts/reputation-scoring-contract/src/events.rs

@@ -0,0 +1,15 @@
+#![allow(deprecated)]
+use soroban_sdk::{symbol_short, Address, Env};
+
+/// Emitted when the contract is paused or unpaused.
+pub fn contract_paused(env: &Env, paused: bool) {
+    let topics = (symbol_short!("paused"),);
+    env.events().publish(topics, paused);
+}
+
+/// Emitted when admin authority is transferred to a new address.
+pub fn admin_transferred(env: &Env, old_admin: &Address, new_admin: &Address) {
+    let topics = (symbol_short!("adm_xfer"),);
+    env.events()
+        .publish(topics, (old_admin.clone(), new_admin.clone()));
+}

contracts/reputation-scoring-contract/src/lib.rs

@@ -2,13 +2,14 @@
 
 mod contract;
 mod error;
+mod events;
 mod storage;
 #[cfg(test)]
 mod test;
 mod types;
 
 use crate::error::ReputationError;
-use soroban_sdk::{contract, contractimpl, Address, Env};
+use soroban_sdk::{contract, contractimpl, Address, BytesN, Env};
 
 #[contract]
 pub struct ReputationScoringContract;
@@ -18,4 +19,20 @@ impl ReputationScoringContract {
     pub fn init(env: Env, admin: Address, vault_address: Address) -> Result<(), ReputationError> {
         contract::initialize(&env, &admin, &vault_address)
     }
+
+    pub fn pause(env: Env) -> Result<(), ReputationError> {
+        contract::pause(&env)
+    }
+
+    pub fn unpause(env: Env) -> Result<(), ReputationError> {
+        contract::unpause(&env)
+    }
+
+    pub fn transfer_admin(env: Env, new_admin: Address) -> Result<(), ReputationError> {
+        contract::transfer_admin(&env, &new_admin)
+    }
+
+    pub fn upgrade_contract(env: Env, new_wasm_hash: BytesN<32>) -> Result<(), ReputationError> {
+        contract::upgrade_contract(&env, new_wasm_hash)
+    }
 }

contracts/reputation-scoring-contract/src/storage.rs

@@ -5,6 +5,7 @@ use soroban_sdk::{contracttype, Address, Env};
 pub enum DataKey {
     Admin,
     VaultAddress,
+    IsPaused,
 }
 
 pub fn has_admin(env: &Env) -> bool {
@@ -15,6 +16,21 @@ pub fn set_admin(env: &Env, admin: &Address) {
     env.storage().instance().set(&DataKey::Admin, admin);
 }
 
+pub fn get_admin(env: &Env) -> Option<Address> {
+    env.storage().instance().get(&DataKey::Admin)
+}
+
 pub fn set_vault_address(env: &Env, vault: &Address) {
     env.storage().instance().set(&DataKey::VaultAddress, vault);
 }
+
+pub fn is_paused(env: &Env) -> bool {
+    env.storage()
+        .instance()
+        .get(&DataKey::IsPaused)
+        .unwrap_or(false)
+}
+
+pub fn set_paused(env: &Env, paused: bool) {
+    env.storage().instance().set(&DataKey::IsPaused, &paused);
+}

contracts/reputation-scoring-contract/src/test.rs

@@ -3,13 +3,52 @@
 use super::*;
 use soroban_sdk::{testutils::Address as _, Address, Env};
 
-#[test]
-fn test_initialize() {
+fn setup() -> (Env, Address, Address, ReputationScoringContractClient<'static>) {
     let env = Env::default();
     env.mock_all_auths();
     let contract_id = env.register(ReputationScoringContract, ());
     let client = ReputationScoringContractClient::new(&env, &contract_id);
     let admin = Address::generate(&env);
     let vault = Address::generate(&env);
+    (env, admin, vault, client)
+}
+
+#[test]
+fn test_initialize() {
+    let (_env, admin, vault, client) = setup();
+    client.init(&admin, &vault);
+}
+
+#[test]
+fn test_initialize_twice_fails() {
+    let (_env, admin, vault, client) = setup();
+    client.init(&admin, &vault);
+    assert!(client.try_init(&admin, &vault).is_err());
+}
+
+#[test]
+fn test_transfer_admin() {
+    let (env, admin, vault, client) = setup();
+    client.init(&admin, &vault);
+    let new_admin = Address::generate(&env);
+    client.transfer_admin(&new_admin);
+}
+
+#[test]
+fn test_pause_blocks_transfer_admin() {
+    let (env, admin, vault, client) = setup();
+    client.init(&admin, &vault);
+    client.pause();
+    let new_admin = Address::generate(&env);
+    assert!(client.try_transfer_admin(&new_admin).is_err());
+}
+
+#[test]
+fn test_unpause_restores_transfer_admin() {
+    let (env, admin, vault, client) = setup();
     client.init(&admin, &vault);
+    client.pause();
+    client.unpause();
+    let new_admin = Address::generate(&env);
+    client.transfer_admin(&new_admin);
 }