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
|
extern crate clipboard;
extern crate rustty;
use clipboard::ClipboardContext;
use rustty::{Terminal, Event, Cell, Color, Attr};
use rustty::ui::Painter;
use std::env;
use std::io::{self, BufRead};
use std::process::{self, Command};
use std::time::Duration;
struct Point {
x: usize,
y: usize,
}
/// Strips the part of a line before and including the first colon.
///
/// # Examples
///
/// ```
/// let value = strip_key("e: email@example.com");
/// assert_eq!(value, "email@example.com");
/// ```
fn strip_key(line: &str) -> &str {
let strings: Vec<&str> = line.split(": ").collect();
strings[1]
}
fn move_selection(term: &mut Terminal, selection: &mut Point, style: Cell, amount: isize) {
term.printline(selection.x, selection.y, " ");
let y = selection.y as isize;
selection.y = (y + amount) as usize;
term.printline_with_cell(selection.x, selection.y, "->", style);
}
fn parse_options(filename: &str) -> Vec<String> {
fn push_option(options: &mut Vec<String>, line: String) {
if line.starts_with("e: ") ||
line.starts_with("u: ") ||
line.starts_with("p: ") {
options.push(line);
}
}
let mut options = Vec::new();
if filename == "-" {
let stdin = io::stdin();
for line in stdin.lock().lines() {
let line = line.expect("Error reading from STDIN");
push_option(&mut options, line.to_owned());
}
} else {
let file = Command::new("pass")
.arg("show")
.arg(filename)
.output()
.expect("Error executing `pass`")
.stdout;
for line in String::from_utf8_lossy(&file).lines() {
push_option(&mut options, line.to_owned());
}
}
options
}
fn main() {
let args: Vec<String> = env::args().collect();
let input = if args.len() > 1 {
&args[1]
} else {
"-"
};
let options = parse_options(input);
if options.is_empty() {
process::exit(1);
}
let mut term = Terminal::new().unwrap();
term.swap_buffers().unwrap();
let knockout_cell = Cell::with_style(Color::White, Color::Black, Attr::Default);
let red_cell = Cell::with_style(Color::White, Color::Red, Attr::Default);
let green_cell = Cell::with_style(Color::White, Color::Green, Attr::Default);
let mut selection = Point { x: 0, y: 2 };
let mut clipboard_ctx = ClipboardContext::new().unwrap();
loop {
term.printline_with_cell(0, 0, "Passextract (Press q or Ctrl-C to quit)", knockout_cell);
term.printline_with_cell(selection.x, selection.y, "->", knockout_cell);
for (i, s) in options.iter().enumerate() {
term.printline(5, i + 2, s)
}
let evt = term.get_event(Duration::from_millis(100)).unwrap();
if let Some(Event::Key(ch)) = evt {
match ch {
'q' | '\x03' => {
break;
}
'j' => {
if selection.y < options.len() + 1 {
move_selection(&mut term, &mut selection, knockout_cell, 1)
}
}
'k' => {
if selection.y > 2 {
move_selection(&mut term, &mut selection, knockout_cell, -1)
}
}
'\x0D' => {
match clipboard_ctx.set_contents(strip_key(&options[selection.y - 2]).to_owned()) {
Ok(_) => {
term.printline_with_cell(selection.x, selection.y, "->", green_cell);
},
Err(_) => {
term.printline_with_cell(selection.x, selection.y, "->", red_cell);
}
}
}
_ => { continue }
}
}
term.swap_buffers().unwrap();
}
}
|