use std::collections::HashMap;
use chrono::{DateTime, Utc};
use ironflow_store::models::{Run, RunStatus, TriggerKind};
use rust_decimal::Decimal;
use serde::{Deserialize, Serialize};
use uuid::Uuid;
use super::StepResponse;
#[cfg_attr(feature = "openapi", derive(utoipa::ToSchema))]
#[derive(Debug, Serialize, Deserialize)]
pub struct RunResponse {
pub id: Uuid,
pub workflow_name: String,
pub status: RunStatus,
pub trigger: TriggerKind,
pub error: Option<String>,
pub retry_count: u32,
pub max_retries: u32,
#[cfg_attr(feature = "openapi", schema(value_type = f64))]
pub cost_usd: Decimal,
pub duration_ms: u64,
pub created_at: DateTime<Utc>,
pub updated_at: DateTime<Utc>,
pub started_at: Option<DateTime<Utc>>,
pub completed_at: Option<DateTime<Utc>>,
pub handler_version: Option<String>,
#[serde(default, skip_serializing_if = "HashMap::is_empty")]
pub labels: HashMap<String, String>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub scheduled_at: Option<DateTime<Utc>>,
}
impl From<Run> for RunResponse {
fn from(run: Run) -> Self {
RunResponse {
id: run.id,
workflow_name: run.workflow_name,
status: run.status.state,
trigger: run.trigger,
error: run.error,
retry_count: run.retry_count,
max_retries: run.max_retries,
cost_usd: run.cost_usd,
duration_ms: run.duration_ms,
created_at: run.created_at,
updated_at: run.updated_at,
started_at: run.started_at,
completed_at: run.completed_at,
handler_version: run.handler_version,
labels: run.labels,
scheduled_at: run.scheduled_at,
}
}
}
#[cfg_attr(feature = "openapi", derive(utoipa::ToSchema))]
#[derive(Debug, Serialize)]
pub struct RunDetailResponse {
pub run: RunResponse,
pub steps: Vec<StepResponse>,
pub payload: serde_json::Value,
}
#[cfg_attr(feature = "openapi", derive(utoipa::IntoParams, utoipa::ToSchema))]
#[derive(Debug, Deserialize)]
pub struct ListRunsQuery {
pub workflow: Option<String>,
pub status: Option<RunStatus>,
pub has_steps: Option<bool>,
pub label: Option<String>,
pub page: Option<u32>,
pub per_page: Option<u32>,
}
impl ListRunsQuery {
pub fn parse_labels(&self) -> Option<HashMap<String, String>> {
self.label.as_ref().and_then(|raw| {
let mut map = HashMap::new();
for entry in raw.split(',') {
let entry = entry.trim();
if let Some((k, v)) = entry.split_once(':') {
map.insert(k.to_string(), v.to_string());
}
}
if map.is_empty() { None } else { Some(map) }
})
}
}