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
|
package revolt
// Permissions struct
type Permissions struct {
Bitvise uint
Mode string // can ben CHANNEL, SERVER or USER
Permissions map[string]uint
}
// Init all of the perms for channel.
func (p *Permissions) InitChannel() *Permissions {
p.Permissions = map[string]uint{
"VIEW": 1 << 0,
"SEND_MESSAGE": 1 << 1,
"MANAGE_MESSAGES": 1 << 2,
"MANAGE_CHANNEL": 1 << 3,
"VOICE_CALL": 1 << 4,
"INVITE_OTHERS": 1 << 5,
"EMBED_LINKS": 1 << 6,
"UPLOAD_FILES": 1 << 7,
}
p.Mode = "CHANNEL"
return p
}
// Init all of the perms for user.
func (p *Permissions) InitUser() *Permissions {
p.Permissions = map[string]uint{
"ACCESS": 1 << 0,
"VIEW_PROFILE": 1 << 1,
"SEND_MESSAGE": 1 << 2,
"INVITE": 1 << 3,
}
p.Mode = "USER"
return p
}
// Init all of the perms for server.
func (p *Permissions) InitServer() *Permissions {
p.Permissions = map[string]uint{
"VIEW": 1 << 0,
"MANAGE_ROLES": 1 << 1,
"MANAGE_CHANNELS": 1 << 2,
"MANAGE_SERVER": 1 << 3,
"KICK_MEMBERS": 1 << 4,
"BAN_MEMBERS": 1 << 5,
"TIMEOUT_MEMBERS": 1 << 6,
// 6 bits of space
"CHANGE_NICKNAME": 1 << 12,
"CHANGE_NICKNAMES": 1 << 13,
"CHANGE_AVATAR": 1 << 14,
"REMOVE_AVATARS": 1 << 15,
}
p.Mode = "SERVER"
return p
}
// Calculate if bitvise has permission
func (p Permissions) Has(perm string) bool {
if value, ok := p.Permissions[perm]; ok {
return p.Bitvise&value != 0
}
return false
}
// Add new permission(s).
func (p *Permissions) Add(perms ...string) *Permissions {
for _, perm := range perms {
if value, ok := p.Permissions[perm]; ok {
p.Bitvise = p.Bitvise | value
}
}
return p
}
// Remove permission(s).
func (p *Permissions) Remove(perms ...string) *Permissions {
for _, perm := range perms {
if value, ok := p.Permissions[perm]; ok {
p.Bitvise = p.Bitvise - value
}
}
return p
}
// Calculate perms and return unsigned int.
func (p Permissions) Calculate(perms ...string) uint {
var total uint
for _, perm := range perms {
if value, ok := p.Permissions[perm]; ok {
total = total | value
}
}
return total
}
|
