robespierre-models 0.3.1

robespierre - model declarations
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115

use crate::{
    channels::{
        Channel, ChannelPermissions, GroupChannel, ServerChannel, TextChannel, VoiceChannel,
    },
    id::{RoleId, UserId},
    servers::{Member, Role, Server, ServerPermissions},
};

fn perms_or(server: &Server, roles: &[RoleId]) -> (ServerPermissions, ChannelPermissions) {
    if let Some(ref roles_obj) = server.roles {
        roles_obj.iter().filter(|it| roles.contains(it.0)).fold(
            server.default_permissions,
            |p, (_, Role { permissions, .. })| (p.0 | permissions.0, p.1 | permissions.1),
        )
    } else {
        server.default_permissions
    }
}

fn perms_or_in_channel(
    server: &Server,
    roles: &[RoleId],
    channel: &Channel,
) -> (ServerPermissions, ChannelPermissions) {
    let base = perms_or(server, roles);

    match channel {
        Channel::TextChannel(TextChannel {
            server_channel:
                ServerChannel {
                    default_permissions,
                    role_permissions,
                    ..
                },
            ..
        })
        | Channel::VoiceChannel(VoiceChannel {
            server_channel:
                ServerChannel {
                    default_permissions,
                    role_permissions,
                    ..
                },
            ..
        }) => {
            let new = (
                base.0,
                base.1 | default_permissions.unwrap_or(ChannelPermissions::empty()),
            );

            let new = role_permissions
                .iter()
                .filter(|it| roles.contains(it.0))
                .fold(new, |p, (_, new)| (p.0, p.1 | *new));

            new
        }

        _ => unreachable!(),
    }
}

pub fn member_has_permissions(
    member: &Member,
    server_permissions: ServerPermissions,
    server: &Server,
) -> bool {
    if server.owner == member.id.user {
        // owner has all perms
        return true;
    }

    perms_or(server, &member.roles)
        .0
        .contains(server_permissions)
}

pub fn member_has_permissions_in_channel(
    member: &Member,
    server_permissions: ServerPermissions,
    server: &Server,
    channel_permissions: ChannelPermissions,
    channel: &Channel,
) -> bool {
    if server.owner == member.id.user {
        // owner has all perms
        return true;
    }

    let (sp, cp) = perms_or_in_channel(server, &member.roles, channel);

    sp.contains(server_permissions) && cp.contains(channel_permissions)
}

pub fn user_has_permissions_in_group(
    user: UserId,
    group: &Channel,
    check_permissions: ChannelPermissions,
) -> bool {
    match group {
        Channel::Group(GroupChannel {
            owner, permissions, ..
        }) => {
            if user == *owner {
                // owner has all perms
                return true;
            }

            permissions
                .unwrap_or(ChannelPermissions::empty())
                .contains(check_permissions)
        }
        _ => false,
    }
}