Add bookmarklet to our range-diff page for easy of use

Urgau · Urgau · commit ac70dd94f097 · 2025-08-10T19:23:58.000+02:00
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -50,6 +50,7 @@ subtle = "2.6.1"
 sha2 = "0.10.9"
 imara-diff = "0.2.0"
 pulldown-cmark-escape = "0.11.0"
+axum-extra = { version = "0.10.1", default-features = false }
 
 [dependencies.serde]
 version = "1"
diff --git a/src/gh_range_diff.rs b/src/gh_range_diff.rs
@@ -8,6 +8,7 @@ use axum::{
     http::HeaderValue,
     response::IntoResponse,
 };
+use axum_extra::extract::Host;
 use hyper::header::CACHE_CONTROL;
 use hyper::{
     HeaderMap, StatusCode,
@@ -23,6 +24,7 @@ use crate::{github, handlers::Context, utils::AppError};
 pub async fn gh_range_diff(
     Path((owner, repo, basehead)): Path<(String, String, String)>,
     State(ctx): State<Arc<Context>>,
+    Host(host): Host,
 ) -> axum::response::Result<impl IntoResponse, AppError> {
     let Some((oldhead, newhead)) = basehead.split_once("..") else {
         return Ok((
@@ -137,6 +139,9 @@ pub async fn gh_range_diff(
     // Create the HTML buffer with a very rough approximation for the capacity
     let mut html: String = String::with_capacity(800 + old.files.len() * 100);
 
+    // Compute the bookmarklet for the current host
+    let bookmarklet = bookmarklet(&host);
+
     // Write HTML header, style, ...
     writeln!(
         &mut html,
@@ -170,6 +175,7 @@ pub async fn gh_range_diff(
 </head>
 <body>
 <h3>range-diff of {oldbase}..{oldhead} {newbase}..{newhead}</h3>
+<p>Bookmarklet: <a href="{bookmarklet}" title="Drag-and-drop me on the bookmarks bar, so I can be used on GitHub compare page">range-diff</a></p>
 "#
     )?;
 
@@ -311,3 +317,21 @@ impl UnifiedDiffPrinter for HtmlDiffPrinter<'_> {
         Ok(())
     }
 }
+
+// Create the javascript bookmarklet based on the host
+fn bookmarklet(host: &str) -> String {
+    let protocol = if host.starts_with("localhost:") {
+        "http"
+    } else {
+        "https"
+    };
+
+    format!(
+        r"javascript:(() => {{
+    const githubUrlPattern = /^https:\/\/github\.com\/([^\/]+)\/([^\/]+)\/compare\/([^\/]+[.]{{2}}[^\/]+)$/;
+    const match = window.location.href.match(githubUrlPattern);
+    if (!match) {{alert('Invalid GitHub Compare URL format.\nExpected: https://github.com/ORG_NAME/REPO_NAME/compare/BASESHA..HEADSHA'); return;}}
+    const [, orgName, repoName, basehead] = match; window.location = `{protocol}://{host}/gh-range-diff/${{orgName}}/${{repoName}}/${{basehead}}`;
+}})();"
+    )
+}
