Bump version number

[ci skip]

.drone.yml

@@ -0,0 +1,22 @@
+---
+kind: pipeline
+name: build
+
+steps:
+  - name: docker
+    image: plugins/docker
+    settings:
+      repo: vbrandl/hits-of-code
+      auto_tag: true
+      username:
+        from_secret: docker_username
+      password:
+        from_secret: docker_password
+    ref:
+      - refs/heads/master
+      - refs/tags/v*
+---
+kind: signature
+hmac: 36c14cd35c5abe274025760c9ca34f198d392be0c31435ad58cb9ecaa0abef2a
+
+...

.gitlab-ci.yml

@@ -12,9 +12,9 @@ variables:
   CONTAINER_BUILDER_IMAGE: $CI_REGISTRY_IMAGE:builder-latest
   CONTAINER_TEST_IMAGE: $CI_REGISTRY_IMAGE:$CI_COMMIT_REF_SLUG
   CONTAINER_RELEASE_IMAGE: $CI_REGISTRY_IMAGE:latest
-  CONTAINER_TAG_IMAGE: $CI_REGISTRY_IMAGE:${CI_COMMIT_TAG:1}
 
 before_script:
+  - export CONTAINER_TAG_IMAGE="$CI_REGISTRY_IMAGE:${CI_COMMIT_TAG:1}"
   - docker login -u $CI_REGISTRY_USER -p $CI_REGISTRY_PASSWORD $CI_REGISTRY
 
 build:
@@ -42,7 +42,7 @@ release-tag:
   stage: release
   script:
     - docker pull $CONTAINER_TEST_IMAGE
-    - docker tag $CONTAINER_TEST_IMAGE $CONTAINER_RELEASE_IMAGE
+    - docker tag $CONTAINER_TEST_IMAGE $CONTAINER_TAG_IMAGE
     - docker push $CONTAINER_TAG_IMAGE
   only:
     - /^v\d+\.\d+\.\d+/

Cargo.toml

@@ -1,22 +1,22 @@
 [package]
 name = "hoc"
-version = "0.2.0"
+version = "0.4.0"
 authors = ["Valentin Brandl <vbrandl@riseup.net>"]
 edition = "2018"
 build = "build.rs"
 
 [dependencies]
-actix-web = "1.0.0-alpha.4"
+actix-web = "1.0.0-beta.2"
 badge = "0.2.0"
 bytes = "0.4.12"
-futures = "0.1.25"
+futures = "0.1.26"
 git2 = "0.8.0"
 lazy_static = "1.3.0"
 log = "0.4.6"
 number_prefix = "0.3.0"
 openssl-probe = "0.1.2"
 pretty_env_logger = "0.3.0"
-reqwest = "0.9.15"
+reqwest = "0.9.16"
 serde = "1.0.90"
 serde_derive = "1.0.90"
 serde_json = "1.0.39"

README.md

@@ -1,14 +1,16 @@
 # Hits-of-Code
 
 [![Hits-of-Code](https://hitsofcode.com/github/vbrandl/hoc)](https://hitsofcode.com/view/github/vbrandl/hoc)
-[![Docker build](https://img.shields.io/docker/cloud/build/vbrandl/hits-of-code.svg)](https://hub.docker.com/r/vbrandl/hits-of-code)
+[![Drone build](https://drone.vbrandl.net/api/badges/vbrandl/hoc/status.svg)](https://drone.vbrandl.net/vbrandl/hoc)
 [![Gitlab build](https://gitlab.com/vbrandl/hoc/badges/master/pipeline.svg)](https://gitlab.com/vbrandl/hoc/pipelines)
+[![Travis build](https://travis-ci.org/vbrandl/hoc.svg?branch=master)](https://travis-ci.org/vbrandl/hoc)
+[![dependency status](https://deps.rs/repo/github/vbrandl/hoc/status.svg)](https://deps.rs/repo/github/vbrandl/hoc)
 
 Small webservice, that returns a badge of the Hits-of-Code of a git repository, as described by [Yegor
 Bugayenko](https://www.yegor256.com/2014/11/14/hits-of-code.html). It is implemented in
 [Rust](https://www.rust-lang.org/), using the [actix-web](https://actix.rs/) web framework.
 
-A live version of this API can be found on [hitsofcode.com](https://hitsofcode.com/).
+A live version of this webservice can be found on [hitsofcode.com](https://hitsofcode.com/).
 
 ## API
 
@@ -47,12 +49,8 @@ from [Docker Hub](https://hub.docker.com/r/vbrandl/hits-of-code)
 $ docker run -it --rm vbrandl/hits-of-code --help
 ```
 
+When running the binary directly, you need a git binary in your `PATH`.
 
-## TODO
-
-* [x] Customization of badges (e.g. colors)
-* [x] Support other platforms beside GitHub (GitLab and Bitbucket)
-* [ ] Allow exclusion of certain files/globs from the HoC count
 
 ## License
 

src/error.rs

@@ -2,6 +2,8 @@ use crate::P500;
 use actix_web::{HttpResponse, ResponseError};
 use std::fmt;
 
+pub(crate) type Result<T> = std::result::Result<T, Error>;
+
 #[derive(Debug)]
 pub(crate) enum Error {
     Badge(String),

src/main.rs

@@ -13,8 +13,8 @@ mod service;
 
 use crate::{
     cache::CacheState,
-    error::Error,
-    service::{Bitbucket, GitHub, Gitlab, Service},
+    error::{Error, Result},
+    service::{Bitbucket, FormService, GitHub, Gitlab, Service},
 };
 use actix_web::{
     error::ErrorBadRequest,
@@ -27,6 +27,7 @@ use futures::{unsync::mpsc, Stream};
 use git2::Repository;
 use number_prefix::{NumberPrefix, Prefixed, Standalone};
 use std::{
+    borrow::Cow,
     fs::create_dir_all,
     path::{Path, PathBuf},
     process::Command,
@@ -42,6 +43,13 @@ pub struct VersionInfo<'a> {
     pub version: &'a str,
 }
 
+#[derive(Deserialize, Serialize)]
+struct GeneratorForm<'a> {
+    service: FormService,
+    user: Cow<'a, str>,
+    repo: Cow<'a, str>,
+}
+
 const VERSION_INFO: VersionInfo = VersionInfo {
     commit: env!("VERGEN_SHA_SHORT"),
     version: env!("CARGO_PKG_VERSION"),
@@ -73,6 +81,7 @@ struct State {
 }
 
 const CSS: &str = include_str!("../static/tacit-css.min.css");
+const FAVICON: &[u8] = include_bytes!("../static/favicon32.png");
 
 #[derive(StructOpt, Debug)]
 struct Opt {
@@ -101,16 +110,19 @@ struct Opt {
     #[structopt(short = "d", long = "domain", default_value = "hitsofcode.com")]
     /// Interface to listen on
     domain: String,
+    #[structopt(short = "w", long = "workers", default_value = "4")]
+    /// Number of worker threads
+    workers: usize,
 }
 
-fn pull(path: impl AsRef<Path>) -> Result<(), Error> {
+fn pull(path: impl AsRef<Path>) -> Result<()> {
     let repo = Repository::open_bare(path)?;
     let mut origin = repo.find_remote("origin")?;
     origin.fetch(&["refs/heads/*:refs/heads/*"], None, None)?;
     Ok(())
 }
 
-fn hoc(repo: &str, repo_dir: &str, cache_dir: &str) -> Result<(u64, String), Error> {
+fn hoc(repo: &str, repo_dir: &str, cache_dir: &str) -> Result<(u64, String)> {
     let repo_dir = format!("{}/{}", repo_dir, repo);
     let cache_dir = format!("{}/{}.json", cache_dir, repo);
     let cache_dir = Path::new(&cache_dir);
@@ -161,7 +173,7 @@ fn hoc(repo: &str, repo_dir: &str, cache_dir: &str) -> Result<(u64, String), Err
             s.split_whitespace()
                 .take(2)
                 .map(str::parse::<u64>)
-                .filter_map(Result::ok)
+                .filter_map(std::result::Result::ok)
                 .sum::<u64>()
         })
         .sum();
@@ -172,7 +184,7 @@ fn hoc(repo: &str, repo_dir: &str, cache_dir: &str) -> Result<(u64, String), Err
     Ok((cache.count, head))
 }
 
-fn remote_exists(url: &str) -> Result<bool, Error> {
+fn remote_exists(url: &str) -> Result<bool> {
     Ok(CLIENT.head(url).send()?.status() == reqwest::StatusCode::OK)
 }
 
@@ -192,10 +204,10 @@ fn handle_hoc_request<T, F>(
     state: web::Data<Arc<State>>,
     data: web::Path<(String, String)>,
     mapper: F,
-) -> Result<HttpResponse, Error>
+) -> Result<HttpResponse>
 where
     T: Service,
-    F: Fn(HocResult) -> Result<HttpResponse, Error>,
+    F: Fn(HocResult) -> Result<HttpResponse>,
 {
     hoc_request::<T>(state, data).and_then(mapper)
 }
@@ -203,8 +215,8 @@ where
 fn hoc_request<T: Service>(
     state: web::Data<Arc<State>>,
     data: web::Path<(String, String)>,
-) -> Result<HocResult, Error> {
-    let repo = format!("{}/{}", data.0, data.1);
+) -> Result<HocResult> {
+    let repo = format!("{}/{}", data.0.to_lowercase(), data.1.to_lowercase());
     let service_path = format!("{}/{}", T::domain(), repo);
     let path = format!("{}/{}", state.repos, service_path);
     let file = Path::new(&path);
@@ -239,7 +251,7 @@ fn hoc_request<T: Service>(
 fn calculate_hoc<T: Service>(
     state: web::Data<Arc<State>>,
     data: web::Path<(String, String)>,
-) -> Result<HttpResponse, Error> {
+) -> Result<HttpResponse> {
     let mapper = |r| match r {
         HocResult::NotFound => Ok(p404()),
         HocResult::Hoc { hoc_pretty, .. } => {
@@ -272,7 +284,7 @@ fn calculate_hoc<T: Service>(
 fn overview<T: Service>(
     state: web::Data<Arc<State>>,
     data: web::Path<(String, String)>,
-) -> Result<HttpResponse, Error> {
+) -> Result<HttpResponse> {
     let mapper = |r| match r {
         HocResult::NotFound => Ok(p404()),
         HocResult::Hoc {
@@ -314,6 +326,26 @@ fn index() -> HttpResponse {
         .body(INDEX.as_slice())
 }
 
+#[post("/generate")]
+fn generate(params: web::Form<GeneratorForm>) -> Result<HttpResponse> {
+    let repo = format!("{}/{}", params.user, params.repo);
+    let mut buf = Vec::new();
+    templates::generate(
+        &mut buf,
+        VERSION_INFO,
+        &OPT.domain,
+        params.service.url(),
+        params.service.service(),
+        &repo,
+    )?;
+    let (tx, rx_body) = mpsc::unbounded();
+    let _ = tx.unbounded_send(Bytes::from(buf));
+
+    Ok(HttpResponse::Ok()
+        .content_type("text/html")
+        .streaming(rx_body.map_err(|_| ErrorBadRequest("bad request"))))
+}
+
 fn p404() -> HttpResponse {
     HttpResponse::NotFound()
         .content_type("text/html")
@@ -325,6 +357,11 @@ fn css() -> HttpResponse {
     HttpResponse::Ok().content_type("text/css").body(CSS)
 }
 
+#[get("/favicon.ico")]
+fn favicon32() -> HttpResponse {
+    HttpResponse::Ok().content_type("image/png").body(FAVICON)
+}
+
 fn main() -> std::io::Result<()> {
     std::env::set_var("RUST_LOG", "actix_web=info,hoc=info");
     pretty_env_logger::init();
@@ -340,6 +377,8 @@ fn main() -> std::io::Result<()> {
             .wrap(middleware::Logger::default())
             .service(index)
             .service(css)
+            .service(favicon32)
+            .service(generate)
             .service(web::resource("/github/{user}/{repo}").to(calculate_hoc::<GitHub>))
             .service(web::resource("/gitlab/{user}/{repo}").to(calculate_hoc::<Gitlab>))
             .service(web::resource("/bitbucket/{user}/{repo}").to(calculate_hoc::<Bitbucket>))
@@ -348,6 +387,7 @@ fn main() -> std::io::Result<()> {
             .service(web::resource("/view/bitbucket/{user}/{repo}").to(overview::<Bitbucket>))
             .default_service(web::resource("").route(web::get().to(p404)))
     })
+    .workers(OPT.workers)
     .bind(interface)?
     .run()
 }

src/service.rs

@@ -4,6 +4,34 @@ pub(crate) trait Service {
     fn commit_url(repo: &str, commit_ref: &str) -> String;
 }
 
+#[derive(Deserialize, Serialize)]
+pub(crate) enum FormService {
+    #[serde(rename = "github")]
+    GitHub,
+    #[serde(rename = "gitlab")]
+    Gitlab,
+    #[serde(rename = "bitbucket")]
+    Bitbucket,
+}
+
+impl FormService {
+    pub(crate) fn url(&self) -> &str {
+        match self {
+            FormService::GitHub => "github.com",
+            FormService::Gitlab => "gitlab.com",
+            FormService::Bitbucket => "bitbucket.org",
+        }
+    }
+
+    pub(crate) fn service(&self) -> &str {
+        match self {
+            FormService::GitHub => "github",
+            FormService::Gitlab => "gitlab",
+            FormService::Bitbucket => "bitbucket",
+        }
+    }
+}
+
 pub(crate) struct GitHub;
 
 impl Service for GitHub {

templates/generate.rs.html

@@ -0,0 +1,23 @@
+@use super::base;
+@use crate::VersionInfo;
+
+@(version_info: VersionInfo, domain: &str, url: &str, service: &str, path: &str)
+
+@:base("Hits-of-Code Badges", "Badge Generator", {
+
+<p>
+Here is the markdown for the badge for <a href="https://@url/@path">@url/@path</a>
+</p>
+
+<pre>
+[![Hits-of-Code](https://@domain/@service/@path)](https://@domain/view/@service/@path)
+</pre>
+
+<p>
+It will be rendered like this
+</p>
+
+<pre>
+<a href="https://@domain/view/@service/@path"><img src="https://@domain/@service/@path" alt="example badge" /></a>
+</pre>
+}, version_info)

templates/index.rs.html

@@ -45,6 +45,21 @@ would render this badge:
 alt="example badge" /></a>
 </pre>
 
+<h2>Badge Generator</h2>
+
+<form method="post" action="/generate">
+    <select name="service" id="service">
+        <option value="github">GitHub</option>
+        <option value="gitlab">Gitlab</option>
+        <option value="bitbucket">Bitbucket</option>
+    </select>
+    <label>/</label>
+    <input name="user" id="user" type="text" placeholder="user" />
+    <label>/</label>
+    <input name="repo" id="repo" type="text" placeholder="repository" />
+    <button type="submit">Generate</button>
+</form>
+
 <h2>Source Code</h2>
 
 <p>