waves_rust/model/transaction/
set_script_transaction.rs

1use crate::error::{Error, Result};
2use crate::model::{Base64String, ByteString};
3use crate::util::JsonDeserializer;
4use crate::waves_proto::SetScriptTransactionData;
5use serde_json::{Map, Value};
6
7const TYPE: u8 = 13;
8
9#[derive(Clone, Eq, PartialEq, Debug)]
10pub struct SetScriptTransactionInfo {
11    script: Base64String,
12}
13
14impl SetScriptTransactionInfo {
15    pub fn new(script: Base64String) -> Self {
16        Self { script }
17    }
18
19    pub fn script(&self) -> Base64String {
20        self.script.clone()
21    }
22}
23
24impl TryFrom<&Value> for SetScriptTransactionInfo {
25    type Error = Error;
26
27    fn try_from(value: &Value) -> Result<Self> {
28        let script = JsonDeserializer::safe_to_string_from_field(value, "script")?;
29
30        Ok(SetScriptTransactionInfo {
31            script: Base64String::from_string(&script)?,
32        })
33    }
34}
35
36#[derive(Clone, Eq, PartialEq, Debug)]
37pub struct SetScriptTransaction {
38    script: Base64String,
39}
40
41impl SetScriptTransaction {
42    pub fn new(script: Base64String) -> Self {
43        Self { script }
44    }
45
46    pub fn script(&self) -> Base64String {
47        self.script.clone()
48    }
49
50    pub fn tx_type() -> u8 {
51        TYPE
52    }
53}
54
55impl TryFrom<&Value> for SetScriptTransaction {
56    type Error = Error;
57
58    fn try_from(value: &Value) -> Result<Self> {
59        let script = JsonDeserializer::safe_to_string_from_field(value, "script")?;
60
61        Ok(SetScriptTransaction {
62            script: Base64String::from_string(&script)?,
63        })
64    }
65}
66
67impl TryFrom<&SetScriptTransaction> for Map<String, Value> {
68    type Error = Error;
69
70    fn try_from(value: &SetScriptTransaction) -> Result<Self> {
71        let mut set_script_tx_json = Map::new();
72        set_script_tx_json.insert("script".to_owned(), value.script.encoded().into());
73        Ok(set_script_tx_json)
74    }
75}
76
77impl TryFrom<&SetScriptTransaction> for SetScriptTransactionData {
78    type Error = Error;
79
80    fn try_from(value: &SetScriptTransaction) -> Result<Self> {
81        Ok(SetScriptTransactionData {
82            script: value.script.bytes(),
83        })
84    }
85}
86
87#[cfg(test)]
88mod tests {
89    use crate::error::Result;
90    use crate::model::{Base64String, ByteString, SetScriptTransaction, SetScriptTransactionInfo};
91    use crate::waves_proto::SetScriptTransactionData;
92    use serde_json::{json, Map, Value};
93    use std::borrow::Borrow;
94    use std::fs;
95
96    const COMPILED_SCRIPT: &str = "base64:AAIFAAAAAAAAAAsIAhIHCgUCBAEIEQAAAAAAAAABAAAAA2ludgEAAAAEY2FsbAAAAAUAAAACYnYAAAABYgAAAANpbnQAAAADc3RyAAAABGxpc3QEAAAABWFzc2V0CQAEQgAAAAUCAAAABUFzc2V0AgAAAAAAAAAAAAAAAAEAAAAAAAAAAAAGBAAAAAdhc3NldElkCQAEOAAAAAEFAAAABWFzc2V0BAAAAAVsZWFzZQkABEQAAAACCAUAAAADaW52AAAABmNhbGxlcgAAAAAAAAAABwQAAAAHbGVhc2VJZAkABDkAAAABBQAAAAVsZWFzZQkABEwAAAACCQEAAAALQmluYXJ5RW50cnkAAAACAgAAAANiaW4FAAAAB2Fzc2V0SWQJAARMAAAAAgkBAAAADEJvb2xlYW5FbnRyeQAAAAICAAAABGJvb2wGCQAETAAAAAIJAQAAAAxJbnRlZ2VyRW50cnkAAAACAgAAAANpbnQAAAAAAAABiJQJAARMAAAAAgkBAAAAC1N0cmluZ0VudHJ5AAAAAgIAAAAHYXNzZXRJZAkAAlgAAAABBQAAAAdhc3NldElkCQAETAAAAAIJAQAAAAtTdHJpbmdFbnRyeQAAAAICAAAAB2xlYXNlSWQJAAJYAAAAAQUAAAAHbGVhc2VJZAkABEwAAAACCQEAAAALU3RyaW5nRW50cnkAAAACAgAAAANkZWwCAAAAAAkABEwAAAACCQEAAAALRGVsZXRlRW50cnkAAAABAgAAAANkZWwJAARMAAAAAgUAAAAFYXNzZXQJAARMAAAAAgkBAAAAClNwb25zb3JGZWUAAAACBQAAAAdhc3NldElkAAAAAAAAAAABCQAETAAAAAIJAQAAAAdSZWlzc3VlAAAAAwUAAAAHYXNzZXRJZAAAAAAAAAAABAcJAARMAAAAAgkBAAAABEJ1cm4AAAACBQAAAAdhc3NldElkAAAAAAAAAAADCQAETAAAAAIJAQAAAA5TY3JpcHRUcmFuc2ZlcgAAAAMIBQAAAANpbnYAAAAGY2FsbGVyAAAAAAAAAAACBQAAAAdhc3NldElkCQAETAAAAAIFAAAABWxlYXNlCQAETAAAAAIJAQAAAAtMZWFzZUNhbmNlbAAAAAEJAAQ5AAAAAQUAAAAFbGVhc2UFAAAAA25pbAAAAAD/oHwO";
97
98    #[test]
99    fn test_json_to_set_script_transaction() {
100        let data = fs::read_to_string("./tests/resources/set_script_rs.json")
101            .expect("Unable to read file");
102        let json: Value = serde_json::from_str(&data).expect("failed to generate json from str");
103
104        let set_script_from_json: SetScriptTransactionInfo = json.borrow().try_into().unwrap();
105
106        assert_eq!(
107            set_script_from_json.script().encoded_with_prefix(),
108            COMPILED_SCRIPT
109        );
110    }
111
112    #[test]
113    fn test_set_script_to_proto() -> Result<()> {
114        let set_script_tx = &SetScriptTransaction::new(Base64String::from_string(COMPILED_SCRIPT)?);
115        let proto: SetScriptTransactionData = set_script_tx.try_into()?;
116
117        assert_eq!(proto.script, set_script_tx.script().bytes());
118
119        Ok(())
120    }
121
122    #[test]
123    fn test_set_script_to_json() -> Result<()> {
124        let set_script_tx = &SetScriptTransaction::new(Base64String::from_string(COMPILED_SCRIPT)?);
125
126        let map: Map<String, Value> = set_script_tx.try_into()?;
127        let json: Value = map.into();
128        let expected_json = json!({
129            "script": "AAIFAAAAAAAAAAsIAhIHCgUCBAEIEQAAAAAAAAABAAAAA2ludgEAAAAEY2FsbAAAAAUAAAACYnYAAAABYgAAAANpbnQAAAADc3RyAAAABGxpc3QEAAAABWFzc2V0CQAEQgAAAAUCAAAABUFzc2V0AgAAAAAAAAAAAAAAAAEAAAAAAAAAAAAGBAAAAAdhc3NldElkCQAEOAAAAAEFAAAABWFzc2V0BAAAAAVsZWFzZQkABEQAAAACCAUAAAADaW52AAAABmNhbGxlcgAAAAAAAAAABwQAAAAHbGVhc2VJZAkABDkAAAABBQAAAAVsZWFzZQkABEwAAAACCQEAAAALQmluYXJ5RW50cnkAAAACAgAAAANiaW4FAAAAB2Fzc2V0SWQJAARMAAAAAgkBAAAADEJvb2xlYW5FbnRyeQAAAAICAAAABGJvb2wGCQAETAAAAAIJAQAAAAxJbnRlZ2VyRW50cnkAAAACAgAAAANpbnQAAAAAAAABiJQJAARMAAAAAgkBAAAAC1N0cmluZ0VudHJ5AAAAAgIAAAAHYXNzZXRJZAkAAlgAAAABBQAAAAdhc3NldElkCQAETAAAAAIJAQAAAAtTdHJpbmdFbnRyeQAAAAICAAAAB2xlYXNlSWQJAAJYAAAAAQUAAAAHbGVhc2VJZAkABEwAAAACCQEAAAALU3RyaW5nRW50cnkAAAACAgAAAANkZWwCAAAAAAkABEwAAAACCQEAAAALRGVsZXRlRW50cnkAAAABAgAAAANkZWwJAARMAAAAAgUAAAAFYXNzZXQJAARMAAAAAgkBAAAAClNwb25zb3JGZWUAAAACBQAAAAdhc3NldElkAAAAAAAAAAABCQAETAAAAAIJAQAAAAdSZWlzc3VlAAAAAwUAAAAHYXNzZXRJZAAAAAAAAAAABAcJAARMAAAAAgkBAAAABEJ1cm4AAAACBQAAAAdhc3NldElkAAAAAAAAAAADCQAETAAAAAIJAQAAAA5TY3JpcHRUcmFuc2ZlcgAAAAMIBQAAAANpbnYAAAAGY2FsbGVyAAAAAAAAAAACBQAAAAdhc3NldElkCQAETAAAAAIFAAAABWxlYXNlCQAETAAAAAIJAQAAAAtMZWFzZUNhbmNlbAAAAAEJAAQ5AAAAAQUAAAAFbGVhc2UFAAAAA25pbAAAAAD/oHwO"
130        });
131        assert_eq!(expected_json, json);
132        Ok(())
133    }
134}