use std::time::Duration;
use crate::errors::ApexeError;
use regex::Regex;
use serde::{Deserialize, Serialize};
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ResolvedTool {
pub name: String,
pub binary_path: String,
pub version: Option<String>,
}
pub struct ToolResolver;
impl ToolResolver {
pub fn resolve(&self, tool_name: &str, timeout: Duration) -> Result<ResolvedTool, ApexeError> {
let binary_path = which::which(tool_name)
.map_err(|_| ApexeError::ToolNotFound {
tool_name: tool_name.to_string(),
})?
.to_string_lossy()
.to_string();
let version = self.get_version(&binary_path, tool_name, timeout);
Ok(ResolvedTool {
name: tool_name.to_string(),
binary_path,
version,
})
}
fn get_version(
&self,
binary_path: &str,
_tool_name: &str,
timeout: Duration,
) -> Option<String> {
let output = super::exec::run_with_timeout(binary_path, &["--version"], timeout).ok()?;
let stdout = String::from_utf8_lossy(&output.stdout);
let first_line = stdout.lines().next()?;
extract_version_from_line(first_line)
}
}
pub fn extract_version_from_line(line: &str) -> Option<String> {
let re = Regex::new(r"(\d+\.\d+[\.\d]*)").ok()?;
re.captures(line)
.and_then(|c| c.get(1))
.map(|m| m.as_str().to_string())
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_resolved_tool_serde_with_version() {
let tool = ResolvedTool {
name: "git".into(),
binary_path: "/usr/bin/git".into(),
version: Some("2.43.0".into()),
};
let json = serde_json::to_string(&tool).unwrap();
let back: ResolvedTool = serde_json::from_str(&json).unwrap();
assert_eq!(back.name, "git");
assert_eq!(back.version, Some("2.43.0".into()));
}
#[test]
fn test_resolved_tool_serde_no_version() {
let tool = ResolvedTool {
name: "mytool".into(),
binary_path: "/usr/bin/mytool".into(),
version: None,
};
let json = serde_json::to_string(&tool).unwrap();
let back: ResolvedTool = serde_json::from_str(&json).unwrap();
assert!(back.version.is_none());
}
#[test]
fn test_resolve_known_tool() {
let resolver = ToolResolver;
let result = resolver.resolve("sh", std::time::Duration::from_secs(5));
assert!(result.is_ok());
let resolved = result.unwrap();
assert_eq!(resolved.name, "sh");
assert!(!resolved.binary_path.is_empty());
}
#[test]
fn test_resolve_unknown_tool() {
let resolver = ToolResolver;
let result = resolver.resolve("zzz_no_such_tool_xyz", std::time::Duration::from_secs(5));
assert!(result.is_err());
match result.unwrap_err() {
ApexeError::ToolNotFound { tool_name } => {
assert_eq!(tool_name, "zzz_no_such_tool_xyz");
}
other => panic!("Expected ToolNotFound, got: {other:?}"),
}
}
#[test]
fn test_extract_version_git_style() {
assert_eq!(
extract_version_from_line("git version 2.43.0"),
Some("2.43.0".into())
);
}
#[test]
fn test_extract_version_curl_style() {
assert_eq!(
extract_version_from_line("curl 8.1.2 (x86_64-apple-darwin)"),
Some("8.1.2".into())
);
}
#[test]
fn test_extract_version_no_version() {
assert_eq!(extract_version_from_line("no version here"), None);
}
#[test]
fn test_extract_version_empty() {
assert_eq!(extract_version_from_line(""), None);
}
}