Early work on tab completion

src/main.rs

@@ -1,4 +1,4 @@
-#![allow(unused_unsafe)]
+#![allow(static_mut_refs,unused_unsafe)]
 
 pub mod libsh;
 pub mod shellenv;
@@ -20,6 +20,14 @@ use crate::prelude::*;
 
 pub static mut SAVED_TERMIOS: Option<Option<Termios>> = None;
 
+bitflags! {
+	pub struct FernFlags: u32 {
+		const NO_RC = 0b000001;
+		const NO_HIST = 0b000010;
+		const INTERACTIVE = 0b000100;
+	}
+}
+
 pub fn save_termios() {
 	unsafe {
 		SAVED_TERMIOS = Some(if isatty(std::io::stdin().as_raw_fd()).unwrap() {
@@ -34,6 +42,8 @@ pub fn save_termios() {
 }
 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()
 	}
 }
@@ -44,22 +54,68 @@ fn set_termios() {
 		termios::tcsetattr(std::io::stdin(), nix::sys::termios::SetArg::TCSANOW, &termios).unwrap();
 	}
 }
+fn parse_args(shenv: &mut ShEnv) {
+	let mut args = std::env::args().skip(1);
+	let mut script_path: Option<PathBuf> = None;
+	let mut command: Option<String> = None;
+	let mut flags = FernFlags::empty();
+
+	log!(DEBUG, args);
+	while let Some(mut arg) = args.next() {
+		log!(DEBUG, arg);
+		if arg.starts_with("--") {
+			arg = arg.strip_prefix("--").unwrap().to_string();
+			match arg.as_str() {
+				"no-rc" => flags |= FernFlags::NO_RC,
+				"no-hist" => flags |= FernFlags::NO_HIST,
+				_ => eprintln!("Warning - Unrecognized option: {arg}")
+			}
+		} else if arg.starts_with('-') {
+			arg = arg.strip_prefix('-').unwrap().to_string();
+			match arg.as_str() {
+				"c" => command = args.next(),
+				_ => eprintln!("Warning - Unrecognized option: {arg}")
+			}
+		} else {
+			let path_check = PathBuf::from(&arg);
+			if path_check.is_file() {
+				script_path = Some(path_check);
+			}
+		}
+	}
+
+	if !flags.contains(FernFlags::NO_RC) {
+		let _ = shenv.source_rc().eprint();
+	}
+
+	if let Some(cmd) = command {
+		let input = clean_string(cmd);
+		let _ = exec_input(input, shenv).eprint();
+
+	} else if let Some(script) = script_path {
+		let _ = shenv.source_file(script).eprint();
+
+	} else {
+		interactive(shenv);
+	}
+}
 
 pub fn main() {
 	sig_setup();
 	save_termios();
 	set_termios();
 	let mut shenv = ShEnv::new();
-	if let Err(e) = shenv.source_rc() {
-		eprintln!("Error sourcing rc file: {}", e.to_string());
-	}
 
+	parse_args(&mut shenv);
+}
+
+fn interactive(shenv: &mut ShEnv) {
 	loop {
 		log!(TRACE, "Entered loop");
-		match prompt::read_line(&mut shenv) {
+		match prompt::read_line(shenv) {
 			Ok(line) => {
 				shenv.meta_mut().start_timer();
-				let _ = exec_input(line, &mut shenv).eprint();
+				let _ = exec_input(line, shenv).eprint();
 			}
 			Err(e) => {
 				eprintln!("{}",e);