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
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
use crate::error::{Error, Result};
use crate::model::{AssetId, Base64String, ByteString};
use crate::util::JsonDeserializer;
use crate::waves_proto::SetAssetScriptTransactionData;
use serde_json::{Map, Value};

const TYPE: u8 = 15;

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

impl SetAssetScriptTransactionInfo {
    pub fn new(asset_id: AssetId, script: Base64String) -> Self {
        Self { asset_id, script }
    }

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

    pub fn asset_id(&self) -> AssetId {
        self.asset_id.clone()
    }
}

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

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

        Ok(SetAssetScriptTransactionInfo {
            asset_id: AssetId::from_string(&asset_id)?,
            script: Base64String::from_string(&script)?,
        })
    }
}

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

impl SetAssetScriptTransaction {
    pub fn new(asset_id: AssetId, script: Base64String) -> Self {
        Self { asset_id, script }
    }

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

    pub fn asset_id(&self) -> AssetId {
        self.asset_id.clone()
    }

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

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

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

        Ok(SetAssetScriptTransaction {
            asset_id: AssetId::from_string(&asset_id)?,
            script: Base64String::from_string(&script)?,
        })
    }
}

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

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

impl TryFrom<&SetAssetScriptTransaction> for SetAssetScriptTransactionData {
    type Error = Error;

    fn try_from(value: &SetAssetScriptTransaction) -> Result<Self> {
        Ok(SetAssetScriptTransactionData {
            asset_id: value.asset_id.bytes(),
            script: value.script.bytes(),
        })
    }
}

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

    const COMPILED_ASSET_SCRIPT: &str = "base64:AgQAAAAHbWFzdGVyMQkBAAAAEWFkZHJlc3NGcm9tU3RyaW5nAAAAAQIAAAAQMzMzbWFzdGVyQWRkcmVzcwQAAAAHJG1hdGNoMAUAAAACdHgDCQAAAQAAAAIFAAAAByRtYXRjaDACAAAAE1RyYW5zZmVyVHJhbnNhY3Rpb24EAAAAAXQFAAAAByRtYXRjaDADCQAAAAAAAAIIBQAAAAF0AAAABnNlbmRlcgUAAAAHbWFzdGVyMQYJAAAAAAAAAggFAAAAAXQAAAAJcmVjaXBpZW50BQAAAAdtYXN0ZXIxAwkAAAEAAAACBQAAAAckbWF0Y2gwAgAAABdNYXNzVHJhbnNmZXJUcmFuc2FjdGlvbgQAAAACbXQFAAAAByRtYXRjaDAJAAAAAAAAAggFAAAAAm10AAAABnNlbmRlcgUAAAAHbWFzdGVyMQMJAAABAAAAAgUAAAAHJG1hdGNoMAIAAAATRXhjaGFuZ2VUcmFuc2FjdGlvbgcGFLbwIw==";

    #[test]
    fn test_json_to_set_asset_script_transaction() {
        let data = fs::read_to_string("./tests/resources/set_asset_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: SetAssetScriptTransaction = json.borrow().try_into().unwrap();

        assert_eq!(
            "CVwsbXjXmdYF2q4RCPuQKf7sLGpzhk7BNnYsxGZZJMym",
            set_script_from_json.asset_id.encoded()
        );

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

    #[test]
    fn test_set_asset_script_to_proto() -> Result<()> {
        let set_asset_script_tx = &SetAssetScriptTransaction::new(
            AssetId::from_string("CVwsbXjXmdYF2q4RCPuQKf7sLGpzhk7BNnYsxGZZJMym")?,
            Base64String::from_string(COMPILED_ASSET_SCRIPT)?,
        );
        let proto: SetAssetScriptTransactionData = set_asset_script_tx.try_into()?;

        assert_eq!(proto.asset_id, set_asset_script_tx.asset_id().bytes());
        assert_eq!(proto.script, set_asset_script_tx.script().bytes());

        Ok(())
    }

    #[test]
    fn test_set_asset_script_to_json() -> Result<()> {
        let set_asset_script_tx = &SetAssetScriptTransaction::new(
            AssetId::from_string("CVwsbXjXmdYF2q4RCPuQKf7sLGpzhk7BNnYsxGZZJMym")?,
            Base64String::from_string(COMPILED_ASSET_SCRIPT)?,
        );

        let map: Map<String, Value> = set_asset_script_tx.try_into()?;
        let json: Value = map.into();
        let expected_json = json!({
            "assetId": "CVwsbXjXmdYF2q4RCPuQKf7sLGpzhk7BNnYsxGZZJMym",
            "script": "AgQAAAAHbWFzdGVyMQkBAAAAEWFkZHJlc3NGcm9tU3RyaW5nAAAAAQIAAAAQMzMzbWFzdGVyQWRkcmVzcwQAAAAHJG1hdGNoMAUAAAACdHgDCQAAAQAAAAIFAAAAByRtYXRjaDACAAAAE1RyYW5zZmVyVHJhbnNhY3Rpb24EAAAAAXQFAAAAByRtYXRjaDADCQAAAAAAAAIIBQAAAAF0AAAABnNlbmRlcgUAAAAHbWFzdGVyMQYJAAAAAAAAAggFAAAAAXQAAAAJcmVjaXBpZW50BQAAAAdtYXN0ZXIxAwkAAAEAAAACBQAAAAckbWF0Y2gwAgAAABdNYXNzVHJhbnNmZXJUcmFuc2FjdGlvbgQAAAACbXQFAAAAByRtYXRjaDAJAAAAAAAAAggFAAAAAm10AAAABnNlbmRlcgUAAAAHbWFzdGVyMQMJAAABAAAAAgUAAAAHJG1hdGNoMAIAAAATRXhjaGFuZ2VUcmFuc2FjdGlvbgcGFLbwIw=="
        });
        assert_eq!(expected_json, json);
        Ok(())
    }
}