aboutsummaryrefslogtreecommitdiff
path: root/src/parser/command.rs
blob: 496614f263d813feff7af43a9250fa81a0424888 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
use std::io::{Error, ErrorKind};
use std::path::{Path, PathBuf};

#[derive(Debug)]
pub enum Redirect {
    Std,
    FileOverwrite(String),
    FileAppend(String),
}

#[derive(Debug, PartialEq)]
pub enum RunOn {
    Always,
    ExitSuccess,
    ExitFailure,
}

pub enum RunResult {
    Command(std::process::ExitStatus),
    Builtin,
}

#[derive(Debug)]
pub struct CommandInfo {
    pub args: Vec<String>,
    pub stdout: Redirect,
    pub when: RunOn,
}

impl CommandInfo {
    pub fn new() -> CommandInfo {
        CommandInfo {
            args: Vec::new(),
            stdout: Redirect::Std,
            when: RunOn::Always,
        }
    }

    pub fn run(
        &self,
        home: &PathBuf,
        cwd: &mut PathBuf,
        status: &Option<std::process::ExitStatus>,
    ) -> Result<RunResult, Error> {
        match self.args[0].as_str() {
            "!" => {
                println!("{:?}", status);
                Ok(RunResult::Builtin)
            }
            "cd" => match cd(&self.args[1..], home) {
                Ok(p) => {
                    *cwd = p;
                    Ok(RunResult::Builtin)
                }
                Err(e) => Err(e),
            },
            "exit" => {
                std::process::exit(0);
            }
            _ => {
                let mut child = std::process::Command::new(&self.args[0])
                    .args(&self.args[1..])
                    .spawn()?;
                Ok(RunResult::Command(child.wait().unwrap()))
            }
        }
    }
}

fn cd(args: &[String], home: &PathBuf) -> Result<PathBuf, Error> {
    if args.len() > 1 {
        return Err(Error::new(
            ErrorKind::InvalidInput,
            "Too many arguments passed to cd",
        ));
    }

    if args.len() == 0 {
        return Ok(home.to_path_buf());
    }

    let root = match std::fs::canonicalize(Path::new(args[0].as_str())) {
        Ok(p) => p,
        Err(_) => Path::new("/").to_path_buf(),
    };

    match std::env::set_current_dir(&root) {
        Ok(_) => Ok(root.to_path_buf()),
        Err(e) => Err(e),
    }
}