From d97c5ea1f33dedad79da2e9b0810d11aadaf51da Mon Sep 17 00:00:00 2001
From: ismailarilik <arilik.ismail@gmail.com>
Date: Tue, 5 Nov 2024 20:04:42 +0300
Subject: [PATCH] feat(notify-zulip): support `notify-zulip.<label>` getting
 mapped to multiple actions

---
 src/config.rs                |  10 +-
 src/handlers/notify_zulip.rs | 179 +++++++++++++++++++++++++++--------
 2 files changed, 149 insertions(+), 40 deletions(-)

diff --git a/src/config.rs b/src/config.rs
index 523836fb..a29e8743 100644
--- a/src/config.rs
+++ b/src/config.rs
@@ -221,7 +221,15 @@ pub(crate) struct AutolabelLabelConfig {
 #[derive(PartialEq, Eq, Debug, serde::Deserialize)]
 pub(crate) struct NotifyZulipConfig {
     #[serde(flatten)]
-    pub(crate) labels: HashMap<String, NotifyZulipLabelConfig>,
+    pub(crate) labels: HashMap<String, NotifyZulipNameConfig>,
+}
+
+#[derive(PartialEq, Eq, Debug, serde::Deserialize)]
+pub(crate) struct NotifyZulipNameConfig {
+    #[serde(flatten)]
+    pub(crate) default: Option<NotifyZulipLabelConfig>,
+    #[serde(flatten)]
+    pub(crate) others: Option<HashMap<String, NotifyZulipLabelConfig>>,
 }
 
 #[derive(PartialEq, Eq, Debug, serde::Deserialize)]
diff --git a/src/handlers/notify_zulip.rs b/src/handlers/notify_zulip.rs
index ba697813..8727eae2 100644
--- a/src/handlers/notify_zulip.rs
+++ b/src/handlers/notify_zulip.rs
@@ -1,5 +1,5 @@
 use crate::{
-    config::{NotifyZulipConfig, NotifyZulipLabelConfig},
+    config::{NotifyZulipConfig, NotifyZulipLabelConfig, NotifyZulipNameConfig},
     github::{Issue, IssuesAction, IssuesEvent, Label},
     handlers::Context,
 };
@@ -12,6 +12,8 @@ pub(super) struct NotifyZulipInput {
     /// For example, if an `I-prioritize` issue is closed,
     /// this field will be `I-prioritize`.
     label: Label,
+    is_default_valid: bool,
+    names: Vec<String>,
 }
 
 pub(super) enum NotificationType {
@@ -52,24 +54,67 @@ pub(super) async fn parse_input(
 fn parse_label_change_input(
     event: &IssuesEvent,
     label: Label,
-    config: &NotifyZulipLabelConfig,
+    config: &NotifyZulipNameConfig,
 ) -> Option<NotifyZulipInput> {
-    if !has_all_required_labels(&event.issue, config) {
-        // Issue misses a required label, ignore this event
+    let mut is_default_valid = false;
+    let mut names: Vec<String> = vec![];
+
+    match &config.default {
+        Some(label_config) => {
+            if has_all_required_labels(&event.issue, &label_config) {
+                match event.action {
+                    IssuesAction::Labeled { .. } if !label_config.messages_on_add.is_empty() => {
+                        is_default_valid = true;
+                    }
+                    IssuesAction::Unlabeled { .. } if !label_config.messages_on_remove.is_empty() => {
+                        is_default_valid = true;
+                    }
+                    _ => (),
+                }
+            }
+        }
+        None => ()
+    }
+
+    match &config.others {
+        Some(other_configs) => {
+            for (name, label_config) in other_configs {
+                if has_all_required_labels(&event.issue, &label_config) {
+                    match event.action {
+                        IssuesAction::Labeled { .. } if !label_config.messages_on_add.is_empty() => {
+                            names.push(name.to_string());
+                        }
+                        IssuesAction::Unlabeled { .. } if !label_config.messages_on_remove.is_empty() => {
+                            names.push(name.to_string());
+                        }
+                        _ => (),
+                    }
+                }
+            }
+        }
+        None => ()
+    }
+
+    if !is_default_valid && names.is_empty() {
+        // It seems that there is no match between this event and any notify-zulip config, ignore this event
         return None;
     }
 
     match event.action {
-        IssuesAction::Labeled { .. } if !config.messages_on_add.is_empty() => {
+        IssuesAction::Labeled { .. } => {
             Some(NotifyZulipInput {
                 notification_type: NotificationType::Labeled,
                 label,
+                is_default_valid,
+                names,
             })
         }
-        IssuesAction::Unlabeled { .. } if !config.messages_on_remove.is_empty() => {
+        IssuesAction::Unlabeled { .. } => {
             Some(NotifyZulipInput {
                 notification_type: NotificationType::Unlabeled,
                 label,
+                is_default_valid,
+                names,
             })
         }
         _ => None,
@@ -92,22 +137,65 @@ fn parse_close_reopen_input(
                 .map(|config| (label, config))
         })
         .flat_map(|(label, config)| {
-            if !has_all_required_labels(&event.issue, config) {
-                // Issue misses a required label, ignore this event
+            let mut is_default_valid = false;
+            let mut names: Vec<String> = vec![];
+
+            match &config.default {
+                Some(label_config) => {
+                    if has_all_required_labels(&event.issue, &label_config) {
+                        match event.action {
+                            IssuesAction::Closed if !label_config.messages_on_close.is_empty() => {
+                                is_default_valid = true;
+                            }
+                            IssuesAction::Reopened if !label_config.messages_on_reopen.is_empty() => {
+                                is_default_valid = true;
+                            }
+                            _ => (),
+                        }
+                    }
+                }
+                None => ()
+            }
+
+            match &config.others {
+                Some(other_configs) => {
+                    for (name, label_config) in other_configs {
+                        if has_all_required_labels(&event.issue, &label_config) {
+                            match event.action {
+                                IssuesAction::Closed if !label_config.messages_on_close.is_empty() => {
+                                    names.push(name.to_string());
+                                }
+                                IssuesAction::Reopened if !label_config.messages_on_reopen.is_empty() => {
+                                    names.push(name.to_string());
+                                }
+                                _ => (),
+                            }
+                        }
+                    }
+                }
+                None => ()
+            }
+
+            if !is_default_valid && names.is_empty() {
+                // It seems that there is no match between this event and any notify-zulip config, ignore this event
                 return None;
             }
 
             match event.action {
-                IssuesAction::Closed if !config.messages_on_close.is_empty() => {
+                IssuesAction::Closed => {
                     Some(NotifyZulipInput {
                         notification_type: NotificationType::Closed,
                         label,
+                        is_default_valid,
+                        names,
                     })
                 }
-                IssuesAction::Reopened if !config.messages_on_reopen.is_empty() => {
+                IssuesAction::Reopened => {
                     Some(NotifyZulipInput {
                         notification_type: NotificationType::Reopened,
                         label,
+                        is_default_valid,
+                        names,
                     })
                 }
                 _ => None,
@@ -140,41 +228,54 @@ pub(super) async fn handle_input<'a>(
     inputs: Vec<NotifyZulipInput>,
 ) -> anyhow::Result<()> {
     for input in inputs {
-        let config = &config.labels[&input.label.name];
-
-        let topic = &config.topic;
-        let topic = topic.replace("{number}", &event.issue.number.to_string());
-        let mut topic = topic.replace("{title}", &event.issue.title);
-        // Truncate to 60 chars (a Zulip limitation)
-        let mut chars = topic.char_indices().skip(59);
-        if let (Some((len, _)), Some(_)) = (chars.next(), chars.next()) {
-            topic.truncate(len);
-            topic.push('…');
+        let name_config = &config.labels[&input.label.name];
+
+        // Get valid label configs
+        let mut label_configs: Vec<&NotifyZulipLabelConfig> = vec![];
+        if input.is_default_valid {
+            label_configs.push(name_config.default.as_ref().unwrap());
+        }
+        for name in input.names {
+            label_configs.push(&name_config.others.as_ref().unwrap()[&name]);
         }
 
-        let msgs = match input.notification_type {
-            NotificationType::Labeled => &config.messages_on_add,
-            NotificationType::Unlabeled => &config.messages_on_remove,
-            NotificationType::Closed => &config.messages_on_close,
-            NotificationType::Reopened => &config.messages_on_reopen,
-        };
+        for label_config in label_configs {
+            let config = label_config;
 
-        let recipient = crate::zulip::Recipient::Stream {
-            id: config.zulip_stream,
-            topic: &topic,
-        };
+            let topic = &config.topic;
+            let topic = topic.replace("{number}", &event.issue.number.to_string());
+            let mut topic = topic.replace("{title}", &event.issue.title);
+            // Truncate to 60 chars (a Zulip limitation)
+            let mut chars = topic.char_indices().skip(59);
+            if let (Some((len, _)), Some(_)) = (chars.next(), chars.next()) {
+                topic.truncate(len);
+                topic.push('…');
+            }
 
-        for msg in msgs {
-            let msg = msg.replace("{number}", &event.issue.number.to_string());
-            let msg = msg.replace("{title}", &event.issue.title);
-            let msg = replace_team_to_be_nominated(&event.issue.labels, msg);
+            let msgs = match input.notification_type {
+                NotificationType::Labeled => &config.messages_on_add,
+                NotificationType::Unlabeled => &config.messages_on_remove,
+                NotificationType::Closed => &config.messages_on_close,
+                NotificationType::Reopened => &config.messages_on_reopen,
+            };
 
-            crate::zulip::MessageApiRequest {
-                recipient,
-                content: &msg,
+            let recipient = crate::zulip::Recipient::Stream {
+                id: config.zulip_stream,
+                topic: &topic,
+            };
+
+            for msg in msgs {
+                let msg = msg.replace("{number}", &event.issue.number.to_string());
+                let msg = msg.replace("{title}", &event.issue.title);
+                let msg = replace_team_to_be_nominated(&event.issue.labels, msg);
+
+                crate::zulip::MessageApiRequest {
+                    recipient,
+                    content: &msg,
+                }
+                .send(&ctx.github.raw())
+                .await?;
             }
-            .send(&ctx.github.raw())
-            .await?;
         }
     }