use core::borrow::Borrow;
use crate::double_key;
use fuel_storage::Mappable;
use fuel_types::{
Bytes32,
ContractId,
};
use alloc::vec::Vec;
use educe::Educe;
use fuel_types::bytes::Bytes;
#[cfg(feature = "random")]
use rand::{
Rng,
distributions::{
Distribution,
Standard,
},
};
pub struct ContractsState;
impl Mappable for ContractsState {
type Key = Self::OwnedKey;
type OwnedKey = ContractsStateKey;
type OwnedValue = ContractsStateData;
type Value = [u8];
}
double_key!(
ContractsStateKey,
ContractId,
contract_id,
Bytes32,
state_key
);
#[derive(Educe, Clone, PartialEq, Eq, Hash)]
#[educe(Debug)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct ContractsStateData(pub Bytes);
impl From<Vec<u8>> for ContractsStateData {
fn from(c: Vec<u8>) -> Self {
Self(c.into())
}
}
impl From<ContractsStateData> for Vec<u8> {
fn from(c: ContractsStateData) -> Vec<u8> {
c.0.into_inner()
}
}
impl From<&[u8]> for ContractsStateData {
fn from(c: &[u8]) -> Self {
Self(c.to_vec().into())
}
}
impl From<&mut [u8]> for ContractsStateData {
fn from(c: &mut [u8]) -> Self {
Self(c.to_vec().into())
}
}
impl Borrow<[u8]> for ContractsStateData {
fn borrow(&self) -> &[u8] {
&self.0
}
}
impl AsRef<[u8]> for ContractsStateData {
fn as_ref(&self) -> &[u8] {
self.0.as_ref()
}
}
impl AsMut<[u8]> for ContractsStateData {
fn as_mut(&mut self) -> &mut [u8] {
self.0.as_mut()
}
}
#[cfg(feature = "random")]
impl Distribution<ContractsStateData> for Standard {
fn sample<R: Rng + ?Sized>(&self, rng: &mut R) -> ContractsStateData {
ContractsStateData(rng.r#gen::<Bytes32>().to_vec().into())
}
}
impl IntoIterator for ContractsStateData {
type IntoIter = alloc::vec::IntoIter<Self::Item>;
type Item = u8;
fn into_iter(self) -> Self::IntoIter {
self.0.into_inner().into_iter()
}
}