use crate::models::{PermissionOverwrite, Permissions, Role};
#[must_use]
pub fn calculate_permissions(
user_id: u64,
base: Permissions,
mut roles: impl AsMut<[Role]>,
overwrites: Option<&[PermissionOverwrite]>,
) -> Permissions {
let roles = roles.as_mut();
roles.sort_unstable_by_key(|r| r.position);
calculate_permissions_sorted(user_id, base, roles, overwrites)
}
#[must_use]
pub fn calculate_permissions_sorted(
user_id: u64,
base: Permissions,
roles: impl AsRef<[Role]>,
overwrites: Option<&[PermissionOverwrite]>,
) -> Permissions {
let roles = roles.as_ref();
let mut perms = roles
.iter()
.fold(base, |acc, role| acc | role.permissions.allow);
perms &= !roles
.iter()
.fold(base, |acc, role| acc | role.permissions.deny);
if perms.contains(Permissions::ADMINISTRATOR) {
return Permissions::all();
}
if let Some(overwrites) = overwrites {
let mut role_overwrites = overwrites
.iter()
.filter_map(|o| roles.iter().find(|r| r.id == o.id).map(|r| (o, r.position)))
.collect::<Vec<_>>();
role_overwrites.sort_by_key(|(_, pos)| *pos);
for (overwrite, _) in role_overwrites {
perms |= overwrite.permissions.allow;
perms &= !overwrite.permissions.deny;
}
if let Some(o) = overwrites.iter().find(|o| o.id == user_id) {
perms |= o.permissions.allow;
perms &= !o.permissions.deny;
}
}
perms
}