floe-core 0.3.7

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

use crate::io::storage::{paths, CloudClient, Target};
use crate::report::{JsonReportFormatter, ReportFormatter};
use crate::{config, report, FloeResult};

pub struct ReportOutput<'a> {
    target: &'a Target,
    formatter: &'a dyn ReportFormatter,
    cloud: &'a mut CloudClient,
    resolver: &'a config::StorageResolver,
}

impl<'a> ReportOutput<'a> {
    pub fn new(
        target: &'a Target,
        formatter: &'a dyn ReportFormatter,
        cloud: &'a mut CloudClient,
        resolver: &'a config::StorageResolver,
    ) -> Self {
        Self {
            target,
            formatter,
            cloud,
            resolver,
        }
    }

    pub fn write_entity_report(
        &mut self,
        run_id: &str,
        entity: &config::EntityConfig,
        report: &report::RunReport,
    ) -> FloeResult<String> {
        let relative = report::ReportWriter::report_relative_path(run_id, &entity.name);
        write_report(
            self.target,
            &relative,
            self.formatter,
            ReportPayload::Entity(report),
            self.cloud,
            self.resolver,
            &format!("entity.name={}", entity.name),
        )
    }

    pub fn write_summary_report(
        &mut self,
        run_id: &str,
        report: &report::RunSummaryReport,
    ) -> FloeResult<String> {
        let relative = report::ReportWriter::summary_relative_path(run_id);
        write_report(
            self.target,
            &relative,
            self.formatter,
            ReportPayload::Summary(report),
            self.cloud,
            self.resolver,
            "report",
        )
    }
}

pub fn write_entity_report(
    target: &Target,
    run_id: &str,
    entity: &config::EntityConfig,
    report: &report::RunReport,
    cloud: &mut CloudClient,
    resolver: &config::StorageResolver,
) -> FloeResult<String> {
    let formatter = JsonReportFormatter;
    let mut output = ReportOutput::new(target, &formatter, cloud, resolver);
    output.write_entity_report(run_id, entity, report)
}

pub fn write_summary_report(
    target: &Target,
    run_id: &str,
    report: &report::RunSummaryReport,
    cloud: &mut CloudClient,
    resolver: &config::StorageResolver,
) -> FloeResult<String> {
    let formatter = JsonReportFormatter;
    let mut output = ReportOutput::new(target, &formatter, cloud, resolver);
    output.write_summary_report(run_id, report)
}

enum ReportPayload<'a> {
    Entity(&'a report::RunReport),
    Summary(&'a report::RunSummaryReport),
}

fn write_report(
    target: &Target,
    relative: &str,
    formatter: &dyn ReportFormatter,
    payload: ReportPayload<'_>,
    cloud: &mut CloudClient,
    resolver: &config::StorageResolver,
    context: &str,
) -> FloeResult<String> {
    // Storage-agnostic report write: local file or temp upload for cloud.
    let content = match payload {
        ReportPayload::Entity(report) => formatter.serialize_run(report)?,
        ReportPayload::Summary(report) => formatter.serialize_summary(report)?,
    };

    match target {
        Target::Local { base_path, .. } => {
            let output_path =
                paths::normalize_local_path(&paths::resolve_output_dir_path(base_path, relative));
            write_text_file(&output_path, &content)?;
            Ok(output_path.display().to_string())
        }
        _ => {
            let uri = target.join_relative(relative);
            let temp_dir = tempfile::tempdir()?;
            let filename = Path::new(relative)
                .file_name()
                .and_then(|name| name.to_str())
                .unwrap_or("report.json");
            let temp_path = temp_dir.path().join(filename);
            write_text_file(&temp_path, &content)?;
            let client = cloud.client_for_context(resolver, target.storage(), context)?;
            client.upload_from_path(&temp_path, &uri)?;
            Ok(uri)
        }
    }
}

fn write_text_file(path: &Path, content: &str) -> FloeResult<()> {
    let path = crate::io::storage::paths::normalize_local_path(path);
    if let Some(parent) = path.parent() {
        std::fs::create_dir_all(parent)?;
    }
    let tmp_path = temp_path(&path);
    let mut file = std::fs::File::create(&tmp_path)?;
    use std::io::Write;
    file.write_all(content.as_bytes())?;
    file.sync_all()?;
    std::fs::rename(&tmp_path, &path)?;
    Ok(())
}

fn temp_path(path: &Path) -> PathBuf {
    let file_name = path
        .file_name()
        .and_then(|name| name.to_str())
        .unwrap_or("report.json");
    let tmp_name = format!("{file_name}.tmp-{}", unique_suffix());
    path.parent().unwrap_or(path).join(tmp_name)
}

fn unique_suffix() -> String {
    use std::time::{SystemTime, UNIX_EPOCH};
    let nanos = SystemTime::now()
        .duration_since(UNIX_EPOCH)
        .map(|duration| duration.as_nanos())
        .unwrap_or(0);
    format!("{}-{}", std::process::id(), nanos)
}