Do not apply labels in autolabel when opening a draft PR

Author: Kobzol

src/handlers/autolabel.rs

@@ -1,3 +1,4 @@
+use crate::db::issue_data::IssueData;
 use crate::{
     config::AutolabelConfig,
     github::{IssuesAction, IssuesEvent, Label},
@@ -6,6 +7,15 @@ use crate::{
 use anyhow::Context as _;
 use tracing as log;
 
+/// Key for the state in the database
+const AUTOLABEL_KEY: &str = "autolabel";
+
+/// State stored in the database
+#[derive(Debug, Default, serde::Deserialize, serde::Serialize)]
+struct AutolabelState {
+    new_pr_labels_applied: bool,
+}
+
 pub(super) struct AutolabelInput {
     add: Vec<Label>,
     remove: Vec<Label>,
@@ -27,6 +37,12 @@ pub(super) async fn parse_input(
     // remove. Not much can be done about that currently; the before/after on
     // synchronize may be straddling a rebase, which will break diff generation.
     if event.action == IssuesAction::Opened || event.action == IssuesAction::Synchronize {
+        let mut db = ctx.db.get().await;
+        let mut state: IssueData<'_, AutolabelState> =
+            IssueData::load(&mut db, &event.issue, AUTOLABEL_KEY)
+                .await
+                .map_err(|e| e.to_string())?;
+
         let files = event
             .issue
             .diff(&ctx.github)
@@ -69,11 +85,18 @@ pub(super) async fn parse_input(
                         name: label.to_owned(),
                     });
                 }
-                if cfg.new_pr && event.action == IssuesAction::Opened {
-                    autolabels.push(Label {
-                        name: label.to_owned(),
-                    });
-                }
+            }
+
+            // Newly opened non-draft PRs or PRs that were opened as draft and that were marked as
+            // "ready for review" for the first time are treated as "new PRs".
+            let treat_as_new_pr = (event.action == IssuesAction::Opened && !event.issue.draft)
+                || (event.action == IssuesAction::ReadyForReview
+                    && !state.data.new_pr_labels_applied);
+            if cfg.new_pr && treat_as_new_pr {
+                autolabels.push(Label {
+                    name: label.to_owned(),
+                });
+                state.data.new_pr_labels_applied = true;
             }
 
             if event.issue.pull_request.is_none()
@@ -86,6 +109,8 @@ pub(super) async fn parse_input(
             }
         }
 
+        state.save().await.map_err(|e| e.to_string())?;
+
         if !autolabels.is_empty() {
             return Ok(Some(AutolabelInput {
                 add: autolabels,