canary_probe 0.2.1

Canary Probe is a simple tool to help students in Computer Programming I (CSU0001, NTNU) to test their code.
Documentation 
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

use anyhow::bail;
use bollard::exec;
use bollard::Docker;
use futures_util::TryStreamExt;

pub(crate) async fn exec(
    docker: &Docker,
    container_name: &str,
    cmd: &str,
    working_dir: Option<&str>,
) -> anyhow::Result<String> {
    let error_code = uuid::Uuid::new_v4();
    let command = format!("{} || echo {}", cmd, error_code);

    let exec_config = exec::CreateExecOptions {
        attach_stdout: Some(true),
        attach_stderr: Some(true),
        cmd: Some(vec!["/bin/sh", "-c", command.as_str()]),
        working_dir,
        ..Default::default()
    };

    let exec = docker.create_exec(container_name, exec_config).await?;
    let output = if let exec::StartExecResults::Attached { output, .. } =
        docker.start_exec(&exec.id, None).await?
    {
        output
            .try_collect::<Vec<_>>()
            .await?
            .into_iter()
            .map(|msg| msg.to_string())
            .collect::<Vec<_>>()
            .join("\n")
    } else {
        unreachable!();
    };

    if output.contains(&error_code.to_string()) {
        bail!(
            "Failed to execute command: {}\n{}",
            cmd,
            output.replace(&error_code.to_string(), "")
        );
    }

    Ok(output)
}

/// export a directory from a container in tar format
pub(crate) async fn export(
    docker: &Docker,
    container_name: &str,
    dir: &str,
) -> anyhow::Result<Vec<u8>> {
    let command = format!("tar -C {} -cf - .", dir);

    let exec_config = exec::CreateExecOptions {
        attach_stdout: Some(true),
        cmd: Some(vec!["/bin/sh", "-c", command.as_str()]),
        ..Default::default()
    };

    let exec = docker.create_exec(container_name, exec_config).await?;
    let output = if let exec::StartExecResults::Attached { output, .. } =
        docker.start_exec(&exec.id, None).await?
    {
        output
            .try_collect::<Vec<_>>()
            .await?
            .into_iter()
            .flat_map(|msg| msg.into_bytes())
            .collect::<Vec<_>>()
    } else {
        unreachable!();
    };

    Ok(output)
}