vdev 0.3.1

CLI utilities for Vector (vector.dev) development and CI workflows
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105

use crate::utils::git;
use anyhow::Result;
use sha2::Digest;
use std::{env, fs, path::Path};
use tempfile::TempDir;

/// Releases latest version to the vectordotdev homebrew tap
#[derive(clap::Args, Debug)]
#[command()]
pub struct Cli {
    /// GitHub username for the repository.
    #[arg(long, default_value = "vectordotdev")]
    username: String,

    /// Vector version, defaults to $`VECTOR_VERSION` if not specified.
    #[arg(long, env = "VECTOR_VERSION")]
    vector_version: String,
}

impl Cli {
    pub fn exec(self) -> Result<()> {
        // Create temporary directory for cloning the homebrew-brew repository
        let td = TempDir::new()?;
        env::set_current_dir(td.path())?;

        debug!(
            "Cloning the homebrew repository for username: {}",
            self.username
        );
        clone_and_setup_git(&self.username)?;

        let vector_version = env::var("VECTOR_VERSION")?;
        debug!("Updating the vector.rb formula for VECTOR_VERSION={vector_version}.");
        update_formula("Formula/vector.rb", &vector_version)?;

        debug!("Committing and pushing changes (if any).");
        commit_and_push_changes(&vector_version)?;

        td.close()?;
        Ok(())
    }
}

/// Clones the repository and sets up Git configuration
fn clone_and_setup_git(username: &str) -> Result<()> {
    let github_token = env::var("HOMEBREW_PAT").or_else(|_| env::var("GITHUB_TOKEN"))?;
    let homebrew_repo =
        format!("https://{username}:{github_token}@github.com/{username}/homebrew-brew.git");

    git::clone(&homebrew_repo)?;
    env::set_current_dir("homebrew-brew")?;
    git::set_config_value("user.name", "vic")?;
    git::set_config_value("user.email", "vector@datadoghq.com")?;
    Ok(())
}

/// Updates the vector.rb formula with new URLs, SHA256s, and version
fn update_formula<P>(file_path: P, vector_version: &str) -> Result<()>
where
    P: AsRef<Path>,
{
    // URLs and SHA256s for both architectures
    let arm_package_url = format!(
        "https://packages.timber.io/vector/{vector_version}/vector-{vector_version}-arm64-apple-darwin.tar.gz"
    );
    let arm_package_sha256 = hex::encode(sha2::Sha256::digest(
        reqwest::blocking::get(&arm_package_url)?.bytes()?,
    ));

    let content = fs::read_to_string(&file_path)?;

    // Replace the lines with updated URLs and SHA256s
    let updated_content = content
        .lines()
        .map(|line| {
            if line.trim_start().starts_with("version \"") {
                format!("  version \"{vector_version}\"")
            } else if line.contains("# arm64 url") {
                format!("      url \"{arm_package_url}\" # arm64 url")
            } else if line.contains("# arm64 sha256") {
                format!("      sha256 \"{arm_package_sha256}\" # arm64 sha256")
            } else {
                line.to_string()
            }
        })
        .collect::<Vec<_>>()
        .join("\n");

    fs::write(file_path, updated_content)?;
    Ok(())
}

/// Commits and pushes changes if any exist
fn commit_and_push_changes(vector_version: &str) -> Result<()> {
    let has_changes = !git::check_git_repository_clean()?;
    if has_changes {
        debug!("Modified lines {:?}", git::get_modified_files());
        let commit_message = format!("Release Vector {vector_version}");
        git::commit(&commit_message)?;
        git::push()?;
    } else {
        debug!("No changes to push.");
    }
    Ok(())
}