tga 1.0.13

//! Report pipeline — orchestrates aggregation and formatter dispatch.

use std::path::{Path, PathBuf};

use tracing::{info, warn};

use crate::core::config::Config;
use crate::core::db::Database;
use crate::report::aggregator::Aggregator;
use crate::report::errors::{ReportError, Result};
use crate::report::formatters::{csv as csv_fmt, json as json_fmt, markdown as md_fmt};

/// Default output directory when [`crate::core::config::OutputConfig::directory`] is unset.
const DEFAULT_OUTPUT_DIR: &str = "./reports";

/// Supported format identifiers.
const FORMAT_CSV: &str = "csv";
const FORMAT_JSON: &str = "json";
const FORMAT_MARKDOWN: &str = "markdown";

/// Stage 3 orchestrator.
pub struct ReportPipeline {
    config: Config,
}

/// Summary of a [`ReportPipeline::run`] invocation.
#[derive(Debug, Clone)]
pub struct ReportStats {
    /// Total commits that appeared in the report.
    pub total_commits: usize,
    /// Total distinct authors that appeared in the report.
    pub total_authors: usize,
    /// Absolute paths of every file written.
    pub files_written: Vec<PathBuf>,
}

impl ReportPipeline {
    /// Construct a new pipeline bound to `config`.
    pub fn new(config: Config) -> Self {
        Self { config }
    }

    /// Aggregate the database and write all configured report formats.
    ///
    /// If `config.output.formats` is empty (and no legacy `format` is set),
    /// all three formats are emitted. The output directory is created if it
    /// does not already exist.
    ///
    /// # Errors
    ///
    /// Propagates any aggregator, formatter, or I/O failure.
    pub fn run(&self, db: &Database) -> Result<ReportStats> {
        let data = Aggregator::build(db, &self.config)?;
        let output_dir = self.resolve_output_dir();
        std::fs::create_dir_all(&output_dir)?;
        info!(dir = %output_dir.display(), "writing reports");

        let formats = self.resolve_formats();
        let mut files_written = Vec::new();

        for fmt in &formats {
            match fmt.as_str() {
                FORMAT_CSV => {
                    files_written.push(csv_fmt::write_author_csv(&data, &output_dir)?);
                    files_written.push(csv_fmt::write_weekly_csv(&data, &output_dir)?);
                    files_written.push(csv_fmt::write_weekly_metrics_csv(&data, &output_dir)?);
                    files_written.push(csv_fmt::write_developer_activity_csv(&data, &output_dir)?);
                    files_written.push(csv_fmt::write_summary_csv(&data, &output_dir)?);
                    files_written.push(csv_fmt::write_untracked_csv(&data, &output_dir)?);
                    files_written.push(csv_fmt::write_weekly_categorization_csv(
                        &data,
                        &output_dir,
                    )?);
                    files_written.push(csv_fmt::write_weekly_velocity_csv(&data, &output_dir)?);
                    files_written.push(csv_fmt::write_weekly_dora_csv(&data, &output_dir)?);
                }
                FORMAT_JSON => {
                    files_written.push(json_fmt::write_json(&data, &output_dir)?);
                    files_written.push(json_fmt::write_velocity_json(&data, &output_dir)?);
                    files_written.push(json_fmt::write_quality_json(&data, &output_dir)?);
                    files_written.push(json_fmt::write_dora_json(&data, &output_dir)?);
                }
                FORMAT_MARKDOWN | "md" => {
                    files_written.push(md_fmt::write_markdown(&data, &output_dir)?);
                }
                other => {
                    warn!(format = %other, "ignoring unknown output format");
                    return Err(ReportError::Report(format!(
                        "unknown output format: {other}"
                    )));
                }
            }
        }

        Ok(ReportStats {
            total_commits: data.total_commits,
            total_authors: data.total_authors,
            files_written,
        })
    }

    fn resolve_output_dir(&self) -> PathBuf {
        match self
            .config
            .output
            .as_ref()
            .and_then(|o| o.directory.as_ref())
        {
            Some(p) => crate::core::config::expand_path(p),
            None => Path::new(DEFAULT_OUTPUT_DIR).to_path_buf(),
        }
    }

    fn resolve_formats(&self) -> Vec<String> {
        if let Some(out) = &self.config.output {
            if !out.formats.is_empty() {
                return out.formats.iter().map(|s| s.to_lowercase()).collect();
            }
            if let Some(single) = &out.format {
                return vec![single.to_lowercase()];
            }
        }
        vec![
            FORMAT_CSV.to_string(),
            FORMAT_JSON.to_string(),
            FORMAT_MARKDOWN.to_string(),
        ]
    }
}