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
-
+
}, 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
+
+