multiversx-sc 0.66.0

use core::{borrow::Borrow, marker::PhantomData};

pub use super::{
    StorageMapper, StorageMapperFromAddress,
    source::{CurrentStorage, StorageAddress},
};
use crate::{
    abi::{TypeAbi, TypeAbiFrom, TypeDescriptionContainer, TypeName},
    api::StorageMapperApi,
    codec::{
        DecodeErrorHandler, EncodeErrorHandler, TopDecode, TopDecodeInput, TopEncode,
        TopEncodeMulti, TopEncodeMultiOutput, TopEncodeOutput, multi_types::PlaceholderOutput,
    },
    storage::{StorageKey, storage_clear, storage_set},
    types::{ManagedAddress, ManagedType},
};

/// A storage mapper for managing a single serializable value with atomic read/write operations.
///
/// # Storage Layout
///
/// The `SingleValueMapper` stores a single value directly at its storage key:
///
/// - `base_key` → encoded value of type `T`
///
/// This is the simplest storage mapper - just one key storing one value.
///
/// # Main Operations
///
/// - **Write**: `set(value)` - Stores a value (accepts owned or borrowed). O(1).
/// - **Read**: `get()` - Retrieves the stored value. O(1).
/// - **Conditional write**: `set_if_empty(value)` - Stores only if empty. O(1).
/// - **Update**: `update(|v| {...})` - Read-modify-write in one operation. O(1).
/// - **Replace**: `replace(new_value)` - Swaps value and returns old one. O(1).
/// - **Take**: `take()` - Retrieves value and clears storage. O(1).
/// - **Clear**: `clear()` - Removes value from storage. O(1).
/// - **Check**: `is_empty()` - Returns true if no value stored. O(1).
///
/// # Value Semantics
///
/// - Empty storage: Returns the default/zero value when calling `get()`
/// - Setting a value: Overwrites any existing value
/// - Clearing: Removes the value entirely from storage
/// - Zero values: A value set to its zero/default may be indistinguishable from empty storage
///
/// # Trade-offs
///
/// - **Pros**: Simplest storage pattern; minimal overhead; direct key-value mapping; very efficient.
/// - **Cons**: Only one value per mapper; no built-in indexing or collections.
///
/// # Use Cases
///
/// - Configuration values (flags, thresholds, addresses)
/// - Global counters or accumulators
/// - Contract state variables (owner, paused status, etc.)
/// - Cached computed values
/// - Simple on-chain variables that don't need collection semantics
///
/// # Example
///
/// ```rust
/// # use multiversx_sc::storage::mappers::{StorageMapper, SingleValueMapper};
/// # use multiversx_sc::types::ManagedAddress;
/// # fn example<SA: multiversx_sc::api::StorageMapperApi>(owner: ManagedAddress<SA>) {
/// # let mapper = SingleValueMapper::<SA, ManagedAddress<SA>>::new(
/// #     multiversx_sc::storage::StorageKey::new(&b"owner"[..])
/// # );
/// // Set a value
/// mapper.set(&owner);
///
/// // Check if empty
/// assert!(!mapper.is_empty());
///
/// // Get the value
/// let current_owner = mapper.get();
/// assert_eq!(current_owner, owner);
///
/// // Conditional set (only if empty)
/// mapper.set_if_empty(&owner);  // Does nothing, already set
///
/// // Update in place
/// mapper.update(|addr| {
///     // Modify the value
///     *addr = owner.clone();
/// });
///
/// // Replace and get old value
/// # let new_owner = owner.clone();
/// let old_owner = mapper.replace(&new_owner);
/// assert_eq!(old_owner, owner);
///
/// // Take value (get and clear)
/// let taken = mapper.take();
/// assert!(mapper.is_empty());
///
/// // Clear storage
/// mapper.set(&owner);
/// mapper.clear();
/// assert!(mapper.is_empty());
/// # }
/// ```
///
/// # Numeric Counter Example
///
/// ```rust
/// # use multiversx_sc::storage::mappers::{StorageMapper, SingleValueMapper};
/// # fn counter_example<SA: multiversx_sc::api::StorageMapperApi>() {
/// # let counter = SingleValueMapper::<SA, u64>::new(
/// #     multiversx_sc::storage::StorageKey::new(&b"counter"[..])
/// # );
/// // Initialize counter
/// counter.set(0u64);
///
/// // Increment using update
/// counter.update(|value| *value += 1);
/// assert_eq!(counter.get(), 1);
///
/// // Increment and return new value
/// let new_value = counter.update(|value| {
///     *value += 1;
///     *value
/// });
/// assert_eq!(new_value, 2);
/// # }
/// ```
pub struct SingleValueMapper<SA, T, A = CurrentStorage>
where
    SA: StorageMapperApi,
    A: StorageAddress<SA>,
    T: TopEncode + TopDecode + 'static,
{
    address: A,
    key: StorageKey<SA>,
    _phantom_api: PhantomData<SA>,
    _phantom_item: PhantomData<T>,
}

impl<SA, T> StorageMapper<SA> for SingleValueMapper<SA, T, CurrentStorage>
where
    SA: StorageMapperApi,
    T: TopEncode + TopDecode,
{
    #[inline]
    fn new(base_key: StorageKey<SA>) -> Self {
        SingleValueMapper {
            address: CurrentStorage,
            key: base_key,
            _phantom_api: PhantomData,
            _phantom_item: PhantomData,
        }
    }
}

impl<SA, T> StorageMapperFromAddress<SA> for SingleValueMapper<SA, T, ManagedAddress<SA>>
where
    SA: StorageMapperApi,
    T: TopEncode + TopDecode,
{
    #[inline]
    fn new_from_address(address: ManagedAddress<SA>, base_key: StorageKey<SA>) -> Self {
        SingleValueMapper {
            address,
            key: base_key,
            _phantom_api: PhantomData,
            _phantom_item: PhantomData,
        }
    }
}

impl<SA, T, A> SingleValueMapper<SA, T, A>
where
    SA: StorageMapperApi,
    A: StorageAddress<SA>,
    T: TopEncode + TopDecode,
{
    /// Retrieves current value from storage.
    pub fn get(&self) -> T {
        self.address.address_storage_get(self.key.as_ref())
    }

    /// Returns whether the storage managed by this mapper is empty.
    pub fn is_empty(&self) -> bool {
        self.raw_byte_length() == 0
    }

    pub fn raw_byte_length(&self) -> usize {
        self.address.address_storage_get_len(self.key.as_ref())
    }
}

impl<SA, T> SingleValueMapper<SA, T, CurrentStorage>
where
    SA: StorageMapperApi,
    T: TopEncode + TopDecode,
{
    /// Saves argument to storage.
    ///
    /// Accepts owned item of type `T`, or any borrowed form of it, such as `&T`.
    #[inline]
    pub fn set<BT>(&self, new_value: BT)
    where
        BT: Borrow<T>,
    {
        storage_set(self.key.as_ref(), new_value.borrow());
    }

    /// Saves argument to storage only if the storage is empty.
    /// Does nothing otherwise.
    pub fn set_if_empty<BT>(&self, value: BT)
    where
        BT: Borrow<T>,
    {
        if self.is_empty() {
            self.set(value);
        }
    }

    /// Clears the storage for this mapper.
    pub fn clear(&self) {
        storage_clear(self.key.as_ref());
    }

    /// Syntactic sugar, to more compactly express a get, update and set in one line.
    /// Takes whatever lies in storage, apples the given closure and saves the final value back to storage.
    /// Propagates the return value of the given function.
    pub fn update<R, F: FnOnce(&mut T) -> R>(&self, f: F) -> R {
        let mut value = self.get();
        let result = f(&mut value);
        self.set(value);
        result
    }

    /// Takes the value out of the storage, clearing it in the process.
    pub fn take(&self) -> T {
        let value = self.get();
        self.clear();
        value
    }

    // Replaces the actual value in the storage by the value given in parameter, returning the old value.
    pub fn replace<BT>(&self, new_value: BT) -> T
    where
        BT: Borrow<T>,
    {
        let value = self.get();
        self.set(new_value);
        value
    }
}

impl<SA, T> TopEncodeMulti for SingleValueMapper<SA, T, CurrentStorage>
where
    SA: StorageMapperApi,
    T: TopEncode + TopDecode,
{
    fn multi_encode_or_handle_err<O, H>(&self, output: &mut O, h: H) -> Result<(), H::HandledErr>
    where
        O: TopEncodeMultiOutput,
        H: EncodeErrorHandler,
    {
        output.push_single_value(&self.get(), h)
    }
}

/// Intermediary type for deserializing the result of an endpoint that returns a `SingleValueMapper`.
///
/// Necessary because we cannot implement `TypeAbiFrom` directly on `T`.
pub struct SingleValue<T: TopDecode>(T);

impl<T: TopEncode + TopDecode> TopEncode for SingleValue<T> {
    fn top_encode_or_handle_err<O, H>(&self, output: O, h: H) -> Result<(), H::HandledErr>
    where
        O: TopEncodeOutput,
        H: EncodeErrorHandler,
    {
        self.0.top_encode_or_handle_err(output, h)
    }
}

impl<T: TopDecode> TopDecode for SingleValue<T> {
    fn top_decode_or_handle_err<I, H>(input: I, h: H) -> Result<Self, H::HandledErr>
    where
        I: TopDecodeInput,
        H: DecodeErrorHandler,
    {
        Ok(SingleValue::<T>(T::top_decode_or_handle_err(input, h)?))
    }
}

impl<T: TopDecode> From<T> for SingleValue<T> {
    fn from(value: T) -> Self {
        SingleValue::<T>(value)
    }
}

impl<T: TopDecode> SingleValue<T> {
    #[inline]
    pub fn into(self) -> T {
        self.0
    }
}

impl<SA, T, R> TypeAbiFrom<SingleValueMapper<SA, T, CurrentStorage>> for SingleValue<R>
where
    SA: StorageMapperApi,
    T: TopEncode + TopDecode,
    R: TopDecode + TypeAbiFrom<T>,
{
}

impl<SA, T> TypeAbiFrom<SingleValueMapper<SA, T>> for PlaceholderOutput
where
    SA: StorageMapperApi,
    T: TopEncode + TopDecode,
{
}

impl<SA, T> TypeAbiFrom<Self> for SingleValueMapper<SA, T, CurrentStorage>
where
    SA: StorageMapperApi,
    T: TopEncode + TopDecode + TypeAbi,
{
}

impl<SA, T> TypeAbi for SingleValueMapper<SA, T, CurrentStorage>
where
    SA: StorageMapperApi,
    T: TopEncode + TopDecode + TypeAbi,
{
    type Unmanaged = T::Unmanaged;

    fn type_name() -> TypeName {
        T::type_name()
    }

    fn type_name_rust() -> TypeName {
        T::type_name_rust()
    }

    fn provide_type_descriptions<TDC: TypeDescriptionContainer>(accumulator: &mut TDC) {
        T::provide_type_descriptions(accumulator)
    }
}