runkon-flow 0.6.1-alpha

use std::collections::HashMap;
use std::sync::Arc;
use std::time::Duration;

use serde::{Deserialize, Serialize};

use crate::extensions::Extensions;
use crate::status::{WorkflowRunStatus, WorkflowStepStatus};

/// A step key is a `(name, iteration)` pair used for skip-set and step-map lookups.
#[allow(dead_code)]
pub(crate) type StepKey = (String, u32);

/// Describes what a workflow run is currently blocked on when in `Waiting` status.
#[cfg_attr(feature = "utoipa", derive(utoipa::ToSchema))]
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(tag = "type", rename_all = "snake_case")]
pub enum BlockedOn {
    HumanApproval {
        gate_name: String,
        prompt: Option<String>,
        #[serde(default)]
        options: Vec<String>,
    },
    HumanReview {
        gate_name: String,
        prompt: Option<String>,
        #[serde(default)]
        options: Vec<String>,
    },
    PrApproval {
        gate_name: String,
        approvals_needed: u32,
    },
    PrChecks {
        gate_name: String,
    },
}

/// A workflow run record from the database.
#[cfg_attr(feature = "utoipa", derive(utoipa::ToSchema))]
#[derive(Debug, Clone, Serialize)]
pub struct WorkflowRun {
    pub id: String,
    pub workflow_name: String,
    pub parent_run_id: String,
    pub status: WorkflowRunStatus,
    pub dry_run: bool,
    pub trigger: String,
    pub started_at: String,
    pub ended_at: Option<String>,
    pub result_summary: Option<String>,
    pub error: Option<String>,
    pub definition_snapshot: Option<String>,
    pub inputs: HashMap<String, String>,
    pub parent_workflow_run_id: Option<String>,
    pub iteration: i64,
    pub blocked_on: Option<BlockedOn>,
    pub workflow_title: Option<String>,
    pub total_duration_ms: Option<i64>,
    pub dismissed: bool,
    #[serde(skip)]
    pub extensions: Extensions,
    #[serde(skip)]
    pub owner_token: Option<String>,
    #[serde(skip)]
    pub lease_until: Option<String>,
    #[serde(skip)]
    pub generation: i64,
}

/// Extract the human-readable title from a workflow definition snapshot JSON string.
pub fn extract_workflow_title(snapshot: Option<&str>) -> Option<String> {
    let s = snapshot?;
    match serde_json::from_str::<serde_json::Value>(s) {
        Ok(v) => v["title"].as_str().map(String::from),
        Err(e) => {
            tracing::warn!(
                "Malformed definition_snapshot JSON — could not extract workflow title: {e}"
            );
            None
        }
    }
}

impl WorkflowRun {
    /// Whether this run was triggered by a workflow hook (prevents infinite chains).
    pub fn is_triggered_by_hook(&self) -> bool {
        self.trigger == "hook"
    }

    /// Returns the human-readable display name for this run.
    pub fn display_name(&self) -> &str {
        self.workflow_title
            .as_deref()
            .unwrap_or(&self.workflow_name)
    }
}

/// A workflow step execution record from the database.
#[cfg_attr(feature = "utoipa", derive(utoipa::ToSchema))]
#[derive(Debug, Clone, Default, Serialize)]
pub struct WorkflowRunStep {
    pub id: String,
    pub workflow_run_id: String,
    pub step_name: String,
    pub role: String,
    pub can_commit: bool,
    pub condition_expr: Option<String>,
    pub status: WorkflowStepStatus,
    pub child_run_id: Option<String>,
    pub position: i64,
    pub started_at: Option<String>,
    pub ended_at: Option<String>,
    pub result_text: Option<String>,
    pub condition_met: Option<bool>,
    pub iteration: i64,
    pub parallel_group_id: Option<String>,
    pub context_out: Option<String>,
    pub markers_out: Option<String>,
    pub retry_count: i64,
    pub gate_type: Option<String>,
    pub gate_prompt: Option<String>,
    pub gate_timeout: Option<String>,
    pub gate_approved_by: Option<String>,
    pub gate_approved_at: Option<String>,
    pub gate_feedback: Option<String>,
    pub structured_output: Option<String>,
    pub output_file: Option<String>,
    pub gate_options: Option<String>,
    pub gate_selections: Option<String>,
    pub fan_out_total: Option<i64>,
    pub fan_out_completed: i64,
    pub fan_out_failed: i64,
    pub fan_out_skipped: i64,
    pub step_error: Option<String>,
}

/// Lightweight summary of the currently-running step for a workflow run.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct WorkflowStepSummary {
    pub step_name: String,
    pub iteration: i64,
    pub workflow_chain: Vec<String>,
}

/// Configuration for workflow execution.
#[derive(Clone)]
pub struct WorkflowExecConfig {
    pub poll_interval: Duration,
    pub step_timeout: Duration,
    pub fail_fast: bool,
    pub dry_run: bool,
    pub shutdown: Option<std::sync::Arc<std::sync::atomic::AtomicBool>>,
    /// Event sinks that receive observability events after each state transition.
    /// Defaults to empty (no sinks). Use `..WorkflowExecConfig::default()` spread
    /// syntax to leave this unset when you don't need events.
    pub event_sinks: Vec<Arc<dyn crate::events::EventSink>>,
    /// Lease TTL passed to `acquire_lease`. Defaults to 30s (LEASE_TTL_SECONDS).
    /// Override in tests to use a shorter TTL.
    pub lease_ttl_secs: i64,
    /// How often the background refresh thread renews the lease. Defaults to 10s.
    /// Override in tests to exercise refresh behaviour without long waits.
    pub lease_refresh_interval: Duration,
}

impl std::fmt::Debug for WorkflowExecConfig {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("WorkflowExecConfig")
            .field("poll_interval", &self.poll_interval)
            .field("step_timeout", &self.step_timeout)
            .field("fail_fast", &self.fail_fast)
            .field("dry_run", &self.dry_run)
            .field("shutdown", &self.shutdown)
            .field(
                "event_sinks",
                &format_args!("[{} sink(s)]", self.event_sinks.len()),
            )
            .field("lease_ttl_secs", &self.lease_ttl_secs)
            .field("lease_refresh_interval", &self.lease_refresh_interval)
            .finish()
    }
}

impl Default for WorkflowExecConfig {
    fn default() -> Self {
        Self {
            poll_interval: Duration::from_secs(5),
            step_timeout: Duration::from_secs(12 * 60 * 60),
            fail_fast: true,
            dry_run: false,
            shutdown: None,
            event_sinks: vec![],
            lease_ttl_secs: 30,
            lease_refresh_interval: Duration::from_secs(10),
        }
    }
}

/// Result of executing a workflow.
#[derive(Debug, Clone)]
pub struct WorkflowResult {
    pub workflow_run_id: String,
    pub workflow_name: String,
    pub all_succeeded: bool,
    pub total_duration_ms: i64,
    pub extensions: Extensions,
}

/// Input describing a successfully completed step, passed to `record_step_success`.
///
/// Groups the step output data that previously made call sites unwieldy.
/// Does not include `step_key` — that is an execution bookkeeping concern kept
/// as a separate parameter. `iteration` is included because it is needed to
/// populate `ContextEntry`.
#[derive(Debug, Clone, Default)]
pub struct StepSuccess {
    pub step_name: String,
    pub result_text: Option<String>,
    /// Executor-provided metric metadata (keys from `runkon_flow::constants::metadata_keys`).
    pub metadata: HashMap<String, String>,
    pub markers: Vec<String>,
    pub context: String,
    pub child_run_id: Option<String>,
    pub iteration: u32,
    pub structured_output: Option<String>,
    pub output_file: Option<String>,
}

impl StepSuccess {
    /// Build a [`StepSuccess`] from an [`ActionOutput`] plus execution bookkeeping.
    pub fn from_action_output(
        output: &crate::traits::action_executor::ActionOutput,
        step_name: String,
        context: String,
        iteration: u32,
        output_file: Option<String>,
    ) -> Self {
        Self {
            step_name,
            result_text: output.result_text.clone(),
            metadata: output.metadata.clone(),
            markers: output.markers.clone(),
            context,
            child_run_id: output.child_run_id.clone(),
            iteration,
            structured_output: output.structured_output.clone(),
            output_file,
        }
    }

    /// Build a [`StepSuccess`] from a database [`WorkflowRunStep`] record.
    pub fn from_workflow_run_step(
        step_name: String,
        step: &WorkflowRunStep,
        markers: Vec<String>,
        context: String,
        iteration: u32,
    ) -> Self {
        Self {
            step_name,
            result_text: step.result_text.clone(),
            markers,
            context,
            child_run_id: step.child_run_id.clone(),
            structured_output: step.structured_output.clone(),
            output_file: step.output_file.clone(),
            iteration,
            metadata: HashMap::new(),
        }
    }
}

/// Result of a single step execution (kept in memory during execution).
#[derive(Debug, Clone, Default)]
pub struct StepResult {
    pub step_name: String,
    pub status: WorkflowStepStatus,
    pub result_text: Option<String>,
    pub markers: Vec<String>,
    pub context: String,
    pub child_run_id: Option<String>,
    pub structured_output: Option<String>,
    pub output_file: Option<String>,
}

impl StepResult {
    /// Create a failed StepResult with the given error text.
    pub fn failed(step_name: &str, result_text: String) -> Self {
        Self {
            step_name: step_name.to_string(),
            status: WorkflowStepStatus::Failed,
            result_text: Some(result_text),
            ..Self::default()
        }
    }

    /// Create a skipped StepResult.
    pub fn skipped(step_name: &str) -> Self {
        Self {
            step_name: step_name.to_string(),
            status: WorkflowStepStatus::Skipped,
            ..Self::default()
        }
    }

    /// Create a completed StepResult without per-step metrics (for resume paths).
    pub fn completed_without_metrics(success: &StepSuccess) -> Self {
        Self::completed(success)
    }

    /// Create a completed StepResult from a [`StepSuccess`] description.
    pub fn completed(success: &StepSuccess) -> Self {
        Self {
            step_name: success.step_name.clone(),
            status: WorkflowStepStatus::Completed,
            result_text: success.result_text.clone(),
            markers: success.markers.clone(),
            context: success.context.clone(),
            child_run_id: success.child_run_id.clone(),
            structured_output: success.structured_output.clone(),
            output_file: success.output_file.clone(),
        }
    }
}

/// An entry in the accumulated context history.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ContextEntry {
    pub step: String,
    pub iteration: u32,
    pub context: String,
    #[serde(default)]
    pub markers: Vec<String>,
    #[serde(default)]
    pub structured_output: Option<String>,
    #[serde(default)]
    pub output_file: Option<String>,
}

impl From<StepSuccess> for ContextEntry {
    fn from(success: StepSuccess) -> Self {
        Self {
            step: success.step_name,
            iteration: success.iteration,
            context: success.context,
            markers: success.markers,
            structured_output: success.structured_output,
            output_file: success.output_file,
        }
    }
}

/// A single row in the `workflow_run_step_fan_out_items` table.
#[derive(Debug, Clone, Serialize, Deserialize)]
#[cfg_attr(feature = "utoipa", derive(utoipa::ToSchema))]
pub struct FanOutItemRow {
    pub id: String,
    pub step_run_id: String,
    pub item_type: String,
    pub item_id: String,
    pub item_ref: String,
    pub child_run_id: Option<String>,
    pub status: String,
    pub dispatched_at: Option<String>,
    pub completed_at: Option<String>,
    /// Per-item context map, merged into child workflow inputs as `item.<key>`.
    #[serde(default)]
    pub context: std::collections::HashMap<String, String>,
}

#[cfg(test)]
mod tests {
    use std::collections::HashMap;

    use super::{StepResult, StepSuccess, WorkflowRunStep};
    use crate::status::WorkflowStepStatus;

    #[test]
    fn step_result_failed_sets_status_and_text() {
        let r = StepResult::failed("plan", "out of tokens".to_string());
        assert_eq!(r.step_name, "plan");
        assert_eq!(r.status, WorkflowStepStatus::Failed);
        assert_eq!(r.result_text, Some("out of tokens".to_string()));
        assert!(r.markers.is_empty());
        assert_eq!(r.context, "");
    }

    #[test]
    fn step_result_skipped_sets_status_and_defaults() {
        let r = StepResult::skipped("lint");
        assert_eq!(r.step_name, "lint");
        assert_eq!(r.status, WorkflowStepStatus::Skipped);
        assert!(r.result_text.is_none());
        assert!(r.markers.is_empty());
        assert_eq!(r.context, "");
    }

    #[test]
    fn step_result_completed_sets_all_fields() {
        let success = StepSuccess {
            step_name: "review".to_string(),
            result_text: Some("looks good".to_string()),
            markers: vec!["approved".to_string()],
            context: "ctx".to_string(),
            child_run_id: Some("child-1".to_string()),
            structured_output: Some(r#"{"ok":true}"#.to_string()),
            output_file: Some("/tmp/out".to_string()),
            ..StepSuccess::default()
        };
        let r = StepResult::completed(&success);
        assert_eq!(r.step_name, "review");
        assert_eq!(r.status, WorkflowStepStatus::Completed);
        assert_eq!(r.result_text, Some("looks good".to_string()));
        assert_eq!(r.markers, vec!["approved"]);
        assert_eq!(r.context, "ctx");
        assert_eq!(r.child_run_id, Some("child-1".to_string()));
        assert_eq!(r.structured_output, Some(r#"{"ok":true}"#.to_string()));
        assert_eq!(r.output_file, Some("/tmp/out".to_string()));
    }

    #[test]
    fn completed_without_metrics_delegates_to_completed() {
        let success = StepSuccess {
            step_name: "restore".to_string(),
            result_text: Some("ok".to_string()),
            markers: vec!["done".to_string()],
            context: "restored".to_string(),
            ..StepSuccess::default()
        };
        let r = StepResult::completed_without_metrics(&success);
        assert_eq!(r.step_name, "restore");
        assert_eq!(r.status, WorkflowStepStatus::Completed);
        assert_eq!(r.result_text, Some("ok".to_string()));
        assert_eq!(r.markers, vec!["done"]);
        assert_eq!(r.context, "restored");
    }

    #[test]
    fn step_success_into_context_entry_maps_all_fields() {
        let success = StepSuccess {
            step_name: "my-step".to_string(),
            iteration: 7,
            context: "ctx-body".to_string(),
            markers: vec!["m1".to_string(), "m2".to_string()],
            structured_output: Some(r#"{"k":"v"}"#.to_string()),
            output_file: Some("/tmp/out".to_string()),
            result_text: Some("rt".to_string()),
            child_run_id: Some("child-1".to_string()),
            ..StepSuccess::default()
        };
        let entry: super::ContextEntry = success.into();
        assert_eq!(entry.step, "my-step", "step should come from step_name");
        assert_eq!(entry.iteration, 7);
        assert_eq!(entry.context, "ctx-body");
        assert_eq!(entry.markers, vec!["m1", "m2"]);
        assert_eq!(entry.structured_output, Some(r#"{"k":"v"}"#.to_string()));
        assert_eq!(entry.output_file, Some("/tmp/out".to_string()));
    }

    #[test]
    fn from_workflow_run_step_maps_fields() {
        let step = WorkflowRunStep {
            result_text: Some("all good".to_string()),
            child_run_id: Some("child-1".to_string()),
            structured_output: Some(r#"{"ok":true}"#.to_string()),
            output_file: Some("/tmp/out".to_string()),
            ..WorkflowRunStep::default()
        };
        let success = StepSuccess::from_workflow_run_step(
            "child-step".to_string(),
            &step,
            vec!["m1".to_string(), "m2".to_string()],
            "ctx-body".to_string(),
            7,
        );
        assert_eq!(success.step_name, "child-step");
        assert_eq!(success.result_text, Some("all good".to_string()));
        assert_eq!(success.markers, vec!["m1", "m2"]);
        assert_eq!(success.context, "ctx-body");
        assert_eq!(success.child_run_id, Some("child-1".to_string()));
        assert_eq!(
            success.structured_output,
            Some(r#"{"ok":true}"#.to_string())
        );
        assert_eq!(success.output_file, Some("/tmp/out".to_string()));
        assert_eq!(success.iteration, 7);
        assert!(success.metadata.is_empty());
    }

    #[test]
    fn from_action_output_maps_all_fields() {
        use crate::constants::metadata_keys;
        let mut metadata = HashMap::new();
        metadata.insert(metadata_keys::COST_USD.to_string(), "0.05".to_string());
        metadata.insert(metadata_keys::NUM_TURNS.to_string(), "3".to_string());
        metadata.insert(metadata_keys::DURATION_MS.to_string(), "1200".to_string());
        metadata.insert(metadata_keys::INPUT_TOKENS.to_string(), "100".to_string());
        metadata.insert(metadata_keys::OUTPUT_TOKENS.to_string(), "200".to_string());
        metadata.insert(
            metadata_keys::CACHE_READ_INPUT_TOKENS.to_string(),
            "50".to_string(),
        );
        metadata.insert(
            metadata_keys::CACHE_CREATION_INPUT_TOKENS.to_string(),
            "25".to_string(),
        );
        let output = crate::traits::action_executor::ActionOutput {
            markers: vec!["m1".to_string()],
            context: Some("ctx".to_string()),
            result_text: Some("rt".to_string()),
            metadata: metadata.clone(),
            child_run_id: Some("child-1".to_string()),
            structured_output: Some(r#"{"ok":true}"#.to_string()),
        };
        let success = StepSuccess::from_action_output(
            &output,
            "review".to_string(),
            "ctx".to_string(),
            5,
            Some("/tmp/out".to_string()),
        );
        assert_eq!(success.step_name, "review");
        assert_eq!(success.result_text, Some("rt".to_string()));
        assert_eq!(
            success.metadata.get(metadata_keys::COST_USD),
            Some(&"0.05".to_string())
        );
        assert_eq!(
            success.metadata.get(metadata_keys::NUM_TURNS),
            Some(&"3".to_string())
        );
        assert_eq!(
            success.metadata.get(metadata_keys::DURATION_MS),
            Some(&"1200".to_string())
        );
        assert_eq!(success.markers, vec!["m1"]);
        assert_eq!(success.context, "ctx");
        assert_eq!(success.child_run_id, Some("child-1".to_string()));
        assert_eq!(success.iteration, 5);
        assert_eq!(
            success.structured_output,
            Some(r#"{"ok":true}"#.to_string())
        );
        assert_eq!(success.output_file, Some("/tmp/out".to_string()));
    }
}