// Usage: your_docker.sh run <image> <command> <arg1> <arg2> ...
fn main() {
    // You can use print statements as follows for debugging, they'll be visible when running tests.
    // println!("Logs from your program will appear here!");

    // Uncomment this block to pass the first stage!
    let args: Vec<_> = std::env::args().collect();
    let command = &args[3];
    let command_args = &args[4..];
    let output = std::process::Command::new(command)
        .args(command_args)
        .output()
        .unwrap();
    
    if output.status.success() {
        let std_out = std::str::from_utf8(&output.stdout).unwrap();
        if std_out.len() > 0 {
            println!("{}", std_out);
        }
        let std_err = std::str::from_utf8(&output.stderr).unwrap();
        if std_err.len() > 0 {
            println!("{}", std_err);
        }
    } else {
        std::process::exit(1);
    }
}