From dbbbdad3faa18d1e5e12c911fbe2a94612fb73d8 Mon Sep 17 00:00:00 2001
From: Valentin Brandl <valentin.brandl@ebsnet.de>
Date: Thu, 18 Aug 2022 14:07:07 +0200
Subject: [PATCH] Allow providing a branch name in generator

---
 src/lib.rs                 | 6 ++++++
 templates/generate.rs.html | 6 +++---
 templates/index.rs.html    | 2 ++
 3 files changed, 11 insertions(+), 3 deletions(-)

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<Cow<'a, str>>,
 }
 
 #[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 <a href="https://@url/@path">@url/@path</
 </p>
 
 <pre>
-[![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)
 </pre>
 
 <p>
@@ -18,6 +18,6 @@ It will be rendered like this
 </p>
 
 <pre>
-<a href="@base_url/@service/@path/view"><img src="@base_url/@service/@path" alt="example badge" /></a>
+<a href="@base_url/@service/@path/view?branch=@branch"><img src="@base_url/@service/@path?branch=@branch" alt="example badge" /></a>
 </pre>
 }, 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
     <input name="user" id="user" type="text" placeholder="user" />
     <label>/</label>
     <input name="repo" id="repo" type="text" placeholder="repository" />
+    <label>:</label>
+    <input name="branch" id="branch" type="text" placeholder="branch (defaults to `master`)" />
     <button type="submit">Generate</button>
 </form>