refactor: move reverse into the State

This commit is contained in:
graelo 2020-05-30 22:23:33 +02:00
parent 56106f4e38
commit 7e96bef522
3 changed files with 45 additions and 33 deletions

View file

@ -19,7 +19,7 @@ pub mod view;
pub fn run(buffer: String, opt: &CliOpt) -> Option<(String, bool)> { pub fn run(buffer: String, opt: &CliOpt) -> Option<(String, bool)> {
let lines: Vec<&str> = buffer.split('\n').collect(); let lines: Vec<&str> = buffer.split('\n').collect();
let mut state = state::State::new(&lines, &opt.alphabet, &opt.custom_regex); let mut state = state::State::new(&lines, &opt.alphabet, &opt.custom_regex, opt.reverse);
let hint_style = match &opt.hint_style { let hint_style = match &opt.hint_style {
None => None, None => None,
@ -37,7 +37,6 @@ pub fn run(buffer: String, opt: &CliOpt) -> Option<(String, bool)> {
let selection: Option<(String, bool)> = { let selection: Option<(String, bool)> = {
let mut viewbox = view::View::new( let mut viewbox = view::View::new(
&mut state, &mut state,
opt.reverse,
opt.unique_hint, opt.unique_hint,
&opt.hint_alignment, &opt.hint_alignment,
&opt.colors, &opt.colors,

View file

@ -38,6 +38,7 @@ pub struct State<'a> {
pub lines: &'a Vec<&'a str>, pub lines: &'a Vec<&'a str>,
alphabet: &'a Alphabet, alphabet: &'a Alphabet,
custom_regexes: &'a Vec<String>, custom_regexes: &'a Vec<String>,
pub reverse: bool,
} }
impl<'a> State<'a> { impl<'a> State<'a> {
@ -45,15 +46,17 @@ impl<'a> State<'a> {
lines: &'a Vec<&'a str>, lines: &'a Vec<&'a str>,
alphabet: &'a Alphabet, alphabet: &'a Alphabet,
custom_regexes: &'a Vec<String>, custom_regexes: &'a Vec<String>,
reverse: bool,
) -> State<'a> { ) -> State<'a> {
State { State {
lines, lines,
alphabet, alphabet,
custom_regexes, custom_regexes,
reverse,
} }
} }
pub fn matches(&self, reverse: bool, unique: bool) -> Vec<Match<'a>> { pub fn matches(&self, unique: bool) -> Vec<Match<'a>> {
let mut matches = Vec::new(); let mut matches = Vec::new();
let exclude_patterns = EXCLUDE_PATTERNS let exclude_patterns = EXCLUDE_PATTERNS
@ -126,7 +129,7 @@ impl<'a> State<'a> {
let mut hints = self.alphabet.make_hints(matches.len()); let mut hints = self.alphabet.make_hints(matches.len());
// This looks wrong but we do a pop after // This looks wrong but we do a pop after
if !reverse { if !self.reverse {
hints.reverse(); hints.reverse();
} else { } else {
matches.reverse(); matches.reverse();
@ -152,7 +155,7 @@ impl<'a> State<'a> {
} }
} }
if reverse { if self.reverse {
matches.reverse(); matches.reverse();
} }
@ -174,7 +177,7 @@ mod tests {
let lines = split("lorem 127.0.0.1 lorem 255.255.255.255 lorem 127.0.0.1 lorem"); let lines = split("lorem 127.0.0.1 lorem 255.255.255.255 lorem 127.0.0.1 lorem");
let custom = [].to_vec(); let custom = [].to_vec();
let alphabet = Alphabet("abcd".to_string()); let alphabet = Alphabet("abcd".to_string());
let results = State::new(&lines, &alphabet, &custom).matches(false, false); let results = State::new(&lines, &alphabet, &custom, false).matches(false);
assert_eq!(results.len(), 3); assert_eq!(results.len(), 3);
assert_eq!(results.first().unwrap().hint.clone().unwrap(), "a"); assert_eq!(results.first().unwrap().hint.clone().unwrap(), "a");
@ -186,7 +189,7 @@ mod tests {
let lines = split("lorem 127.0.0.1 lorem 255.255.255.255 lorem 127.0.0.1 lorem"); let lines = split("lorem 127.0.0.1 lorem 255.255.255.255 lorem 127.0.0.1 lorem");
let custom = [].to_vec(); let custom = [].to_vec();
let alphabet = Alphabet("abcd".to_string()); let alphabet = Alphabet("abcd".to_string());
let results = State::new(&lines, &alphabet, &custom).matches(false, true); let results = State::new(&lines, &alphabet, &custom, false).matches(true);
assert_eq!(results.len(), 3); assert_eq!(results.len(), 3);
assert_eq!(results.first().unwrap().hint.clone().unwrap(), "a"); assert_eq!(results.first().unwrap().hint.clone().unwrap(), "a");
@ -198,7 +201,7 @@ mod tests {
let lines = split("latest sha256:30557a29d5abc51e5f1d5b472e79b7e296f595abcf19fe6b9199dbbc809c6ff4 20 hours ago"); let lines = split("latest sha256:30557a29d5abc51e5f1d5b472e79b7e296f595abcf19fe6b9199dbbc809c6ff4 20 hours ago");
let custom = [].to_vec(); let custom = [].to_vec();
let alphabet = Alphabet("abcd".to_string()); let alphabet = Alphabet("abcd".to_string());
let results = State::new(&lines, &alphabet, &custom).matches(false, false); let results = State::new(&lines, &alphabet, &custom, false).matches(false);
assert_eq!(results.len(), 1); assert_eq!(results.len(), 1);
assert_eq!( assert_eq!(
@ -212,7 +215,7 @@ mod tests {
let lines = split("path: /var/log/nginx.log\npath: test/log/nginx-2.log:32folder/.nginx@4df2.log"); let lines = split("path: /var/log/nginx.log\npath: test/log/nginx-2.log:32folder/.nginx@4df2.log");
let custom = [].to_vec(); let custom = [].to_vec();
let alphabet = Alphabet("abcd".to_string()); let alphabet = Alphabet("abcd".to_string());
let results = State::new(&lines, &alphabet, &custom).matches(false, false); let results = State::new(&lines, &alphabet, &custom, false).matches(false);
assert_eq!(results.len(), 3); assert_eq!(results.len(), 3);
assert_eq!(results.get(0).unwrap().text, "/var/log/nginx.log"); assert_eq!(results.get(0).unwrap().text, "/var/log/nginx.log");
@ -225,7 +228,7 @@ mod tests {
let lines = split("Lorem /tmp/foo/bar_lol, lorem\n Lorem /var/log/boot-strap.log lorem ../log/kern.log lorem"); let lines = split("Lorem /tmp/foo/bar_lol, lorem\n Lorem /var/log/boot-strap.log lorem ../log/kern.log lorem");
let custom = [].to_vec(); let custom = [].to_vec();
let alphabet = Alphabet("abcd".to_string()); let alphabet = Alphabet("abcd".to_string());
let results = State::new(&lines, &alphabet, &custom).matches(false, false); let results = State::new(&lines, &alphabet, &custom, false).matches(false);
assert_eq!(results.len(), 3); assert_eq!(results.len(), 3);
assert_eq!(results.get(0).unwrap().text.clone(), "/tmp/foo/bar_lol"); assert_eq!(results.get(0).unwrap().text.clone(), "/tmp/foo/bar_lol");
@ -241,7 +244,7 @@ mod tests {
let lines = split("Lorem ~/.gnu/.config.txt, lorem"); let lines = split("Lorem ~/.gnu/.config.txt, lorem");
let custom = [].to_vec(); let custom = [].to_vec();
let alphabet = Alphabet("abcd".to_string()); let alphabet = Alphabet("abcd".to_string());
let results = State::new(&lines, &alphabet, &custom).matches(false, false); let results = State::new(&lines, &alphabet, &custom, false).matches(false);
assert_eq!(results.len(), 1); assert_eq!(results.len(), 1);
assert_eq!(results.get(0).unwrap().text.clone(), "~/.gnu/.config.txt"); assert_eq!(results.get(0).unwrap().text.clone(), "~/.gnu/.config.txt");
@ -253,7 +256,7 @@ mod tests {
split("Lorem ipsum 123e4567-e89b-12d3-a456-426655440000 lorem\n Lorem lorem lorem"); split("Lorem ipsum 123e4567-e89b-12d3-a456-426655440000 lorem\n Lorem lorem lorem");
let custom = [].to_vec(); let custom = [].to_vec();
let alphabet = Alphabet("abcd".to_string()); let alphabet = Alphabet("abcd".to_string());
let results = State::new(&lines, &alphabet, &custom).matches(false, false); let results = State::new(&lines, &alphabet, &custom, false).matches(false);
assert_eq!(results.len(), 1); assert_eq!(results.len(), 1);
} }
@ -263,7 +266,7 @@ mod tests {
let lines = split("Lorem fd70b5695 5246ddf f924213 lorem\n Lorem 973113963b491874ab2e372ee60d4b4cb75f717c lorem"); let lines = split("Lorem fd70b5695 5246ddf f924213 lorem\n Lorem 973113963b491874ab2e372ee60d4b4cb75f717c lorem");
let custom = [].to_vec(); let custom = [].to_vec();
let alphabet = Alphabet("abcd".to_string()); let alphabet = Alphabet("abcd".to_string());
let results = State::new(&lines, &alphabet, &custom).matches(false, false); let results = State::new(&lines, &alphabet, &custom, false).matches(false);
assert_eq!(results.len(), 4); assert_eq!(results.len(), 4);
assert_eq!(results.get(0).unwrap().text.clone(), "fd70b5695"); assert_eq!(results.get(0).unwrap().text.clone(), "fd70b5695");
@ -281,7 +284,7 @@ mod tests {
split("Lorem ipsum 127.0.0.1 lorem\n Lorem 255.255.10.255 lorem 127.0.0.1 lorem"); split("Lorem ipsum 127.0.0.1 lorem\n Lorem 255.255.10.255 lorem 127.0.0.1 lorem");
let custom = [].to_vec(); let custom = [].to_vec();
let alphabet = Alphabet("abcd".to_string()); let alphabet = Alphabet("abcd".to_string());
let results = State::new(&lines, &alphabet, &custom).matches(false, false); let results = State::new(&lines, &alphabet, &custom, false).matches(false);
assert_eq!(results.len(), 3); assert_eq!(results.len(), 3);
assert_eq!(results.get(0).unwrap().text.clone(), "127.0.0.1"); assert_eq!(results.get(0).unwrap().text.clone(), "127.0.0.1");
@ -294,7 +297,7 @@ mod tests {
let lines = split("Lorem ipsum fe80::2:202:fe4 lorem\n Lorem 2001:67c:670:202:7ba8:5e41:1591:d723 lorem fe80::2:1 lorem ipsum fe80:22:312:fe::1%eth0"); let lines = split("Lorem ipsum fe80::2:202:fe4 lorem\n Lorem 2001:67c:670:202:7ba8:5e41:1591:d723 lorem fe80::2:1 lorem ipsum fe80:22:312:fe::1%eth0");
let custom = [].to_vec(); let custom = [].to_vec();
let alphabet = Alphabet("abcd".to_string()); let alphabet = Alphabet("abcd".to_string());
let results = State::new(&lines, &alphabet, &custom).matches(false, false); let results = State::new(&lines, &alphabet, &custom, false).matches(false);
assert_eq!(results.len(), 4); assert_eq!(results.len(), 4);
assert_eq!(results.get(0).unwrap().text.clone(), "fe80::2:202:fe4"); assert_eq!(results.get(0).unwrap().text.clone(), "fe80::2:202:fe4");
@ -316,7 +319,7 @@ mod tests {
); );
let custom = [].to_vec(); let custom = [].to_vec();
let alphabet = Alphabet("abcd".to_string()); let alphabet = Alphabet("abcd".to_string());
let results = State::new(&lines, &alphabet, &custom).matches(false, false); let results = State::new(&lines, &alphabet, &custom, false).matches(false);
assert_eq!(results.len(), 2); assert_eq!(results.len(), 2);
assert_eq!(results.get(0).unwrap().pattern.clone(), "markdown_url"); assert_eq!(results.get(0).unwrap().pattern.clone(), "markdown_url");
@ -336,7 +339,7 @@ mod tests {
let lines = split("Lorem ipsum https://www.rust-lang.org/tools lorem\n Lorem ipsumhttps://crates.io lorem https://github.io?foo=bar lorem ssh://github.io"); let lines = split("Lorem ipsum https://www.rust-lang.org/tools lorem\n Lorem ipsumhttps://crates.io lorem https://github.io?foo=bar lorem ssh://github.io");
let custom = [].to_vec(); let custom = [].to_vec();
let alphabet = Alphabet("abcd".to_string()); let alphabet = Alphabet("abcd".to_string());
let results = State::new(&lines, &alphabet, &custom).matches(false, false); let results = State::new(&lines, &alphabet, &custom, false).matches(false);
assert_eq!(results.len(), 4); assert_eq!(results.len(), 4);
assert_eq!( assert_eq!(
@ -360,7 +363,7 @@ mod tests {
let lines = split("Lorem 0xfd70b5695 0x5246ddf lorem\n Lorem 0x973113tlorem"); let lines = split("Lorem 0xfd70b5695 0x5246ddf lorem\n Lorem 0x973113tlorem");
let custom = [].to_vec(); let custom = [].to_vec();
let alphabet = Alphabet("abcd".to_string()); let alphabet = Alphabet("abcd".to_string());
let results = State::new(&lines, &alphabet, &custom).matches(false, false); let results = State::new(&lines, &alphabet, &custom, false).matches(false);
assert_eq!(results.len(), 3); assert_eq!(results.len(), 3);
assert_eq!(results.get(0).unwrap().text.clone(), "0xfd70b5695"); assert_eq!(results.get(0).unwrap().text.clone(), "0xfd70b5695");
@ -374,7 +377,7 @@ mod tests {
split("Lorem #fd7b56 lorem #FF00FF\n Lorem #00fF05 lorem #abcd00 lorem #afRR00"); split("Lorem #fd7b56 lorem #FF00FF\n Lorem #00fF05 lorem #abcd00 lorem #afRR00");
let custom = [].to_vec(); let custom = [].to_vec();
let alphabet = Alphabet("abcd".to_string()); let alphabet = Alphabet("abcd".to_string());
let results = State::new(&lines, &alphabet, &custom).matches(false, false); let results = State::new(&lines, &alphabet, &custom, false).matches(false);
assert_eq!(results.len(), 4); assert_eq!(results.len(), 4);
assert_eq!(results.get(0).unwrap().text.clone(), "#fd7b56"); assert_eq!(results.get(0).unwrap().text.clone(), "#fd7b56");
@ -388,7 +391,7 @@ mod tests {
let lines = split("Lorem QmRdbNSxDJBXmssAc9fvTtux4duptMvfSGiGuq6yHAQVKQ lorem Qmfoobar"); let lines = split("Lorem QmRdbNSxDJBXmssAc9fvTtux4duptMvfSGiGuq6yHAQVKQ lorem Qmfoobar");
let custom = [].to_vec(); let custom = [].to_vec();
let alphabet = Alphabet("abcd".to_string()); let alphabet = Alphabet("abcd".to_string());
let results = State::new(&lines, &alphabet, &custom).matches(false, false); let results = State::new(&lines, &alphabet, &custom, false).matches(false);
assert_eq!(results.len(), 1); assert_eq!(results.len(), 1);
assert_eq!( assert_eq!(
@ -403,7 +406,7 @@ mod tests {
split("Lorem 5695 52463 lorem\n Lorem 973113 lorem 99999 lorem 8888 lorem\n 23456 lorem 5432 lorem 23444"); split("Lorem 5695 52463 lorem\n Lorem 973113 lorem 99999 lorem 8888 lorem\n 23456 lorem 5432 lorem 23444");
let custom = [].to_vec(); let custom = [].to_vec();
let alphabet = Alphabet("abcd".to_string()); let alphabet = Alphabet("abcd".to_string());
let results = State::new(&lines, &alphabet, &custom).matches(false, false); let results = State::new(&lines, &alphabet, &custom, false).matches(false);
assert_eq!(results.len(), 8); assert_eq!(results.len(), 8);
} }
@ -413,7 +416,7 @@ mod tests {
let lines = split("Lorem lorem\n--- a/src/main.rs"); let lines = split("Lorem lorem\n--- a/src/main.rs");
let custom = [].to_vec(); let custom = [].to_vec();
let alphabet = Alphabet("abcd".to_string()); let alphabet = Alphabet("abcd".to_string());
let results = State::new(&lines, &alphabet, &custom).matches(false, false); let results = State::new(&lines, &alphabet, &custom, false).matches(false);
assert_eq!(results.len(), 1); assert_eq!(results.len(), 1);
assert_eq!(results.get(0).unwrap().text.clone(), "src/main.rs"); assert_eq!(results.get(0).unwrap().text.clone(), "src/main.rs");
@ -424,7 +427,7 @@ mod tests {
let lines = split("Lorem lorem\n+++ b/src/main.rs"); let lines = split("Lorem lorem\n+++ b/src/main.rs");
let custom = [].to_vec(); let custom = [].to_vec();
let alphabet = Alphabet("abcd".to_string()); let alphabet = Alphabet("abcd".to_string());
let results = State::new(&lines, &alphabet, &custom).matches(false, false); let results = State::new(&lines, &alphabet, &custom, false).matches(false);
assert_eq!(results.len(), 1); assert_eq!(results.len(), 1);
assert_eq!(results.get(0).unwrap().text.clone(), "src/main.rs"); assert_eq!(results.get(0).unwrap().text.clone(), "src/main.rs");
@ -439,7 +442,7 @@ mod tests {
.map(|&s| s.to_string()) .map(|&s| s.to_string())
.collect(); .collect();
let alphabet = Alphabet("abcd".to_string()); let alphabet = Alphabet("abcd".to_string());
let results = State::new(&lines, &alphabet, &custom).matches(false, false); let results = State::new(&lines, &alphabet, &custom, false).matches(false);
assert_eq!(results.len(), 9); assert_eq!(results.len(), 9);
assert_eq!(results.get(0).unwrap().text.clone(), "http://foo.bar"); assert_eq!(results.get(0).unwrap().text.clone(), "http://foo.bar");

View file

@ -111,14 +111,13 @@ enum Event {
impl<'a> View<'a> { impl<'a> View<'a> {
pub fn new( pub fn new(
state: &'a mut state::State<'a>, state: &'a mut state::State<'a>,
reversed: bool,
unique_hint: bool, unique_hint: bool,
hint_alignment: &'a HintAlignment, hint_alignment: &'a HintAlignment,
rendering_colors: &'a ViewColors, rendering_colors: &'a ViewColors,
hint_style: Option<HintStyle>, hint_style: Option<HintStyle>,
) -> View<'a> { ) -> View<'a> {
let matches = state.matches(reversed, unique_hint); let matches = state.matches(unique_hint);
let focus_index = if reversed { matches.len() - 1 } else { 0 }; let focus_index = if state.reverse { matches.len() - 1 } else { 0 };
View { View {
state, state,
@ -426,8 +425,20 @@ impl<'a> View<'a> {
event::Key::Left => self.prev(), event::Key::Left => self.prev(),
event::Key::Right => self.next(), event::Key::Right => self.next(),
event::Key::Char(_ch @ 'n') => self.next(), event::Key::Char(_ch @ 'n') => {
event::Key::Char(_ch @ 'N') => self.prev(), if self.state.reverse {
self.prev()
} else {
self.next()
}
}
event::Key::Char(_ch @ 'N') => {
if self.state.reverse {
self.next()
} else {
self.prev()
}
}
// TODO: use a Trie or another data structure to determine // TODO: use a Trie or another data structure to determine
// if the entered key belongs to a longer hint. // if the entered key belongs to a longer hint.
@ -753,7 +764,7 @@ Barcelona https://en.wikipedia.org/wiki/Barcelona - ";
let custom_regexes = [].to_vec(); let custom_regexes = [].to_vec();
let alphabet = alphabets::Alphabet("abcd".to_string()); let alphabet = alphabets::Alphabet("abcd".to_string());
let mut state = state::State::new(&lines, &alphabet, &custom_regexes); let mut state = state::State::new(&lines, &alphabet, &custom_regexes, false);
let rendering_colors = ViewColors { let rendering_colors = ViewColors {
text_fg: Box::new(color::Black), text_fg: Box::new(color::Black),
text_bg: Box::new(color::White), text_bg: Box::new(color::White),
@ -813,8 +824,8 @@ Barcelona https://en.wikipedia.org/wiki/Barcelona - ";
let custom_regexes = [].to_vec(); let custom_regexes = [].to_vec();
let alphabet = alphabets::Alphabet("abcd".to_string()); let alphabet = alphabets::Alphabet("abcd".to_string());
let mut state = state::State::new(&lines, &alphabet, &custom_regexes); let reverse = true;
let reversed = true; let mut state = state::State::new(&lines, &alphabet, &custom_regexes, reverse);
let unique_hint = false; let unique_hint = false;
let rendering_colors = ViewColors { let rendering_colors = ViewColors {
@ -832,7 +843,6 @@ Barcelona https://en.wikipedia.org/wiki/Barcelona - ";
let view = View::new( let view = View::new(
&mut state, &mut state,
reversed,
unique_hint, unique_hint,
&hint_alignment, &hint_alignment,
&rendering_colors, &rendering_colors,