stoatchat/src/database/permissions/server.rs

use crate::util::result::Result;

use super::PermissionCalculator;

use num_enum::TryFromPrimitive;
use std::ops;

#[derive(Debug, PartialEq, Eq, TryFromPrimitive, Copy, Clone)]
#[repr(u32)]
pub enum ServerPermission {
    View            = 0b00000000000000000000000000000001, // 1
    ManageMembers   = 0b00000000000000000000000000000010, // 2
    ManageChannels  = 0b00000000000000000000000000000100, // 4
    ManageServer    = 0b00000000000000000000000000001000, // 8
    // 8 bits of space
    ChangeNickname  = 0b00000000000000000001000000000000, // 4096
    ManageNicknames = 0b00000000000000000010000000000000, // 8192
    ChangeAvatar    = 0b00000000000000000100000000000000, // 16392
    RemoveAvatars   = 0b00000000000000001000000000000000, // 32784
    // 16 bits of space
}

impl_op_ex!(+ |a: &ServerPermission, b: &ServerPermission| -> u32 { *a as u32 | *b as u32 });
impl_op_ex_commutative!(+ |a: &u32, b: &ServerPermission| -> u32 { *a | *b as u32 });

bitfield! {
    pub struct ServerPermissions(MSB0 [u32]);
    u32;
    pub get_view, _: 31;
    pub get_manage_members, _: 30;
    pub get_manage_channels, _: 29;
    pub get_manage_server, _: 28;

    pub get_change_nickname, _: 19;
    pub get_manage_nicknames, _: 18;
    pub get_change_avatar, _: 17;
    pub get_remove_avatars, _: 16;
}

impl<'a> PermissionCalculator<'a> {
    pub async fn calculate_server(self) -> Result<u32> {
        let server = if let Some(server) = self.server {
            server
        } else {
            unreachable!()
        };

        if self.perspective.id == server.owner {
            Ok(u32::MAX)
        } else {
            Ok(ServerPermission::View + ServerPermission::ChangeNickname + ServerPermission::ChangeAvatar)
        }
    }

    pub async fn for_server(self) -> Result<ServerPermissions<[u32; 1]>> {
        Ok(ServerPermissions([self.calculate_server().await?]))
    }
}