#!/usr/bin/env rust-script
use regex::Regex;
use serde::Serialize;
use std::env;
use std::fs;
use std::io::Write;
use std::path::Path;
use std::process::{exit, Command, Stdio};
fn get_arg(name: &str) -> Option<String> {
let args: Vec<String> = env::args().collect();
let flag = format!("--{}", name);
if let Some(idx) = args.iter().position(|a| a == &flag) {
return args.get(idx + 1).cloned();
}
let env_name = name.to_uppercase().replace('-', "_");
env::var(&env_name).ok().filter(|s| !s.is_empty())
}
const MAX_RELEASE_BODY: usize = 125_000;
fn fit_release_body(body: String, tag: &str, repository: &str) -> String {
if body.len() <= MAX_RELEASE_BODY {
return body;
}
let notice = format!(
"\n\n---\n\n*This release note was truncated to fit GitHub's {MAX_RELEASE_BODY}-character \
limit. The complete entry is in [CHANGELOG.md](https://github.com/{repository}/blob/{tag}/CHANGELOG.md).*\n"
);
let budget = MAX_RELEASE_BODY.saturating_sub(notice.len());
let mut end = 0;
for (index, byte) in body.bytes().enumerate() {
if index >= budget {
break;
}
if byte == b'\n' {
end = index + 1;
}
}
if end == 0 {
end = body
.char_indices()
.map(|(index, _)| index)
.take_while(|index| *index <= budget)
.last()
.unwrap_or(0);
}
format!("{}{}", &body[..end], notice)
}
fn get_changelog_for_version(version: &str) -> String {
let changelog_path = "CHANGELOG.md";
if !Path::new(changelog_path).exists() {
return format!("Release v{}", version);
}
let content = match fs::read_to_string(changelog_path) {
Ok(c) => c,
Err(_) => return format!("Release v{}", version),
};
let escaped_version = regex::escape(version);
let header_pattern = format!(r"(?m)^## \[{escaped_version}\]");
let header_re = Regex::new(&header_pattern).unwrap();
if let Some(version_header) = header_re.find(&content) {
let after_header = &content[version_header.end()..];
let body_start = after_header
.find('\n')
.map_or(after_header.len(), |i| i + 1);
let body = &after_header[body_start..];
let next_section_re = Regex::new(r"(?m)^## \[").unwrap();
let section_body = if let Some(next) = next_section_re.find(body) {
&body[..next.start()]
} else {
body
};
let trimmed = section_body.trim();
if trimmed.is_empty() {
format!("Release v{}", version)
} else {
trimmed.to_string()
}
} else {
format!("Release v{}", version)
}
}
fn badge_escape(value: &str) -> String {
value
.replace('-', "--")
.replace('_', "__")
.replace(' ', "%20")
.replace('/', "%2F")
.replace(':', "%3A")
}
fn crates_io_badge(url: &str, version: &str) -> String {
let version_url = format!("{}/{}", url.trim_end_matches('/'), version);
format!(
"[]({}) []({})",
url,
version,
badge_escape(version),
version_url
)
}
fn docker_hub_badge(url: &str, version: &str) -> String {
let image = url
.trim_end_matches('/')
.strip_prefix("https://hub.docker.com/r/")
.unwrap_or("konard/link-assistant-router");
let tag_url = format!("{}/tags?name={}", url.trim_end_matches('/'), version);
let image_tag = format!("{}:{}", image, version);
format!(
"[]({})",
version,
badge_escape(&image_tag),
tag_url
)
}
#[derive(Serialize)]
struct ReleasePayload {
tag_name: String,
name: String,
body: String,
}
fn main() {
let version = match get_arg("release-version") {
Some(v) => v,
None => {
eprintln!("Error: Missing required argument --release-version");
eprintln!("Usage: rust-script scripts/create-github-release.rs --release-version <version> --repository <repository>");
exit(1);
}
};
let repository = match get_arg("repository") {
Some(r) => r,
None => {
eprintln!("Error: Missing required argument --repository");
eprintln!("Usage: rust-script scripts/create-github-release.rs --release-version <version> --repository <repository>");
exit(1);
}
};
let tag_prefix = get_arg("tag-prefix").unwrap_or_else(|| "v".to_string());
let crates_io_url = get_arg("crates-io-url");
let docker_hub_url = get_arg("docker-hub-url");
let tag = format!("{}{}", tag_prefix, version);
println!("Creating GitHub release for {}...", tag);
let mut release_notes = get_changelog_for_version(&version);
let mut badges = Vec::new();
if let Some(url) = crates_io_url {
badges.push(crates_io_badge(&url, &version));
}
if let Some(url) = docker_hub_url {
badges.push(docker_hub_badge(&url, &version));
}
if !badges.is_empty() {
release_notes = format!("{}\n\n{}", badges.join("\n"), release_notes);
}
let payload = ReleasePayload {
tag_name: tag.clone(),
name: format!("{}{}", tag_prefix, version),
body: fit_release_body(release_notes, &tag, &repository),
};
let payload_json = serde_json::to_string(&payload).expect("Failed to serialize payload");
let mut child = Command::new("gh")
.args([
"api",
&format!("repos/{}/releases", repository),
"-X",
"POST",
"--input",
"-",
])
.stdin(Stdio::piped())
.stdout(Stdio::piped())
.stderr(Stdio::piped())
.spawn()
.expect("Failed to execute gh command");
if let Some(ref mut stdin) = child.stdin {
stdin
.write_all(payload_json.as_bytes())
.expect("Failed to write to stdin");
}
let output = child
.wait_with_output()
.expect("Failed to wait on gh command");
if output.status.success() {
println!("Created GitHub release: {}", tag);
} else {
let stderr = String::from_utf8_lossy(&output.stderr);
if stderr.contains("already exists") {
println!("Release {} already exists, skipping", tag);
} else {
eprintln!("Error creating release: {}", stderr);
exit(1);
}
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn an_ordinary_release_note_is_left_alone() {
let body = "## [1.2.3]\n\n- one fix\n".to_string();
assert_eq!(
fit_release_body(body.clone(), "v1.2.3", "owner/repo"),
body
);
}
#[test]
fn an_oversized_release_note_is_published_truncated_and_says_so() {
let body = "- a changelog entry that repeats\n".repeat(8_000);
assert!(body.len() > MAX_RELEASE_BODY, "the fixture must overflow");
let fitted = fit_release_body(body, "v0.116.0", "owner/repo");
assert!(
fitted.len() <= MAX_RELEASE_BODY,
"the API rejects anything larger: {} bytes",
fitted.len()
);
assert!(
fitted.contains("truncated"),
"a shortened body must not read as a complete one"
);
assert!(
fitted.contains("owner/repo/blob/v0.116.0/CHANGELOG.md"),
"and must name where the rest is: {}",
&fitted[fitted.len() - 300..]
);
let kept = fitted.split("\n\n---\n\n").next().expect("body");
assert!(kept.ends_with('\n'), "the cut lands between entries");
}
#[test]
fn a_body_of_multibyte_characters_is_cut_safely() {
let body = "é".repeat(MAX_RELEASE_BODY);
let fitted = fit_release_body(body, "v1.0.0", "owner/repo");
assert!(fitted.len() <= MAX_RELEASE_BODY);
assert!(fitted.contains("truncated"));
}
}