90 lines
2.0 KiB
Rust
90 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;
|
|
pub mod getopt;
|
|
|
|
use std::collections::HashSet;
|
|
|
|
use expand::expand_aliases;
|
|
use libsh::error::ShResult;
|
|
use parse::{execute::Dispatcher, ParsedSrc};
|
|
use signal::sig_setup;
|
|
use state::{source_rc, write_meta};
|
|
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
|
|
});
|
|
}
|
|
}
|
|
|
|
#[allow(static_mut_refs)]
|
|
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: String) -> ShResult<()> {
|
|
write_meta(|m| m.start_timer());
|
|
let input = expand_aliases(input, HashSet::new());
|
|
let mut parser = ParsedSrc::new(Arc::new(input));
|
|
parser.parse_src()?;
|
|
|
|
let mut dispatcher = Dispatcher::new(parser.extract_nodes());
|
|
dispatcher.begin_dispatch()
|
|
}
|
|
|
|
fn main() {
|
|
save_termios();
|
|
set_termios();
|
|
sig_setup();
|
|
|
|
if let Err(e) = source_rc() {
|
|
eprintln!("{e}");
|
|
}
|
|
|
|
loop {
|
|
let input = match prompt::read_line() {
|
|
Ok(line) => line,
|
|
Err(e) => {
|
|
eprintln!("{e}");
|
|
continue
|
|
}
|
|
};
|
|
|
|
if let Err(e) = exec_input(input) {
|
|
eprintln!("{e}");
|
|
}
|
|
}
|
|
}
|