sentry-options-cli 1.0.13

use std::fs;
use std::io::Write;
use std::path::Path;
use std::process::{Command, Stdio};
use std::thread;
use std::time::Duration;

use crate::repo_schema_config::{RepoSchemaConfig, RepoSchemaConfigs};
use crate::{AppError, Result};

pub fn fetch_all_schemas(config: &RepoSchemaConfigs, out_dir: &Path, quiet: bool) -> Result<()> {
    if out_dir.exists() {
        return Err(AppError::Validation(format!(
            "Output directory already exists: {}",
            out_dir.display()
        )));
    }
    fs::create_dir_all(out_dir)?;

    // Sort repo names for deterministic ordering
    let mut repo_names: Vec<_> = config.repos.keys().collect();
    repo_names.sort();

    if !quiet {
        println!("Fetching schemas...");
    }

    // Fetch all repos in parallel, collecting results
    let results: Vec<_> = thread::scope(|s| {
        let handles: Vec<_> = repo_names
            .iter()
            .map(|repo_name| {
                let source = &config.repos[*repo_name];
                let name = repo_name.as_str();
                (
                    name,
                    s.spawn(move || fetch_repo_schemas(name, source, out_dir)),
                )
            })
            .collect();

        handles
            .into_iter()
            .map(|(name, h)| {
                let result: std::result::Result<(), String> = match h.join() {
                    Ok(Ok(())) => Ok(()),
                    Ok(Err(e)) => Err(e.to_string()),
                    Err(_) => Err(format!("{}: thread panicked", name)),
                };
                (name, result)
            })
            .collect::<Vec<_>>()
    });

    // Report results in deterministic order
    let mut errors: Vec<String> = Vec::new();
    for (repo_name, result) in results {
        match result {
            Ok(()) => {
                if !quiet {
                    println!("  Fetched {}", repo_name);
                }
            }
            Err(e) => errors.push(e),
        }
    }

    if !errors.is_empty() {
        return Err(AppError::Git(errors.join("\n")));
    }

    Ok(())
}

const MAX_RETRIES: u32 = 3;
const INITIAL_RETRY_DELAY: Duration = Duration::from_secs(3);

/// fetches repo schema using exponential backoff
/// waits a maximum of 3 + 6 + 12 = 21 seconds
/// will only propagate the last error from the last retry
fn fetch_repo_schemas(repo_name: &str, source: &RepoSchemaConfig, out_dir: &Path) -> Result<()> {
    let mut delay = INITIAL_RETRY_DELAY;
    let mut last_err = None;

    for attempt in 0..=MAX_RETRIES {
        if attempt > 0 {
            thread::sleep(delay);
            delay *= 2;
        }

        match try_fetch_repo_schemas(repo_name, source, out_dir) {
            Ok(()) => return Ok(()),
            Err(e) => last_err = Some(e),
        }
    }

    Err(last_err.unwrap())
}

/// calls the github cli to sparse checkout a schema file from a repo
fn try_fetch_repo_schemas(
    repo_name: &str,
    source: &RepoSchemaConfig,
    out_dir: &Path,
) -> Result<()> {
    let temp_dir = tempfile::tempdir()?;
    let repo_path = temp_dir.path().join("repo");
    let repo_str = repo_path.to_str().ok_or_else(|| {
        AppError::Validation(format!("{}: temp path contains invalid UTF-8", repo_name))
    })?;

    // Partial clone to reduce download size, and only fetch the schemas we need
    git(
        &[
            "clone",
            "--filter=blob:none",
            "--sparse",
            &source.url,
            repo_str,
        ],
        None,
        repo_name,
    )?;
    git(
        &["sparse-checkout", "set", &source.path],
        Some(&repo_path),
        repo_name,
    )?;
    git(&["checkout", &source.sha], Some(&repo_path), repo_name)?;

    copy_schemas(&repo_path.join(&source.path), out_dir)?;
    Ok(())
}

fn git(args: &[&str], cwd: Option<&Path>, repo_name: &str) -> Result<()> {
    let mut cmd = Command::new("git");
    cmd.args(args);
    if let Some(dir) = cwd {
        cmd.current_dir(dir);
    }
    let output = cmd.output()?;
    if !output.status.success() {
        return Err(AppError::Git(format!(
            "{}: {}",
            repo_name,
            String::from_utf8_lossy(&output.stderr).trim()
        )));
    }
    Ok(())
}

fn copy_schemas(src: &Path, dest: &Path) -> Result<()> {
    if !src.exists() {
        return Err(AppError::Validation(format!(
            "Schemas path does not exist: {}",
            src.display()
        )));
    }

    for entry in fs::read_dir(src)? {
        let entry = entry?;
        if entry.file_type()?.is_dir() {
            let namespace = entry.file_name();
            let src_schema = entry.path().join("schema.json");

            if src_schema.exists() {
                let dest_ns = dest.join(&namespace);
                fs::create_dir_all(&dest_ns)?;
                fs::copy(&src_schema, dest_ns.join("schema.json"))?;
            }
        }
    }
    Ok(())
}

/// Fetch multiple SHAs from origin
pub fn fetch_shas(shas: &[&str]) -> Result<()> {
    for sha in shas {
        let output = Command::new("git")
            .args(["fetch", "origin", sha])
            .output()?;

        if !output.status.success() {
            return Err(AppError::Git(format!(
                "Failed to fetch {}: {}",
                sha,
                String::from_utf8_lossy(&output.stderr).trim()
            )));
        }
    }
    Ok(())
}

/// Extract schemas directory from the current git repository at a specific SHA
/// SHA must already be fetched (call fetch_shas to ensure this)
pub fn extract_schemas_at_sha(sha: &str, schemas_path: &str, out_dir: &Path) -> Result<()> {
    fs::create_dir_all(out_dir)?;

    // Use `git ls-tree` to check path existence in a single command.
    // Unlike cat-file, it exits non-zero for an invalid/unreachable SHA and
    // exits 0 with empty stdout when the SHA is valid but the path isn't in
    // the tree.
    let ls_tree = Command::new("git")
        .args(["ls-tree", sha, schemas_path])
        .output()?;

    if !ls_tree.status.success() {
        return Err(AppError::Git(format!(
            "git ls-tree failed for {} at {}: {}",
            schemas_path,
            sha,
            String::from_utf8_lossy(&ls_tree.stderr).trim()
        )));
    }

    if ls_tree.stdout.is_empty() {
        // The schemas directory didn't exist at this SHA (e.g. the repo is
        // onboarding sentry-options for the first time). Create the expected
        // subdirectory so callers see an empty schema set.
        fs::create_dir_all(out_dir.join(schemas_path))?;
        return Ok(());
    }

    // git archive will output a tarball of the specified directory
    let archive_output = Command::new("git")
        .args(["archive", sha, schemas_path])
        .output()?;

    if !archive_output.status.success() {
        return Err(AppError::Git(format!(
            "Failed to archive {} at {}: {}",
            schemas_path,
            sha,
            String::from_utf8_lossy(&archive_output.stderr).trim()
        )));
    }

    let mut tar_child = Command::new("tar")
        .args(["-xf", "-", "-C"])
        .arg(out_dir)
        .stdin(Stdio::piped())
        .spawn()?;

    // avoids writing to a temp file between the git archive and tar -xf
    if let Some(mut stdin) = tar_child.stdin.take() {
        stdin.write_all(&archive_output.stdout)?;
    }

    let status = tar_child.wait()?;
    if !status.success() {
        return Err(AppError::Validation(format!(
            "Failed to extract tar archive for {} at {}",
            schemas_path, sha
        )));
    }

    Ok(())
}

#[cfg(test)]
mod tests {
    use super::*;
    use tempfile::TempDir;

    #[test]
    fn test_copy_schemas_valid_structure() {
        let src_dir = TempDir::new().unwrap();
        let dest_dir = TempDir::new().unwrap();

        let ns1 = src_dir.path().join("relay");
        let ns2 = src_dir.path().join("billing");
        fs::create_dir_all(&ns1).unwrap();
        fs::create_dir_all(&ns2).unwrap();
        fs::write(ns1.join("schema.json"), r#"{"version": "1.0"}"#).unwrap();
        fs::write(ns2.join("schema.json"), r#"{"version": "1.0"}"#).unwrap();

        let dest = dest_dir.path().join("sentry");
        copy_schemas(src_dir.path(), &dest).unwrap();

        assert!(dest.join("relay/schema.json").exists());
        assert!(dest.join("billing/schema.json").exists());
    }

    #[test]
    fn test_copy_schemas_missing_src() {
        let dest_dir = TempDir::new().unwrap();
        let err = copy_schemas(Path::new("/nonexistent/path"), dest_dir.path()).unwrap_err();
        assert!(err.to_string().contains("Schemas path does not exist"));
    }

    #[test]
    fn test_copy_schemas_skips_non_directories() {
        let src_dir = TempDir::new().unwrap();
        let dest_dir = TempDir::new().unwrap();

        fs::write(src_dir.path().join("README.md"), "# Schemas").unwrap();
        let ns = src_dir.path().join("relay");
        fs::create_dir_all(&ns).unwrap();
        fs::write(ns.join("schema.json"), r#"{"version": "1.0"}"#).unwrap();

        let dest = dest_dir.path().join("sentry");
        copy_schemas(src_dir.path(), &dest).unwrap();

        assert!(dest.join("relay/schema.json").exists());
        assert!(!dest.join("README.md").exists());
    }

    #[test]
    fn test_copy_schemas_skips_dirs_without_schema() {
        let src_dir = TempDir::new().unwrap();
        let dest_dir = TempDir::new().unwrap();

        fs::create_dir_all(src_dir.path().join("empty")).unwrap();
        let ns = src_dir.path().join("relay");
        fs::create_dir_all(&ns).unwrap();
        fs::write(ns.join("schema.json"), r#"{"version": "1.0"}"#).unwrap();

        let dest = dest_dir.path().join("sentry");
        copy_schemas(src_dir.path(), &dest).unwrap();

        assert!(dest.join("relay/schema.json").exists());
        assert!(!dest.join("empty").exists());
    }

    #[test]
    fn test_fetch_all_schemas_errors_if_output_exists() {
        use crate::repo_schema_config::RepoSchemaConfigs;

        let out_dir = TempDir::new().unwrap();
        let existing_dir = out_dir.path().join("schemas");
        fs::create_dir_all(&existing_dir).unwrap();

        let config: RepoSchemaConfigs = serde_json::from_str(r#"{"repos": {}}"#).unwrap();
        let err = fetch_all_schemas(&config, &existing_dir, true).unwrap_err();
        assert!(err.to_string().contains("Output directory already exists"));
    }
}