Add enhanced GitHub Actions raw logs viewer

Urgau · Urgau · commit 8800ee60fd72 · 2025-06-29T14:51:31.000+02:00
diff --git a/src/gha_logs.rs b/src/gha_logs.rs
@@ -0,0 +1,114 @@
+use crate::github;
+use crate::handlers::Context;
+use anyhow::Context as _;
+use hyper::header::{CONTENT_SECURITY_POLICY, CONTENT_TYPE};
+use hyper::{Body, Response, StatusCode};
+use std::str::FromStr;
+use std::sync::Arc;
+use uuid::Uuid;
+
+const ANSI_UP_URL: &str = "https://cdn.jsdelivr.net/npm/ansi_up@6.0.6/+esm";
+
+pub async fn gha_logs(
+    ctx: Arc<Context>,
+    owner: &str,
+    repo: &str,
+    log_id: &str,
+) -> Result<Response<Body>, hyper::Error> {
+    let res = process_logs(ctx, owner, repo, log_id).await;
+    let res = match res {
+        Ok(r) => r,
+        Err(e) => {
+            tracing::error!("gha_logs: unable to serve logs for {owner}/{repo}#{log_id}: {e:?}");
+            return Ok(Response::builder()
+                .status(StatusCode::INTERNAL_SERVER_ERROR)
+                .body(Body::from(format!("{:?}", e)))
+                .unwrap());
+        }
+    };
+
+    Ok(res)
+}
+
+async fn process_logs(
+    ctx: Arc<Context>,
+    owner: &str,
+    repo: &str,
+    log_id: &str,
+) -> anyhow::Result<Response<Body>> {
+    let log_id = u128::from_str(log_id).context("log_id is not a number")?;
+
+    let repos = ctx
+        .team
+        .repos()
+        .await
+        .context("unable to retrieve team repos")?;
+
+    let Some(repos) = repos.repos.get(owner) else {
+        anyhow::bail!("Organization `{owner}` is not part of team repos")
+    };
+
+    if !repos.iter().any(|r| r.name == repo) {
+        anyhow::bail!("Repository `{repo}` is not part of team repos");
+    }
+
+    let logs = ctx
+        .github
+        .raw_job_logs(
+            &github::IssueRepository {
+                organization: owner.to_string(),
+                repository: repo.to_string(),
+            },
+            log_id,
+        )
+        .await
+        .context("unable to get the raw logs")?;
+
+    let json_logs = serde_json::to_string(&logs).context("unable to JSON-ify the raw logs")?;
+
+    let nonce = Uuid::new_v4().to_hyphenated().to_string();
+
+    let html = format!(
+        r#"<!DOCTYPE html>
+<html>
+<head>
+    <title>{owner}/{repo}${log_id} - triagebot</title>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <link rel="icon" sizes="32x32" type="image/png" href="https://rust-lang.org/static/images/favicon-32x32.png">    
+    <style>
+        body {{
+            font: 14px SFMono-Regular, Consolas, Liberation Mono, Menlo, monospace;
+            background: #0C0C0C;
+            color: #CCCCCC;
+            white-space: pre;
+        }}
+    </style>
+    <script type="module" nonce="{nonce}">
+        import {{ AnsiUp }} from '{ANSI_UP_URL}'
+
+        var logs = {json_logs};
+        var ansi_up = new AnsiUp();
+
+        var html = ansi_up.ansi_to_html(logs);
+
+        var cdiv = document.getElementById("console");
+        cdiv.innerHTML = html;
+    </script>
+</head>
+<body id="console">
+</body>
+</html>"#,
+    );
+
+    tracing::info!("gha_logs: serving logs for {owner}/{repo}#{log_id}");
+
+    return Ok(Response::builder()
+        .status(StatusCode::OK)
+        .header(CONTENT_TYPE, "text/html; charset=utf-8")
+        .header(
+            CONTENT_SECURITY_POLICY,
+            format!("script-src 'nonce-{nonce}' {ANSI_UP_URL}"),
+        )
+        .body(Body::from(html))?);
+}
diff --git a/src/lib.rs b/src/lib.rs
@@ -13,6 +13,7 @@ pub mod agenda;
 mod changelogs;
 mod config;
 pub mod db;
+pub mod gha_logs;
 pub mod github;
 pub mod handlers;
 mod interactions;
diff --git a/src/main.rs b/src/main.rs
@@ -44,6 +44,7 @@ async fn serve_req(
     let mut router = Router::new();
     router.add("/triage", "index".to_string());
     router.add("/triage/:owner/:repo", "pulls".to_string());
+    router.add("/gha-logs/:owner/:repo/:log-id", "gha-logs".to_string());
     let (req, body_stream) = req.into_parts();
 
     if let Ok(matcher) = router.recognize(req.uri.path()) {
@@ -52,8 +53,14 @@ async fn serve_req(
             let owner = params.find("owner");
             let repo = params.find("repo");
             return triagebot::triage::pulls(ctx, owner.unwrap(), repo.unwrap()).await;
-        } else {
+        } else if matcher.handler().as_str() == "index" {
             return triagebot::triage::index();
+        } else if matcher.handler().as_str() == "gha-logs" {
+            let params = matcher.params();
+            let owner = params.find("owner").unwrap();
+            let repo = params.find("repo").unwrap();
+            let log_id = params.find("log-id").unwrap();
+            return triagebot::gha_logs::gha_logs(ctx, owner, repo, log_id).await;
         }
     }
 
