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
140
141
142
|
use crate::cheat;
use crate::filesystem;
use std::collections::HashMap;
use std::process;
use std::process::{Command, Stdio};
pub struct Opts<'a> {
pub query: Option<String>,
pub filter: Option<String>,
pub prompt: Option<String>,
pub preview: bool,
pub autoselect: bool,
pub overrides: Option<&'a String>, // TODO: remove &'a
pub header_lines: u8,
pub multi: bool,
pub copyable: bool,
pub suggestions: bool,
pub nth: Option<u8>,
}
impl Default for Opts<'_> {
fn default() -> Self {
Self {
query: None,
filter: None,
preview: true,
autoselect: true,
overrides: None,
header_lines: 0,
prompt: None,
multi: false,
copyable: false,
suggestions: true,
nth: None,
}
}
}
pub fn call<F>(opts: Opts, stdin_fn: F) -> (String, Option<HashMap<String, cheat::Value>>)
where
F: Fn(&mut process::ChildStdin) -> Option<HashMap<String, cheat::Value>>,
{
let mut c = Command::new("fzf");
c.args(&[
"--preview-window",
"up:2",
"--with-nth",
"1,2,3",
"--delimiter",
"\t",
"--ansi",
"--bind",
"ctrl-j:down,ctrl-k:up",
"--exact",
]);
if opts.autoselect {
c.arg("--select-1");
}
if opts.multi {
c.arg("--multi");
}
if opts.copyable {
c.args(&["--expect", "ctrl-y,enter"]);
}
if opts.preview {
c.args(&[
"--preview",
format!("{} preview {{}}", filesystem::exe_string()).as_str(),
]);
}
if let Some(q) = opts.query {
c.args(&["--query", &q]);
}
if let Some(f) = opts.filter {
c.args(&["--filter", &f]);
}
if let Some(p) = opts.prompt {
c.args(&["--prompt", &p]);
}
if let Some(n) = opts.nth {
c.args(&["--nth", &n.to_string()]);
}
if opts.header_lines > 0 {
c.args(&["--header-lines", format!("{}", opts.header_lines).as_str()]);
}
if let Some(o) = opts.overrides {
o.as_str()
.split(' ')
.map(|s| s.to_string())
.filter(|s| !s.is_empty())
.for_each(|s| {
c.arg(s);
});
}
if !opts.suggestions {
c.args(&["--print-query", "--no-select-1", "--height", "1"]);
}
let child = c
.stdin(Stdio::piped())
.stdout(Stdio::piped())
.stderr(Stdio::inherit())
.spawn();
let mut child = match child {
Ok(x) => x,
Err(_) => {
eprintln!("navi was unable to call fzf.\nPlease make sure it's correctly installed\nRefer to https://github.com/junegunn/fzf for more info.");
process::exit(33)
}
};
let stdin = child.stdin.as_mut().unwrap();
let result = stdin_fn(stdin);
let out = child.wait_with_output().unwrap();
let text = match out.status.code() {
Some(0) | Some(1) => String::from_utf8(out.stdout).unwrap(),
Some(130) => process::exit(130),
_ => {
let err = String::from_utf8(out.stderr)
.unwrap_or_else(|_| "<stderr contains invalid UTF-8>".to_owned());
panic!("External command failed:\n {}", err)
}
};
(text, result)
}
|
