twilight-http 0.5.0

use crate::{
    client::Client,
    error::Error,
    request::{
        application::{InteractionError, InteractionErrorType},
        validate, Pending, Request,
    },
    routing::Route,
};
use serde::Serialize;
use std::collections::HashMap;
use twilight_model::{
    application::command::permissions::CommandPermissions,
    id::{ApplicationId, CommandId, GuildId},
};

#[derive(Serialize)]
struct PartialGuildCommandPermissions {
    id: CommandId,
    permissions: CommandPermissions,
}

/// Update command permissions for all commands in a guild.
///
/// This overwrites the command permissions so the full set of permissions
/// have to be sent every time.
pub struct SetCommandPermissions<'a> {
    application_id: ApplicationId,
    guild_id: GuildId,
    fields: Vec<PartialGuildCommandPermissions>,
    fut: Option<Pending<'a, CommandPermissions>>,
    http: &'a Client,
}

impl<'a> SetCommandPermissions<'a> {
    pub(crate) fn new(
        http: &'a Client,
        application_id: ApplicationId,
        guild_id: GuildId,
        permissions: impl Iterator<Item = (CommandId, CommandPermissions)>,
    ) -> Result<Self, InteractionError> {
        let fields = permissions
            .map(
                |(command_id, command_permissions)| PartialGuildCommandPermissions {
                    id: command_id,
                    permissions: command_permissions,
                },
            )
            .collect::<Vec<PartialGuildCommandPermissions>>();

        if !fields
            .iter()
            .fold(HashMap::new(), |mut acc, permission| {
                acc.entry(permission.id)
                    .and_modify(|p| *p += 1)
                    .or_insert(1_usize);

                acc
            })
            .iter()
            .all(|permission| validate::command_permissions(*permission.1))
        {
            return Err(InteractionError {
                kind: InteractionErrorType::TooManyCommandPermissions,
            });
        }

        Ok(Self {
            application_id,
            guild_id,
            fields,
            fut: None,
            http,
        })
    }

    fn start(&mut self) -> Result<(), Error> {
        let request = Request::builder(Route::SetCommandPermissions {
            application_id: self.application_id.0,
            guild_id: self.guild_id.0,
        })
        .json(&self.fields)?;

        self.fut
            .replace(Box::pin(self.http.request(request.build())));

        Ok(())
    }
}

poll_req!(SetCommandPermissions<'_>, CommandPermissions);

#[cfg(test)]
mod tests {
    use super::SetCommandPermissions;
    use crate::Client;
    use std::iter;
    use twilight_model::{
        application::command::permissions::{CommandPermissions, CommandPermissionsType},
        id::{ApplicationId, CommandId, GuildId, RoleId},
    };

    fn make_iter() -> impl Iterator<Item = (CommandId, CommandPermissions)> {
        iter::repeat((
            CommandId(3),
            CommandPermissions {
                id: CommandPermissionsType::Role(RoleId(4)),
                permission: true,
            },
        ))
    }

    #[test]
    fn test_correct_validation() {
        let http = Client::new("token");

        let permissions = make_iter().take(4);

        let request = SetCommandPermissions::new(&http, ApplicationId(1), GuildId(2), permissions);

        assert!(request.is_ok());
    }

    #[test]
    fn test_incorrect_validation() {
        let http = Client::new("token");

        let permissions = make_iter().take(11);

        let request = SetCommandPermissions::new(&http, ApplicationId(1), GuildId(2), permissions);

        assert!(request.is_err());
    }
}