mirror of
https://github.com/diamondburned/arikawa.git
synced 2024-12-14 01:15:00 +00:00
diamondburned
c6679dc52c
This commit refactors the Store interface in State into smaller interfaces in package store. These interfaces are combined into one structure called a "Cabinet". The default implementation of those interfaces have been rewritten in package defaultstore, while the old no-op implementation stays with the store package. This commit also omitted several state handlers for user events, as it is unclear what they are actually structured like.
100 lines
1.7 KiB
Go
100 lines
1.7 KiB
Go
package defaultstore
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/diamondburned/arikawa/v2/discord"
|
|
"github.com/diamondburned/arikawa/v2/internal/moreatomic"
|
|
"github.com/diamondburned/arikawa/v2/state/store"
|
|
)
|
|
|
|
type Role struct {
|
|
guilds moreatomic.Map
|
|
}
|
|
|
|
var _ store.RoleStore = (*Role)(nil)
|
|
|
|
type roles struct {
|
|
mut sync.Mutex
|
|
roles map[discord.RoleID]discord.Role
|
|
}
|
|
|
|
func NewRole() *Role {
|
|
return &Role{
|
|
guilds: *moreatomic.NewMap(func() interface{} {
|
|
return &roles{
|
|
roles: make(map[discord.RoleID]discord.Role, 1),
|
|
}
|
|
}),
|
|
}
|
|
}
|
|
|
|
func (s *Role) Reset() error {
|
|
return s.guilds.Reset()
|
|
}
|
|
|
|
func (s *Role) Role(guildID discord.GuildID, roleID discord.RoleID) (*discord.Role, error) {
|
|
iv, ok := s.guilds.Load(guildID)
|
|
if !ok {
|
|
return nil, store.ErrNotFound
|
|
}
|
|
|
|
rs := iv.(*roles)
|
|
|
|
rs.mut.Lock()
|
|
defer rs.mut.Unlock()
|
|
|
|
r, ok := rs.roles[roleID]
|
|
if ok {
|
|
return &r, nil
|
|
}
|
|
|
|
return nil, store.ErrNotFound
|
|
}
|
|
|
|
func (s *Role) Roles(guildID discord.GuildID) ([]discord.Role, error) {
|
|
iv, ok := s.guilds.Load(guildID)
|
|
if !ok {
|
|
return nil, store.ErrNotFound
|
|
}
|
|
|
|
rs := iv.(*roles)
|
|
|
|
rs.mut.Lock()
|
|
defer rs.mut.Unlock()
|
|
|
|
var roles = make([]discord.Role, 0, len(rs.roles))
|
|
for _, role := range rs.roles {
|
|
roles = append(roles, role)
|
|
}
|
|
|
|
return roles, nil
|
|
}
|
|
|
|
func (s *Role) RoleSet(guildID discord.GuildID, role discord.Role) error {
|
|
iv, _ := s.guilds.LoadOrStore(guildID)
|
|
|
|
rs := iv.(*roles)
|
|
|
|
rs.mut.Lock()
|
|
rs.roles[role.ID] = role
|
|
rs.mut.Unlock()
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *Role) RoleRemove(guildID discord.GuildID, roleID discord.RoleID) error {
|
|
iv, ok := s.guilds.Load(guildID)
|
|
if !ok {
|
|
return nil
|
|
}
|
|
|
|
rs := iv.(*roles)
|
|
|
|
rs.mut.Lock()
|
|
delete(rs.roles, roleID)
|
|
rs.mut.Unlock()
|
|
|
|
return nil
|
|
}
|