triagebot/src/handlers/nominate.rs

//! Purpose: Allow team members to nominate issues or PRs.

use crate::{
    config::NominateConfig,
    github::{self, Event},
    handlers::{Context, Handler},
    interactions::ErrorComment,
};
use futures::future::{BoxFuture, FutureExt};
use parser::command::nominate::{NominateCommand, Style};
use parser::command::{Command, Input};

pub(super) struct NominateHandler;

impl Handler for NominateHandler {
    type Input = NominateCommand;
    type Config = NominateConfig;

    fn parse_input(
        &self,
        ctx: &Context,
        event: &Event,
        _: Option<&Self::Config>,
    ) -> Result<Option<Self::Input>, String> {
        let body = if let Some(b) = event.comment_body() {
            b
        } else {
            // not interested in other events
            return Ok(None);
        };

        if let Event::Issue(e) = event {
            if !matches!(e.action, github::IssuesAction::Opened | github::IssuesAction::Edited) {
                // skip events other than opening or editing the issue to avoid retriggering commands in the
                // issue body
                return Ok(None);
            }
        }

        let mut input = Input::new(&body, &ctx.username);
        let command = input.parse_command();
        
        if let Some(previous) = event.comment_from() {
            let mut prev_input = Input::new(&previous, &ctx.username);
            let prev_command = prev_input.parse_command();
            if command == prev_command {
                return Ok(None);
            }
        }
        
        match command {
            Command::Nominate(Ok(command)) => Ok(Some(command)),
            Command::Nominate(Err(err)) => {
                return Err(format!(
                    "Parsing nominate command in [comment]({}) failed: {}",
                    event.html_url().expect("has html url"),
                    err
                ));
            }
            _ => Ok(None),
        }
    }

    fn handle_input<'a>(
        &self,
        ctx: &'a Context,
        config: &'a Self::Config,
        event: &'a Event,
        input: Self::Input,
    ) -> BoxFuture<'a, anyhow::Result<()>> {
        handle_input(ctx, config, event, input).boxed()
    }
}

async fn handle_input(
    ctx: &Context,
    config: &NominateConfig,
    event: &Event,
    cmd: NominateCommand,
) -> anyhow::Result<()> {
    let is_team_member = if let Err(_) | Ok(false) = event.user().is_team_member(&ctx.github).await
    {
        false
    } else {
        true
    };

    if !is_team_member {
        let cmnt = ErrorComment::new(
            &event.issue().unwrap(),
            format!(
                "Nominating and approving issues and pull requests is restricted to members of\
                 the Rust teams."
            ),
        );
        cmnt.post(&ctx.github).await?;
        return Ok(());
    }

    let mut issue_labels = event.issue().unwrap().labels().to_owned();
    if cmd.style == Style::BetaApprove {
        if !issue_labels.iter().any(|l| l.name == "beta-nominated") {
            let cmnt = ErrorComment::new(
                &event.issue().unwrap(),
                format!(
                    "This pull request is not beta-nominated, so it cannot be approved yet.\
                     Perhaps try to beta-nominate it by using `@{} beta-nominate <team>`?",
                    ctx.username,
                ),
            );
            cmnt.post(&ctx.github).await?;
            return Ok(());
        }

        // Add the beta-accepted label, but don't attempt to remove beta-nominated or the team
        // label.
        if !issue_labels.iter().any(|l| l.name == "beta-accepted") {
            issue_labels.push(github::Label {
                name: "beta-accepted".into(),
            });
        }
    } else {
        if !config.teams.contains_key(&cmd.team) {
            let cmnt = ErrorComment::new(
                &event.issue().unwrap(),
                format!(
                    "This team (`{}`) cannot be nominated for via this command;\
                     it may need to be added to `triagebot.toml` on the master branch.",
                    cmd.team,
                ),
            );
            cmnt.post(&ctx.github).await?;
            return Ok(());
        }

        let label = config.teams[&cmd.team].clone();
        if !issue_labels.iter().any(|l| l.name == label) {
            issue_labels.push(github::Label { name: label });
        }

        let style_label = match cmd.style {
            Style::Decision => "I-nominated",
            Style::Beta => "beta-nominated",
            Style::BetaApprove => unreachable!(),
        };
        if !issue_labels.iter().any(|l| l.name == style_label) {
            issue_labels.push(github::Label {
                name: style_label.into(),
            });
        }
    }

    if &issue_labels[..] != event.issue().unwrap().labels() {
        event
            .issue()
            .unwrap()
            .set_labels(&ctx.github, issue_labels)
            .await?;
    }

    Ok(())
}