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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134

use crate::error::{Error, Result};
use crate::model::{Base64String, ByteString};
use crate::util::JsonDeserializer;
use crate::waves_proto::SetScriptTransactionData;
use serde_json::{Map, Value};

const TYPE: u8 = 13;

#[derive(Clone, Eq, PartialEq, Debug)]
pub struct SetScriptTransactionInfo {
    script: Base64String,
}

impl SetScriptTransactionInfo {
    pub fn new(script: Base64String) -> Self {
        Self { script }
    }

    pub fn script(&self) -> Base64String {
        self.script.clone()
    }
}

impl TryFrom<&Value> for SetScriptTransactionInfo {
    type Error = Error;

    fn try_from(value: &Value) -> Result<Self> {
        let script = JsonDeserializer::safe_to_string_from_field(value, "script")?;

        Ok(SetScriptTransactionInfo {
            script: Base64String::from_string(&script)?,
        })
    }
}

#[derive(Clone, Eq, PartialEq, Debug)]
pub struct SetScriptTransaction {
    script: Base64String,
}

impl SetScriptTransaction {
    pub fn new(script: Base64String) -> Self {
        Self { script }
    }

    pub fn script(&self) -> Base64String {
        self.script.clone()
    }

    pub fn tx_type() -> u8 {
        TYPE
    }
}

impl TryFrom<&Value> for SetScriptTransaction {
    type Error = Error;

    fn try_from(value: &Value) -> Result<Self> {
        let script = JsonDeserializer::safe_to_string_from_field(value, "script")?;

        Ok(SetScriptTransaction {
            script: Base64String::from_string(&script)?,
        })
    }
}

impl TryFrom<&SetScriptTransaction> for Map<String, Value> {
    type Error = Error;

    fn try_from(value: &SetScriptTransaction) -> Result<Self> {
        let mut set_script_tx_json = Map::new();
        set_script_tx_json.insert("script".to_owned(), value.script.encoded().into());
        Ok(set_script_tx_json)
    }
}

impl TryFrom<&SetScriptTransaction> for SetScriptTransactionData {
    type Error = Error;

    fn try_from(value: &SetScriptTransaction) -> Result<Self> {
        Ok(SetScriptTransactionData {
            script: value.script.bytes(),
        })
    }
}

#[cfg(test)]
mod tests {
    use crate::error::Result;
    use crate::model::{Base64String, ByteString, SetScriptTransaction, SetScriptTransactionInfo};
    use crate::waves_proto::SetScriptTransactionData;
    use serde_json::{json, Map, Value};
    use std::borrow::Borrow;
    use std::fs;

    const COMPILED_SCRIPT: &str = "base64:AAIFAAAAAAAAAAsIAhIHCgUCBAEIEQAAAAAAAAABAAAAA2ludgEAAAAEY2FsbAAAAAUAAAACYnYAAAABYgAAAANpbnQAAAADc3RyAAAABGxpc3QEAAAABWFzc2V0CQAEQgAAAAUCAAAABUFzc2V0AgAAAAAAAAAAAAAAAAEAAAAAAAAAAAAGBAAAAAdhc3NldElkCQAEOAAAAAEFAAAABWFzc2V0BAAAAAVsZWFzZQkABEQAAAACCAUAAAADaW52AAAABmNhbGxlcgAAAAAAAAAABwQAAAAHbGVhc2VJZAkABDkAAAABBQAAAAVsZWFzZQkABEwAAAACCQEAAAALQmluYXJ5RW50cnkAAAACAgAAAANiaW4FAAAAB2Fzc2V0SWQJAARMAAAAAgkBAAAADEJvb2xlYW5FbnRyeQAAAAICAAAABGJvb2wGCQAETAAAAAIJAQAAAAxJbnRlZ2VyRW50cnkAAAACAgAAAANpbnQAAAAAAAABiJQJAARMAAAAAgkBAAAAC1N0cmluZ0VudHJ5AAAAAgIAAAAHYXNzZXRJZAkAAlgAAAABBQAAAAdhc3NldElkCQAETAAAAAIJAQAAAAtTdHJpbmdFbnRyeQAAAAICAAAAB2xlYXNlSWQJAAJYAAAAAQUAAAAHbGVhc2VJZAkABEwAAAACCQEAAAALU3RyaW5nRW50cnkAAAACAgAAAANkZWwCAAAAAAkABEwAAAACCQEAAAALRGVsZXRlRW50cnkAAAABAgAAAANkZWwJAARMAAAAAgUAAAAFYXNzZXQJAARMAAAAAgkBAAAAClNwb25zb3JGZWUAAAACBQAAAAdhc3NldElkAAAAAAAAAAABCQAETAAAAAIJAQAAAAdSZWlzc3VlAAAAAwUAAAAHYXNzZXRJZAAAAAAAAAAABAcJAARMAAAAAgkBAAAABEJ1cm4AAAACBQAAAAdhc3NldElkAAAAAAAAAAADCQAETAAAAAIJAQAAAA5TY3JpcHRUcmFuc2ZlcgAAAAMIBQAAAANpbnYAAAAGY2FsbGVyAAAAAAAAAAACBQAAAAdhc3NldElkCQAETAAAAAIFAAAABWxlYXNlCQAETAAAAAIJAQAAAAtMZWFzZUNhbmNlbAAAAAEJAAQ5AAAAAQUAAAAFbGVhc2UFAAAAA25pbAAAAAD/oHwO";

    #[test]
    fn test_json_to_set_script_transaction() {
        let data = fs::read_to_string("./tests/resources/set_script_rs.json")
            .expect("Unable to read file");
        let json: Value = serde_json::from_str(&data).expect("failed to generate json from str");

        let set_script_from_json: SetScriptTransactionInfo = json.borrow().try_into().unwrap();

        assert_eq!(
            set_script_from_json.script().encoded_with_prefix(),
            COMPILED_SCRIPT
        );
    }

    #[test]
    fn test_set_script_to_proto() -> Result<()> {
        let set_script_tx = &SetScriptTransaction::new(Base64String::from_string(COMPILED_SCRIPT)?);
        let proto: SetScriptTransactionData = set_script_tx.try_into()?;

        assert_eq!(proto.script, set_script_tx.script().bytes());

        Ok(())
    }

    #[test]
    fn test_set_script_to_json() -> Result<()> {
        let set_script_tx = &SetScriptTransaction::new(Base64String::from_string(COMPILED_SCRIPT)?);

        let map: Map<String, Value> = set_script_tx.try_into()?;
        let json: Value = map.into();
        let expected_json = json!({
            "script": "AAIFAAAAAAAAAAsIAhIHCgUCBAEIEQAAAAAAAAABAAAAA2ludgEAAAAEY2FsbAAAAAUAAAACYnYAAAABYgAAAANpbnQAAAADc3RyAAAABGxpc3QEAAAABWFzc2V0CQAEQgAAAAUCAAAABUFzc2V0AgAAAAAAAAAAAAAAAAEAAAAAAAAAAAAGBAAAAAdhc3NldElkCQAEOAAAAAEFAAAABWFzc2V0BAAAAAVsZWFzZQkABEQAAAACCAUAAAADaW52AAAABmNhbGxlcgAAAAAAAAAABwQAAAAHbGVhc2VJZAkABDkAAAABBQAAAAVsZWFzZQkABEwAAAACCQEAAAALQmluYXJ5RW50cnkAAAACAgAAAANiaW4FAAAAB2Fzc2V0SWQJAARMAAAAAgkBAAAADEJvb2xlYW5FbnRyeQAAAAICAAAABGJvb2wGCQAETAAAAAIJAQAAAAxJbnRlZ2VyRW50cnkAAAACAgAAAANpbnQAAAAAAAABiJQJAARMAAAAAgkBAAAAC1N0cmluZ0VudHJ5AAAAAgIAAAAHYXNzZXRJZAkAAlgAAAABBQAAAAdhc3NldElkCQAETAAAAAIJAQAAAAtTdHJpbmdFbnRyeQAAAAICAAAAB2xlYXNlSWQJAAJYAAAAAQUAAAAHbGVhc2VJZAkABEwAAAACCQEAAAALU3RyaW5nRW50cnkAAAACAgAAAANkZWwCAAAAAAkABEwAAAACCQEAAAALRGVsZXRlRW50cnkAAAABAgAAAANkZWwJAARMAAAAAgUAAAAFYXNzZXQJAARMAAAAAgkBAAAAClNwb25zb3JGZWUAAAACBQAAAAdhc3NldElkAAAAAAAAAAABCQAETAAAAAIJAQAAAAdSZWlzc3VlAAAAAwUAAAAHYXNzZXRJZAAAAAAAAAAABAcJAARMAAAAAgkBAAAABEJ1cm4AAAACBQAAAAdhc3NldElkAAAAAAAAAAADCQAETAAAAAIJAQAAAA5TY3JpcHRUcmFuc2ZlcgAAAAMIBQAAAANpbnYAAAAGY2FsbGVyAAAAAAAAAAACBQAAAAdhc3NldElkCQAETAAAAAIFAAAABWxlYXNlCQAETAAAAAIJAQAAAAtMZWFzZUNhbmNlbAAAAAEJAAQ5AAAAAQUAAAAFbGVhc2UFAAAAA25pbAAAAAD/oHwO"
        });
        assert_eq!(expected_json, json);
        Ok(())
    }
}