refactor: consolidate color_status into util, remove duplicate in queries/results

Author: eddietejeda

src/queries.rs

@@ -1,5 +1,5 @@
 use crate::api::ApiClient;
-use crossterm::style::{Color, Stylize};
+use crossterm::style::Stylize;
 use serde::{Deserialize, Serialize};
 
 const SQL_KEYWORDS: &[&str] = &[
@@ -141,16 +141,6 @@ struct ListResponse {
     next_cursor: Option<String>,
 }
 
-fn color_status(status: &str) -> String {
-    let color = match status {
-        "succeeded" => Color::Green,
-        "failed" => Color::Red,
-        "running" | "queued" | "pending" => Color::Yellow,
-        _ => Color::Reset,
-    };
-    status.with(color).to_string()
-}
-
 fn truncate_sql(sql: &str, max: usize) -> String {
     let flat = sql.split_whitespace().collect::<Vec<_>>().join(" ");
     if flat.chars().count() <= max {
@@ -193,7 +183,7 @@ pub fn list(
                     .map(|r| {
                         vec![
                             r.id.clone(),
-                            color_status(&r.status),
+                            crate::util::color_status(&r.status),
                             crate::util::format_date(&r.created_at),
                             r.execution_time_ms
                                 .map(|ms| ms.to_string())
@@ -241,7 +231,7 @@ fn print_detail(r: &QueryRun, format: &str) {
         "table" => {
             let label = |l: &str| format!("{:<14}", l).dark_grey().to_string();
             println!("{}{}", label("id:"), r.id);
-            println!("{}{}", label("status:"), color_status(&r.status));
+            println!("{}{}", label("status:"), crate::util::color_status(&r.status));
             println!(
                 "{}{}",
                 label("created:"),

src/results.rs

@@ -1,5 +1,5 @@
 use crate::api::ApiClient;
-use crossterm::style::{Color, Stylize};
+use crossterm::style::Stylize;
 use serde::{Deserialize, Serialize};
 
 #[derive(Deserialize, Serialize)]
@@ -22,17 +22,6 @@ struct ListResponse {
     has_more: bool,
 }
 
-fn color_status(status: &str) -> String {
-    let color = match status {
-        "ready" => Color::Green,
-        "failed" => Color::Red,
-        "pending" | "processing" => Color::Yellow,
-        "expired" => Color::DarkGrey,
-        _ => Color::Reset,
-    };
-    status.with(color).to_string()
-}
-
 pub fn list(workspace_id: &str, limit: Option<u32>, offset: Option<u32>, format: &str) {
     let api = ApiClient::new(Some(workspace_id));
 
@@ -59,7 +48,7 @@ pub fn list(workspace_id: &str, limit: Option<u32>, offset: Option<u32>, format:
                     .map(|r| {
                         let mut row = vec![
                             r.id.clone(),
-                            color_status(&r.status),
+                            crate::util::color_status(&r.status),
                             crate::util::format_date(&r.created_at),
                         ];
                         if has_rows {

src/util.rs

@@ -328,6 +328,20 @@ fn colorize_json_value(v: &str) -> String {
     format!("{colored}{comma}")
 }
 
+/// Color a status string for terminal output. Covers vocabulary from both
+/// query runs (succeeded/failed/running/queued/pending) and results (ready/expired/processing).
+pub fn color_status(status: &str) -> String {
+    use crossterm::style::{Color, Stylize};
+    let color = match status {
+        "succeeded" | "ready" => Color::Green,
+        "failed" => Color::Red,
+        "running" | "queued" | "pending" | "processing" => Color::Yellow,
+        "expired" => Color::DarkGrey,
+        _ => Color::Reset,
+    };
+    status.with(color).to_string()
+}
+
 /// Format an ISO date string compactly: "2024-03-15 14:23" (no seconds, no timezone).
 pub fn format_date(s: &str) -> String {
     let s = s.split('.').next().unwrap_or(s).trim_end_matches('Z');