TerraOceanPlugin/GoBot/discord/modules/vote.go

package modules

import (
	"github.com/bwmarrin/discordgo"

	"TerraOceanBot/discord/message"
)

type voteType struct {
	//TODO Add FormatName(name string) to clean up vote name
	EmbedBuilder  func(id int, name string) *message.Embed
	ResultHandler func(s *discordgo.Session, id int, name string, isPositive bool)
}

var voteTypes = map[string]voteType{
	"custom": voteType{
		EmbedBuilder:  createCustomEmbed,
		ResultHandler: announceCustomResult,
	},
	"invite": voteType{
		EmbedBuilder:  createInviteEmbed,
		ResultHandler: handleInviteResult,
	},
}