cchat-discord/internal/discord/channel/channel.go

package channel

import (
	"github.com/diamondburned/arikawa/discord"
	"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/text"
	"github.com/diamondburned/cchat/utils/empty"
	"github.com/pkg/errors"
)

type Channel struct {
	*empty.Server
	*shared.Channel
}

var _ cchat.Server = (*Channel)(nil)

func New(s *state.Instance, ch discord.Channel) (cchat.Server, error) {
	// Ensure the state keeps the channel's permission.
	_, err := s.Permissions(ch.ID, s.UserID)
	if err != nil {
		return nil, errors.Wrap(err, "Failed to get permission")
	}

	return Channel{
		Channel: &shared.Channel{
			ID:      ch.ID,
			GuildID: ch.GuildID,
			State:   s,
		},
	}, nil
}

// self does not do IO.
func (ch Channel) self() (*discord.Channel, error) {
	return ch.State.Store.Channel(ch.Channel.ID)
}

// messages does not do IO.
func (ch Channel) messages() ([]discord.Message, error) {
	return ch.State.Store.Messages(ch.Channel.ID)
}

func (ch Channel) guild() (*discord.Guild, error) {
	if ch.GuildID.IsValid() {
		return ch.State.Store.Guild(ch.GuildID)
	}
	return nil, errors.New("channel not in a guild")
}

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.Channel.ID.String()}
	}

	if c.NSFW {
		return text.Rich{Content: "#!" + c.Name}
	} else {
		return text.Rich{Content: "#" + c.Name}
	}
}

func (ch Channel) AsMessenger() cchat.Messenger {
	if !ch.HasPermission(discord.PermissionViewChannel) {
		return nil
	}

	return message.New(ch.Channel)
}