vibe-action 0.1.4

//! Output trait, registry, and level enum.

use serde::{Deserialize, Serialize};

use crate::output::{
    cli::CliOutput, json::JsonOutput, msg::OutputMsg, plain::PlainOutput, test::TestOutput,
    tracing::TracingOutput,
};

/// Output type key.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub enum OutputType {
    Cli,
    Tracing,
    Plain,
    Json,
    Test,
}

#[derive(Serialize, Deserialize, Debug, Clone, Copy, PartialEq, Eq, Hash)]
#[serde(rename_all = "lowercase")]
pub enum OutputKind {
    Plain,
    Info,
    Success,
    Warning,
    Error,
    Debug,
    Trace,
    Progress,
}

/// Export context for plugin integration.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub enum ExportContext {
    Actions,
    Status,
    Success,
}

impl ExportContext {
    pub fn as_str(&self) -> &'static str {
        match self {
            Self::Actions => "actions",
            Self::Status => "status",
            Self::Success => "success",
        }
    }
}

/// Output trait.
pub trait Output: Send + Sync {
    fn output_type(&self) -> OutputType;
    fn plain(&self, msg: &OutputMsg);
    fn error(&self, msg: &OutputMsg);
    fn warning(&self, msg: &OutputMsg);
    fn info(&self, msg: &OutputMsg);
    fn success(&self, msg: &OutputMsg);
    fn debug(&self, msg: &OutputMsg);
    fn trace(&self, msg: &OutputMsg);
    fn progress(&self, msg: &OutputMsg);
}

/// Registry of outputs with a single active output.
pub struct OutputRegistry {
    current: Box<dyn Output>,
}

impl OutputRegistry {
    /// Create registry based on VIBE_LOG_TYPE and VIBE_TRACE_LEVEL.
    pub fn new(log_type: &str, trace_level: &str) -> Self {
        let output_type = match log_type {
            "cli" => OutputType::Cli,
            "tracing" => OutputType::Tracing,
            "plain" => OutputType::Plain,
            "json" => OutputType::Json,
            "test" => OutputType::Test,
            _ => OutputType::Cli,
        };
        let formatter = super::format::FormatOutput::new(output_type);
        let current: Box<dyn Output> = match output_type {
            OutputType::Cli => Box::new(CliOutput::new(formatter)),
            OutputType::Tracing => Box::new(TracingOutput::new(trace_level, formatter)),
            OutputType::Plain => Box::new(PlainOutput::new(formatter)),
            OutputType::Json => Box::new(JsonOutput::new(formatter)),
            OutputType::Test => Box::new(TestOutput::new(formatter)),
        };

        Self { current }
    }

    /// Universally dispatches any structural log message to its specific strategy method.
    pub fn write(&self, msg: &OutputMsg) {
        match msg.kind {
            OutputKind::Plain => self.current.plain(msg),
            OutputKind::Info => self.current.info(msg),
            OutputKind::Success => self.current.success(msg),
            OutputKind::Warning => self.current.warning(msg),
            OutputKind::Error => self.current.error(msg),
            OutputKind::Debug => self.current.debug(msg),
            OutputKind::Trace => self.current.trace(msg),
            OutputKind::Progress => self.current.progress(msg),
        }
    }

    pub fn output_type(&self) -> OutputType {
        self.current.output_type()
    }
}