feat: add priority option

Author: Sakib25800

.sqlx/query-32c1cd799f8a1e8c451493959419570892d76e7adbc3e5c7b50e97e1bfee27bd.json

@@ -0,0 +1,2 @@
+-- Add down migration script here
+ALTER TABLE pull_request DROP COLUMN priority;

.sqlx/query-3f7f3b1f86aaa6acdf76f4d597aff4b4c7ac77b9498f77617086b2c12582b971.json

@@ -0,0 +1,2 @@
+-- Add up migration script here
+ALTER TABLE pull_request ADD COLUMN priority INT;

.sqlx/query-6410d5ac20c9b301fc7b9e639eabc7b82a2a1d01cde48ba01fdc5cd8ce2ce128.json renamed to .sqlx/query-491dd6278535e1015a8071f416a56112a153713a71af8d62917e2ff9873ecec1.json

@@ -23,10 +23,15 @@ pub enum Approver {
 #[derive(Debug, PartialEq)]
 pub enum BorsCommand {
     /// Approve a commit.
-    Approve(Approver),
+    Approve {
+        /// Who is approving the commit.
+        approver: Approver,
+        /// Priority of the commit.
+        priority: Option<u32>,
+    },
     /// Unapprove a commit.
     Unapprove,
-    /// Print help
+    /// Print help.
     Help,
     /// Ping the bot.
     Ping,

.sqlx/query-9531efa35aabadfb04c06f23375eafe1c8f10444e706940fa0f20a50b824044d.json renamed to .sqlx/query-67511f5f15c07537d3d4f2d011abedd2173ab301e8a06c740b587738b140795a.json

@@ -119,11 +119,17 @@ fn parse_parts(input: &str) -> Result<Vec<CommandPart>, CommandParseError> {
 /// Parsers
 
 /// Parses "@bors r+"
-fn parse_self_approve<'a>(command: &'a str, _parts: &[CommandPart<'a>]) -> ParseResult<'a> {
-    if command == "r+" {
-        Some(Ok(BorsCommand::Approve(Approver::Myself)))
-    } else {
-        None
+fn parse_self_approve<'a>(command: &'a str, parts: &[CommandPart<'a>]) -> ParseResult<'a> {
+    if command != "r+" {
+        return None;
+    }
+
+    match parse_priority(parts) {
+        Ok(priority) => Some(Ok(BorsCommand::Approve {
+            approver: Approver::Myself,
+            priority,
+        })),
+        Err(e) => Some(Err(e)),
     }
 }
 
@@ -135,9 +141,13 @@ fn parse_approve_on_behalf<'a>(parts: &[CommandPart<'a>]) -> ParseResult<'a> {
         } else if value.is_empty() {
             return Some(Err(CommandParseError::MissingArgValue { arg: "r" }));
         } else {
-            Some(Ok(BorsCommand::Approve(Approver::Specified(
-                value.to_string(),
-            ))))
+            match parse_priority(parts) {
+                Ok(priority) => Some(Ok(BorsCommand::Approve {
+                    approver: Approver::Specified(value.to_string()),
+                    priority,
+                })),
+                Err(e) => Some(Err(e)),
+            }
         }
     } else {
         Some(Err(CommandParseError::MissingArgValue { arg: "r" }))
@@ -241,6 +251,29 @@ fn parser_try_cancel<'a>(command: &'a str, parts: &[CommandPart<'a>]) -> ParseRe
     }
 }
 
+/// Parses "p=<priority>"
+fn parse_priority<'a>(parts: &[CommandPart<'a>]) -> Result<Option<u32>, CommandParseError<'a>> {
+    for part in parts {
+        match part {
+            CommandPart::Bare(key) => {
+                return Err(CommandParseError::UnknownArg(key));
+            }
+            CommandPart::KeyValue { key, value } => {
+                if *key == "p" {
+                    return match value.parse::<u32>() {
+                        Ok(p) => Ok(Some(p)),
+                        Err(_) => Err(CommandParseError::ValidationError(
+                            "Priority must be a valid number".to_string(),
+                        )),
+                    };
+                }
+            }
+        }
+    }
+
+    Ok(None)
+}
+
 #[cfg(test)]
 mod tests {
     use crate::bors::command::parser::{CommandParseError, CommandParser};
@@ -293,38 +326,43 @@ mod tests {
         assert_eq!(cmds.len(), 1);
         assert!(matches!(
             cmds[0],
-            Ok(BorsCommand::Approve(Approver::Myself))
+            Ok(BorsCommand::Approve {
+                approver: Approver::Myself,
+                priority: None,
+            })
         ));
     }
 
     #[test]
     fn parse_approve_on_behalf() {
         let cmds = parse_commands("@bors r=user1");
         assert_eq!(cmds.len(), 1);
-        insta::assert_debug_snapshot!(cmds[0], @r###"
+        insta::assert_debug_snapshot!(cmds[0], @r#"
         Ok(
-            Approve(
-                Specified(
+            Approve {
+                approver: Specified(
                     "user1",
                 ),
-            ),
+                priority: None,
+            },
         )
-        "###);
+        "#);
     }
 
     #[test]
     fn parse_approve_on_behalf_of_only_one_approver() {
         let cmds = parse_commands("@bors r=user1,user2");
         assert_eq!(cmds.len(), 1);
-        insta::assert_debug_snapshot!(cmds[0], @r###"
+        insta::assert_debug_snapshot!(cmds[0], @r#"
         Ok(
-            Approve(
-                Specified(
+            Approve {
+                approver: Specified(
                     "user1,user2",
                 ),
-            ),
+                priority: None,
+            },
         )
-        "###);
+        "#);
     }
 
     #[test]

.sqlx/query-7a9718cb973d09f68a6854a4fb08871f29d60b8617caf94dfa439f9ef8453caa.json

@@ -9,8 +9,14 @@ pub(super) async fn command_help(
     pr: &PullRequest,
 ) -> anyhow::Result<()> {
     let help = [
-        BorsCommand::Approve(Approver::Myself),
-        BorsCommand::Approve(Approver::Specified("".to_string())),
+        BorsCommand::Approve {
+            approver: Approver::Myself,
+            priority: None,
+        },
+        BorsCommand::Approve {
+            approver: Approver::Specified("".to_string()),
+            priority: None,
+        },
         BorsCommand::Unapprove,
         BorsCommand::Try {
             parent: None,
@@ -34,11 +40,11 @@ pub(super) async fn command_help(
 fn get_command_help(command: BorsCommand) -> String {
     // !!! When modifying this match, also update the command list above (in [`command_help`]) !!!
     let help = match command {
-        BorsCommand::Approve(Approver::Myself) => {
-            "`r+`: Approve this PR"
+        BorsCommand::Approve { approver: Approver::Myself, .. } => {
+            "`r+ [p=<priority>]`: Approve this PR. Optionally, you can specify a `<priority>`."
         }
-        BorsCommand::Approve(Approver::Specified(_)) => {
-            "`r=<user>`: Approve this PR on behalf of `<user>`"
+        BorsCommand::Approve {approver: Approver::Specified(_), ..} => {
+            "`r=<user> [p=<priority>]`: Approve this PR on behalf of `<user>`. Optionally, you can specify a `<priority>`."
         }
         BorsCommand::Unapprove => {
             "`r-`: Unapprove this PR"
@@ -68,8 +74,8 @@ mod tests {
         run_test(pool, |mut tester| async {
             tester.post_comment("@bors help").await?;
             insta::assert_snapshot!(tester.get_comment().await?, @r"
-            - `r+`: Approve this PR
-            - `r=<user>`: Approve this PR on behalf of `<user>`
+            - `r+ [p=<priority>]`: Approve this PR. Optionally, you can specify a `<priority>`.
+            - `r=<user> [p=<priority>]`: Approve this PR on behalf of `<user>`. Optionally, you can specify a `<priority>`.
             - `r-`: Unapprove this PR
             - `try [parent=<parent>] [jobs=<jobs>]`: Start a try build. Optionally, you can specify a `<parent>` SHA or a list of `<jobs>` to run
             - `try cancel`: Cancel a running try build

.sqlx/query-cf60a06c7355a3c94a9728af149a708b0a9c29b8f183d332eab720a53989a598.json

@@ -205,11 +205,18 @@ async fn handle_comment(
                 let repo = Arc::clone(&repo);
                 let database = Arc::clone(&database);
                 let result = match command {
-                    BorsCommand::Approve(approver) => {
+                    BorsCommand::Approve { approver, priority } => {
                         let span = tracing::info_span!("Approve");
-                        command_approve(repo, database, &pull_request, &comment.author, &approver)
-                            .instrument(span)
-                            .await
+                        command_approve(
+                            repo,
+                            database,
+                            &pull_request,
+                            &comment.author,
+                            &approver,
+                            priority,
+                        )
+                        .instrument(span)
+                        .await
                     }
                     BorsCommand::Unapprove => {
                         let span = tracing::info_span!("Unapprove");