use std::fs::{self, File};
use std::path::Path;
use anyhow::{anyhow, Context, Result};
use regex::Regex;
use reqwest::blocking::Client;
use reqwest::header;
use semver::Version;
use serde_json::{self, Value};
use clyde::package::Package;
use clyde::ui::Ui;
use crate::add_assets::select_best_urls;
use crate::fetch::FetchStatus;
fn get_repo_owner(package: &Package) -> Result<Option<String>> {
let rx = Regex::new("https://github.com/(?P<repo_owner>([^/]+)/([^/]+))").unwrap();
let repo_owner = rx
.captures(&package.repository)
.map(|captures| captures["repo_owner"].to_string());
Ok(repo_owner)
}
fn get_latest_release(release_file: &Path, repo_owner: &str) -> Result<()> {
let url = format!("https://api.github.com/repos/{repo_owner}/releases/latest");
let client = Client::new();
let mut response = client
.get(url)
.header(header::USER_AGENT, "clydetools")
.send()?;
if !response.status().is_success() {
let code = response.status().as_u16();
let body = response
.text()
.unwrap_or_else(|_| "[unreadable]".to_string());
return Err(anyhow!("Request failed with error {code}:\n{body}"));
}
let mut file = File::create(release_file)?;
response.copy_to(&mut file)?;
Ok(())
}
fn get_release_json(ui: &Ui, out_dir: &Path, package: &Package, repo_owner: &str) -> Result<Value> {
let release_file = out_dir.join(format!("{}.json", package.name));
if !release_file.exists() {
ui.info("Querying GitHub");
get_latest_release(&release_file, repo_owner)?;
}
let file = File::open(release_file)?;
let json = serde_json::from_reader(&file)?;
Ok(json)
}
fn count_chars(txt: &str, wanted: char) -> u32 {
let mut count = 0;
for ch in txt.chars() {
if ch == wanted {
count += 1;
}
}
count
}
fn extract_version(value: &Value) -> Result<Version> {
let tag = value["tag_name"]
.as_str()
.expect("No 'tag_name' in release JSON");
let version_str = if tag.starts_with('v') {
tag.get(1..).unwrap()
} else {
tag
};
let mut version_str = version_str.to_string();
while count_chars(&version_str, '.') < 2 {
version_str.push_str(".0");
}
let version = Version::parse(&version_str)?;
Ok(version)
}
fn extract_build_urls(value: &Value) -> Result<Vec<String>> {
let assets = value["assets"]
.as_array()
.expect("No 'assets' element in release JSON");
let urls = assets
.iter()
.map(|asset| {
asset["browser_download_url"]
.as_str()
.expect("No 'browser_download_url' in release JSON asset")
.to_string()
})
.collect();
Ok(urls)
}
pub fn is_hosted_on_github(package: &Package) -> Result<bool> {
let repo_owner = get_repo_owner(package)?;
Ok(repo_owner.is_some())
}
pub fn gh_fetch(ui: &Ui, package: &Package) -> Result<FetchStatus> {
let out_dir = Path::new("out");
if !out_dir.exists() {
ui.info(&format!("Creating {} dir", out_dir.display()));
fs::create_dir(out_dir)
.with_context(|| format!("Cannot create {} dir", out_dir.display()))?;
}
let repo_owner = match get_repo_owner(package)? {
Some(x) => x,
None => panic!("gh_fetch() should not be called on a package not hosted on GitHub"),
};
let release_json = get_release_json(ui, out_dir, package, &repo_owner)?;
let github_latest_version = extract_version(&release_json)?;
let package_latest_version = package
.get_latest_version()
.ok_or_else(|| anyhow!("Can't get latest version of {}", package.name))?;
if package_latest_version >= &github_latest_version {
return Ok(FetchStatus::UpToDate);
}
let urls = select_best_urls(ui, &extract_build_urls(&release_json)?)?;
Ok(FetchStatus::NeedUpdate {
version: github_latest_version,
urls,
})
}