From d57419d64857773cdb27da85eb354c0c2d28f45b Mon Sep 17 00:00:00 2001
From: slinkydeveloper <francescoguard@gmail.com>
Date: Thu, 21 Nov 2024 15:40:25 +0100
Subject: [PATCH] Add Killed invocation status to CLI

---
 cli/src/clients/datafusion_helpers.rs | 3 +++
 cli/src/ui/invocations.rs             | 1 +
 2 files changed, 4 insertions(+)

diff --git a/cli/src/clients/datafusion_helpers.rs b/cli/src/clients/datafusion_helpers.rs
index f7b781bf06..b72013e0ed 100644
--- a/cli/src/clients/datafusion_helpers.rs
+++ b/cli/src/clients/datafusion_helpers.rs
@@ -128,6 +128,7 @@ pub enum InvocationState {
     Running,
     Suspended,
     BackingOff,
+    Killed,
     Completed,
 }
 
@@ -142,6 +143,7 @@ impl FromStr for InvocationState {
             "suspended" => Self::Suspended,
             "backing-off" => Self::BackingOff,
             "completed" => Self::Completed,
+            "killed" => Self::Killed,
             _ => Self::Unknown,
         })
     }
@@ -157,6 +159,7 @@ impl Display for InvocationState {
             InvocationState::Running => write!(f, "running"),
             InvocationState::Suspended => write!(f, "suspended"),
             InvocationState::BackingOff => write!(f, "backing-off"),
+            InvocationState::Killed => write!(f, "killed"),
             InvocationState::Completed => write!(f, "completed"),
         }
     }
diff --git a/cli/src/ui/invocations.rs b/cli/src/ui/invocations.rs
index d1af069b53..0ba03447ea 100644
--- a/cli/src/ui/invocations.rs
+++ b/cli/src/ui/invocations.rs
@@ -113,6 +113,7 @@ pub fn invocation_status_style(status: InvocationState) -> Style {
         InvocationState::Suspended => DStyle::new().dim(),
         InvocationState::BackingOff => DStyle::new().red(),
         InvocationState::Completed => DStyle::new().blue(),
+        InvocationState::Killed => DStyle::new().red(),
     }
 }