ahc-evaluation 0.1.1

Evaluates the submission code for AHC (AtCoder Heuristic Contest).
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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114

use std::process::Command;

use anyhow::{ensure, Context};

use crate::config::Config;

/// Builds the submission code.
pub fn build_submission(config: &Config) -> anyhow::Result<()> {
    let cmd_args = &config.command.build.submission;

    // Skips build if build command for submission code is empty.
    let Some(program) = cmd_args.first() else {
        return Ok(());
    };

    let process_handle = Command::new(program)
        .args(&cmd_args[1..])
        .spawn()
        .with_context(|| {
            format!(
                "
Failed to start the child process that builds the submission code.
List of arguments: {:?}
",
                cmd_args
            )
        })?;

    let output = process_handle
        .wait_with_output()
        .with_context(|| "Failed to building the submission code.")?;

    ensure!(
        output.status.success(),
        BuildCommandError {
            cmd_args: config.command.build.submission.clone(),
            output
        }
    );

    Ok(())
}

/// Builds the local tester.
pub fn build_tester(config: &Config) -> anyhow::Result<()> {
    let cmd_args = &config.command.build.tester;

    // Skips build if build command for local tester is empty.
    let Some(program) = cmd_args.first() else {
        return Ok(());
    };

    let process_handle = Command::new(program)
        .args(&cmd_args[1..])
        .spawn()
        .with_context(|| {
            format!(
                "
Failed to start the child process that builds the local tester.
List of arguments: {:?}
",
                cmd_args
            )
        })?;

    let output = process_handle
        .wait_with_output()
        .with_context(|| "Failed to build the local tester.")?;

    ensure!(
        output.status.success(),
        BuildCommandError {
            cmd_args: config.command.build.tester.clone(),
            output
        }
    );

    Ok(())
}

#[derive(Debug)]
pub struct BuildCommandError {
    pub cmd_args: Vec<String>,
    pub output: std::process::Output,
}

impl std::fmt::Display for BuildCommandError {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        let stdout = String::from_utf8(self.output.stdout.clone()).unwrap();
        let stderr = String::from_utf8(self.output.stderr.clone()).unwrap();

        write!(
            f,
            "
Failed to execute command.

Exit code: {}

Command line arguments: {:?},

------------------------------- Standard Output --------------------------------
{}
--------------------------------------------------------------------------------

---------------------------- Standard Error Output -----------------------------
{}
--------------------------------------------------------------------------------
",
            self.output.status, self.cmd_args, stdout, stderr,
        )
    }
}

impl std::error::Error for BuildCommandError {}