-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
16 changed files
with
2,185 additions
and
606 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file was deleted.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,227 @@ | ||
| use std::{collections::HashMap, sync::Arc}; | ||
|
|
||
| use adw::prelude::*; | ||
| use relm4::{adw, gtk, Component, ComponentParts, RelmWidgetExt}; | ||
| use testangel::{ | ||
| action_loader::ActionMap, | ||
| ipc::EngineList, | ||
| report_generation::{self, ReportGenerationError}, | ||
| types::{FlowError, Action}, | ||
| }; | ||
| use testangel_ipc::prelude::{Evidence, EvidenceContent, ParameterValue}; | ||
|
|
||
| use crate::next_ui::{file_filters, lang}; | ||
|
|
||
| #[derive(Debug)] | ||
| pub enum ExecutionDialogCommandOutput { | ||
| /// Execution completed with the resulting evidence | ||
| Complete(Vec<Evidence>), | ||
|
|
||
| /// Execution failed at the given step and for the given reason | ||
| Failed(usize, FlowError), | ||
| } | ||
|
|
||
| #[derive(Debug)] | ||
| pub struct ExecutionDialogInit { | ||
| pub action: Action, | ||
| pub engine_list: Arc<EngineList>, | ||
| pub action_map: Arc<ActionMap>, | ||
| } | ||
|
|
||
| #[derive(Debug)] | ||
| pub enum ExecutionDialogInput { | ||
| Close, | ||
| FailedToGenerateReport(ReportGenerationError), | ||
| } | ||
|
|
||
| #[derive(Debug)] | ||
| pub struct ExecutionDialog; | ||
|
|
||
| impl ExecutionDialog { | ||
| /// Create the absolute barebones of a message dialog, allowing for custom button and response mapping. | ||
| fn create_message_dialog<S>(&self, title: S, message: S) -> adw::MessageDialog | ||
| where | ||
| S: AsRef<str>, | ||
| { | ||
| adw::MessageDialog::builder() | ||
| .title(title.as_ref()) | ||
| .heading(title.as_ref()) | ||
| .body(message.as_ref()) | ||
| .modal(true) | ||
| .build() | ||
| } | ||
| } | ||
|
|
||
| #[relm4::component(pub)] | ||
| impl Component for ExecutionDialog { | ||
| type Init = ExecutionDialogInit; | ||
| type Input = ExecutionDialogInput; | ||
| type Output = (); | ||
| type CommandOutput = ExecutionDialogCommandOutput; | ||
|
|
||
| view! { | ||
| #[root] | ||
| adw::Window { | ||
| set_modal: true, | ||
| set_resizable: false, | ||
|
|
||
| gtk::Box { | ||
| set_orientation: gtk::Orientation::Vertical, | ||
| set_spacing: 5, | ||
| set_margin_all: 50, | ||
|
|
||
| gtk::Spinner { | ||
| set_spinning: true, | ||
| }, | ||
| gtk::Label { | ||
| set_label: &lang::lookup("flow-execution-running"), | ||
| }, | ||
| }, | ||
| }, | ||
| } | ||
|
|
||
| fn init( | ||
| init: Self::Init, | ||
| root: &Self::Root, | ||
| sender: relm4::ComponentSender<Self>, | ||
| ) -> relm4::ComponentParts<Self> { | ||
| let model = ExecutionDialog; | ||
| let widgets = view_output!(); | ||
| let action = init.action; | ||
| let engine_list = init.engine_list.clone(); | ||
| let action_map = init.action_map.clone(); | ||
|
|
||
| // TODO: ask for input, then build `ActionConfiguration` and trigger | ||
| // sender.spawn_oneshot_command(move || { | ||
| // let mut outputs: Vec<HashMap<usize, ParameterValue>> = Vec::new(); | ||
| // let mut evidence = Vec::new(); | ||
|
|
||
| // for engine in engine_list.inner() { | ||
| // if engine.reset_state().is_err() { | ||
| // evidence.push(Evidence { | ||
| // label: String::from("WARNING: State Warning"), | ||
| // content: EvidenceContent::Textual(String::from("For this test execution, the state couldn't be correctly reset. Some results may not be accurate.")) | ||
| // }); | ||
| // } | ||
| // } | ||
|
|
||
| // for (step, action_config) in action.actions.iter().enumerate() { | ||
| // match action_config.execute( | ||
| // action_map.clone(), | ||
| // engine_list.clone(), | ||
| // outputs.clone(), | ||
| // ) { | ||
| // Ok((output, ev)) => { | ||
| // outputs.push(output); | ||
| // evidence = [evidence, ev].concat(); | ||
| // } | ||
| // Err(e) => { | ||
| // return ExecutionDialogCommandOutput::Failed(step + 1, e); | ||
| // } | ||
| // } | ||
| // } | ||
|
|
||
| // ExecutionDialogCommandOutput::Complete(evidence) | ||
| // }); | ||
|
|
||
| ComponentParts { model, widgets } | ||
| } | ||
|
|
||
| fn update( | ||
| &mut self, | ||
| message: Self::Input, | ||
| sender: relm4::ComponentSender<Self>, | ||
| root: &Self::Root, | ||
| ) { | ||
| match message { | ||
| ExecutionDialogInput::Close => root.destroy(), | ||
| ExecutionDialogInput::FailedToGenerateReport(reason) => { | ||
| let dialog = self.create_message_dialog( | ||
| lang::lookup("report-failed"), | ||
| lang::lookup_with_args("report-failed-message", { | ||
| let mut map = HashMap::new(); | ||
| map.insert("reason", reason.to_string().into()); | ||
| map | ||
| }), | ||
| ); | ||
| dialog.set_transient_for(Some(root)); | ||
| dialog.add_response("ok", &lang::lookup("ok")); | ||
| dialog.set_default_response(Some("ok")); | ||
| let sender_c = sender.clone(); | ||
| dialog.connect_response(None, move |dlg, _response| { | ||
| sender_c.input(ExecutionDialogInput::Close); | ||
| dlg.close(); | ||
| }); | ||
| dialog.set_visible(true); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| fn update_cmd( | ||
| &mut self, | ||
| message: Self::CommandOutput, | ||
| sender: relm4::ComponentSender<Self>, | ||
| root: &Self::Root, | ||
| ) { | ||
| match message { | ||
| ExecutionDialogCommandOutput::Complete(evidence) => { | ||
| log::info!("Execution complete."); | ||
|
|
||
| // Present save dialog | ||
| let dialog = gtk::FileDialog::builder() | ||
| .modal(true) | ||
| .title(lang::lookup("report-save-title")) | ||
| .initial_name(lang::lookup("report-default-name")) | ||
| .filters(&file_filters::filter_list(vec![ | ||
| file_filters::pdfs(), | ||
| file_filters::all(), | ||
| ])) | ||
| .build(); | ||
|
|
||
| let sender_c = sender.clone(); | ||
| dialog.save( | ||
| Some(root), | ||
| Some(&relm4::gtk::gio::Cancellable::new()), | ||
| move |res| { | ||
| if let Ok(file) = res { | ||
| let path = file.path().unwrap(); | ||
| if let Err(e) = report_generation::save_report( | ||
| path.with_extension("pdf"), | ||
| evidence.clone(), | ||
| ) { | ||
| // Failed to generate report | ||
| sender_c.input(ExecutionDialogInput::FailedToGenerateReport(e)); | ||
| return; | ||
| } else if let Err(e) = opener::open(path.with_extension("pdf")) { | ||
| log::warn!("Failed to open evidence: {e}"); | ||
| } | ||
| } | ||
| sender_c.input(ExecutionDialogInput::Close); | ||
| }, | ||
| ); | ||
| } | ||
|
|
||
| ExecutionDialogCommandOutput::Failed(step, reason) => { | ||
| log::warn!("Execution failed"); | ||
| let dialog = self.create_message_dialog( | ||
| lang::lookup("flow-execution-failed"), | ||
| lang::lookup_with_args("flow-execution-failed-message", { | ||
| let mut map = HashMap::new(); | ||
| map.insert("step", step.into()); | ||
| map.insert("reason", reason.to_string().into()); | ||
| map | ||
| }), | ||
| ); | ||
| dialog.set_transient_for(Some(root)); | ||
| dialog.add_response("ok", &lang::lookup("ok")); | ||
| dialog.set_default_response(Some("ok")); | ||
| let sender_c = sender.clone(); | ||
| dialog.connect_response(None, move |dlg, _response| { | ||
| sender_c.input(ExecutionDialogInput::Close); | ||
| dlg.close(); | ||
| }); | ||
| dialog.set_visible(true); | ||
| } | ||
| } | ||
| } | ||
| } |
Oops, something went wrong.