Configure rotation mode through Zulip

Author: Kobzol

src/zulip.rs

@@ -1,6 +1,6 @@
 use crate::db::notifications::add_metadata;
 use crate::db::notifications::{self, delete_ping, move_indices, record_ping, Identifier};
-use crate::db::review_prefs::{get_review_prefs, upsert_review_prefs};
+use crate::db::review_prefs::{get_review_prefs, upsert_review_prefs, RotationMode};
 use crate::github::{get_id_for_username, GithubClient, User};
 use crate::handlers::docs_update::docs_update;
 use crate::handlers::pr_tracking::get_assigned_prs;
@@ -270,6 +270,15 @@ async fn workqueue_commands(
 
     let db_client = ctx.db.get().await;
 
+    let gh_username = username_from_gh_id(&ctx.github, gh_id)
+        .await?
+        .ok_or_else(|| anyhow::anyhow!("Cannot find your GitHub username in the team database"))?;
+    let user = User {
+        login: gh_username.clone(),
+        id: gh_id,
+    };
+    let review_prefs = get_review_prefs(&db_client, gh_id).await?;
+
     let response = match subcommand {
         "show" => {
             let mut assigned_prs = get_assigned_prs(ctx, gh_id)
@@ -287,17 +296,26 @@ async fn workqueue_commands(
             let review_prefs = get_review_prefs(&db_client, gh_id)
                 .await
                 .context("cannot get review preferences")?;
-            let capacity = match review_prefs.and_then(|p| p.max_assigned_prs) {
+            let capacity = match review_prefs.as_ref().and_then(|p| p.max_assigned_prs) {
                 Some(max) => max.to_string(),
                 None => String::from("Not set (i.e. unlimited)"),
             };
+            let rotation_mode = review_prefs
+                .as_ref()
+                .map(|p| p.rotation_mode)
+                .unwrap_or_default();
+            let rotation_mode = match rotation_mode {
+                RotationMode::OnRotation => "on rotation",
+                RotationMode::OffRotation => "off rotation",
+            };
 
             let mut response = format!(
                 "`rust-lang/rust` PRs in your review queue: {prs} ({} {})\n",
                 assigned_prs.len(),
                 pluralize("PR", assigned_prs.len())
             );
             writeln!(response, "Review capacity: {capacity}\n")?;
+            writeln!(response, "Rotation mode: *{rotation_mode}*\n")?;
             writeln!(response, "*Note that only certain PRs that are assigned to you are included in your review queue.*")?;
             response
         }
@@ -315,21 +333,16 @@ async fn workqueue_commands(
                         )?)
                     }
                 }
-                None => anyhow::bail!("Missing parameter."),
-            };
-            let gh_username = username_from_gh_id(&ctx.github, gh_id)
-                .await?
-                .ok_or_else(|| {
-                    anyhow::anyhow!("Cannot find your GitHub username in the team database")
-                })?;
-
-            let user = User {
-                login: gh_username.clone(),
-                id: gh_id,
+                None => anyhow::bail!("Missing parameter. See `work help` for more information."),
             };
-            upsert_review_prefs(&db_client, user, max_assigned_prs)
-                .await
-                .context("Error occurred while setting review preferences.")?;
+            upsert_review_prefs(
+                &db_client,
+                user,
+                max_assigned_prs,
+                review_prefs.map(|p| p.rotation_mode).unwrap_or_default(),
+            )
+            .await
+            .context("Error occurred while setting review preferences.")?;
             tracing::info!("Setting max assignment PRs of `{gh_username}` to {max_assigned_prs:?}");
             format!(
                 "Review capacity set to {}",
@@ -339,8 +352,37 @@ async fn workqueue_commands(
                 }
             )
         }
-        "help" => r"work show => show your assigned PRs
-work set-pr-limit <number>|unlimited => set the maximum number of PRs you can be assigned to"
+        "set-rotation-mode" => {
+            let rotation_mode = match words.next() {
+                Some(value) => {
+                    if words.next().is_some() {
+                        anyhow::bail!("Too many parameters.");
+                    }
+                    match value {
+                        "on" => RotationMode::OnRotation,
+                        "off" => RotationMode::OffRotation,
+                        _ => anyhow::bail!("Unknown rotation mode {value}. Use `on` or `off`.")
+                    }
+                }
+                None => anyhow::bail!("Missing parameter. See `work help` for more information."),
+            };
+            upsert_review_prefs(
+                &db_client,
+                user,
+                review_prefs.and_then(|p| p.max_assigned_prs.map(|v| v as u32)),
+                rotation_mode,
+            )
+                .await
+                .context("Error occurred while setting review preferences.")?;
+            tracing::info!("Setting rotation mode `{gh_username}` to {rotation_mode:?}");
+            format!(
+                "Rotation mode set to {rotation_mode:?}"
+            )
+        }
+        "help" => r"`work show`: show your assigned PRs
+`work set-pr-limit <number>|unlimited`: set the maximum number of PRs you can be assigned to
+`work set-rotation-mode <off|on>`: configure if you are *on* rotation or *off* rotation (e.g. when you are on a vacation)
+"
             .to_string(),
         _ => anyhow::bail!("Invalid subcommand."),
     };