cardano_serialization_lib/builders/script_structs/
native_script_source.rs

1use crate::*;
2
3#[derive(Clone, Debug, Eq, Ord, PartialEq, PartialOrd)]
4pub(crate) enum NativeScriptSourceEnum {
5    NativeScript(NativeScript, Option<RequiredSigners>),
6    RefInput(TransactionInput, ScriptHash, Option<RequiredSigners>, usize),
7}
8
9impl NativeScriptSourceEnum {
10    pub fn script_hash(&self) -> ScriptHash {
11        match self {
12            NativeScriptSourceEnum::NativeScript(script, _) => script.hash(),
13            NativeScriptSourceEnum::RefInput(_, script_hash, _, _) => script_hash.clone(),
14        }
15    }
16
17    pub fn required_signers(&self) -> Option<Ed25519KeyHashes> {
18        match self {
19            NativeScriptSourceEnum::NativeScript(script, required_signers) => {
20                match required_signers {
21                    Some(signers) => Some(signers.clone()),
22                    None => Some(script.into())
23                }
24            }
25            NativeScriptSourceEnum::RefInput(_, _, required_signers, _) => required_signers.clone(),
26        }
27    }
28
29    pub fn set_required_signers(&mut self, key_hashes: &Ed25519KeyHashes) {
30        match self {
31            NativeScriptSourceEnum::NativeScript(_, required_signers) => {
32                *required_signers = Some(key_hashes.clone());
33            }
34            NativeScriptSourceEnum::RefInput(_, _, required_signers, _) => {
35                *required_signers = Some(key_hashes.clone());
36            }
37        }
38    }
39}
40
41#[wasm_bindgen]
42#[derive(Clone, Debug, Eq, Ord, PartialEq, PartialOrd)]
43pub struct NativeScriptSource(pub(crate) NativeScriptSourceEnum);
44
45#[wasm_bindgen]
46impl NativeScriptSource {
47    pub fn new(script: &NativeScript) -> Self {
48        Self(NativeScriptSourceEnum::NativeScript(script.clone(), None))
49    }
50
51    pub fn new_ref_input(
52        script_hash: &ScriptHash,
53        input: &TransactionInput,
54        script_size: usize,
55    ) -> Self {
56        Self(NativeScriptSourceEnum::RefInput(
57            input.clone(),
58            script_hash.clone(),
59            None,
60            script_size
61        ))
62    }
63
64    pub fn set_required_signers(&mut self, key_hashes: &Ed25519KeyHashes) {
65        self.0.set_required_signers(key_hashes)
66    }
67
68    pub(crate) fn script_hash(&self) -> ScriptHash {
69        self.0.script_hash()
70    }
71
72    pub fn get_ref_script_size(&self) -> Option<usize> {
73        match &self.0 {
74            NativeScriptSourceEnum::NativeScript(..) => None,
75            NativeScriptSourceEnum::RefInput(.., size) => Some(*size)
76        }
77    }
78}