Improved role selection

discord/component.go

@@ -30,3 +30,27 @@ func (discord *Discord) CreateMessageComponent(channelID string, content string,
 
 	return result.ID
 }
+
+// UpdateMessageComponent updates a discord component
+func (discord *Discord) UpdateMessageComponent(messageID string, channelID string, content string, components []Component) string {
+	dComponents := make([]discordgo.MessageComponent, len(components))
+	for i, v := range components {
+		dComponents[i] = v.toMessageComponent()
+	}
+
+	result, err := discord.session.ChannelMessageEditComplex(&discordgo.MessageEdit{
+		Components: dComponents,
+		Content:    &content,
+		Channel:    channelID,
+		ID:         messageID,
+	})
+	if err != nil {
+		log.Printf("Error updating message component: %s", err)
+	}
+
+	if result != nil {
+		return result.ID
+	}
+
+	return ""
+}

discord/component_button.go

@@ -9,6 +9,7 @@ type Button struct {
 	Label string
 	ID    string
 
+	Emoji   *Emoji
 	discord *Discord
 }
 
@@ -18,6 +19,17 @@ func (discord *Discord) NewButton(id string, label string) *Button {
 		discord: discord,
 		ID:      id,
 		Label:   label,
+		Emoji:   nil,
+	}
+}
+
+// NewButtonWithEmoji creates a new button component with a emoji
+func (discord *Discord) NewButtonWithEmoji(id string, label string, emoji *Emoji) *Button {
+	return &Button{
+		discord: discord,
+		ID:      id,
+		Label:   label,
+		Emoji:   emoji,
 	}
 }
 
@@ -41,9 +53,16 @@ func (button *Button) OnClick(action func(user common.User)) {
 }
 
 func (button *Button) toMessageComponent() discordgo.MessageComponent {
-	return discordgo.Button{
+	cmp := discordgo.Button{
 		Label:    button.Label,
 		CustomID: button.ID,
 		Style:    discordgo.PrimaryButton,
 	}
+	if button.Emoji != nil {
+		cmp.Emoji = &discordgo.ComponentEmoji{
+			Name: button.Emoji.Name,
+			ID:   button.Emoji.ID,
+		}
+	}
+	return cmp
 }

discord/emoji.go

@@ -0,0 +1,78 @@
+package discord
+
+import (
+	"image"
+	"log"
+
+	"github.com/bwmarrin/discordgo"
+	"github.com/yeslayla/birdbot/core"
+)
+
+type Emoji struct {
+	discord *Discord
+	ID      string
+
+	Name  string
+	Roles []string
+}
+
+// GetEmoji returns a emoji that exists on Discord
+func (discord *Discord) GetEmoji(name string) *Emoji {
+
+	emojis, err := discord.session.GuildEmojis(discord.guildID)
+	if err != nil {
+		log.Printf("Error occured listing roles: %s", err)
+		return nil
+	}
+
+	for _, emoji := range emojis {
+		if emoji.Managed {
+			continue
+		}
+		if emoji.Name == name {
+
+			return &Emoji{
+				ID:      emoji.ID,
+				discord: discord,
+				Name:    emoji.Name,
+				Roles:   emoji.Roles,
+			}
+		}
+	}
+
+	return nil
+}
+
+// CreateEmoji creates a new emoji on Discord
+func (discord *Discord) CreateEmoji(name string, image image.Image) *Emoji {
+	result, err := discord.session.GuildEmojiCreate(discord.guildID, &discordgo.EmojiParams{
+		Name:  name,
+		Image: core.ImageToBase64(image),
+	})
+	if err != nil {
+		log.Printf("Failed to create emoji: %s", err)
+		return nil
+	}
+
+	if result == nil {
+		log.Print("Failed to create emoji: result is nil")
+		return nil
+	}
+
+	return &Emoji{
+		ID:      result.ID,
+		Name:    result.Name,
+		Roles:   result.Roles,
+		discord: discord,
+	}
+}
+
+// Save updates the emoji on Discord
+func (emoji *Emoji) Save() {
+	if _, err := emoji.discord.session.GuildEmojiEdit(emoji.discord.guildID, emoji.ID, &discordgo.EmojiParams{
+		Name:  emoji.Name,
+		Roles: emoji.Roles,
+	}); err != nil {
+		log.Printf("Failed to save role: %s", err)
+	}
+}

discord/user.go

@@ -1,15 +1,12 @@
 package discord
 
 import (
-	"bytes"
-	"encoding/base64"
-	"fmt"
 	"image"
-	"image/jpeg"
 	"log"
 
 	"github.com/bwmarrin/discordgo"
 	"github.com/yeslayla/birdbot-common/common"
+	"github.com/yeslayla/birdbot/core"
 )
 
 // NewUser creates a new user object from a discordgo.User object
@@ -49,13 +46,7 @@ func (discord *Discord) GetAvatar(user common.User) image.Image {
 func (discord *Discord) GetAvatarBase64(user common.User) string {
 	avatar := discord.GetAvatar(user)
 
-	fmtAvatar := &bytes.Buffer{}
-	if err := jpeg.Encode(fmtAvatar, avatar, nil); err != nil {
-		log.Println("Error encoding avatar: ", err)
-		return ""
-	}
-
-	return fmt.Sprintf("data:image/png;base64,%s", base64.StdEncoding.EncodeToString(fmtAvatar.Bytes()))
+	return core.ImageToBase64(avatar)
 }
 
 // AssignRole adds a role to a user