diff --git a/src/lib.rs b/src/lib.rs index baa9c7e..412b0be 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -55,6 +55,7 @@ struct GeneratorForm<'a> { service: FormService, user: Cow<'a, str>, repo: Cow<'a, str>, + branch: Option>, } #[derive(Debug)] @@ -447,6 +448,11 @@ async fn generate( params.service.url(), params.service.service(), &repo, + params + .branch + .as_deref() + .filter(|s| !s.is_empty()) + .unwrap_or("master"), )?; Ok(HttpResponse::Ok().content_type("text/html").body(buf)) diff --git a/templates/generate.rs.html b/templates/generate.rs.html index f4ad6dc..78194ec 100644 --- a/templates/generate.rs.html +++ b/templates/generate.rs.html @@ -1,7 +1,7 @@ @use super::base; @use crate::statics::VersionInfo; -@(version_info: VersionInfo, repo_count: usize, base_url: &str, url: &str, service: &str, path: &str) +@(version_info: VersionInfo, repo_count: usize, base_url: &str, url: &str, service: &str, path: &str, branch: &str) @:base("Hits-of-Code Badges", "Badge Generator", { @@ -10,7 +10,7 @@ Here is the markdown for the badge for @url/@path
-[![Hits-of-Code](@base_url/@service/@path)](@base_url/@service/@path/view)
+[![Hits-of-Code](@base_url/@service/@path?branch=@branch)](@base_url/@service/@path/view?branch=@branch)
 

@@ -18,6 +18,6 @@ It will be rendered like this

-example badge
+example badge
 
}, version_info, repo_count) diff --git a/templates/index.rs.html b/templates/index.rs.html index 9dd2520..aebc661 100644 --- a/templates/index.rs.html +++ b/templates/index.rs.html @@ -80,6 +80,8 @@ the lines of + +