sui_types/
randomness_state.rs

1// Copyright (c) Mysten Labs, Inc.
2// SPDX-License-Identifier: Apache-2.0
3
4use move_binary_format::{CompiledModule, file_format::SignatureToken};
5use move_bytecode_utils::resolve_struct;
6use move_core_types::{account_address::AccountAddress, ident_str, identifier::IdentStr};
7
8use crate::base_types::SequenceNumber;
9
10use crate::{
11    SUI_FRAMEWORK_ADDRESS, SUI_RANDOMNESS_STATE_OBJECT_ID, error::SuiResult, object::Owner,
12    storage::ObjectStore,
13};
14
15pub const RANDOMNESS_MODULE_NAME: &IdentStr = ident_str!("random");
16pub const RANDOMNESS_STATE_STRUCT_NAME: &IdentStr = ident_str!("Random");
17pub const RANDOMNESS_STATE_UPDATE_FUNCTION_NAME: &IdentStr = ident_str!("update_randomness_state");
18pub const RANDOMNESS_STATE_CREATE_FUNCTION_NAME: &IdentStr = ident_str!("create");
19pub const RESOLVED_SUI_RANDOMNESS_STATE: (&AccountAddress, &IdentStr, &IdentStr) = (
20    &SUI_FRAMEWORK_ADDRESS,
21    RANDOMNESS_MODULE_NAME,
22    RANDOMNESS_STATE_STRUCT_NAME,
23);
24
25pub fn get_randomness_state_obj_initial_shared_version(
26    object_store: &dyn ObjectStore,
27) -> SuiResult<Option<SequenceNumber>> {
28    Ok(object_store
29        .get_object(&SUI_RANDOMNESS_STATE_OBJECT_ID)
30        .map(|obj| match obj.owner {
31            Owner::Shared {
32                initial_shared_version,
33            } => initial_shared_version,
34            _ => unreachable!("Randomness state object must be shared"),
35        }))
36}
37
38pub fn is_mutable_random(view: &CompiledModule, s: &SignatureToken) -> bool {
39    match s {
40        SignatureToken::MutableReference(inner) => is_mutable_random(view, inner),
41        SignatureToken::Datatype(idx) => {
42            resolve_struct(view, *idx) == RESOLVED_SUI_RANDOMNESS_STATE
43        }
44        _ => false,
45    }
46}