Add a filter to the triage dashboard

giraffate · giraffate · commit 7656c322f16d · 2021-05-17T22:32:19.000+09:00
diff --git a/src/main.rs b/src/main.rs
@@ -21,7 +21,14 @@ async fn serve_req(req: Request<Body>, ctx: Arc<Context>) -> Result<Response<Bod
             let params = matcher.params();
             let owner = params.find("owner");
             let repo = params.find("repo");
-            return triagebot::triage::pulls(ctx, owner.unwrap(), repo.unwrap()).await;
+            let filter = if let Some(q) = req.uri.query() {
+                let pair = url::form_urlencoded::parse(q.as_bytes()).find(|(k, _)| k == "q");
+                pair.map(|(_, filter)| filter.to_string())
+            } else {
+                None
+            };
+
+            return triagebot::triage::pulls(ctx, owner.unwrap(), repo.unwrap(), filter).await;
         } else {
             return triagebot::triage::index();
         }
diff --git a/src/triage.rs b/src/triage.rs
@@ -21,6 +21,7 @@ pub async fn pulls(
     ctx: Arc<Context>,
     owner: &str,
     repo: &str,
+    filter: Option<String>,
 ) -> Result<Response<Body>, hyper::Error> {
     let octocrab = &ctx.octocrab;
     let res = octocrab
@@ -51,6 +52,24 @@ pub async fn pulls(
         next_page = page.next;
     }
 
+    let mut label_filter = Vec::new();
+    let mut author_filter = Vec::new();
+    let mut assignee_filter = Vec::new();
+    if let Some(filter) = filter.clone() {
+        let kvs: Vec<&str> = filter.split_whitespace().collect();
+        for kv in kvs {
+            match kv.split_once(':') {
+                Some((k, v)) if !v.is_empty() => match k {
+                    "label" => label_filter.push(v.to_string()),
+                    "author" => author_filter.push(v.to_string()),
+                    "assignee" => assignee_filter.push(v.to_string()),
+                    _ => {}
+                },
+                _ => {}
+            }
+        }
+    }
+
     let mut pulls: Vec<Value> = Vec::new();
     for base_pull in base_pulls.into_iter() {
         let assignee = base_pull.assignee.map_or("".to_string(), |v| v.login);
@@ -71,20 +90,39 @@ pub async fn pulls(
             (Utc::now() - base_pull.created_at).num_days()
         };
 
-        let labels = base_pull.labels.map_or("".to_string(), |labels| {
+        let labels = base_pull.labels.map_or(Vec::new(), |labels| {
             labels
                 .iter()
                 .map(|label| label.name.clone())
                 .collect::<Vec<_>>()
-                .join(", ")
         });
+        if !label_filter.is_empty() {
+            let mut flg = false;
+            for filter in label_filter.iter() {
+                if !labels.iter().any(|label| *label == *filter) {
+                    flg = true;
+                }
+            }
+            if flg {
+                continue;
+            }
+        }
+
+        let labels = labels.join(", ");
         let wait_for_author = labels.contains("S-waiting-on-author");
         let wait_for_review = labels.contains("S-waiting-on-review");
         let html_url = base_pull.html_url;
         let number = base_pull.number;
         let title = base_pull.title;
         let author = base_pull.user.login;
 
+        if !author_filter.is_empty() && !author_filter.iter().all(|s| *s == author) {
+            continue;
+        }
+        if !assignee_filter.is_empty() && !assignee_filter.iter().all(|s| *s == assignee) {
+            continue;
+        }
+
         let pull = PullRequest {
             html_url,
             number,
@@ -105,6 +143,11 @@ pub async fn pulls(
     context.insert("pulls", &pulls);
     context.insert("owner", &owner);
     context.insert("repo", &repo);
+    if let Some(filter) = filter {
+        context.insert("filter", &filter);
+    } else {
+        context.insert("filter", "");
+    }
 
     let tera = tera::Tera::new("templates/triage/**/*").unwrap();
     let body = Body::from(tera.render("pulls.html", &context).unwrap());
diff --git a/templates/triage/pulls.html b/templates/triage/pulls.html
@@ -42,6 +42,10 @@
 
     <body>
         <h1>Triage dashboard - <a href="https://github.com/{{ owner }}/{{ repo }}">{{ owner }}/{{ repo }}</a></h1>
+        <form action="/triage/{{ owner }}/{{ repo }}" method="get">
+            <input type="search" name="q" value="{{ filter }}" style="width: 400px;">
+            <input type="submit" value="filter">
+        </form>
         <table>
             <thead>
                 <tr>
