add retry command

Author: zeegomo

src/server/routes/webhooks/args.rs

@@ -122,6 +122,10 @@ generate_parser!(pub enum Command {
         name: Option<String> = "name",
     })
 
+    "retry" => Retry(RetryArgs {
+        name: Option<String> = "name",
+    })
+
     "reload-acl" => ReloadACL(ReloadACLArgs {})
 
     _ => Edit(EditArgs {

src/server/routes/webhooks/commands.rs

@@ -4,7 +4,9 @@ use crate::experiments::{CapLints, CrateSelect, Experiment, GitHubIssue, Mode, S
 use crate::prelude::*;
 use crate::server::github::Issue;
 use crate::server::messages::{Label, Message};
-use crate::server::routes::webhooks::args::{AbortArgs, EditArgs, RetryReportArgs, RunArgs};
+use crate::server::routes::webhooks::args::{
+    AbortArgs, EditArgs, RetryArgs, RetryReportArgs, RunArgs,
+};
 use crate::server::Data;
 
 pub fn ping(data: &Data, issue: &Issue) -> Fallible<()> {
@@ -108,6 +110,31 @@ pub fn retry_report(data: &Data, issue: &Issue, args: RetryReportArgs) -> Fallib
     }
 }
 
+pub fn retry(data: &Data, issue: &Issue, args: RetryArgs) -> Fallible<()> {
+    let name = get_name(&data.db, issue, args.name)?;
+
+    if let Some(mut experiment) = Experiment::get(&data.db, &name)? {
+        if experiment.status != Status::Failed {
+            bail!("Experiment **`{}`** didn't fail!", name);
+        }
+
+        experiment.set_status(&data.db, Status::Queued)?;
+        data.reports_worker.wake();
+
+        Message::new()
+            .line(
+                "hammer_and_wrench",
+                format!("Experiment **`{}`** queued again.", name),
+            )
+            .set_label(Label::ExperimentQueued)
+            .send(&issue.url, data)?;
+
+        Ok(())
+    } else {
+        bail!("an experiment named **`{}`** doesn't exist!", name);
+    }
+}
+
 pub fn abort(data: &Data, issue: &Issue, args: AbortArgs) -> Fallible<()> {
     let name = get_name(&data.db, issue, args.name)?;
 

src/server/routes/webhooks/mod.rs

@@ -111,6 +111,10 @@ fn process_command(
                 commands::retry_report(data, issue, args)?;
             }
 
+            Command::Retry(args) => {
+                commands::retry(data, issue, args)?;
+            }
+
             Command::Abort(args) => {
                 commands::abort(data, issue, args)?;
             }