Export merge bots in the REST API

Author: Kobzol

rust_team_data/src/v1.rs

@@ -219,12 +219,19 @@ pub enum BranchProtectionMode {
     PrNotRequired,
 }
 
+#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
+#[serde(rename_all = "snake_case")]
+pub enum MergeBot {
+    Homu,
+}
+
 #[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
 pub struct BranchProtection {
     pub pattern: String,
     pub dismiss_stale_review: bool,
     pub mode: BranchProtectionMode,
     pub allowed_merge_teams: Vec<String>,
+    pub merge_bots: Vec<MergeBot>,
 }
 
 #[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]

src/static_api.rs

@@ -1,5 +1,7 @@
 use crate::data::Data;
-use crate::schema::{Bot, Email, Permissions, RepoPermission, TeamKind, ZulipGroupMember};
+use crate::schema::{
+    Bot, Email, MergeBot, Permissions, RepoPermission, TeamKind, ZulipGroupMember,
+};
 use anyhow::{ensure, Context as _, Error};
 use indexmap::IndexMap;
 use log::info;
@@ -59,6 +61,13 @@ impl<'a> Generator<'a> {
                         BranchProtectionMode::PrNotRequired
                     },
                     allowed_merge_teams: b.allowed_merge_teams.clone(),
+                    merge_bots: b
+                        .merge_bots
+                        .iter()
+                        .map(|bot| match bot {
+                            MergeBot::Homu => v1::MergeBot::Homu,
+                        })
+                        .collect(),
                 })
                 .collect();
             let managed_by_bors = r.bots.contains(&Bot::Bors);

tests/static-api/_expected/v1/repos.json

@@ -25,7 +25,8 @@
               "required_approvals": 1
             }
           },
-          "allowed_merge_teams": []
+          "allowed_merge_teams": [],
+          "merge_bots": []
         }
       ],
       "archived": true,
@@ -59,7 +60,8 @@
           },
           "allowed_merge_teams": [
             "foo"
-          ]
+          ],
+          "merge_bots": []
         }
       ],
       "archived": false,

tests/static-api/_expected/v1/repos/archived_repo.json

@@ -23,7 +23,8 @@
           "required_approvals": 1
         }
       },
-      "allowed_merge_teams": []
+      "allowed_merge_teams": [],
+      "merge_bots": []
     }
   ],
   "archived": true,

tests/static-api/_expected/v1/repos/some_repo.json

@@ -25,7 +25,8 @@
       },
       "allowed_merge_teams": [
         "foo"
-      ]
+      ],
+      "merge_bots": []
     }
   ],
   "archived": false,