command : mise à jour de la modularité

This commit is contained in:
Ahurac 2024-04-23 16:47:38 +02:00
parent a05ce8981c
commit 8c517ae8c7
7 changed files with 180 additions and 175 deletions

112
src/command/builtin.rs Normal file
View file

@ -0,0 +1,112 @@
use super::Command;
use crate::error;
use crate::error::CdError;
use crate::error::IllegalNumberError;
use crate::exit_code::ExitCode;
use crate::variables::Variables;
use std::{env::set_current_dir, path::PathBuf};
type BuiltinFunction = fn(&Vec<String>, &mut Variables, &mut ExitCode);
fn cd(args: &Vec<String>, variables: &mut Variables, exit_code: &mut ExitCode) {
let path: Option<PathBuf>;
if !args.is_empty() {
path = Some(PathBuf::from(&args[0]));
} else {
path = match variables.get("HOME").as_str() {
"" => None,
path => Some(PathBuf::from(path)),
};
}
exit_code.set_success();
if path.is_some() {
let path = path.unwrap();
let result = set_current_dir(&path);
if result.is_err() {
let error = CdError::new(&args[0]);
error::print(Box::new(error));
exit_code.set(2);
}
}
}
fn colon(_args: &Vec<String>, _variables: &mut Variables, exit_code: &mut ExitCode) {
exit_code.set_success();
}
fn exit(args: &Vec<String>, _variables: &mut Variables, exit_code: &mut ExitCode) {
if !args.is_empty() {
let raw_exit_code = match args[0].parse::<usize>() {
Ok(parsed) => Ok(u8::try_from(parsed % 256).unwrap()),
Err(_e) => Err(IllegalNumberError::new("exit", args[0].as_str())),
};
if raw_exit_code.is_ok() {
exit_code.set(raw_exit_code.unwrap());
} else {
error::print(Box::new(raw_exit_code.unwrap_err()));
exit_code.set(2);
()
}
}
crate::control::exit(exit_code);
}
fn unset(args: &Vec<String>, variables: &mut Variables, exit_code: &mut ExitCode) {
for variable_name in args.iter() {
variables.unset(variable_name);
}
exit_code.set_success();
}
fn export(args: &Vec<String>, variables: &mut Variables, exit_code: &mut ExitCode) {
for variable in args.iter() {
let assignation: Vec<&str> = variable.split('=').collect();
if assignation.len() == 2 {
variables.set(&assignation[0], &assignation[1]);
}
variables.export(&assignation[0]);
}
exit_code.set_success();
}
pub struct Builtin {
function: BuiltinFunction,
args: Vec<String>,
}
impl Builtin {
pub fn new(argv: &Vec<String>) -> Result<Self, ()> {
let mut args = argv.clone();
let program = args.remove(0);
let function: Option<BuiltinFunction> = match program.as_str() {
"cd" => Some(cd),
"exit" => Some(exit),
"unset" => Some(unset),
"export" => Some(export),
":" => Some(colon),
_ => None,
};
match function {
Some(function) => Ok(Self { function, args }),
None => Err(()),
}
}
}
impl Command for Builtin {
fn spawn(&mut self, variables: &mut Variables, exit_code: &mut ExitCode) {
(self.function)(&self.args, variables, exit_code)
}
}

View file

@ -1,79 +0,0 @@
use crate::error;
use crate::error::CdError;
use crate::error::IllegalNumberError;
use crate::exit_code::ExitCode;
use crate::variables::Variables;
use std::env;
use std::{env::set_current_dir, path::PathBuf};
pub(super) fn cd(args: &Vec<String>, _variables: &mut Variables, exit_code: &mut ExitCode) {
let path: Option<PathBuf>;
if !args.is_empty() {
path = Some(PathBuf::from(&args[0]));
} else {
path = match env::var("HOME") {
Ok(var) => Some(PathBuf::from(var)),
Err(_e) => None,
};
}
exit_code.set_success();
if path.is_some() {
let path = path.unwrap();
let result = set_current_dir(&path);
if result.is_err() {
let error = CdError::new(&args[0]);
error::print(Box::new(error));
exit_code.set(2);
}
}
}
pub(super) fn colon(_args: &Vec<String>, _variables: &mut Variables, exit_code: &mut ExitCode) {
exit_code.set_success();
}
pub(crate) fn exit(args: &Vec<String>, _variables: &mut Variables, exit_code: &mut ExitCode) {
let raw_exit_code: Result<i32, IllegalNumberError>;
if args.is_empty() {
raw_exit_code = Ok(i32::from(exit_code.get()));
} else {
raw_exit_code = match args[0].parse::<usize>() {
Ok(parsed) => Ok(i32::try_from(parsed % 256).unwrap()),
Err(_e) => Err(IllegalNumberError::new("exit", args[0].as_str())),
};
}
if raw_exit_code.is_ok() {
println!("exit");
std::process::exit(raw_exit_code.unwrap());
} else {
error::print(Box::new(raw_exit_code.unwrap_err()));
}
}
pub(super) fn unset(args: &Vec<String>, variables: &mut Variables, exit_code: &mut ExitCode) {
for variable_name in args.iter() {
variables.unset(variable_name);
}
exit_code.set_success();
}
pub(super) fn export(args: &Vec<String>, variables: &mut Variables, exit_code: &mut ExitCode) {
for variable in args.iter() {
let assignation: Vec<&str> = variable.split('=').collect();
if assignation.len() == 2 {
variables.set(&assignation[0], &assignation[1]);
}
variables.export(&assignation[0]);
}
exit_code.set_success();
}

View file

@ -1,26 +0,0 @@
use super::Builtin;
use super::Command;
use super::UnixProgram;
pub struct CommandBuilder {
argv: Vec<String>,
}
impl CommandBuilder {
pub fn new(argv: Vec<String>) -> Self {
Self { argv }
}
pub fn build(&self) -> Box<dyn Command> {
let builtin = Builtin::new(&self.argv);
let command: Box<dyn Command>;
if builtin.is_err() {
command = Box::from(UnixProgram::new(&self.argv));
} else {
command = Box::from(builtin.unwrap());
}
command
}
}

View file

@ -1,85 +1,35 @@
pub mod builtins; pub mod builtin;
pub mod command_builder;
pub mod command_sequence; pub mod command_sequence;
pub mod unix_program;
use crate::error;
use crate::error::CommandNotFoundError;
use crate::exit_code::ExitCode; use crate::exit_code::ExitCode;
use crate::variables::Variables; use crate::variables::Variables;
use builtin::Builtin;
type BuiltinFunction = fn(&Vec<String>, &mut Variables, &mut ExitCode); use unix_program::UnixProgram;
pub trait Command { pub trait Command {
fn spawn(&mut self, variables: &mut Variables, exit_code: &mut ExitCode); fn spawn(&mut self, variables: &mut Variables, exit_code: &mut ExitCode);
} }
pub struct Builtin { pub struct CommandBuilder {
function: BuiltinFunction, argv: Vec<String>,
args: Vec<String>,
} }
impl Builtin { impl CommandBuilder {
pub fn new(argv: &Vec<String>) -> Result<Self, ()> { pub fn new(argv: Vec<String>) -> Self {
let mut args = argv.clone(); Self { argv }
let program = args.remove(0);
let function: Option<BuiltinFunction> = match program.as_str() {
"cd" => Some(builtins::cd),
"exit" => Some(builtins::exit),
"unset" => Some(builtins::unset),
"export" => Some(builtins::export),
":" => Some(builtins::colon),
_ => None,
};
match function {
Some(function) => Ok(Self { function, args }),
None => Err(()),
}
} }
}
impl Command for Builtin { pub fn build(&self) -> Box<dyn Command> {
fn spawn(&mut self, variables: &mut Variables, exit_code: &mut ExitCode) { let builtin = Builtin::new(&self.argv);
(self.function)(&self.args, variables, exit_code) let command: Box<dyn Command>;
}
}
pub struct UnixProgram { if builtin.is_err() {
command: std::process::Command, command = Box::from(UnixProgram::new(&self.argv));
}
impl UnixProgram {
pub fn new(argv: &Vec<String>) -> Self {
let mut argv = argv.clone();
let program = argv.remove(0);
let mut command = std::process::Command::new(&program);
command.args(argv);
Self { command }
}
}
impl Command for UnixProgram {
fn spawn(&mut self, _variables: &mut Variables, exit_code: &mut ExitCode) {
let handle = self.command.spawn();
if handle.is_ok() {
let raw_exit_code = handle
.unwrap()
.wait()
.expect("error waiting for the child")
.code()
.unwrap_or(1);
exit_code.set(match u8::try_from(raw_exit_code) {
Ok(code) => code,
Err(_e) => u8::MAX,
});
} else { } else {
let error = CommandNotFoundError::new(self.command.get_program().to_str().unwrap()); command = Box::from(builtin.unwrap());
error::print(Box::new(error));
exit_code.set_command_not_found();
} }
command
} }
} }

View file

@ -0,0 +1,44 @@
use super::Command;
use crate::error;
use crate::error::CommandNotFoundError;
use crate::exit_code::ExitCode;
use crate::variables::Variables;
pub struct UnixProgram {
command: std::process::Command,
}
impl UnixProgram {
pub fn new(argv: &Vec<String>) -> Self {
let mut argv = argv.clone();
let program = argv.remove(0);
let mut command = std::process::Command::new(&program);
command.args(argv);
Self { command }
}
}
impl Command for UnixProgram {
fn spawn(&mut self, _variables: &mut Variables, exit_code: &mut ExitCode) {
let handle = self.command.spawn();
if handle.is_ok() {
let raw_exit_code = handle
.unwrap()
.wait()
.expect("error waiting for the child")
.code()
.unwrap_or(1);
exit_code.set(match u8::try_from(raw_exit_code) {
Ok(code) => code,
Err(_e) => u8::MAX,
});
} else {
let error = CommandNotFoundError::new(self.command.get_program().to_str().unwrap());
error::print(Box::new(error));
exit_code.set_command_not_found();
}
}
}

View file

@ -1,10 +1,14 @@
use crate::command::builtins;
use crate::command::Command; use crate::command::Command;
use crate::exit_code::ExitCode; use crate::exit_code::ExitCode;
use crate::interface::get_user_input; use crate::interface::get_user_input;
use crate::parser::parse; use crate::parser::parse;
use crate::variables::Variables; use crate::variables::Variables;
pub fn exit(exit_code: &mut ExitCode) {
println!("exit");
std::process::exit(i32::from(exit_code.get()));
}
pub fn run() { pub fn run() {
let mut current_exit_code = ExitCode::new(); let mut current_exit_code = ExitCode::new();
let mut variables = Variables::new(); let mut variables = Variables::new();
@ -20,7 +24,7 @@ pub fn run() {
} }
} else { } else {
println!(); println!();
builtins::exit(&vec![], &mut variables, &mut current_exit_code); exit(&mut current_exit_code);
} }
} }
} }

View file

@ -1,5 +1,5 @@
use crate::command::command_builder::CommandBuilder;
use crate::command::command_sequence::CommandSequence; use crate::command::command_sequence::CommandSequence;
use crate::command::CommandBuilder;
use crate::interface::get_user_input; use crate::interface::get_user_input;
use crate::variables::Variables; use crate::variables::Variables;
use std::collections::VecDeque; use std::collections::VecDeque;