chore(vdev): Rewrite release-homebrew into vdev

vdev/Cargo.toml

@@ -20,6 +20,7 @@ directories = "5.0.0"
 # remove this when stabilized https://doc.rust-lang.org/stable/std/path/fn.absolute.html
 dunce = "1.0.3"
 hashlink = { version = "0.8.1", features = ["serde_impl"] }
+hex = "0.4.3"
 indicatif = { version = "0.17.3", features = ["improved_unicode"] }
 itertools = "0.10.5"
 log = "0.4.17"
@@ -28,8 +29,11 @@ os_info = { version = "3.7.0", default-features = false }
 # watch https://github.com/epage/anstyle for official interop with Clap
 owo-colors = { version = "3.5.0", features = ["supports-colors"] }
 paste = "1.0.12"
-regex = { version = "1.7.3", default-features = false, features = ["std", "unicode-perl"] }
+regex = { version = "1.7.1", default-features = false, features = ["std", "perf"] }
+reqwest = { version = "0.11", features = ["json", "blocking"] }
 serde = { version = "1.0", features = ["derive"] }
 serde_json = "1.0.95"
 serde_yaml = "0.9.19"
-toml = { version = "0.7.3", default-features = false, features = ["parse"] }
+sha2 = "0.10.6"
+tempfile = "3.4.0"
+toml = { version = "0.7.2", default-features = false, features = ["parse"] }

vdev/src/commands/release/homebrew.rs

@@ -0,0 +1,79 @@
+use anyhow::{Result};
+use std::env;
+use tempfile::TempDir;
+use crate::git;
+use hex;
+use sha2::Digest;
+use reqwest;
+use std::path::Path;
+use regex;
+
+/// Releases latest version to the vectordotdev homebrew tap
+#[derive(clap::Args, Debug)]
+#[command()]
+pub struct Cli {}
+
+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())?;
+
+        // Set git configurations
+        let config_values = &[
+            ("user.name", "vic"),
+            ("user.email", "vector@datadoghq.com"),
+        ];
+        git::set_config_values(config_values)?;
+        let github_token = env::var("GITHUB_TOKEN")?;
+
+        // Clone the homebrew-brew repository
+        let homebrew_repo = format!("https://{github_token}:x-oauth-basic@github.com/vectordotdev/homebrew-brew");
+        git::clone(&homebrew_repo)?;
+        env::set_current_dir("homebrew-brew")?;
+
+        // Get package details for updating Formula/vector.rb
+        let vector_version = env::var("VECTOR_VERSION")?;
+        let package_url = format!("https://packages.timber.io/vector/{vector_version}/vector-{vector_version}-x86_64-apple-darwin.tar.gz");
+        let package_sha256 = hex::encode(sha2::Sha256::digest(reqwest::blocking::get(&package_url)?.bytes()?));
+
+        // Update content of Formula/vector.rb
+        update_content("Formula/vector.rb", &package_url, &package_sha256, &vector_version)?;
+
+        // Check if there is any change in git index
+        let has_changes = !git::check_git_repository_clean()?;
+        if has_changes {
+            let commit_message = format!("Release Vector {vector_version}");
+            git::commit(&commit_message)?;
+        }
+        git::push()?;
+
+        // Remove temporary directory
+        td.close()?;
+        Ok(())
+    }
+}
+
+/// Open the vector.rb file and update the new content
+fn update_content<P>(file_path: P, package_url: &str, package_sha256: &str, vector_version: &str) -> Result<()>
+where
+    P: AsRef<Path>,
+{
+    let content = std::fs::read_to_string(&file_path)?;
+    let patterns = [
+        (format!(r#"url "{package_url}""#), r#"url ".*""#),
+        (format!(r#"sha256 "{package_sha256}""#), r#"sha256 ".*""#),
+        (format!(r#"version "{vector_version}""#), r#"version ".*""#),
+    ];
+    let new_content = substitute(content, &patterns);
+    std::fs::write(file_path, new_content)?;
+    Ok(())
+}
+
+fn substitute(mut content: String, patterns: &[(String, &str)]) -> String {
+    for (value, pattern) in patterns {
+        let re = regex::Regex::new(pattern).unwrap();
+        content = re.replace_all(&content, value.as_str()).to_string();
+    }
+    content
+}

vdev/src/commands/release/mod.rs

@@ -5,7 +5,7 @@ crate::cli_subcommands! {
     commit,
     docker,
     mod github,
-    homebrew,
+    mod homebrew,
     mod prepare,
     mod push,
     s3,
@@ -23,10 +23,6 @@ crate::script_wrapper! {
     docker = "Build the Vector docker images and optionally push it to the registry"
         => "build-docker.sh"
 }
-crate::script_wrapper! {
-    homebrew = "Releases latest version to the vectordotdev homebrew tap"
-        => "release-homebrew.sh"
-}
 crate::script_wrapper! {
     s3 = "Uploads archives and packages to AWS S3"
         => "release-s3.sh"

vdev/src/git.rs

@@ -79,7 +79,7 @@ pub fn get_git_sha() -> Result<String> {
     check_output(&["rev-parse", "--short", "HEAD"]).map(|output| output.trim_end().to_string())
 }
 
-// Get a list of files that have been modified, as a vector of strings
+/// Get a list of files that have been modified, as a vector of strings
 pub fn get_modified_files() -> Result<Vec<String>> {
     let args = vec![
         "ls-files",
@@ -91,6 +91,41 @@ pub fn get_modified_files() -> Result<Vec<String>> {
     Ok(check_output(&args)?.lines().map(str::to_owned).collect())
 }
 
+pub fn set_config_values(config_values: &[(&str, &str)]) -> Result<String> {
+    let mut args = vec!["config"];
+
+    for (key, value) in config_values {
+        args.push(key);
+        args.push(value);
+    }
+
+    check_output(&args)
+}
+
+/// Checks if the current directory's repo is clean
+pub fn check_git_repository_clean() -> Result<bool> {
+    Ok(Command::new("git")
+        .args(["diff-index", "--quiet", "HEAD"])
+        .stdout(std::process::Stdio::null())
+        .status()
+        .map(|status| status.success())?)
+}
+
+/// Commits changes from the current repo
+pub fn commit(commit_message: &str) -> Result<String> {
+    check_output(&["commit", "--all", "--message", commit_message])
+}
+
+/// Pushes changes from the current repo
+pub fn push() -> Result<String> {
+    check_output(&["push"])
+}
+
+pub fn clone(repo_url: &str) -> Result<String> {
+    // We cannot use capture_output since this will need to run in the CWD
+    Command::new("git").args(["clone", repo_url]).check_output()
+}
+
 pub fn branch_exists(branch_name: &str) -> Result<bool> {
     let output = check_output(&["rev-parse", "--verify", branch_name])?;
     Ok(!output.is_empty())