use crate::{expand::{arithmetic::expand_arith_string, tilde::expand_tilde_string, vars::expand_string}, prelude::*};
use shellenv::jobs::{ChildProc, JobBldr};

pub mod shellcmd;


pub fn exec_input<S: Into<String>>(input: S, shenv: &mut ShEnv) -> ShResult<()> {
	let input = input.into();
	shenv.new_input(&input);
	let total_time = std::time::Instant::now();

	let token_time = std::time::Instant::now();
	let token_stream = Lexer::new(input,shenv).lex();

	let token_stream = expand_aliases(token_stream, shenv);
	for token in &token_stream {
		log!(TRACE, token);
		log!(TRACE, "{}",token.as_raw(shenv));
	}
	log!(INFO, "Tokenizing done in {:?}", token_time.elapsed());

	let parse_time = std::time::Instant::now();
	let syn_tree = Parser::new(token_stream,shenv).parse()?;
	log!(TRACE,syn_tree);
	log!(INFO, "Parsing done in {:?}", parse_time.elapsed());
	if !shenv.ctx().flags().contains(ExecFlags::IN_FUNC) {
		shenv.save_io()?;
	}

	let exec_time = std::time::Instant::now();
	if let Err(e) = Executor::new(syn_tree, shenv).walk() {
		if let ShErrKind::CleanExit = e.kind() {
			let code = shenv.get_code();
			sh_quit(code);
		} else {
			if !shenv.ctx().flags().contains(ExecFlags::IN_FUNC) {
				shenv.reset_io()?;
			}
			return Err(e.into())
		}
	}
	log!(INFO, "Executing done in {:?}", exec_time.elapsed());
	log!(INFO, "Total time spent: {:?}", total_time.elapsed());
	if !shenv.ctx().flags().contains(ExecFlags::IN_FUNC) {
		shenv.reset_io()?;
	}
	log!(INFO, "Io reset");
	Ok(())
}

pub struct Executor<'a> {
	ast: SynTree,
	shenv: &'a mut ShEnv
}

impl<'a> Executor<'a> {
	pub fn new(ast: SynTree, shenv: &'a mut ShEnv) -> Self {
		Self { ast, shenv }
	}
	pub fn walk(&mut self) -> ShResult<()> {
		self.shenv.inputman_mut().push_state();
		log!(TRACE, "Starting walk");
		while let Some(node) = self.ast.next_node() {
			if let NdRule::CmdList { cmds } = node.clone().into_rule() {
				log!(TRACE, "{:?}", cmds);
				exec_list(cmds, self.shenv).try_blame(node.as_raw(self.shenv),node.span())?
			} else { unreachable!() }
		}
		self.shenv.inputman_mut().pop_state();
		log!(TRACE, "passed");
		Ok(())
	}
}

fn exec_list(list: Vec<(Option<CmdGuard>, Node)>, shenv: &mut ShEnv) -> ShResult<()> {
	log!(TRACE, "Executing list");
	let mut list = VecDeque::from(list);
	while let Some(cmd_info) = list.fpop() {
		let guard = cmd_info.0;
		let cmd = cmd_info.1;

		if let Some(guard) = guard {
			let code = shenv.get_code();
			match guard {
				CmdGuard::And => {
					if code != 0 { break; }
				}
				CmdGuard::Or => {
					if code == 0 { break; }
				}
			}
		}
		dispatch_node(cmd, shenv)?;
	}
	Ok(())
}

fn dispatch_node(node: Node, shenv: &mut ShEnv) -> ShResult<()> {
	let node_raw = node.as_raw(shenv);
	let span = node.span();
	match *node.rule() {
		NdRule::Command {..} |
		NdRule::Subshell {..} |
		NdRule::Assignment {..} => dispatch_command(node, shenv).try_blame(node_raw, span)?,
		NdRule::IfThen {..} => shellcmd::exec_if(node, shenv).try_blame(node_raw, span)?,
		NdRule::Loop {..} => shellcmd::exec_loop(node, shenv).try_blame(node_raw, span)?,
		NdRule::ForLoop {..} => shellcmd::exec_for(node, shenv).try_blame(node_raw, span)?,
		NdRule::Case {..} => shellcmd::exec_case(node, shenv).try_blame(node_raw, span)?,
		NdRule::FuncDef {..} => exec_funcdef(node,shenv).try_blame(node_raw, span)?,
		NdRule::Pipeline {..} => exec_pipeline(node, shenv).try_blame(node_raw, span)?,
		_ => unimplemented!("No support for NdRule::{:?} yet", node.rule())
	}
	Ok(())
}

fn dispatch_command(mut node: Node, shenv: &mut ShEnv) -> ShResult<()> {
	let mut is_builtin = false;
	let mut is_func = false;
	let mut is_subsh = false;
	let mut is_assign = false;
	if let NdRule::Command { ref mut argv, redirs: _ } = node.rule_mut() {
		if !shenv.ctx().flags().contains(ExecFlags::NO_EXPAND) {
			*argv = expand_argv(argv.to_vec(), shenv)?;
		}
		let cmd = argv.first().unwrap().as_raw(shenv);
		if shenv.logic().get_function(&cmd).is_some() {
			is_func = true;
		} else if node.flags().contains(NdFlag::BUILTIN) {
			is_builtin = true;
		}
	} else if let NdRule::Subshell { body: _, ref mut argv, redirs: _ } = node.rule_mut() {
		if !shenv.ctx().flags().contains(ExecFlags::NO_EXPAND) {
			*argv = expand_argv(argv.to_vec(), shenv)?;
		}
		is_subsh = true;
	} else if let NdRule::Assignment { assignments: _, cmd: _ } = node.rule() {
		is_assign = true;
	} else { unreachable!() }

	if is_builtin {
		exec_builtin(node, shenv)?;
	} else if is_func {
		exec_func(node, shenv)?;
	} else if is_subsh {
		exec_subshell(node, shenv)?;
	} else if is_assign {
		exec_assignment(node, shenv)?;
	} else {
		exec_cmd(node, shenv)?;
	}
	Ok(())
}

fn exec_func(node: Node, shenv: &mut ShEnv) -> ShResult<()> {
	let rule = node.into_rule();
	if let NdRule::Command { argv, redirs } = rule {
		let mut argv_iter = argv.into_iter();
		let func_name = argv_iter.next().unwrap().as_raw(shenv);
		let body = shenv.logic().get_function(&func_name).unwrap().to_string();
		let snapshot = shenv.clone();
		shenv.vars_mut().reset_params();
		shenv.ctx_mut().set_flag(ExecFlags::IN_FUNC);
		while let Some(arg) = argv_iter.next() {
			let arg_raw = shenv.input_slice(arg.span()).to_string();
			shenv.vars_mut().bpush_arg(&arg_raw);
		}
		shenv.collect_redirs(redirs);

		match exec_input(body, shenv) {
			Ok(()) => {
				*shenv = snapshot;
				return Ok(())
			}
			Err(e) if e.kind() == ShErrKind::FuncReturn => {
				let code = shenv.get_code();
				*shenv = snapshot;
				shenv.set_code(code);
				return Ok(())
			}
			Err(e) => {
				*shenv = snapshot;
				return Err(e.into())
			}
		}
	}
	Ok(())
}

fn exec_funcdef(node: Node, shenv: &mut ShEnv) -> ShResult<()> {
	let rule = node.into_rule();
	if let NdRule::FuncDef { name, body } = rule {
		let name_raw = name.as_raw(shenv);
		let name = name_raw.trim_end_matches("()");
		let body_raw = body.as_raw(shenv);
		let body = body_raw[1..body_raw.len() - 1].trim();

		shenv.logic_mut().set_function(name, body);
	} else { unreachable!() }
	Ok(())
}

fn exec_subshell(node: Node, shenv: &mut ShEnv) -> ShResult<()> {
	let snapshot = shenv.clone();
	shenv.vars_mut().reset_params();
	let is_bg = node.flags().contains(NdFlag::BACKGROUND);
	let rule = node.into_rule();
	if let NdRule::Subshell { body, argv, redirs } = rule {
		if shenv.ctx().flags().contains(ExecFlags::NO_FORK) {
			shenv.ctx_mut().unset_flag(ExecFlags::NO_FORK); // Allow sub-forks in this case
			shenv.collect_redirs(redirs);
			if let Err(e) = shenv.ctx_mut().activate_rdrs() {
				write_err(e)?;
				exit(1);
			}
			for arg in argv {
				let arg_raw = &arg.as_raw(shenv);
				shenv.vars_mut().bpush_arg(arg_raw);
			}
			let body_raw = body.as_raw(shenv);

			match exec_input(body_raw, shenv) {
				Ok(()) => exit(0),
				Err(e) => {
					eprintln!("{}",e);
					exit(1);
				}
			}
		} else {
			match unsafe { fork()? } {
				Child => {
					shenv.collect_redirs(redirs);
					if let Err(e) = shenv.ctx_mut().activate_rdrs() {
						write_err(e)?;
						exit(1);
					}
					for arg in argv {
						let arg_raw = &arg.as_raw(shenv);
						shenv.vars_mut().bpush_arg(arg_raw);
					}
					let body_raw = body.as_raw(shenv);
					match exec_input(body_raw, shenv) {
						Ok(()) => exit(0),
						Err(e) => {
							eprintln!("{}",e);
							exit(1);
						}
					}
				}
				Parent { child } => {
					*shenv = snapshot;
					let children = vec![
						ChildProc::new(child, Some("anonymous subshell"), Some(child))?
					];
					let job = JobBldr::new()
						.with_children(children)
						.with_pgid(child)
						.build();
					dispatch_job(job, is_bg, shenv)?;
				}
			}
		}
	} else { unreachable!() }
	Ok(())
}

fn exec_builtin(node: Node, shenv: &mut ShEnv) -> ShResult<()> {
	log!(TRACE, "Executing builtin");
	let command = if let NdRule::Command { argv, redirs: _ } = node.rule() {
		argv.first().unwrap().as_raw(shenv)
	} else { unreachable!() };

	log!(TRACE, "{}", command.as_str());
	match command.as_str() {
		"echo" => echo(node, shenv)?,
		"cd" => cd(node,shenv)?,
		"pwd" => pwd(node, shenv)?,
		"export" => export(node, shenv)?,
		"jobs" => jobs(node, shenv)?,
		"fg" => continue_job(node, shenv, true)?,
		"bg" => continue_job(node, shenv, false)?,
		"read" => read_builtin(node, shenv)?,
		"alias" => alias(node, shenv)?,
		"exit" => sh_flow(node, shenv, ShErrKind::CleanExit)?,
		"return" => sh_flow(node, shenv, ShErrKind::FuncReturn)?,
		"break" => sh_flow(node, shenv, ShErrKind::LoopBreak)?,
		"continue" => sh_flow(node, shenv, ShErrKind::LoopContinue)?,
		"source" => source(node, shenv)?,
		_ => unimplemented!("Have not yet implemented support for builtin `{}'",command)
	}
	log!(TRACE, "done");
	Ok(())
}

fn exec_assignment(node: Node, shenv: &mut ShEnv) -> ShResult<()> {
	log!(TRACE, "Executing assignment");
	let rule = node.into_rule();
	if let NdRule::Assignment { assignments, cmd } = rule {
		log!(TRACE, "Assignments: {:?}", assignments);
		log!(TRACE, "Command: {:?}", cmd);
		let mut assigns = assignments.into_iter();
		if let Some(cmd) = cmd {
			let saved_env = shenv.vars().env().clone();
			while let Some(token) = assigns.next() {
				let raw = token.as_raw(shenv);
				if let Some((var,val)) = raw.split_once('=') {
					let val_rule = Lexer::get_rule(&val);
					if EXPANSIONS.contains(&val_rule) {
						let exp = match val_rule {
							TkRule::ArithSub => expand_arith_string(val,shenv)?,
							TkRule::DQuote => expand_string(val, shenv)?,
							TkRule::TildeSub => expand_tilde_string(val),
							TkRule::VarSub => {
								let val = shenv.vars().get_var(var);
								val.to_string()
							}
							_ => unimplemented!()
						};
						shenv.vars_mut().export(var, &exp);
					} else {
						shenv.vars_mut().export(var, val);
					}
				}
			}
			dispatch_command(*cmd, shenv)?;
			*shenv.vars_mut().env_mut() = saved_env;
		} else {
			while let Some(token) = assigns.next() {
				let raw = token.as_raw(shenv);
				if let Some((var,val)) = raw.split_once('=') {
					let val_rule = Lexer::get_rule(&val);
					if EXPANSIONS.contains(&val_rule) {
						let exp = match val_rule {
							TkRule::ArithSub => expand_arith_string(val,shenv)?,
							TkRule::DQuote => expand_string(val, shenv)?,
							TkRule::TildeSub => expand_tilde_string(val),
							TkRule::VarSub => {
								let val = shenv.vars().get_var(var);
								val.to_string()
							}
							_ => unimplemented!()
						};
						shenv.vars_mut().set_var(var, &exp);
					} else {
						shenv.vars_mut().set_var(var, val);
					}
				}
			}
		}
	} else { unreachable!() }
	Ok(())
}

fn exec_pipeline(node: Node, shenv: &mut ShEnv) -> ShResult<()> {
	log!(TRACE, "Executing pipeline");
	let is_bg = node.flags().contains(NdFlag::BACKGROUND);
	let rule = node.into_rule();
	if let NdRule::Pipeline { cmds } = rule {
		let mut prev_rpipe: Option<i32> = None;
		let mut cmds       = VecDeque::from(cmds);
		let mut pgid       = None;
		let mut cmd_names  = vec![];
		let mut pids       = vec![];

		while let Some(cmd) = cmds.pop_front() {
			let (mut r_pipe, mut w_pipe) = if cmds.is_empty() {
				// If we are on the last command, don't make new pipes
				(None,None)
			} else {
				let (r_pipe, w_pipe) = c_pipe()?;
				(Some(r_pipe),Some(w_pipe))
			};
			if let NdRule::Command { argv, redirs: _ } = cmd.rule() {
				let cmd_name = argv.first().unwrap().as_raw(shenv);
				cmd_names.push(cmd_name);
			} else if let NdRule::Subshell {..} = cmd.rule() {
				cmd_names.push("subshell".to_string());
			} else {
				cmd_names.push("shell cmd".to_string());
			}

			match unsafe { fork()? } {
				Child => {
					// Set NO_FORK since we are already in a fork, to prevent unnecessarily forking again
					shenv.ctx_mut().set_flag(ExecFlags::NO_FORK);
					// We close this r_pipe since it's the one the next command will use, so not useful here
					if let Some(r_pipe) = r_pipe.take() {
						close(r_pipe)?;
					}

					// Create some redirections
					if let Some(w_pipe) = w_pipe.take() {
						if !cmds.is_empty() {
							let wpipe_redir = Redir::output(1, w_pipe);
							shenv.ctx_mut().push_rdr(wpipe_redir);
						}
					}
					// Use the r_pipe created in the last iteration
					if let Some(prev_rpipe) = prev_rpipe.take() {
						let rpipe_redir = Redir::input(0, prev_rpipe);
						shenv.ctx_mut().push_rdr(rpipe_redir);
					}

					if let Err(e) = dispatch_node(cmd, shenv) {
						eprintln!("{}",e);
						exit(1);
					}
					exit(0);
				}
				Parent { child } => {
					// Close the write pipe out here to signal EOF
					if let Some(w_pipe) = w_pipe.take() {
						close(w_pipe)?;
					}
					if pgid.is_none() {
						pgid = Some(child);
					}
					pids.push(child);
					if let Some(pipe) = prev_rpipe {
						close(pipe)?;
					}
					prev_rpipe = r_pipe;
				}
			}
		}
		let mut children = vec![];
		for (i,pid) in pids.iter().enumerate() {
			let command = cmd_names.get(i).unwrap();
			let child = ChildProc::new(*pid, Some(&command), pgid)?;
			children.push(child);
		}
		let job = JobBldr::new()
			.with_children(children)
			.with_pgid(pgid.unwrap())
			.build();
		dispatch_job(job, is_bg, shenv)?;
	} else { unreachable!() }
	Ok(())
}

fn exec_cmd(node: Node, shenv: &mut ShEnv) -> ShResult<()> {
	log!(TRACE, "Executing command");
	let blame = node.span();
	let is_bg = node.flags().contains(NdFlag::BACKGROUND);
	let rule = node.into_rule();

	if let NdRule::Command { argv, redirs } = rule {
		let (argv,envp) = prep_execve(argv, shenv);
		let command = argv.first().unwrap().to_string();
		if get_bin_path(&command, shenv).is_some() {

			log!(TRACE, "{:?}",shenv.ctx().flags());
			if shenv.ctx().flags().contains(ExecFlags::NO_FORK) {
				log!(TRACE, "Not forking");
				shenv.collect_redirs(redirs);
				log!(TRACE, "{:?}",shenv.ctx().redirs());
				if let Err(e) = shenv.ctx_mut().activate_rdrs() {
					eprintln!("{:?}",e);
					exit(1);
				}
				if let Err(errno) = execvpe(command, argv, envp) {
					if errno != Errno::EFAULT {
						exit(errno as i32);
					}
				}
			} else {
				log!(TRACE, "Forking");
				match unsafe { fork()? } {
					Child => {
						log!(TRACE, redirs);
						shenv.collect_redirs(redirs);
						if let Err(e) = shenv.ctx_mut().activate_rdrs() {
							eprintln!("{:?}",e);
							exit(1);
						}
						execvpe(command, argv, envp)?;
						exit(1);
					}
					Parent { child } => {
						let children = vec![
							ChildProc::new(child, Some(&command), Some(child))?
						];
						let job = JobBldr::new()
							.with_children(children)
							.with_pgid(child)
							.build();
						log!(TRACE, "New job: {:?}", job);
						dispatch_job(job, is_bg, shenv)?;
					}
				}
			}
		} else {
			return Err(ShErr::full(ShErrKind::CmdNotFound, format!("{}", command), shenv.get_input(), blame))
		}
	} else { unreachable!("Found this rule in exec_cmd: {:?}", rule) }
	Ok(())
}

fn prep_execve(argv: Vec<Token>, shenv: &mut ShEnv) -> (Vec<String>, Vec<String>) {
	log!(TRACE, "Preparing execvpe args");
	let argv_s = argv.as_strings(shenv);
	log!(TRACE, argv_s);

	let mut envp = vec![];
	let mut env_vars = shenv.vars().env().iter();
	while let Some(entry) = env_vars.next() {
		let key = entry.0;
		let val = entry.1;
		let formatted = format!("{}={}",key,val);
		envp.push(formatted);
	}
	log!(TRACE, argv_s);
	log!(DEBUG, argv_s);
	(argv_s, envp)
}