vbrandl/hoc
1
0
Fork 0
Code Issues 1 Pull Requests Actions Projects Releases Wiki Activity

Check if repos exist before cloning

Browse Source
This commit is contained in:
Valentin Brandl 2019-04-21 20:45:36 +02:00
parent b12ba47be9
commit 2db66ee850
No known key found for this signature in database
GPG Key ID: 30D341DD34118D7D
1 changed files with 17 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/main.rs
View File

@ -1,5 +1,8 @@
#[macro_use] #[macro_use]
extern crate actix_web; extern crate actix_web;
#[macro_use]
extern crate lazy_static;
extern crate reqwest;
extern crate serde_json; extern crate serde_json;
#[macro_use] #[macro_use]
extern crate serde_derive; extern crate serde_derive;
@ -29,6 +32,10 @@ use std::{
}; };
use structopt::StructOpt; use structopt::StructOpt;
lazy_static! {
static ref CLIENT: reqwest::Client = reqwest::Client::new();
}
struct State { struct State {
repos: String, repos: String,
cache: String, cache: String,
@ -128,6 +135,10 @@ fn hoc(repo: &str, repo_dir: &str, cache_dir: &str) -> Result<u64, Error> {
Ok(cache.count) Ok(cache.count)
} }
fn remote_exists(url: &str) -> Result<bool, Error> {
Ok(CLIENT.head(url).send()?.status() == reqwest::StatusCode::OK)
}
fn calculate_hoc( fn calculate_hoc(
service: &str, service: &str,
state: web::Data<Arc<State>>, state: web::Data<Arc<State>>,
@ -137,10 +148,14 @@ fn calculate_hoc(
let path = format!("{}/{}", state.repos, service_path); let path = format!("{}/{}", state.repos, service_path);
let file = Path::new(&path); let file = Path::new(&path);
if !file.exists() { if !file.exists() {
let url = format!("https://{}", service_path);
if !remote_exists(&url)? {
return Ok(p404());
}
create_dir_all(file)?; create_dir_all(file)?;
let repo = Repository::init_bare(file)?; let repo = Repository::init_bare(file)?;
repo.remote_add_fetch("origin", "refs/heads/*:refs/heads/*")?; repo.remote_add_fetch("origin", "refs/heads/*:refs/heads/*")?;
repo.remote_set_url("origin", &format!("https://{}", service_path))?; repo.remote_set_url("origin", &url)?;
} }
pull(&path)?; pull(&path)?;
let hoc = hoc(&service_path, &state.repos, &state.cache)?; let hoc = hoc(&service_path, &state.repos, &state.cache)?;
@ -211,7 +226,7 @@ fn css() -> HttpResponse {
} }
fn main() -> std::io::Result<()> { fn main() -> std::io::Result<()> {
std::env::set_var("RUST_LOG", "actix_web=info"); std::env::set_var("RUST_LOG", "actix_web=warn");
pretty_env_logger::init(); pretty_env_logger::init();
openssl_probe::init_ssl_cert_env_vars(); openssl_probe::init_ssl_cert_env_vars();
let opt = Opt::from_args(); let opt = Opt::from_args();