Making job name a parameter, instead of having the pandas value hardcoded

Author: datapythonista

src/github.rs

@@ -60,14 +60,15 @@ fn extract_run_id_from_detail_url(url: &str) -> Result<u64, PreviewerError> {
 
 async fn run_id_from_commit(client: &Client,
                             base_api_url: &str,
-                            commit_reference: &str) -> Result<u64,
-                                                              PreviewerError> {
+                            commit_reference: &str,
+                            job_name: &str) -> Result<u64,
+                                                      PreviewerError> {
     let url = format!("{base_api_url}commits/{commit_reference}/check-runs");
     let json_obj = fetch_json(client, &url).await?;
 
     match json_obj["check_runs"]
         .as_array()
-        .and_then(|jobs| jobs.iter().find(|job| job["name"] == "Doc Build and Upload")) {
+        .and_then(|jobs| jobs.iter().find(|job| job["name"] == job_name)) {
             Some(job) => {
                 if let Some(detail_url) = job["details_url"].as_str() {
                     extract_run_id_from_detail_url(detail_url)
@@ -138,6 +139,7 @@ async fn download_artifact(client: &Client,
 pub async fn publish_artifact(client: &Client,
                               base_api_url: &str,
                               pull_request_number: u64,
+                              job_name: &str,
                               target_dir: &Path,
                               max_artifact_size: usize) -> Result<(),
                                                            PreviewerError> {
@@ -151,7 +153,8 @@ pub async fn publish_artifact(client: &Client,
 
     let run_id = run_id_from_commit(client,
                                     base_api_url,
-                                    &last_commit).await?;
+                                    &last_commit,
+                                    job_name).await?;
     log::info!("[PR {}] Run id: {}", pull_request_number, run_id);
 
     let artifact_url = artifact_url_from_run_id(client,

src/main.rs

@@ -1,4 +1,5 @@
 use std::path::Path;
+use serde_derive::Deserialize;
 use actix_web::{post, App, HttpResponse, HttpServer, Responder,
                 web, rt, middleware::Logger};
 use awc::Client;
@@ -10,17 +11,24 @@ mod cleanup;
 mod config;
 mod args;
 
+#[derive(Deserialize)]
+struct PreviewParams {
+    job: String,
+}
+
 /// Handler of the HTTP server /preview/ endpoint.
 ///
 /// This gets the parameters from the url and adds a job to the queue
 /// to be processed by the worker.
 #[post("/preview/{github_owner}/{github_repo}/{pull_request_number}/")]
 async fn preview_handler(params: web::Path<(String, String, u64)>,
+                         query_params: web::Query<PreviewParams>,
                          client: web::Data<Client>,
                          settings: web::Data<config::SettingsPerThread>) -> impl Responder {
     let github_owner = &params.0;
     let github_repo = &params.1;
     let pull_request_number = params.2;
+    let job_name = &query_params.job;
 
     if !settings.github_allowed_owners.is_empty()
         && !settings.github_allowed_owners.contains(github_owner)  {
@@ -51,6 +59,7 @@ async fn preview_handler(params: web::Path<(String, String, u64)>,
     match github::publish_artifact(&client,
                                    &base_api_url,
                                    pull_request_number,
+                                   job_name,
                                    &target_dir,
                                    settings.max_artifact_size).await {
         Ok(_) => {