aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
authorZapAnton <[email protected]>2018-07-30 07:10:37 +0100
committerZapAnton <[email protected]>2018-07-30 07:10:37 +0100
commitc417c1efe7a93b6e871f2b2c85296a3a3008af54 (patch)
tree76204d5750a5057c25529c399572631f5a646e6b /src
parent769b323073a523414fd8bbfa1c3a734529196c1d (diff)
Rustfmt-ed source files
Diffstat (limited to 'src')
-rw-r--r--src/content.rs59
-rw-r--r--src/main.rs115
2 files changed, 86 insertions, 88 deletions
diff --git a/src/content.rs b/src/content.rs
index d767f9c..ff01342 100644
--- a/src/content.rs
+++ b/src/content.rs
@@ -1,15 +1,15 @@
1extern crate reqwest;
2extern crate serde_json;
3extern crate cursive; 1extern crate cursive;
4extern crate regex; 2extern crate regex;
3extern crate reqwest;
4extern crate serde_json;
5extern crate urlencoding; 5extern crate urlencoding;
6 6
7use self::regex::Regex;
7use cursive::theme::Effect; 8use cursive::theme::Effect;
8use cursive::utils::markup::StyledString; 9use cursive::utils::markup::StyledString;
9use cursive::Cursive;
10use cursive::views::Dialog; 10use cursive::views::Dialog;
11use cursive::Cursive;
11use serde_json::Value; 12use serde_json::Value;
12use self::regex::Regex;
13use CONFIGURATION; 13use CONFIGURATION;
14 14
15pub fn query_url_gen(title: &str) -> String { 15pub fn query_url_gen(title: &str) -> String {
@@ -57,36 +57,33 @@ pub fn get_extract(v: &Value) -> Result<String, reqwest::Error> {
57 Ok(format!("{}", extract)) 57 Ok(format!("{}", extract))
58 } 58 }
59 // ignore non strings 59 // ignore non strings
60 _ => Ok(format!("This page does not exist anymore")) 60 _ => Ok(format!("This page does not exist anymore")),
61 } 61 }
62} 62}
63 63
64pub fn extract_formatter(extract: String) -> StyledString { 64pub fn extract_formatter(extract: String) -> StyledString {
65 let mut formatted = StyledString::new(); 65 let mut formatted = StyledString::new();
66 66
67 let heading= Regex::new(r"^== (?P<d>.*) ==$").unwrap(); 67 let heading = Regex::new(r"^== (?P<d>.*) ==$").unwrap();
68 let subheading= Regex::new(r"^=== (?P<d>.*) ===$").unwrap(); 68 let subheading = Regex::new(r"^=== (?P<d>.*) ===$").unwrap();
69 let subsubheading= Regex::new(r"^==== (?P<d>.*) ====$").unwrap(); 69 let subsubheading = Regex::new(r"^==== (?P<d>.*) ====$").unwrap();
70 70
71 for line in extract.lines() { 71 for line in extract.lines() {
72 if heading.is_match(line) { 72 if heading.is_match(line) {
73 formatted.append( 73 formatted.append(StyledString::styled(
74 StyledString::styled( 74 heading.replace(line, "$d"),
75 heading.replace(line, "$d"), Effect::Bold 75 Effect::Bold,
76 ) 76 ));
77 );
78 } else if subheading.is_match(line) { 77 } else if subheading.is_match(line) {
79 formatted.append( 78 formatted.append(StyledString::styled(
80 StyledString::styled( 79 subheading.replace(line, "$d"),
81 subheading.replace(line, "$d"), Effect::Italic 80 Effect::Italic,
82 ) 81 ));
83 );
84 } else if subsubheading.is_match(line) { 82 } else if subsubheading.is_match(line) {
85 formatted.append( 83 formatted.append(StyledString::styled(
86 StyledString::styled( 84 subsubheading.replace(line, "$d"),
87 subsubheading.replace(line, "$d"), Effect::Underline 85 Effect::Underline,
88 ) 86 ));
89 );
90 } else { 87 } else {
91 formatted.append(StyledString::plain(line)); 88 formatted.append(StyledString::plain(line));
92 } 89 }
@@ -100,10 +97,9 @@ pub fn extract_formatter(extract: String) -> StyledString {
100pub fn get_search_results(search: &str) -> Result<Vec<String>, reqwest::Error> { 97pub fn get_search_results(search: &str) -> Result<Vec<String>, reqwest::Error> {
101 let url = search_url_gen(search); 98 let url = search_url_gen(search);
102 let mut res = reqwest::get(&url[..])?; 99 let mut res = reqwest::get(&url[..])?;
103 let v: Value = serde_json::from_str(&res.text()?) 100 let v: Value = serde_json::from_str(&res.text()?).unwrap_or_else(|e| {
104 .unwrap_or_else( |e| { 101 panic!("Recieved error {:?}", e);
105 panic!("Recieved error {:?}", e); 102 });
106 } );
107 103
108 let mut results: Vec<String> = vec![]; 104 let mut results: Vec<String> = vec![];
109 for item in v[1].as_array().unwrap() { 105 for item in v[1].as_array().unwrap() {
@@ -129,19 +125,18 @@ pub fn get_links(v: &Value) -> Result<Vec<String>, reqwest::Error> {
129 for item in arr { 125 for item in arr {
130 match item["title"] { 126 match item["title"] {
131 Value::String(ref title) => links.push(title.to_string()), 127 Value::String(ref title) => links.push(title.to_string()),
132 _ => links.push(String::from("lol")) 128 _ => links.push(String::from("lol")),
133 } 129 }
134 } 130 }
135 }, 131 }
136 _ => links.push(String::from("lol")) 132 _ => links.push(String::from("lol")),
137 }; 133 };
138 134
139 Ok(links) 135 Ok(links)
140} 136}
141 137
142pub fn pop_error(s: &mut Cursive, msg: String) { 138pub fn pop_error(s: &mut Cursive, msg: String) {
143 s.add_layer(Dialog::text(format!("{}", msg)) 139 s.add_layer(Dialog::text(format!("{}", msg)).button("Ok", |s| s.quit()));
144 .button("Ok", |s| s.quit()));
145} 140}
146 141
147pub fn handler(e: reqwest::Error) -> String { 142pub fn handler(e: reqwest::Error) -> String {
diff --git a/src/main.rs b/src/main.rs
index ab35f2d..8b7fdca 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,21 +1,19 @@
1extern crate clap;
2extern crate cursive;
1extern crate reqwest; 3extern crate reqwest;
2extern crate serde_json; 4extern crate serde_json;
3extern crate cursive;
4extern crate clap;
5#[macro_use] 5#[macro_use]
6extern crate lazy_static; 6extern crate lazy_static;
7 7
8use cursive::Cursive;
9use cursive::traits::*; 8use cursive::traits::*;
10use cursive::views::{ 9use cursive::views::{
11 TextView, Dialog, EditView, 10 Dialog, DummyView, EditView, LinearLayout, OnEventView, SelectView, TextView,
12 SelectView, OnEventView, LinearLayout,
13 DummyView
14}; 11};
12use cursive::Cursive;
15 13
16use serde_json::Value; 14use serde_json::Value;
17 15
18use clap::{Arg, App}; 16use clap::{App, Arg};
19 17
20pub mod content; 18pub mod content;
21use content::*; 19use content::*;
@@ -24,7 +22,7 @@ pub mod theme;
24use theme::*; 22use theme::*;
25 23
26struct Configuration { 24struct Configuration {
27 wiki_url: String 25 wiki_url: String,
28} 26}
29 27
30lazy_static! { 28lazy_static! {
@@ -33,14 +31,13 @@ lazy_static! {
33 31
34fn main() { 32fn main() {
35 parse_arguments(); 33 parse_arguments();
36 34
37 // Initial setup 35 // Initial setup
38 let mut main = Cursive::default(); 36 let mut main = Cursive::default();
39 37
40 // Set theme 38 // Set theme
41 main.set_theme(theme_gen()); 39 main.set_theme(theme_gen());
42 40
43
44 main.add_global_callback('q', |s| s.quit()); 41 main.add_global_callback('q', |s| s.quit());
45 main.add_global_callback('s', |s| search(s)); 42 main.add_global_callback('s', |s| search(s));
46 43
@@ -52,15 +49,19 @@ fn parse_arguments() -> Configuration {
52 .version("0.1.0") 49 .version("0.1.0")
53 .author("NerdyPepper") 50 .author("NerdyPepper")
54 .about("TUI MediaWiki browser") 51 .about("TUI MediaWiki browser")
55 .arg(Arg::with_name("URL") 52 .arg(
56 .help("The URL of the wiki to be viewed") 53 Arg::with_name("URL")
57 .index(1)) 54 .help("The URL of the wiki to be viewed")
58 .arg(Arg::with_name("lang") 55 .index(1),
59 .short("l") 56 )
60 .long("lang") 57 .arg(
61 .value_name("CODE") 58 Arg::with_name("lang")
62 .help("Choose the language for Wikipedia") 59 .short("l")
63 .takes_value(true)) 60 .long("lang")
61 .value_name("CODE")
62 .help("Choose the language for Wikipedia")
63 .takes_value(true),
64 )
64 .get_matches(); 65 .get_matches();
65 66
66 if matches.is_present("HELP") { 67 if matches.is_present("HELP") {
@@ -68,50 +69,52 @@ fn parse_arguments() -> Configuration {
68 } 69 }
69 70
70 let lang = matches.value_of("lang").unwrap_or("en").to_string(); 71 let lang = matches.value_of("lang").unwrap_or("en").to_string();
71 let wiki_url = matches.value_of("URL").unwrap_or(&format!("https://{}.wikipedia.org", lang)).to_string(); 72 let wiki_url = matches
73 .value_of("URL")
74 .unwrap_or(&format!("https://{}.wikipedia.org", lang))
75 .to_string();
72 76
73 Configuration { 77 Configuration { wiki_url }
74 wiki_url
75 }
76} 78}
77 79
78fn search(s: &mut Cursive) { 80fn search(s: &mut Cursive) {
79
80 fn go(s: &mut Cursive, search: &str) { 81 fn go(s: &mut Cursive, search: &str) {
81 s.pop_layer(); 82 s.pop_layer();
82 let mut result = vec![]; 83 let mut result = vec![];
83 match get_search_results(&search) { 84 match get_search_results(&search) {
84 Ok(x) => result = x, 85 Ok(x) => result = x,
85 Err(e) => pop_error(s,handler(e)), 86 Err(e) => pop_error(s, handler(e)),
86 }; 87 };
87 let choose_result = SelectView::<String>::new() 88 let choose_result = SelectView::<String>::new()
88 .with_all_str(result) 89 .with_all_str(result)
89 .on_submit(|s, name|{ 90 .on_submit(|s, name| {
90 s.pop_layer(); 91 s.pop_layer();
91 on_submit(s, name); 92 on_submit(s, name);
92 }) 93 })
93 .scrollable(); 94 .scrollable();
94 s.add_layer(Dialog::around(choose_result) 95 s.add_layer(
95 .title("Search Results") 96 Dialog::around(choose_result)
96 .button("Cancel", |s| match s.pop_layer() { _ => () }) 97 .title("Search Results")
97 .fixed_size(( 45,10 ))); 98 .button("Cancel", |s| match s.pop_layer() {
99 _ => (),
100 })
101 .fixed_size((45, 10)),
102 );
98 } 103 }
99 104
100 s.add_layer(Dialog::around(EditView::new() 105 s.add_layer(
101 .on_submit(go) 106 Dialog::around(EditView::new().on_submit(go).with_id("search"))
102 .with_id("search") 107 .title("Search for a page")
103 ) 108 .button("Go", |s| {
104 .title("Search for a page") 109 let search_txt = s.call_on_id("search", |v: &mut EditView| v.get_content())
105 .button("Go", |s| { 110 .unwrap();
106 let search_txt = s.call_on_id( "search", |v: &mut EditView| { 111 go(s, &search_txt);
107 v.get_content() 112 })
108 }).unwrap(); 113 .button("Cancel", |s| match s.pop_layer() {
109 go(s, &search_txt); 114 _ => (),
110 }) 115 })
111 .button("Cancel", |s| match s.pop_layer(){ 116 .fixed_size((35, 5)),
112 _ => () 117 );
113 })
114 .fixed_size(( 35, 5 )));
115} 118}
116 119
117fn on_submit(s: &mut Cursive, name: &str) { 120fn on_submit(s: &mut Cursive, name: &str) {
@@ -129,11 +132,11 @@ fn on_submit(s: &mut Cursive, name: &str) {
129 132
130 match get_extract(&v) { 133 match get_extract(&v) {
131 Ok(x) => extract = x, 134 Ok(x) => extract = x,
132 Err(e) => pop_error(s, handler(e)) 135 Err(e) => pop_error(s, handler(e)),
133 }; 136 };
134 match get_links(&v) { 137 match get_links(&v) {
135 Ok(x) => link_vec = x, 138 Ok(x) => link_vec = x,
136 Err(e) => pop_error(s, handler(e)) 139 Err(e) => pop_error(s, handler(e)),
137 }; 140 };
138 141
139 // get the act together 142 // get the act together
@@ -149,12 +152,12 @@ fn on_submit(s: &mut Cursive, name: &str) {
149 Dialog::around( 152 Dialog::around(
150 OnEventView::new( 153 OnEventView::new(
151 LinearLayout::horizontal() 154 LinearLayout::horizontal()
152 .child(article_content.fixed_width(72)) 155 .child(article_content.fixed_width(72))
153 .child(DummyView) 156 .child(DummyView)
154 .child(links) 157 .child(links),
155 ) 158 ).on_event('t', |s| match s.pop_layer() {
156 .on_event('t', |s| match s.pop_layer() { _ => () }) 159 _ => (),
157 ) 160 }),
158 .title(heading) 161 ).title(heading),
159 ); 162 );
160} 163}