1use aluvm::LibSite;
25use amplify::confinement::{SmallBlob, U24 as U24MAX};
26use strict_encoding::StreamReader;
27use strict_types::{SemId, StrictVal, TypeSystem};
28use ultrasonic::RawData;
29
30use crate::{StateBuildError, StateConvertError, LIB_NAME_SONIC};
31
32pub const TOTAL_RAW_BYTES: usize = U24MAX;
33
34#[derive(Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug)]
35#[derive(StrictType, StrictDumb, StrictEncode, StrictDecode)]
36#[strict_type(lib = LIB_NAME_SONIC, tags = custom, dumb = Self::StrictDecode(strict_dumb!()))]
37#[cfg_attr(feature = "serde", derive(Serialize, Deserialize), serde(rename_all = "camelCase"))]
38pub enum RawConvertor {
39 #[strict_type(tag = 0x00)]
41 StrictDecode(SemId),
42 #[strict_type(tag = 0xFF)]
48 AluVM(
49 LibSite,
52 ),
53}
54
55impl RawConvertor {
56 pub fn convert(&self, raw: &RawData, sys: &TypeSystem) -> Result<StrictVal, StateConvertError> {
57 match self {
58 Self::StrictDecode(sem_id) => strict_convert(*sem_id, raw, sys),
59 Self::AluVM(_) => Err(StateConvertError::Unsupported),
60 }
61 }
62}
63
64#[derive(Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug)]
65#[derive(StrictType, StrictDumb, StrictEncode, StrictDecode)]
66#[strict_type(lib = LIB_NAME_SONIC, tags = custom, dumb = Self::StrictEncode(strict_dumb!()))]
67#[cfg_attr(feature = "serde", derive(Serialize, Deserialize), serde(rename_all = "camelCase"))]
68pub enum RawBuilder {
69 #[strict_type(tag = 0x00)]
71 StrictEncode(SemId),
72
73 #[strict_type(tag = 0xFF)]
76 AluVM(
77 LibSite,
80 ),
81}
82
83impl RawBuilder {
84 #[allow(clippy::result_large_err)]
85 pub fn build(&self, val: StrictVal, sys: &TypeSystem) -> Result<RawData, StateBuildError> {
86 match self {
87 Self::StrictEncode(sem_id) => strict_build(*sem_id, val, sys),
88 Self::AluVM(_) => Err(StateBuildError::Unsupported),
89 }
90 }
91}
92
93fn strict_convert(sem_id: SemId, raw: &RawData, sys: &TypeSystem) -> Result<StrictVal, StateConvertError> {
94 let mut reader = StreamReader::cursor::<TOTAL_RAW_BYTES>(&raw[..]);
95 let mut val = sys.strict_read_type(sem_id, &mut reader)?.unbox();
96
97 if reader.into_cursor().position() != raw[..].len() as u64 {
98 return Err(StateConvertError::NotEntirelyConsumed);
99 }
100
101 loop {
102 if let StrictVal::Tuple(ref mut vec) = val {
103 if vec.len() == 1 {
104 val = vec.remove(0);
105 continue;
106 }
107 }
108 break;
109 }
110
111 Ok(val)
112}
113
114#[allow(clippy::result_large_err)]
115fn strict_build(sem_id: SemId, val: StrictVal, sys: &TypeSystem) -> Result<RawData, StateBuildError> {
116 let mut data = SmallBlob::new();
117
118 let typed_val = sys.typify(val, sem_id)?;
119 sys.strict_write_value(&typed_val, &mut data)?;
120
121 Ok(RawData::from(data))
122}