use super::VersionFile;
use crate::error_code::{self, ErrorCodeExt};
use anyhow::{Context, Result};
use regex::Regex;
use std::path::Path;
use std::sync::OnceLock;
pub struct GradleVersionFile;
#[cfg(test)]
mod tests {
use super::*;
use std::io::Write;
use tempfile::NamedTempFile;
fn write_temp(content: &str) -> NamedTempFile {
let mut f = NamedTempFile::new().unwrap();
f.write_all(content.as_bytes()).unwrap();
f
}
#[test]
fn read_gradle_double_quotes() {
let f = write_temp("plugins { id 'java' }\nversion = \"1.2.3\"\n");
assert_eq!(GradleVersionFile.read_version(f.path()).unwrap(), "1.2.3");
}
#[test]
fn read_gradle_single_quotes() {
let f = write_temp("version = '0.5.0'\n");
assert_eq!(GradleVersionFile.read_version(f.path()).unwrap(), "0.5.0");
}
#[test]
fn read_gradle_with_spaces() {
let f = write_temp("version = \"1.0.0\"\n");
assert_eq!(GradleVersionFile.read_version(f.path()).unwrap(), "1.0.0");
}
#[test]
fn read_no_version_fails() {
let f = write_temp("plugins { id 'java' }\n");
assert!(GradleVersionFile.read_version(f.path()).is_err());
}
#[test]
fn write_gradle_version() {
let f = write_temp("version = \"1.0.0\"\n");
GradleVersionFile.write_version(f.path(), "2.0.0").unwrap();
assert_eq!(GradleVersionFile.read_version(f.path()).unwrap(), "2.0.0");
}
#[test]
fn write_preserves_quote_style() {
let f = write_temp("version = '1.0.0'\n");
GradleVersionFile.write_version(f.path(), "2.0.0").unwrap();
let content = std::fs::read_to_string(f.path()).unwrap();
assert!(content.contains("version = '2.0.0'"));
}
}
static VERSION_RE: OnceLock<Regex> = OnceLock::new();
fn version_re() -> &'static Regex {
VERSION_RE
.get_or_init(|| Regex::new(r#"(?m)^(\s*version\s*=\s*)(["'])([^"']+)(["'])"#).unwrap())
}
impl VersionFile for GradleVersionFile {
fn read_version(&self, file_path: &Path) -> Result<String> {
let content = std::fs::read_to_string(file_path)
.with_context(|| format!("Cannot read {}", file_path.display()))
.error_code(error_code::GRADLE_READ)?;
version_re()
.captures(&content)
.map(|c| c[3].to_string())
.ok_or_else(|| anyhow::anyhow!("No version field found in {}", file_path.display()))
.error_code(error_code::GRADLE_VERSION_NOT_FOUND)
}
fn write_version(&self, file_path: &Path, version: &str) -> Result<()> {
let content = std::fs::read_to_string(file_path)
.with_context(|| format!("Cannot read {}", file_path.display()))
.error_code(error_code::GRADLE_READ)?;
if !version_re().is_match(&content) {
Err(anyhow::anyhow!(
"No version field found to update in {}",
file_path.display()
))
.error_code(error_code::GRADLE_VERSION_NOT_FOUND)?;
}
let new_content = version_re().replace(&content, |caps: ®ex::Captures| {
format!("{}{}{}{}", &caps[1], &caps[2], version, &caps[4])
});
std::fs::write(file_path, new_content.as_ref())
.with_context(|| format!("Cannot write {}", file_path.display()))
.error_code(error_code::GRADLE_WRITE)?;
Ok(())
}
fn read_version_from_bytes(&self, content: &[u8], filename: &str) -> Result<String> {
let text = std::str::from_utf8(content)
.with_context(|| format!("Invalid UTF-8 in {filename}"))
.error_code(error_code::GRADLE_INVALID_UTF8)?;
version_re()
.captures(text)
.map(|c| c[3].to_string())
.ok_or_else(|| anyhow::anyhow!("No version field found in {filename}"))
.error_code(error_code::GRADLE_VERSION_NOT_FOUND)
}
}