1use anyhow::{Context, Result};
 2use clap::Parser;
 3use std::fs;
 4use std::path::Path;
 5
 6mod after_release;
 7mod cherry_pick;
 8mod compare_perf;
 9mod danger;
10mod extension_tests;
11mod nix_build;
12mod release_nightly;
13mod run_bundling;
14
15mod release;
16mod run_agent_evals;
17mod run_tests;
18mod runners;
19mod steps;
20mod vars;
21
22#[derive(Parser)]
23pub struct GenerateWorkflowArgs {}
24
25pub fn run_workflows(_: GenerateWorkflowArgs) -> Result<()> {
26    let dir = Path::new(".github/workflows");
27
28    let workflows = vec![
29        ("danger.yml", danger::danger()),
30        ("run_bundling.yml", run_bundling::run_bundling()),
31        ("release_nightly.yml", release_nightly::release_nightly()),
32        ("run_tests.yml", run_tests::run_tests()),
33        ("release.yml", release::release()),
34        ("cherry_pick.yml", cherry_pick::cherry_pick()),
35        ("compare_perf.yml", compare_perf::compare_perf()),
36        ("run_unit_evals.yml", run_agent_evals::run_unit_evals()),
37        (
38            "run_cron_unit_evals.yml",
39            run_agent_evals::run_cron_unit_evals(),
40        ),
41        ("run_agent_evals.yml", run_agent_evals::run_agent_evals()),
42        ("after_release.yml", after_release::after_release()),
43        ("extension_tests.yml", extension_tests::extension_tests()),
44    ];
45    fs::create_dir_all(dir)
46        .with_context(|| format!("Failed to create directory: {}", dir.display()))?;
47
48    for (filename, workflow) in workflows {
49        let content = workflow
50            .to_string()
51            .map_err(|e| anyhow::anyhow!("{}: {:?}", filename, e))?;
52        let content = format!(
53            "# Generated from xtask::workflows::{}\n# Rebuild with `cargo xtask workflows`.\n{}",
54            workflow.name.unwrap(),
55            content
56        );
57        let file_path = dir.join(filename);
58        fs::write(&file_path, content)?;
59    }
60
61    Ok(())
62}