use anyhow::Result;
use clap::Args;
use colored::Colorize;
use crate::client::NsgClient;
use crate::config::Credentials;
#[derive(Debug, Args)]
pub struct ListCommand {
#[arg(long, help = "Fetch detailed status for each job (slower)")]
detailed: bool,
}
impl ListCommand {
pub fn execute(self) -> Result<()> {
let credentials = Credentials::load()?;
let client = NsgClient::new(credentials.clone())?;
println!("{}", "NSG Job List".bold().cyan());
println!("{}", "=".repeat(80).cyan());
println!();
println!("{} Fetching jobs for user: {}", "→".cyan(), credentials.username.bold());
println!();
let jobs = client.list_jobs()?;
if jobs.is_empty() {
println!("{}", "No jobs found".yellow());
println!();
println!("You can submit a test job with:");
println!(" {}", "nsg submit <zip_file> --tool PY_EXPANSE".cyan());
return Ok(());
}
println!("Found {} job(s)", jobs.len().to_string().bold());
println!();
println!("{}", "=".repeat(80));
for (i, job) in jobs.iter().enumerate() {
println!();
println!("Job #{}", (i + 1).to_string().bold());
println!(" ID: {}", job.job_id.cyan());
if self.detailed {
println!(" {}", "Fetching details...".dimmed());
match client.get_job_status(&job.url) {
Ok(status) => {
let stage_icon = get_stage_icon(&status.job_stage);
println!(" Status: {} {}", stage_icon, status.job_stage.bold());
if status.failed {
println!(" Failed: {} YES", "✗".red().bold());
}
if let Some(date) = &status.date_submitted {
println!(" Submitted: {}", format_timestamp(date));
}
if !status.messages.is_empty() {
if let Some(latest) = status.messages.last() {
println!(" Latest: [{}] {}", latest.stage, truncate(&latest.text, 100));
}
}
}
Err(_) => {
println!(" Status: {} (failed to fetch)", "?".yellow());
}
}
} else {
println!(" Status: {} (use --detailed for full status)", "?".dimmed());
}
println!(" URL: {}", job.url.dimmed());
println!("{}", "=".repeat(80));
}
println!();
println!("{}", "Commands:".bold());
println!(" Check job status: {}", "nsg status <JOB_ID>".cyan());
println!(" Download results: {}", "nsg download <JOB_ID>".cyan());
println!();
Ok(())
}
}
fn get_stage_icon(stage: &str) -> String {
match stage {
"COMPLETED" => "✓".green().bold().to_string(),
"RUNNING" | "RUN" => "⟳".yellow().bold().to_string(),
"QUEUE" | "SUBMITTED" => "⏳".cyan().to_string(),
"FAILED" => "✗".red().bold().to_string(),
_ => "?".dimmed().to_string(),
}
}
fn format_timestamp(ts: &str) -> String {
use chrono::{DateTime, Utc};
if let Ok(dt) = ts.parse::<DateTime<Utc>>() {
dt.format("%Y-%m-%d %H:%M:%S UTC").to_string()
} else {
ts.to_string()
}
}
fn truncate(s: &str, max_len: usize) -> String {
if s.len() <= max_len {
s.to_string()
} else {
format!("{}...", &s[..max_len])
}
}