mirror of
https://github.com/diamondburned/cchat-discord.git
synced 2024-11-01 04:14:21 +00:00
132 lines
3.1 KiB
Go
132 lines
3.1 KiB
Go
package guild
|
|
|
|
import (
|
|
"context"
|
|
"sort"
|
|
|
|
"github.com/diamondburned/arikawa/discord"
|
|
"github.com/diamondburned/arikawa/gateway"
|
|
"github.com/diamondburned/cchat"
|
|
"github.com/diamondburned/cchat-discord/internal/discord/category"
|
|
"github.com/diamondburned/cchat-discord/internal/discord/channel"
|
|
"github.com/diamondburned/cchat-discord/internal/discord/state"
|
|
"github.com/diamondburned/cchat-discord/internal/urlutils"
|
|
"github.com/diamondburned/cchat/text"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
type Guild struct {
|
|
id discord.GuildID
|
|
state *state.Instance
|
|
}
|
|
|
|
var (
|
|
_ cchat.Iconer = (*Guild)(nil)
|
|
_ cchat.Server = (*Guild)(nil)
|
|
_ cchat.Lister = (*Guild)(nil)
|
|
)
|
|
|
|
func New(s *state.Instance, g *discord.Guild) *Guild {
|
|
return &Guild{
|
|
id: g.ID,
|
|
state: s,
|
|
}
|
|
}
|
|
|
|
func NewFromID(s *state.Instance, gID discord.GuildID) (*Guild, error) {
|
|
g, err := s.Guild(gID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return New(s, g), nil
|
|
}
|
|
|
|
func (g *Guild) self(ctx context.Context) (*discord.Guild, error) {
|
|
return g.state.WithContext(ctx).Guild(g.id)
|
|
}
|
|
|
|
func (g *Guild) selfState() (*discord.Guild, error) {
|
|
return g.state.Store.Guild(g.id)
|
|
}
|
|
|
|
func (g *Guild) ID() cchat.ID {
|
|
return g.id.String()
|
|
}
|
|
|
|
func (g *Guild) Name() text.Rich {
|
|
s, err := g.selfState()
|
|
if err != nil {
|
|
// This shouldn't happen.
|
|
return text.Rich{Content: g.id.String()}
|
|
}
|
|
|
|
return text.Rich{Content: s.Name}
|
|
}
|
|
|
|
// IsIconer returns true if the guild has an icon.
|
|
func (g *Guild) IsIconer() bool {
|
|
s, err := g.selfState()
|
|
return err == nil && s.Icon != ""
|
|
}
|
|
|
|
func (g *Guild) Icon(ctx context.Context, iconer cchat.IconContainer) (func(), error) {
|
|
s, err := g.self(ctx)
|
|
if err != nil {
|
|
// This shouldn't happen.
|
|
return nil, errors.Wrap(err, "Failed to get guild")
|
|
}
|
|
|
|
// Used for comparison.
|
|
if s.Icon != "" {
|
|
iconer.SetIcon(urlutils.AvatarURL(s.IconURL()))
|
|
}
|
|
|
|
return g.state.AddHandler(func(update *gateway.GuildUpdateEvent) {
|
|
if g.id == update.ID {
|
|
iconer.SetIcon(urlutils.AvatarURL(s.IconURL()))
|
|
}
|
|
}), nil
|
|
}
|
|
|
|
// IsLister returns true.
|
|
func (g *Guild) IsLister() bool { return true }
|
|
|
|
func (g *Guild) Servers(container cchat.ServersContainer) error {
|
|
c, err := g.state.Channels(g.id)
|
|
if err != nil {
|
|
return errors.Wrap(err, "Failed to get channels")
|
|
}
|
|
|
|
// Only get top-level channels (those with category ID being null).
|
|
var toplevels = category.FilterAccessible(g.state, category.FilterCategory(c, 0))
|
|
|
|
// Sort so that positions are correct.
|
|
sort.SliceStable(toplevels, func(i, j int) bool {
|
|
return toplevels[i].Position < toplevels[j].Position
|
|
})
|
|
|
|
// Sort so that channels are before categories.
|
|
sort.SliceStable(toplevels, func(i, _ int) bool {
|
|
return toplevels[i].Type != discord.GuildCategory
|
|
})
|
|
|
|
var chs = make([]cchat.Server, 0, len(toplevels))
|
|
|
|
for _, ch := range toplevels {
|
|
switch ch.Type {
|
|
case discord.GuildCategory:
|
|
chs = append(chs, category.New(g.state, ch))
|
|
case discord.GuildText:
|
|
c, err := channel.New(g.state, ch)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "Failed to make channel %q: %v", ch.Name, err)
|
|
}
|
|
chs = append(chs, c)
|
|
}
|
|
}
|
|
|
|
container.SetServers(chs)
|
|
return nil
|
|
}
|