forked from jmug/stoatchat
Servers: New permissions implementation.
Feature: Roles and group permissions. (server-side)
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
use crate::util::result::Result;
|
||||
use crate::util::result::{Error, Result};
|
||||
|
||||
use super::PermissionCalculator;
|
||||
use super::Ref;
|
||||
|
||||
use num_enum::TryFromPrimitive;
|
||||
use std::ops;
|
||||
@@ -22,6 +23,13 @@ pub enum ServerPermission {
|
||||
// 16 bits of space
|
||||
}
|
||||
|
||||
lazy_static! {
|
||||
pub static ref DEFAULT_PERMISSION: u32 =
|
||||
ServerPermission::View
|
||||
+ ServerPermission::ChangeNickname
|
||||
+ ServerPermission::ChangeAvatar;
|
||||
}
|
||||
|
||||
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 });
|
||||
|
||||
@@ -52,9 +60,26 @@ impl<'a> PermissionCalculator<'a> {
|
||||
if self.perspective.id == server.owner {
|
||||
Ok(u32::MAX)
|
||||
} else {
|
||||
Ok(ServerPermission::View
|
||||
+ ServerPermission::ChangeNickname
|
||||
+ ServerPermission::ChangeAvatar)
|
||||
match Ref::from_unchecked(self.perspective.id.clone()).fetch_member(&server.id).await {
|
||||
Ok(member) => {
|
||||
let mut perm = server.default_permissions.0 as u32;
|
||||
if let Some(roles) = member.roles {
|
||||
for role in roles {
|
||||
if let Some(server_role) = server.roles.get(&role) {
|
||||
perm |= server_role.permissions.0 as u32;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Ok(perm)
|
||||
}
|
||||
Err(error) => {
|
||||
match &error {
|
||||
Error::NotFound => Ok(0),
|
||||
_ => Err(error)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user