hooks_rs/api/
state.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
use core::mem::MaybeUninit;

use crate::c;

use super::*;

/// Retrieve the data pointed to by a Hook State key and write it to an output buffer
/// The keys are always 32 bytes (unsigned 256 bit integer) and the values are variable
/// length with a maximum size determined by validator voting, at time of writing 128 bytes.
///
/// # Example
/// ```
/// #[inline(always)]
/// fn get_count(key: &[u8; ACC_ID_LEN]) -> u64 {
///     match state::<STATE_VALUE_LEN>(key.as_ref()) {
///         Ok(data) => u64::from_be_bytes(data),
///         Err(_err) => {
///             rollback(b"could not get count state", -1);
///         }
///     }
/// }
///
/// #[inline(always)]
/// fn set_count(count: u64, key: &[u8; ACC_ID_LEN]) {
///     match state_set(count.to_be_bytes().as_ref(), key.as_ref()) {
///         Ok(_) => {}
///         Err(_) => {
///             rollback(b"could not set state", -1);
///         }
///     };
/// }
/// ```
#[inline(always)]
pub fn state<const STATE_VALUE_LEN: usize>(key: &[u8]) -> Result<[u8; STATE_VALUE_LEN]>
where
{
    let func = |buffer_mut_ptr: *mut MaybeUninit<u8>| {
        let result: Result<u64> = unsafe {
            c::state(
                buffer_mut_ptr as u32,
                STATE_VALUE_LEN as u32,
                key.as_ptr() as u32,
                key.len() as u32,
            )
            .into()
        };

        result
    };

    init_buffer_mut(func)
}

/// Set the Hook State for a given key and value
///
/// # Example
/// ```
/// #[inline(always)]
/// fn get_count(key: &[u8; ACC_ID_LEN]) -> u64 {
///     match state::<STATE_VALUE_LEN>(key.as_ref()) {
///         Ok(data) => u64::from_be_bytes(data),
///         Err(_err) => {
///             rollback(b"could not get count state", -1);
///         }
///     }
/// }
///
/// #[inline(always)]
/// fn set_count(count: u64, key: &[u8; ACC_ID_LEN]) {
///     match state_set(count.to_be_bytes().as_ref(), key.as_ref()) {
///         Ok(_) => {}
///         Err(_) => {
///             rollback(b"could not set state", -1);
///         }
///     };
/// }
/// ```

#[inline(always)]
pub fn state_set(data: &[u8], key: &[u8]) -> Result<u64> {
    unsafe {
        c::state_set(
            data.as_ptr() as u32,
            data.len() as u32,
            key.as_ptr() as u32,
            key.len() as u32,
        )
    }
    .into()
}

/// Retrieve the data pointed to, on another account, by a Hook State key and write it to an output buffer
#[inline(always)]
pub fn state_foreign(data: &mut [u8], key: &[u8], namespace: &[u8], accid: &[u8]) -> Result<u64> {
    let res = unsafe {
        c::state_foreign(
            data.as_mut_ptr() as u32,
            data.len() as u32,
            key.as_ptr() as u32,
            key.len() as u32,
            namespace.as_ptr() as u32,
            namespace.len() as u32,
            accid.as_ptr() as u32,
            accid.len() as u32,
        )
    };

    res.into()
}