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

Implement gitlab and bitbucket support

Browse Source
This commit is contained in:
Valentin Brandl 2019-04-16 19:52:12 +02:00
parent e8fd78c517
commit 5944681086
No known key found for this signature in database
GPG Key ID: 30D341DD34118D7D
1 changed files with 29 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
src/main.rs
View File

@ -48,7 +48,7 @@ impl std::fmt::Display for Error {
match self {
Error::Git(e) => write!(fmt, "Git({})", e),
Error::Io(e) => write!(fmt, "Io({})", e),
Error::Badge(s) => write!(fmt, "Basge({})", s),
Error::Badge(s) => write!(fmt, "Badge({})", s),
Error::Internal => write!(fmt, "Internal"),
}
}
@ -121,15 +121,16 @@ fn hoc(repo: &str) -> Result<u64, Error> {
Ok(res)
}
fn github(
fn calculate_hoc(
service: &str,
state: web::Data<State>,
data: web::Path<(String, String)>,
) -> Result<HttpResponse, Error> {
let gh_path = format!("github.com/{}/{}", data.0, data.1);
let path = format!("{}/{}", *state, gh_path);
let service_path = format!("{}/{}/{}", service, data.0, data.1);
let path = format!("{}/{}", *state, service_path);
let file = Path::new(&path);
if !file.exists() {
Repository::clone(&format!("https://{}", gh_path), file)?;
Repository::clone(&format!("https://{}", service_path), file)?;
} else {
pull(&path)?;
}
@ -145,6 +146,27 @@ fn github(
.body(badge.to_svg()))
}
fn github(
state: web::Data<State>,
data: web::Path<(String, String)>,
) -> Result<HttpResponse, Error> {
calculate_hoc("github.com", state, data)
}
fn gitlab(
state: web::Data<State>,
data: web::Path<(String, String)>,
) -> Result<HttpResponse, Error> {
calculate_hoc("gitlab.com", state, data)
}
fn bitbucket(
state: web::Data<State>,
data: web::Path<(String, String)>,
) -> Result<HttpResponse, Error> {
calculate_hoc("bitbucket.org", state, data)
}
fn main() -> std::io::Result<()> {
std::env::set_var("RUST_LOG", "actix_web=info");
pretty_env_logger::init();
@ -157,6 +179,8 @@ fn main() -> std::io::Result<()> {
.data(state.clone())
.wrap(middleware::Logger::default())
.service(web::resource("/github/{user}/{repo}").to(github))
.service(web::resource("/gitlab/{user}/{repo}").to(gitlab))
.service(web::resource("/bitbucket/{user}/{repo}").to(bitbucket))
})
.bind(interface)?
.run()