Major Refactor (#2)
* Major reworks * More refactoring * Refactor feature complete! * Comments * Add versioning
This commit is contained in:
146
discord/channel.go
Normal file
146
discord/channel.go
Normal file
@ -0,0 +1,146 @@
|
||||
package discord
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
|
||||
"github.com/bwmarrin/discordgo"
|
||||
"github.com/yeslayla/birdbot/core"
|
||||
)
|
||||
|
||||
// NewChannelFromID creates a channel from a given ID
|
||||
func (discord *Discord) NewChannelFromID(ID string) *core.Channel {
|
||||
channel, err := discord.session.Channel(ID)
|
||||
if err != nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
return &core.Channel{
|
||||
ID: ID,
|
||||
Name: channel.Name,
|
||||
Verified: true,
|
||||
}
|
||||
}
|
||||
|
||||
// NewChannelFromName creates a channel object with its name
|
||||
func (discord *Discord) NewChannelFromName(channel_name string) (*core.Channel, error) {
|
||||
|
||||
// Grab channels to query
|
||||
channels, err := discord.session.GuildChannels(discord.guildID)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to list channels when creating new channel: '%s': %v", channel_name, err)
|
||||
}
|
||||
for _, channel := range channels {
|
||||
|
||||
// Found channel!
|
||||
if channel.Name == channel_name {
|
||||
|
||||
// Channel already exists!
|
||||
return &core.Channel{
|
||||
Name: channel.Name,
|
||||
ID: channel.ID,
|
||||
Verified: true,
|
||||
}, nil
|
||||
}
|
||||
}
|
||||
|
||||
// Since a channel was not found, create one
|
||||
channel, err := discord.session.GuildChannelCreate(discord.guildID, channel_name, discordgo.ChannelTypeGuildText)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to created channel '%s': %v", channel_name, err)
|
||||
}
|
||||
|
||||
log.Printf("Created channel: '%s'", channel_name)
|
||||
return &core.Channel{
|
||||
Name: channel.Name,
|
||||
ID: channel.ID,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// GetVerifiedChannel looks up channel data and returns a verified objec
|
||||
func (discord *Discord) GetVerifiedChannel(channel *core.Channel) *core.Channel {
|
||||
if channel.ID != "" {
|
||||
return discord.NewChannelFromID(channel.ID)
|
||||
}
|
||||
if channel.Name != "" {
|
||||
channel, err := discord.NewChannelFromName(channel.Name)
|
||||
if err != nil {
|
||||
log.Println("Failed to verify channel by name: ", err)
|
||||
}
|
||||
return channel
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// DeleteChannel deletes a channel
|
||||
func (discord *Discord) DeleteChannel(channel *core.Channel) (bool, error) {
|
||||
if channel.Verified == false {
|
||||
channel = discord.GetVerifiedChannel(channel)
|
||||
}
|
||||
|
||||
_, err := discord.session.ChannelDelete(channel.ID)
|
||||
if err != nil {
|
||||
return false, fmt.Errorf("failed to delete channel: %v", err)
|
||||
}
|
||||
return true, nil
|
||||
}
|
||||
|
||||
// getChannelID returns a channel ID from its name
|
||||
func (discord *Discord) getChannelID(channel_name string) (string, error) {
|
||||
|
||||
// Get list of all channels
|
||||
channels, err := discord.session.GuildChannels(discord.guildID)
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("failed to list channels when getting channel id: '%s': %v", channel_name, err)
|
||||
}
|
||||
|
||||
// Loop through to find channel
|
||||
for _, ch := range channels {
|
||||
|
||||
// Find and return ID!
|
||||
if ch.Name == channel_name {
|
||||
return ch.ID, nil
|
||||
}
|
||||
}
|
||||
|
||||
return "", fmt.Errorf("failed to get channel id for '%s': channel not found", channel_name)
|
||||
}
|
||||
|
||||
// SendMessage sends a message to a given channel
|
||||
func (discord *Discord) SendMessage(channel *core.Channel, message string) error {
|
||||
if channel.Verified == false {
|
||||
channel = discord.GetVerifiedChannel(channel)
|
||||
}
|
||||
|
||||
_, err := discord.session.ChannelMessageSend(channel.ID, message)
|
||||
return err
|
||||
}
|
||||
|
||||
// MoveChannelToCategory places a channel in a given category
|
||||
func (discord *Discord) MoveChannelToCategory(channel *core.Channel, categoryID string) error {
|
||||
if channel.Verified == false {
|
||||
channel = discord.GetVerifiedChannel(channel)
|
||||
}
|
||||
|
||||
// Move to archive category
|
||||
if _, err := discord.session.ChannelEdit(channel.ID, &discordgo.ChannelEdit{
|
||||
ParentID: categoryID,
|
||||
}); err != nil {
|
||||
return fmt.Errorf("failed to move channel to archive category: %v", err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// ArchiveChannel archives a channel
|
||||
func (discord *Discord) ArchiveChannel(channel *core.Channel) error {
|
||||
if channel.Verified == false {
|
||||
channel = discord.GetVerifiedChannel(channel)
|
||||
}
|
||||
|
||||
_, err := discord.session.ChannelEdit(channel.ID, &discordgo.ChannelEdit{
|
||||
Archived: core.Bool(true),
|
||||
})
|
||||
|
||||
return err
|
||||
}
|
Reference in New Issue
Block a user