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

// Copyright 2017 LambdaStack All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

use std::process::Command;
use std::process::Output;
use std::io::Result;

/// run_args - pass in the cmd as low_level &str and args as a String slice
///
/// The function allows you to add all of the options to a given command to
/// # Example
///
/// ```
/// let mut args: Vec<String> = Vec::new();
/// args.push("aux".to_string());
///
/// run_args("ps", &args);
/// ```
pub fn run_args(cmd: &str, args: &[String], shell: bool) -> Result<(Output)> {
    let output;

    if args.len() > 0 {
        if !shell {
            output = try!(Command::new(cmd).args(&args).output());
        } else {
            let mut arg_string = String::new();
            arg_string = arg_string + cmd + " ";
            for s in args {
                arg_string = arg_string + s + " ";
            }
            output = try!(run_cli(arg_string));
        }
    } else {
        // output = try!(Command::new(cmd).output());
        output = try!(run_cli(cmd.to_string()));
    }

    Ok(output)
}

/// run_cli - pass in a String of a normal command line
///
/// The function will split the options into words to supply to the low_level std::process::Command
/// which returns Result<(Output)>
/// # Example
///
/// ```
/// run_cli("ps aux");
/// ```

// NOTE: Add Into so a "" can also be passed in...
pub fn run_cli(cmd_line: String) -> Result<(Output)> {
    let output = try!(Command::new("sh").arg("-c").arg(&cmd_line).output());
    Ok(output)

    // Example of word splitting...
    // let output;
    // let mut words = cmd_line.split_whitespace();
    // let cmd = words.next();
    //
    // match cmd {
    // Some(cmd) => {
    // let args = words.collect::<Vec<&str>>();
    // if args.len() > 0 {
    // output = try!(Command::new(cmd).args(&args).output());
    // } else {
    // output = try!(Command::new(cmd).output());
    // }
    // }
    // None => {
    // Let error bubble up
    // output = try!(Command::new(&cmd_line).output());
    // }
    // }
    //
    // Ok(output)
    //
}