85 lines
2.0 KiB
Rust
85 lines
2.0 KiB
Rust
pub mod prelude;
|
|
pub mod libsh;
|
|
pub mod prompt;
|
|
pub mod procio;
|
|
pub mod parse;
|
|
pub mod expand;
|
|
pub mod state;
|
|
pub mod builtin;
|
|
pub mod jobs;
|
|
pub mod signal;
|
|
#[cfg(test)]
|
|
pub mod tests;
|
|
|
|
use libsh::error::ShResult;
|
|
use parse::{execute::Dispatcher, lex::{LexFlags, LexStream}, ParseStream};
|
|
use signal::sig_setup;
|
|
use termios::{LocalFlags, Termios};
|
|
use crate::prelude::*;
|
|
|
|
pub static mut SAVED_TERMIOS: Option<Option<Termios>> = None;
|
|
|
|
pub fn save_termios() {
|
|
unsafe {
|
|
SAVED_TERMIOS = Some(if isatty(std::io::stdin().as_raw_fd()).unwrap() {
|
|
let mut termios = termios::tcgetattr(std::io::stdin()).unwrap();
|
|
termios.local_flags &= !LocalFlags::ECHOCTL;
|
|
termios::tcsetattr(std::io::stdin(), nix::sys::termios::SetArg::TCSANOW, &termios).unwrap();
|
|
Some(termios)
|
|
} else {
|
|
None
|
|
});
|
|
}
|
|
}
|
|
pub fn get_saved_termios() -> Option<Termios> {
|
|
unsafe {
|
|
// This is only used when the shell exits so it's fine
|
|
// SAVED_TERMIOS is only mutated once at the start as well
|
|
SAVED_TERMIOS.clone().flatten()
|
|
}
|
|
}
|
|
fn set_termios() {
|
|
if isatty(std::io::stdin().as_raw_fd()).unwrap() {
|
|
let mut termios = termios::tcgetattr(std::io::stdin()).unwrap();
|
|
termios.local_flags &= !LocalFlags::ECHOCTL;
|
|
termios::tcsetattr(std::io::stdin(), nix::sys::termios::SetArg::TCSANOW, &termios).unwrap();
|
|
}
|
|
}
|
|
|
|
pub fn exec_input(input: &str) -> ShResult<()> {
|
|
let parse_start = Instant::now();
|
|
let mut tokens = vec![];
|
|
for token in LexStream::new(&input, LexFlags::empty()) {
|
|
tokens.push(token?);
|
|
}
|
|
|
|
let mut nodes = vec![];
|
|
for result in ParseStream::new(tokens) {
|
|
nodes.push(result?);
|
|
}
|
|
flog!(INFO, "parse duration: {:?}", parse_start.elapsed());
|
|
|
|
let exec_start = Instant::now();
|
|
let mut dispatcher = Dispatcher::new(nodes);
|
|
dispatcher.begin_dispatch()?;
|
|
flog!(INFO, "cmd duration: {:?}", exec_start.elapsed());
|
|
|
|
flog!(INFO, "total duration: {:?}", parse_start.elapsed());
|
|
Ok(())
|
|
}
|
|
|
|
fn main() {
|
|
save_termios();
|
|
set_termios();
|
|
sig_setup();
|
|
|
|
|
|
loop {
|
|
let input = prompt::read_line().unwrap();
|
|
|
|
if let Err(e) = exec_input(&input) {
|
|
eprintln!("{e}");
|
|
}
|
|
}
|
|
}
|