81 lines
2.0 KiB
Rust
81 lines
2.0 KiB
Rust
use colored::*;
|
|
use std::fmt;
|
|
use std::fmt::Write;
|
|
|
|
use crate::language::{ComputationResponse, DebugAsk};
|
|
use crate::ReplOptions;
|
|
|
|
pub struct ReplResponse {
|
|
label: Option<String>,
|
|
text: String,
|
|
color: Option<Color>,
|
|
}
|
|
|
|
impl fmt::Display for ReplResponse {
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
let mut buf = String::new();
|
|
if let Some(ref label) = self.label {
|
|
write!(buf, "({})", label).unwrap();
|
|
}
|
|
write!(buf, "=> {}", self.text).unwrap();
|
|
write!(
|
|
f,
|
|
"{}",
|
|
match self.color {
|
|
Some(c) => buf.color(c),
|
|
None => buf.normal(),
|
|
}
|
|
)
|
|
}
|
|
}
|
|
|
|
pub fn handle_computation_response(
|
|
response: ComputationResponse,
|
|
options: &ReplOptions,
|
|
) -> Vec<ReplResponse> {
|
|
let mut responses = vec![];
|
|
|
|
if options.show_total_time {
|
|
responses.push(ReplResponse {
|
|
label: Some("Total time".to_string()),
|
|
text: format!("{:?}", response.global_output_stats.total_duration),
|
|
color: None,
|
|
});
|
|
}
|
|
|
|
if options.show_stage_times {
|
|
responses.push(ReplResponse {
|
|
label: Some("Stage times".to_string()),
|
|
text: format!("{:?}", response.global_output_stats.stage_durations),
|
|
color: None,
|
|
});
|
|
}
|
|
|
|
for debug_resp in response.debug_responses {
|
|
let stage_name = match debug_resp.ask {
|
|
DebugAsk::ByStage { stage_name, .. } => stage_name,
|
|
_ => continue,
|
|
};
|
|
responses.push(ReplResponse {
|
|
label: Some(stage_name.to_string()),
|
|
text: debug_resp.value,
|
|
color: Some(Color::Red),
|
|
});
|
|
}
|
|
|
|
responses.push(match response.main_output {
|
|
Ok(s) => ReplResponse {
|
|
label: None,
|
|
text: s,
|
|
color: None,
|
|
},
|
|
Err(e) => ReplResponse {
|
|
label: Some("Error".to_string()),
|
|
text: e,
|
|
color: Some(Color::Red),
|
|
},
|
|
});
|
|
|
|
responses
|
|
}
|