From e49478b9cbaa82206b4b6164299f6173365da57b Mon Sep 17 00:00:00 2001
From: Eike <eikek@posteo.de>
Date: Wed, 17 Jun 2026 16:26:45 +0200
Subject: [PATCH] Follow logs

---
 src/cli/cmd/job/logs.rs | 58 ++++++++++++++++++++++++++++++++---------
 1 file changed, 45 insertions(+), 13 deletions(-)

diff --git a/src/cli/cmd/job/logs.rs b/src/cli/cmd/job/logs.rs
index 6cc4bb4..51b97c3 100644
--- a/src/cli/cmd/job/logs.rs
+++ b/src/cli/cmd/job/logs.rs
@@ -1,5 +1,8 @@
 use super::Context;
-use crate::{cli::sink::Error as SinkError, data::simple_message::SimpleMessage, httpclient};
+use crate::{cli::sink::Error as SinkError, httpclient};
+use std::time::Duration;
+use tokio::signal;
+use tokio::time::sleep;
 
 use clap::{Parser, ValueHint};
 
@@ -10,8 +13,17 @@ use snafu::{ResultExt, Snafu};
 /// List the logs of a job.
 #[derive(Parser, Debug)]
 pub struct Input {
+    /// The job name/id to get logs for
     #[arg(value_hint=ValueHint::Other)]
     pub job_id: String,
+
+    /// Periodically retrieves logs
+    #[arg(long, short, default_value_t = false)]
+    pub follow: bool,
+
+    /// The interval in seconds to wait between calls for logs
+    #[arg(long, default_value_t = 2)]
+    pub follow_interval: u8,
 }
 
 #[derive(Debug, Snafu)]
@@ -25,24 +37,44 @@ pub enum Error {
 
 impl Input {
     pub async fn exec(&self, ctx: Context) -> Result<(), Error> {
+        if self.follow {
+            self.follow_logs(ctx).await
+        } else {
+            self.show_logs(&ctx, 0).await?;
+            Ok(())
+        }
+    }
+
+    async fn show_logs(&self, ctx: &Context, seen: usize) -> Result<usize, Error> {
         let result = ctx
             .client
             .session_logs(&self.job_id)
             .await
             .context(HttpClientSnafu)?;
+        if let Some(lines_blob) = result.0.get("amalthea-session") {
+            let lines: Vec<&str> = lines_blob.lines().collect();
+            if lines.len() > seen {
+                for line in &lines[seen..] {
+                    println!("{}", line);
+                }
+                return Ok(lines.len());
+            }
+        }
+        return Ok(seen);
+    }
 
-        if let Some(lines) = result.0.get("amalthea-session") {
-            ctx.write_result(&SimpleMessage {
-                message: lines.to_string(),
-            })
-            .await
-            .context(WriteResultSnafu)
-        } else {
-            ctx.write_result(&SimpleMessage {
-                message: "No logs available.".to_string(),
-            })
-            .await
-            .context(WriteResultSnafu)
+    async fn follow_logs(&self, ctx: Context) -> Result<(), Error> {
+        let mut seen: usize = self.show_logs(&ctx, 0).await?;
+        loop {
+            tokio::select! {
+                _ = signal::ctrl_c() => {
+                    eprintln!("Interrupted, exiting.");
+                    break Ok(());
+                }
+                _ = sleep(Duration::from_secs(self.follow_interval as u64)) => {
+                    seen = self.show_logs(&ctx, seen).await?;
+                }
+            }
         }
     }
 }