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
|
mod recite;
use recite::path::prefresh;
use recite::Poem;
use std::io::{self, Write};
use std::path::Path;
/// Starts the main shell loop
///
/// # Arguments
/// * `paths` - A reference to a vector that holds a list to the shell $PATHs
/// * `prompt` - A string slice indicating the shell's prompt
///
/// # Examples
/// ```
/// fn main() {
/// let path = vec!["/bin"];
/// let path = path.into_iter().map(Path::new).collect();
/// let prompt = "|> ";
/// ...
/// repl(&path, prompt);
/// }
/// ```
fn repl(path: &Vec<&Path>, prompt: &str) {
// Initial path refresh on startup
let mut bins: Vec<String> = prefresh(path);
// Main shell loop
loop {
// Output the prompt
print!("{}", prompt);
io::stdout().flush().unwrap();
// Wait for user input
let mut poetry = String::new();
let bytes = io::stdin()
.read_line(&mut poetry)
.expect("dwvsh: error: unable to evaluate the input string");
// Check if we've reached EOF (i.e. <C-d>)
if bytes == 0 {
println!("");
break;
}
// Kill zombies
// TODO: Remove this (since it was implemented in incant_quiet())
// unsafe {
// let mut status = -1;
// let mut pid = waitpid(-1, &mut status, WNOHANG);
// while pid > 0 {
// pid = waitpid(-1, &mut status, WNOHANG);
// }
// }
// Trim the input
let poetry = String::from(poetry.trim());
// Skip parsing if there is no poetry
if poetry.is_empty() {
continue;
}
// Parse a poem
let poem = Poem::read(poetry);
match poem {
Some(poem) => match poem.recite(path, &mut bins) {
Ok(_) => {}
Err(e) => eprintln!("dwvsh: {}", e.to_string().to_lowercase()),
},
None => {}
};
}
}
/// Shell entry
///
/// Shell setup and entry
fn main() {
// Define paths
// TODO: Hardcoded path should only be the fallback
let path = vec![
"/bin",
"/sbin",
"/usr/bin",
"/usr/sbin",
"/usr/local/bin",
"/usr/local/sbin",
];
let path = path.into_iter().map(Path::new).collect();
// Set the prompt
let prompt = "|> ";
// Handle signals
unsafe {
signal_hook::low_level::register(signal_hook::consts::SIGINT, move || {
print!("\n{}", prompt);
io::stdout().flush().unwrap();
})
.unwrap();
};
// Begin evaluating commands
repl(&path, prompt);
}
|
