miden-base-sys 0.12.0

use miden_stdlib_sys::{Felt, Word};

use super::StorageSlotId;

#[allow(improper_ctypes)]
unsafe extern "C" {
    // The public SDK models storage slots as `(prefix, suffix)`, but the host ABI expects the slot
    // coordinates in storage order: `(suffix, prefix)`.
    #[link_name = "miden::protocol::active_account::get_item"]
    pub fn extern_get_storage_item(index_suffix: Felt, index_prefix: Felt, ptr: *mut Word);

    #[link_name = "miden::protocol::active_account::get_initial_item"]
    pub fn extern_get_initial_storage_item(index_suffix: Felt, index_prefix: Felt, ptr: *mut Word);

    #[link_name = "miden::protocol::native_account::set_item"]
    pub fn extern_set_storage_item(
        index_suffix: Felt,
        index_prefix: Felt,
        v0: Felt,
        v1: Felt,
        v2: Felt,
        v3: Felt,
        ptr: *mut Word,
    );

    #[link_name = "miden::protocol::active_account::get_map_item"]
    pub fn extern_get_storage_map_item(
        index_suffix: Felt,
        index_prefix: Felt,
        k0: Felt,
        k1: Felt,
        k2: Felt,
        k3: Felt,
        ptr: *mut Word,
    );

    #[link_name = "miden::protocol::active_account::get_initial_map_item"]
    pub fn extern_get_initial_storage_map_item(
        index_suffix: Felt,
        index_prefix: Felt,
        k0: Felt,
        k1: Felt,
        k2: Felt,
        k3: Felt,
        ptr: *mut Word,
    );

    #[link_name = "miden::protocol::native_account::set_map_item"]
    pub fn extern_set_storage_map_item(
        index_suffix: Felt,
        index_prefix: Felt,
        k0: Felt,
        k1: Felt,
        k2: Felt,
        k3: Felt,
        v0: Felt,
        v1: Felt,
        v2: Felt,
        v3: Felt,
        ptr: *mut Word,
    );
}

/// Gets an item from the account storage.
///
/// Inputs: slot_id
/// Outputs: value
///
/// Where:
/// - slot_id identifies the storage slot to access using the public `(prefix, suffix)` shape.
/// - value is the value of the item.
///
/// Panics if:
/// - the requested slot does not exist in the account storage.
#[inline]
pub fn get_item(slot_id: StorageSlotId) -> Word {
    unsafe {
        let mut ret_area = ::core::mem::MaybeUninit::<Word>::uninit();
        let (suffix, prefix) = slot_id.to_suffix_prefix();
        extern_get_storage_item(suffix, prefix, ret_area.as_mut_ptr());
        ret_area.assume_init()
    }
}

/// Gets the initial value of an item from the account storage.
#[inline]
pub fn get_initial_item(slot_id: StorageSlotId) -> Word {
    unsafe {
        let mut ret_area = ::core::mem::MaybeUninit::<Word>::uninit();
        let (suffix, prefix) = slot_id.to_suffix_prefix();
        extern_get_initial_storage_item(suffix, prefix, ret_area.as_mut_ptr());
        ret_area.assume_init()
    }
}

/// Sets an item in the account storage.
///
/// Inputs: slot_id, value
/// Outputs: old_value
///
/// Where:
/// - slot_id identifies the storage slot to update using the public `(prefix, suffix)` shape.
/// - value is the value to set.
/// - old_value is the previous value of the item.
///
/// Panics if:
/// - the requested slot does not exist in the account storage.
#[inline]
pub fn set_item(slot_id: StorageSlotId, value: Word) -> Word {
    unsafe {
        let mut ret_area = ::core::mem::MaybeUninit::<Word>::uninit();
        let (suffix, prefix) = slot_id.to_suffix_prefix();
        extern_set_storage_item(
            suffix,
            prefix,
            value[0],
            value[1],
            value[2],
            value[3],
            ret_area.as_mut_ptr(),
        );
        ret_area.assume_init()
    }
}

/// Gets a map item from the account storage.
///
/// Inputs: slot_id, key
/// Outputs: value
///
/// Where:
/// - slot_id identifies the map slot where the key should be read.
/// - key is the key of the item to get.
/// - value is the value of the item.
///
/// Panics if:
/// - the requested slot does not exist in the account storage.
/// - the slot content is not a map.
#[inline]
pub fn get_map_item(slot_id: StorageSlotId, key: &Word) -> Word {
    unsafe {
        let mut ret_area = ::core::mem::MaybeUninit::<Word>::uninit();
        let (suffix, prefix) = slot_id.to_suffix_prefix();
        extern_get_storage_map_item(
            suffix,
            prefix,
            key[0],
            key[1],
            key[2],
            key[3],
            ret_area.as_mut_ptr(),
        );
        ret_area.assume_init()
    }
}

/// Gets the initial value from a storage map.
#[inline]
pub fn get_initial_map_item(slot_id: StorageSlotId, key: &Word) -> Word {
    unsafe {
        let mut ret_area = ::core::mem::MaybeUninit::<Word>::uninit();
        let (suffix, prefix) = slot_id.to_suffix_prefix();
        extern_get_initial_storage_map_item(
            suffix,
            prefix,
            key[0],
            key[1],
            key[2],
            key[3],
            ret_area.as_mut_ptr(),
        );
        ret_area.assume_init()
    }
}

/// Sets a map item in the account storage.
///
/// Inputs: slot_id, key, value
/// Outputs: old_value
///
/// Where:
/// - slot_id identifies the map slot where the key should be set using the public `(prefix,
///   suffix)` shape.
/// - key is the key to set.
/// - value is the value to set.
/// - old_value is the old value at key.
///
/// Panics if:
/// - the requested slot does not exist in the account storage.
/// - the slot content is not a map.
#[inline]
pub fn set_map_item(slot_id: StorageSlotId, key: Word, value: Word) -> Word {
    unsafe {
        let mut ret_area = ::core::mem::MaybeUninit::<Word>::uninit();
        let (suffix, prefix) = slot_id.to_suffix_prefix();
        extern_set_storage_map_item(
            suffix,
            prefix,
            key[0],
            key[1],
            key[2],
            key[3],
            value[0],
            value[1],
            value[2],
            value[3],
            ret_area.as_mut_ptr(),
        );
        ret_area.assume_init()
    }
}