128 lines
2.8 KiB
Go
128 lines
2.8 KiB
Go
package channel
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/diamondburned/arikawa/v2/discord"
|
|
"github.com/diamondburned/arikawa/v2/gateway"
|
|
"github.com/diamondburned/cchat"
|
|
"github.com/diamondburned/cchat-discord/internal/discord/channel/message"
|
|
"github.com/diamondburned/cchat-discord/internal/discord/channel/shared"
|
|
"github.com/diamondburned/cchat-discord/internal/discord/state"
|
|
"github.com/diamondburned/cchat-discord/internal/urlutils"
|
|
"github.com/diamondburned/cchat/text"
|
|
"github.com/diamondburned/cchat/utils/empty"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
type Channel struct {
|
|
empty.Server
|
|
shared.Channel
|
|
commander cchat.Commander
|
|
}
|
|
|
|
var _ cchat.Server = (*Channel)(nil)
|
|
|
|
func New(s *state.Instance, ch discord.Channel) (cchat.Server, error) {
|
|
channel, err := NewChannel(s, ch)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return channel, nil
|
|
}
|
|
|
|
func NewChannel(s *state.Instance, ch discord.Channel) (Channel, error) {
|
|
// Ensure the state keeps the channel's permission.
|
|
if ch.GuildID.IsValid() {
|
|
_, err := s.Permissions(ch.ID, s.UserID)
|
|
if err != nil {
|
|
return Channel{}, errors.Wrap(err, "failed to get permission")
|
|
}
|
|
}
|
|
|
|
sharedCh := shared.Channel{
|
|
ID: ch.ID,
|
|
GuildID: ch.GuildID,
|
|
State: s,
|
|
}
|
|
|
|
return Channel{
|
|
Channel: sharedCh,
|
|
commander: NewCommander(sharedCh),
|
|
}, nil
|
|
}
|
|
|
|
func (ch Channel) ID() cchat.ID {
|
|
return ch.Channel.ID.String()
|
|
}
|
|
|
|
func (ch Channel) Name() text.Rich {
|
|
c, err := ch.Self()
|
|
if err != nil {
|
|
return text.Rich{Content: ch.ID()}
|
|
}
|
|
|
|
return text.Plain(shared.ChannelName(*c))
|
|
}
|
|
|
|
func (ch Channel) AsCommander() cchat.Commander {
|
|
return ch.commander
|
|
}
|
|
|
|
func (ch Channel) AsMessenger() cchat.Messenger {
|
|
if !ch.HasPermission(discord.PermissionViewChannel) {
|
|
return nil
|
|
}
|
|
|
|
return message.New(ch.Channel)
|
|
}
|
|
|
|
func (ch Channel) AsIconer() cchat.Iconer {
|
|
// Guild channels never have an icon.
|
|
if ch.GuildID.IsValid() {
|
|
return nil
|
|
}
|
|
|
|
c, err := ch.Self()
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
|
|
// Only DM channels should have an icon.
|
|
if c.Type != discord.DirectMessage {
|
|
return nil
|
|
}
|
|
|
|
return PresenceAvatar{
|
|
user: c.DMRecipients[0],
|
|
guild: ch.GuildID,
|
|
state: ch.State,
|
|
}
|
|
}
|
|
|
|
type PresenceAvatar struct {
|
|
user discord.User
|
|
guild discord.GuildID
|
|
state *state.Instance
|
|
}
|
|
|
|
func (avy PresenceAvatar) Icon(ctx context.Context, iconer cchat.IconContainer) (func(), error) {
|
|
if avy.user.Avatar != "" {
|
|
iconer.SetIcon(urlutils.AvatarURL(avy.user.AvatarURL()))
|
|
}
|
|
|
|
// There are so many other places that could be checked, but this is good
|
|
// enough.
|
|
|
|
c, err := avy.state.Presence(avy.guild, avy.user.ID)
|
|
if err == nil && c.User.Avatar != "" {
|
|
iconer.SetIcon(urlutils.AvatarURL(c.User.AvatarURL()))
|
|
}
|
|
|
|
return avy.state.AddHandler(func(update *gateway.PresenceUpdateEvent) {
|
|
if avy.user.ID == update.User.ID {
|
|
iconer.SetIcon(urlutils.AvatarURL(update.User.AvatarURL()))
|
|
}
|
|
}), nil
|
|
}
|