Add shortcut '@rustbot blocked' to apply S-blocked

parser/src/command/shortcut.rs

@@ -17,6 +17,7 @@ use std::fmt;
 pub enum ShortcutCommand {
     Ready,
     Author,
+    Blocked,
 }
 
 #[derive(PartialEq, Eq, Debug)]
@@ -39,6 +40,7 @@ impl ShortcutCommand {
         let mut shortcuts = HashMap::new();
         shortcuts.insert("ready", ShortcutCommand::Ready);
         shortcuts.insert("author", ShortcutCommand::Author);
+        shortcuts.insert("blocked", ShortcutCommand::Blocked);
 
         let mut toks = input.clone();
         if let Some(Token::Word(word)) = toks.peek_token()? {
@@ -92,3 +94,8 @@ fn test_4() {
         Some(&ParseError::ExpectedEnd),
     );
 }
+
+#[test]
+fn test_5() {
+    assert_eq!(parse("blocked"), Ok(Some(ShortcutCommand::Blocked)));
+}

src/handlers/shortcut.rs

@@ -25,17 +25,24 @@ pub(super) async fn handle_command(
         return Ok(());
     }
 
-    let issue_labels = issue.labels().to_owned();
+    let issue_labels = issue.labels();
     let waiting_on_review = "S-waiting-on-review";
     let waiting_on_author = "S-waiting-on-author";
+    let blocked = "S-blocked";
+    let status_labels = [waiting_on_review, waiting_on_author, blocked];
 
-    let (add, remove) = match input {
-        ShortcutCommand::Ready => (waiting_on_review, waiting_on_author),
-        ShortcutCommand::Author => (waiting_on_author, waiting_on_review),
+    let add = match input {
+        ShortcutCommand::Ready => waiting_on_review,
+        ShortcutCommand::Author => waiting_on_author,
+        ShortcutCommand::Blocked => blocked,
     };
 
     if !issue_labels.iter().any(|l| l.name == add) {
-        issue.remove_label(&ctx.github, remove).await?;
+        for remove in status_labels {
+            if remove != add {
+                issue.remove_label(&ctx.github, remove).await?;
+            }
+        }
         issue
             .add_labels(
                 &ctx.github,