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

//! Simple http rpc client for [ckb-indexer](https://github.com/nervosnetwork/ckb-indexer)

use ckb_jsonrpc_types::{
    BlockNumber, Capacity, CellOutput, JsonBytes, OutPoint, Script, Uint32, Uint64,
};
use ckb_types::H256;
use serde::{Deserialize, Serialize};

use crate::traits::{CellQueryOptions, LiveCell, PrimaryScriptType, ValueRangeOption};

#[derive(Serialize, Clone, Debug)]
pub struct SearchKey {
    pub script: Script,
    pub script_type: ScriptType,
    pub filter: Option<SearchKeyFilter>,
}

#[derive(Serialize, Default, Clone, Debug)]
pub struct SearchKeyFilter {
    pub script: Option<Script>,
    pub output_data_len_range: Option<[Uint64; 2]>,
    pub output_capacity_range: Option<[Uint64; 2]>,
    pub block_range: Option<[BlockNumber; 2]>,
}
impl From<CellQueryOptions> for SearchKey {
    fn from(opts: CellQueryOptions) -> SearchKey {
        let convert_range =
            |range: ValueRangeOption| [Uint64::from(range.start), Uint64::from(range.end)];
        let filter = if opts.secondary_script.is_none()
            && opts.data_len_range.is_none()
            && opts.capacity_range.is_none()
            && opts.block_range.is_none()
        {
            None
        } else {
            Some(SearchKeyFilter {
                script: opts.secondary_script.map(|v| v.into()),
                output_data_len_range: opts.data_len_range.map(convert_range),
                output_capacity_range: opts.capacity_range.map(convert_range),
                block_range: opts.block_range.map(convert_range),
            })
        };
        SearchKey {
            script: opts.primary_script.into(),
            script_type: opts.primary_type.into(),
            filter,
        }
    }
}

#[derive(Serialize, Clone, Debug)]
#[serde(rename_all = "snake_case")]
pub enum ScriptType {
    Lock,
    Type,
}
impl From<PrimaryScriptType> for ScriptType {
    fn from(t: PrimaryScriptType) -> ScriptType {
        match t {
            PrimaryScriptType::Lock => ScriptType::Lock,
            PrimaryScriptType::Type => ScriptType::Type,
        }
    }
}

#[derive(Serialize, Clone, Debug)]
#[serde(rename_all = "snake_case")]
pub enum Order {
    Desc,
    Asc,
}

#[derive(Deserialize, Clone, Debug)]
pub struct Tip {
    pub block_hash: H256,
    pub block_number: BlockNumber,
}

#[derive(Deserialize, Clone, Debug)]
pub struct CellsCapacity {
    pub capacity: Capacity,
    pub block_hash: H256,
    pub block_number: BlockNumber,
}

#[derive(Deserialize, Clone, Debug)]
pub struct IndexerInfo {
    pub version: String,
}

#[derive(Deserialize, Clone, Debug)]
pub struct Cell {
    pub output: CellOutput,
    pub output_data: JsonBytes,
    pub out_point: OutPoint,
    pub block_number: BlockNumber,
    pub tx_index: Uint32,
}
impl From<Cell> for LiveCell {
    fn from(cell: Cell) -> LiveCell {
        LiveCell {
            output: cell.output.into(),
            output_data: cell.output_data.into_bytes(),
            out_point: cell.out_point.into(),
            block_number: cell.block_number.value(),
            tx_index: cell.tx_index.value(),
        }
    }
}

#[derive(Deserialize, Clone, Debug)]
pub struct Tx {
    pub tx_hash: H256,
    pub block_number: BlockNumber,
    pub tx_index: Uint32,
    pub io_index: Uint32,
    pub io_type: IOType,
}

#[derive(Deserialize, Clone, Debug)]
#[serde(rename_all = "snake_case")]
pub enum IOType {
    Input,
    Output,
}

#[derive(Deserialize)]
pub struct Pagination<T> {
    pub objects: Vec<T>,
    pub last_cursor: JsonBytes,
}

crate::jsonrpc!(pub struct IndexerRpcClient {
    pub fn get_tip(&mut self) -> Option<Tip>;
    pub fn get_cells(&mut self, search_key: SearchKey, order: Order, limit: Uint32, after: Option<JsonBytes>) -> Pagination<Cell>;
    pub fn get_transactions(&mut self, search_key: SearchKey, order: Order, limit: Uint32, after: Option<JsonBytes>) -> Pagination<Tx>;
    pub fn get_cells_capacity(&mut self, search_key: SearchKey) -> Option<CellsCapacity>;
    pub fn get_indexer_info(&mut self) -> IndexerInfo;
});