support getting stderr

Author: Panaetius

coman/src/app/messages.rs

@@ -49,8 +49,9 @@ pub enum CscsMsg {
 }
 #[derive(Debug, PartialEq)]
 pub enum JobMsg {
-    ShowLog(usize),
-    CloseLog,
+    Show(usize),
+    Switch,
+    Close,
 }
 #[derive(Debug, Clone, Copy, Default, PartialEq, PartialOrd, Eq, Ord, strum::Display)]
 pub enum View {

coman/src/app/model.rs

@@ -26,7 +26,7 @@ use crate::{
     },
     cscs::{
         handlers::{cscs_login, cscs_system_set},
-        ports::TreeAction,
+        ports::{JobLogAction, TreeAction},
     },
     trace_dbg,
     util::ui::{draw_area_in_absolute, draw_area_in_absolute_fixed_height},
@@ -56,7 +56,7 @@ where
 
     /// Triggers watching job logs
     /// sending None stops watching
-    pub job_log_tx: mpsc::Sender<Option<usize>>,
+    pub job_log_tx: mpsc::Sender<JobLogAction>,
 
     /// Allows creating user events based on messages
     pub user_event_tx: mpsc::Sender<UserEvent>,
@@ -74,7 +74,7 @@ where
         bridge: TerminalBridge<T>,
         error_tx: mpsc::Sender<String>,
         select_system_tx: mpsc::Sender<()>,
-        job_log_tx: mpsc::Sender<Option<usize>>,
+        job_log_tx: mpsc::Sender<JobLogAction>,
         user_event_tx: mpsc::Sender<UserEvent>,
         file_tree_tx: mpsc::Sender<TreeAction>,
     ) -> Self {
@@ -305,7 +305,7 @@ where
     }
     fn handle_job_msg(&mut self, msg: JobMsg) -> Option<Msg> {
         match msg {
-            JobMsg::ShowLog(jobid) => {
+            JobMsg::Show(jobid) => {
                 if self.app.mounted(&Id::WorkloadList) {
                     assert!(self.app.umount(&Id::WorkloadList).is_ok());
                 }
@@ -319,11 +319,18 @@ where
                 assert!(self.app.active(&Id::WorkloadLogs).is_ok());
                 let job_log_tx = self.job_log_tx.clone();
                 tokio::spawn(async move {
-                    job_log_tx.send(Some(jobid)).await.unwrap();
+                    job_log_tx.send(JobLogAction::Job(jobid)).await.unwrap();
                 });
                 None
             }
-            JobMsg::CloseLog => {
+            JobMsg::Switch => {
+                let job_log_tx = self.job_log_tx.clone();
+                tokio::spawn(async move {
+                    job_log_tx.send(JobLogAction::SwitchLog).await.unwrap();
+                });
+                None
+            }
+            JobMsg::Close => {
                 if self.app.mounted(&Id::WorkloadLogs) {
                     assert!(self.app.umount(&Id::WorkloadLogs).is_ok());
                 }
@@ -338,7 +345,7 @@ where
                 let job_log_tx = self.job_log_tx.clone();
                 tokio::spawn(async move {
                     // stopp polling for logs
-                    job_log_tx.send(None).await.unwrap();
+                    job_log_tx.send(JobLogAction::Stop).await.unwrap();
                 });
                 None
             }

coman/src/cli.rs

@@ -63,7 +63,11 @@ pub enum CscsJobCommands {
     #[clap(alias("g"), about = "Get metadata for a specific job [aliases: g]")]
     Get { job_id: i64 },
     #[clap(about = "Get the stdout of a job")]
-    Log { job_id: i64 },
+    Log {
+        #[clap(short, long, action, help = "whether to get stderr instead of stdout")]
+        stderr: bool,
+        job_id: i64,
+    },
 
     #[clap(alias("s"), about = "Submit a new compute job [aliases: s]")]
     Submit {

coman/src/components/toolbar.rs

@@ -8,7 +8,7 @@ use crate::{
     },
     trace_dbg,
 };
-const WORKLOAD_TOOLTIP: &str = "q: quit, Esc: close/back, l: logs, f: File view, x: menu, ?: help";
+const WORKLOAD_TOOLTIP: &str = "q: quit, Esc: close/back, l: logs, f: File view, x: menu, tab: switch view, ?: help";
 const FILETREE_TOOLTIP: &str = "q: quit, ↑↓: navigate,←→: collapse/expand, x: menu, ?: help";
 
 #[derive(MockComponent)]

coman/src/components/workload_list.rs

@@ -77,7 +77,7 @@ impl Component<Msg, UserEvent> for WorkloadList {
                     && !self.jobs.is_empty()
                 {
                     let job = self.jobs[index].clone();
-                    return Some(Msg::Job(JobMsg::ShowLog(job.id)));
+                    return Some(Msg::Job(JobMsg::Show(job.id)));
                 }
                 CmdResult::None
             }

coman/src/components/workload_log.rs

@@ -6,35 +6,32 @@ use tuirealm::{
     props::{Alignment, BorderType, Borders, Color, PropPayload, PropValue, TextSpan},
 };
 
-use crate::{
-    app::{
-        messages::{JobMsg, Msg},
-        user_events::{CscsEvent, UserEvent},
-    },
-    trace_dbg,
+use crate::app::{
+    messages::{JobMsg, Msg},
+    user_events::{CscsEvent, UserEvent},
 };
 
 #[derive(MockComponent)]
 pub struct WorkloadLog {
     component: Textarea,
+    stderr: bool,
 }
 
 impl WorkloadLog {
     pub fn new() -> Self {
         Self {
             component: Textarea::default()
                 .borders(Borders::default().modifiers(BorderType::Rounded).color(Color::Yellow))
-                .title("Workload Log", Alignment::Center)
+                .title("Workload Log (stdout)", Alignment::Center)
                 .step(4),
+            stderr: false,
         }
     }
 }
 impl Component<Msg, UserEvent> for WorkloadLog {
     fn on(&mut self, ev: tuirealm::Event<UserEvent>) -> Option<Msg> {
         let _ = match ev {
             Event::User(UserEvent::Cscs(CscsEvent::GotJobLog(log))) => {
-                let _ = trace_dbg!("got log component");
-                let log = trace_dbg!(log);
                 self.attr(
                     Attribute::Text,
                     AttrValue::Payload(PropPayload::Vec(
@@ -51,8 +48,25 @@ impl Component<Msg, UserEvent> for WorkloadLog {
             Event::Keyboard(KeyEvent { code: Key::PageUp, .. }) => self.perform(Cmd::Scroll(Direction::Up)),
             Event::Keyboard(KeyEvent { code: Key::Home, .. }) => self.perform(Cmd::GoTo(Position::Begin)),
             Event::Keyboard(KeyEvent { code: Key::End, .. }) => self.perform(Cmd::GoTo(Position::End)),
+            Event::Keyboard(KeyEvent { code: Key::Tab, .. }) => {
+                self.stderr = !self.stderr;
+                if self.stderr {
+                    self.attr(
+                        Attribute::Title,
+                        AttrValue::Title(("Workload Log (stderr)".to_owned(), Alignment::Center)),
+                    );
+                } else {
+                    self.attr(
+                        Attribute::Title,
+                        AttrValue::Title(("Workload Log (stdout)".to_owned(), Alignment::Center)),
+                    );
+                }
+                // empty log view
+                self.attr(Attribute::Text, AttrValue::Payload(PropPayload::Vec(vec![])));
+                return Some(Msg::Job(JobMsg::Switch));
+            }
             Event::Keyboard(KeyEvent { code: Key::Esc, .. }) => {
-                return Some(Msg::Job(JobMsg::CloseLog));
+                return Some(Msg::Job(JobMsg::Close));
             }
             _ => CmdResult::None,
         };

coman/src/cscs/cli.rs

@@ -87,10 +87,11 @@ pub(crate) async fn cli_cscs_job_detail(
 
 pub(crate) async fn cli_cscs_job_log(
     job_id: i64,
+    stderr: bool,
     system: Option<String>,
     platform: Option<ComputePlatform>,
 ) -> Result<()> {
-    match cscs_job_log(job_id, system, platform).await {
+    match cscs_job_log(job_id, stderr, system, platform).await {
         Ok(content) => {
             println!("{}", content);
             Ok(())

coman/src/cscs/handlers.rs

@@ -117,7 +117,12 @@ pub async fn cscs_job_details(
     }
 }
 
-pub async fn cscs_job_log(job_id: i64, system: Option<String>, platform: Option<ComputePlatform>) -> Result<String> {
+pub async fn cscs_job_log(
+    job_id: i64,
+    stderr: bool,
+    system: Option<String>,
+    platform: Option<ComputePlatform>,
+) -> Result<String> {
     match get_access_token().await {
         Ok(access_token) => {
             let api_client = CscsApi::new(access_token.0, platform).unwrap();
@@ -127,9 +132,12 @@ pub async fn cscs_job_log(job_id: i64, system: Option<String>, platform: Option<
             if job.is_none() {
                 return Err(eyre!("couldn't find job {}", job_id));
             }
-            api_client
-                .tail(current_system, PathBuf::from(job.unwrap().stdout), 100)
-                .await
+            let path = if stderr {
+                PathBuf::from(job.unwrap().stderr)
+            } else {
+                PathBuf::from(job.unwrap().stdout)
+            };
+            api_client.tail(current_system, path, 100).await
         }
         Err(e) => Err(e),
     }

coman/src/cscs/ports.rs

@@ -154,17 +154,25 @@ impl PollAsync<UserEvent> for AsyncSelectSystemPort {
     }
 }
 
+pub enum JobLogAction {
+    Job(usize),
+    SwitchLog,
+    Stop,
+}
+
 /// This port handles polling the logs of a CSCS job
 pub(crate) struct AsyncJobLogPort {
-    receiver: mpsc::Receiver<Option<usize>>,
+    receiver: mpsc::Receiver<JobLogAction>,
     current_job: Option<usize>,
+    stderr: bool,
 }
 
 impl AsyncJobLogPort {
-    pub fn new(receiver: mpsc::Receiver<Option<usize>>) -> Self {
+    pub fn new(receiver: mpsc::Receiver<JobLogAction>) -> Self {
         Self {
             receiver,
             current_job: None,
+            stderr: false,
         }
     }
 }
@@ -176,11 +184,21 @@ impl PollAsync<UserEvent> for AsyncJobLogPort {
         }
         if !self.receiver.is_empty() {
             if let Some(val) = self.receiver.recv().await {
-                self.current_job = val;
+                match val {
+                    JobLogAction::Job(jobid) => {
+                        self.current_job = Some(jobid);
+                    }
+                    JobLogAction::SwitchLog => {
+                        self.stderr = !self.stderr;
+                    }
+                    JobLogAction::Stop => {
+                        self.current_job = None;
+                    }
+                }
             }
             Ok(Some(Event::None))
         } else if let Some(job_id) = self.current_job {
-            match cscs_job_log(job_id as i64, None, None).await {
+            match cscs_job_log(job_id as i64, self.stderr, None, None).await {
                 Ok(log) => {
                     let log = trace_dbg!(log);
                     Ok(Some(Event::User(UserEvent::Cscs(CscsEvent::GotJobLog(log)))))

coman/src/main.rs

@@ -63,7 +63,9 @@ async fn main() -> Result<()> {
                 cli::CscsCommands::Job { command } => match command {
                     cli::CscsJobCommands::List => cli_cscs_job_list(system, platform).await?,
                     cli::CscsJobCommands::Get { job_id } => cli_cscs_job_detail(job_id, system, platform).await?,
-                    cli::CscsJobCommands::Log { job_id } => cli_cscs_job_log(job_id, system, platform).await?,
+                    cli::CscsJobCommands::Log { job_id, stderr } => {
+                        cli_cscs_job_log(job_id, stderr, system, platform).await?
+                    }
                     cli::CscsJobCommands::Submit {
                         script_file,
                         image,