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

use chia_protocol::{Bytes32, Coin};
use clvm_traits::FromClvm;
use clvm_utils::{tree_hash, CurriedProgram};
use clvmr::{Allocator, NodePtr};

use crate::ParseError;

pub struct ParseContext {
    mod_hash: Bytes32,
    args: NodePtr,
    solution: NodePtr,
    parent_coin: Coin,
    coin: Coin,
}

impl ParseContext {
    pub fn mod_hash(&self) -> Bytes32 {
        self.mod_hash
    }

    pub fn args(&self) -> NodePtr {
        self.args
    }

    pub fn solution(&self) -> NodePtr {
        self.solution
    }

    pub fn parent_coin(&self) -> Coin {
        self.parent_coin
    }

    pub fn coin(&self) -> Coin {
        self.coin
    }
}

pub fn parse_puzzle(
    allocator: &mut Allocator,
    parent_puzzle: NodePtr,
    parent_solution: NodePtr,
    parent_coin: Coin,
    coin: Coin,
) -> Result<ParseContext, ParseError> {
    let CurriedProgram { program, args } = CurriedProgram::from_clvm(allocator, parent_puzzle)?;

    Ok(ParseContext {
        mod_hash: tree_hash(allocator, program).into(),
        args,
        solution: parent_solution,
        parent_coin,
        coin,
    })
}