ckb-script 1.1.0

CKB component to run the type/lock scripts
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
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

use crate::{
    types::{
        DataPieceId, ScriptGroup, ScriptGroupType, ScriptVersion, SgData, SgInfo, TxData, TxInfo,
    },
    verify_env::TxVerifyEnv,
};
use ckb_chain_spec::consensus::ConsensusBuilder;
use ckb_traits::{CellDataProvider, ExtensionProvider, HeaderProvider};
use ckb_types::{
    bytes::Bytes,
    core::{
        Capacity, HeaderBuilder, HeaderView,
        cell::{CellMeta, ResolvedTransaction},
    },
    packed::{self, Byte32, CellOutput, OutPoint, Script},
    prelude::*,
};
use std::collections::{BTreeMap, HashMap};
use std::sync::Arc;

#[derive(Default, Clone)]
pub(crate) struct MockDataLoader {
    pub(crate) headers: HashMap<Byte32, HeaderView>,
    pub(crate) extensions: HashMap<Byte32, packed::Bytes>,
}

impl CellDataProvider for MockDataLoader {
    fn get_cell_data(&self, _out_point: &OutPoint) -> Option<Bytes> {
        None
    }

    fn get_cell_data_hash(&self, _out_point: &OutPoint) -> Option<Byte32> {
        None
    }
}

impl HeaderProvider for MockDataLoader {
    fn get_header(&self, block_hash: &Byte32) -> Option<HeaderView> {
        self.headers.get(block_hash).cloned()
    }
}

impl ExtensionProvider for MockDataLoader {
    fn get_block_extension(&self, hash: &Byte32) -> Option<packed::Bytes> {
        self.extensions.get(hash).cloned()
    }
}

pub(crate) fn new_mock_data_loader() -> MockDataLoader {
    MockDataLoader::default()
}

pub(crate) fn build_cell_meta(capacity_bytes: usize, data: Bytes) -> CellMeta {
    let capacity = Capacity::bytes(capacity_bytes).expect("capacity bytes overflow");
    let builder = CellOutput::new_builder().capacity(capacity);
    let data_hash = CellOutput::calc_data_hash(&data);
    CellMeta {
        out_point: OutPoint::default(),
        transaction_info: None,
        cell_output: builder.build(),
        data_bytes: data.len() as u64,
        mem_cell_data: Some(data),
        mem_cell_data_hash: Some(data_hash),
    }
}

fn build_tx_data_with_loader(
    rtx: Arc<ResolvedTransaction>,
    data_loader: MockDataLoader,
) -> TxData<MockDataLoader> {
    let consensus = ConsensusBuilder::default().build();
    let tx_env = TxVerifyEnv::new_commit(&HeaderBuilder::default().build());

    TxData {
        rtx,
        info: Arc::new(TxInfo {
            data_loader,
            consensus: Arc::new(consensus),
            tx_env: Arc::new(tx_env),
            binaries_by_data_hash: HashMap::default(),
            binaries_by_type_hash: HashMap::default(),
            lock_groups: BTreeMap::default(),
            type_groups: BTreeMap::default(),
            outputs: Vec::new(),
        }),
    }
}

pub(crate) fn build_sg_data(
    rtx: Arc<ResolvedTransaction>,
    input_indices: Vec<usize>,
    output_indices: Vec<usize>,
) -> SgData<MockDataLoader> {
    build_sg_data_with_loader(rtx, new_mock_data_loader(), input_indices, output_indices)
}

pub(crate) fn build_sg_data_with_loader(
    rtx: Arc<ResolvedTransaction>,
    data_loader: MockDataLoader,
    input_indices: Vec<usize>,
    output_indices: Vec<usize>,
) -> SgData<MockDataLoader> {
    let tx_data = build_tx_data_with_loader(rtx, data_loader);
    let script_group = ScriptGroup {
        script: Script::default(),
        group_type: ScriptGroupType::Lock,
        input_indices,
        output_indices,
    };
    let script_hash = script_group.script.calc_script_hash();
    SgData {
        rtx: tx_data.rtx,
        tx_info: tx_data.info,
        sg_info: Arc::new(SgInfo {
            script_version: ScriptVersion::latest(),
            script_group,
            script_hash,
            program_data_piece_id: DataPieceId::CellDep(0),
        }),
    }
}

pub(crate) fn update_tx_info<F: Fn(&mut TxInfo<MockDataLoader>)>(
    mut sg_data: SgData<MockDataLoader>,
    f: F,
) -> SgData<MockDataLoader> {
    let mut tx_info = sg_data.tx_info.as_ref().clone();
    f(&mut tx_info);
    sg_data.tx_info = Arc::new(tx_info);
    sg_data
}

pub(crate) fn update_sg_info<F: Fn(&mut SgInfo)>(
    mut sg_data: SgData<MockDataLoader>,
    f: F,
) -> SgData<MockDataLoader> {
    let mut sg_info = sg_data.sg_info.as_ref().clone();
    f(&mut sg_info);
    sg_data.sg_info = Arc::new(sg_info);
    sg_data
}