aboutsummaryrefslogtreecommitdiff
path: root/src/main.rs
blob: bd009e796e415ec1625954bdd195c19e5f76d30f (plain)
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
extern crate reqwest;
extern crate serde_json;
extern crate cursive;

use cursive::Cursive;
use cursive::traits::*;
use cursive::views::{ 
    TextView, Dialog, EditView,
    SelectView, OnEventView, LinearLayout,
    DummyView
};

pub mod content;
use content::*;

pub mod theme;
use theme::*;

fn main() {
    // Initial setup
    let mut main = Cursive::default();

    // set theme
    main.set_theme(theme_gen());

    main.add_global_callback('q', |s| s.quit());
    main.add_global_callback('s', |s| search(s));

    main.run();
}

fn search(s: &mut Cursive){

    fn go(s: &mut Cursive, search: &str) {
        s.pop_layer();
        let mut result = vec![];
        match get_search_results(search) {
            Ok(x) => result = x,
            Err(e) => pop_error(s,handler(e)),
        };
        let choose_result = SelectView::<String>::new()
            .with_all_str(result)
            .on_submit(|s, name|{
                s.pop_layer();
                on_submit(s, name);
            });
        s.add_layer(Dialog::around(choose_result)
                    .title("Search Results")
                    .button("Cancel", |s| match s.pop_layer() { _ => () })
                    .fixed_size(( 45,10 )));
    }

    s.add_layer(Dialog::around(EditView::new()
                               .on_submit(go)
                               .with_id("search")
                               )
                .title("Search for a page")
                .button("Go", |s| {
                    let search_txt = s.call_on_id( "search", |v: &mut EditView| {
                        v.get_content()
                    }).unwrap();
                    go(s, &search_txt);
                })
                .button("Cancel", |s| match s.pop_layer(){
                    _ => ()
                })
                .fixed_size(( 35, 5 )));
}

fn on_submit(s: &mut Cursive, name: &String) {
    // get article data
    let heading: String = name.clone();
    let url = query_url_gen(&name.replace(" ", "_"));
    let mut extract = String::new();
    let mut link_vec: Vec<String> = vec![];

    // handle errors if any
    let res = reqwest::get(&url).unwrap();
    match get_extract(res) {
        Ok(x) => extract = x,
        Err(e) => pop_error(s, handler(e))
    };
    let res = reqwest::get(&url).unwrap();
    match get_links(res) {
        Ok(x) => link_vec = x,
        Err(e) => pop_error(s, handler(e))
    };

    // get the act together
    let mut article_content = TextView::new(heading);
    article_content.append(String::from("\n\n"));
    article_content.append(extract_formatter(extract));

    let links = SelectView::<String>::new()
        .with_all_str(link_vec)
        .on_submit(on_submit);

    s.add_layer(
        OnEventView::new(
            LinearLayout::horizontal()
            .child(article_content.fixed_width(72))
            .child(DummyView)
            .child(links)
            )
        .on_event('t', |s| match s.pop_layer() { _ => () })
        );
}