capricorn 0.1.93

Parse html according to configuration
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

use crate::document_selection::DocumentSelection;
use crate::parse::HashMapSelectParams;
use crate::parse::SelectParams;
use serde::Deserialize;
use serde_json::{Map, Value};

/// Each: Only one of all, fields and one can exist at the same time.
#[derive(Default, Deserialize, Clone, Debug)]
pub struct Each {
    pub all: Box<Option<SelectParams>>,
    pub one: Box<Option<SelectParams>>,
    pub fields: Option<HashMapSelectParams>,
}

impl Each {
    pub fn each(&self, ds: DocumentSelection) -> Value {
        return if self.all.is_some() {
            self.all(ds)
        } else if self.fields.is_some() {
            self.fields(ds)
        } else {
            self.one(ds)
        };
    }

    fn all(&self, mut ds: DocumentSelection) -> Value {
        let params = self.all.as_ref().as_ref().unwrap();
        let nodes = ds.nodes();
        let mut array = Vec::new();
        for node in nodes.iter() {
            ds = DocumentSelection::ParseNode(node.to_owned());
            let v = ds.parse(params);
            if v.is_null() {
                continue;
            }
            array.push(v);
        }
        Value::Array(array)
    }

    fn one(&self, mut ds: DocumentSelection) -> Value {
        let params = self.one.as_ref().as_ref().unwrap();
        let nodes = ds.nodes();
        for node in nodes.iter() {
            ds = DocumentSelection::ParseNode(node.to_owned());
            let v = ds.parse(params);
            if v.is_null() {
                continue;
            }
            return v;
        }
        params.get_default_val()
    }

    fn fields(&self, ds: DocumentSelection) -> Value {
        let each_keys = self.fields.as_ref().unwrap();
        let nodes = ds.nodes();
        let mut array = Vec::new();
        for node in nodes.iter() {
            let mut cur_map = Map::new();
            for (k, v) in each_keys.iter() {
                let ds = DocumentSelection::ParseNode(node.to_owned());
                let c_val = ds.parse(v);
                let val = c_val;
                cur_map.insert(k.to_string(), Value::from(val));
            }
            array.push(Value::Object(cur_map));
        }
        Value::Array(array)
    }
}