chia-sdk-types 0.33.0

Standard Chia types for things such as puzzle info and conditions.
Documentation 
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

use chia_bls::PublicKey;
use chia_protocol::Bytes;
use clvm_traits::{FromClvm, ToClvm};

use super::Condition;

#[derive(Debug, Clone, Copy, PartialEq, Eq, ToClvm, FromClvm, Hash)]
#[repr(u8)]
#[clvm(atom)]
pub enum AggSigKind {
    Parent = 43,
    Puzzle = 44,
    Amount = 45,
    PuzzleAmount = 46,
    ParentAmount = 47,
    ParentPuzzle = 48,
    Unsafe = 49,
    Me = 50,
}

#[derive(Debug, Clone, PartialEq, Eq, ToClvm, FromClvm)]
#[clvm(list)]
pub struct AggSig {
    pub kind: AggSigKind,
    pub public_key: PublicKey,
    pub message: Bytes,
}

impl AggSig {
    pub fn new(kind: AggSigKind, public_key: PublicKey, message: Bytes) -> Self {
        Self {
            kind,
            public_key,
            message,
        }
    }
}

impl<T> Condition<T> {
    pub fn into_agg_sig(self) -> Option<AggSig> {
        match self {
            Condition::AggSigParent(inner) => Some(AggSig::new(
                AggSigKind::Parent,
                inner.public_key,
                inner.message,
            )),
            Condition::AggSigPuzzle(inner) => Some(AggSig::new(
                AggSigKind::Puzzle,
                inner.public_key,
                inner.message,
            )),
            Condition::AggSigAmount(inner) => Some(AggSig::new(
                AggSigKind::Amount,
                inner.public_key,
                inner.message,
            )),
            Condition::AggSigPuzzleAmount(inner) => Some(AggSig::new(
                AggSigKind::PuzzleAmount,
                inner.public_key,
                inner.message,
            )),
            Condition::AggSigParentAmount(inner) => Some(AggSig::new(
                AggSigKind::ParentAmount,
                inner.public_key,
                inner.message,
            )),
            Condition::AggSigParentPuzzle(inner) => Some(AggSig::new(
                AggSigKind::ParentPuzzle,
                inner.public_key,
                inner.message,
            )),
            Condition::AggSigUnsafe(inner) => Some(AggSig::new(
                AggSigKind::Unsafe,
                inner.public_key,
                inner.message,
            )),
            Condition::AggSigMe(inner) => {
                Some(AggSig::new(AggSigKind::Me, inner.public_key, inner.message))
            }
            _ => None,
        }
    }

    pub fn is_agg_sig(&self) -> bool {
        matches!(
            self,
            Condition::AggSigParent(..)
                | Condition::AggSigPuzzle(..)
                | Condition::AggSigAmount(..)
                | Condition::AggSigPuzzleAmount(..)
                | Condition::AggSigParentAmount(..)
                | Condition::AggSigParentPuzzle(..)
                | Condition::AggSigUnsafe(..)
                | Condition::AggSigMe(..)
        )
    }
}