Run CLA retry after any workflow completion

Author: clockwork-labs-bot

.github/workflows/retry-cla-assistant.yml

@@ -11,6 +11,8 @@ name: Retry CLA Assistant
 on:
   pull_request_target:
     types: [opened, reopened, synchronize, ready_for_review]
+  workflow_run:
+    types: [completed]
   schedule:
     - cron: "7,22,37,52 * * * *"
 

tools/ci/src/retry_cla_assistant.rs

@@ -86,6 +86,19 @@ fn candidate_pull_requests(client: &GithubClient, owner: &str, repo: &str, pr_nu
                 pr_numbers.insert(number);
             }
         }
+        "workflow_run" => {
+            if event.pointer("/workflow_run/name").and_then(Value::as_str) == Some("Retry CLA Assistant") {
+                println!("Ignoring completion of this workflow.");
+                return Ok(Vec::new());
+            }
+            if let Some(prs) = event.pointer("/workflow_run/pull_requests").and_then(Value::as_array) {
+                for pr in prs {
+                    if let Some(number) = pr.get("number").and_then(Value::as_u64) {
+                        pr_numbers.insert(number);
+                    }
+                }
+            }
+        }
         "workflow_dispatch" => {}
         other => println!("Unsupported event {other}; skipping."),
     }