Compare commits
7 Commits
v0.4.1
...
option/cha
Author | SHA1 | Date | |
---|---|---|---|
b47c979e3b
|
|||
bfe9601cd3
|
|||
c6d877b101
|
|||
3bf763f196
|
|||
8f9205fbf0
|
|||
e1bae3edea
|
|||
34fdf453c1
|
@@ -58,7 +58,10 @@ import (
|
||||
func main() {
|
||||
//bolt defaults the command indicator to '.' however that can be changed with the options
|
||||
//Example: bolt.New(bolt.WithIndicator('!')) would support commands like !ping
|
||||
b := bolt.New(bolt.WithLogLevel(bolt.LogLevelCmd))
|
||||
b, err := bolt.New(bolt.WithLogLevel(bolt.LogLevelCmd))
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
b.AddCommands(
|
||||
// basic ping pong command, .ping can be run at anytime by anyone and will reply "pong"
|
||||
@@ -95,11 +98,12 @@ func main() {
|
||||
)
|
||||
|
||||
//start is a blocking call that handles safe-shutdown, all configuration and setup should be done before calling Start()
|
||||
err := b.Start()
|
||||
err = b.Start()
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
|
||||
```
|
||||
|
||||
## Development
|
||||
|
47
bolt.go
47
bolt.go
@@ -14,21 +14,24 @@ import (
|
||||
)
|
||||
|
||||
const (
|
||||
TOKEN_ENV_VAR = "DISCORD_TOKEN" //label for token environment variable
|
||||
//Discord auth token must be saved as an environment variable using this label/name
|
||||
//in order for bolt to detect it
|
||||
TOKEN_ENV_VAR = "DISCORD_TOKEN"
|
||||
|
||||
BOT_INTENTS = dg.IntentGuilds |
|
||||
dg.IntentGuildMembers |
|
||||
dg.IntentGuildPresences |
|
||||
dg.IntentMessageContent |
|
||||
dg.IntentsGuildMessages
|
||||
dg.IntentsGuildMessages |
|
||||
dg.IntentGuildMessageReactions
|
||||
)
|
||||
|
||||
// basic bot structure containing discordgo connection as well as the command map
|
||||
type bolt struct {
|
||||
*dg.Session //holds discordgo internals
|
||||
commands map[string]Command //maps trigger phrase to command struct for fast lookup
|
||||
indicator string //the indicator used to detect whether a message is a command
|
||||
logLvl LogLevel //determines how much the bot logs
|
||||
channels []string //optional list of channels that the bot listens in, all channels are used if list is empty
|
||||
}
|
||||
|
||||
type Bolt interface {
|
||||
@@ -44,7 +47,7 @@ type Bolt interface {
|
||||
timeoutCheck(msg *dg.MessageCreate, s *dg.Session, run Command) (bool, error)
|
||||
}
|
||||
|
||||
// create a new bolt interface
|
||||
// creates and returns a new bolt interface after ensuring token is present and applying option functions
|
||||
func New(opts ...Option) (Bolt, error) {
|
||||
_, check := os.LookupEnv(TOKEN_ENV_VAR)
|
||||
if !check {
|
||||
@@ -74,9 +77,9 @@ func New(opts ...Option) (Bolt, error) {
|
||||
}
|
||||
|
||||
// starts the bot, commands are added and the connection to Discord is opened, this is a BLOCKING
|
||||
// call that handles safe shutdown of the bot
|
||||
// call that handles safe shutdown
|
||||
func (b *bolt) Start() error {
|
||||
//register commands and open connection
|
||||
//registers the message handler used internally to detect commands
|
||||
b.AddHandler(b.messageHandler)
|
||||
|
||||
err := b.Open()
|
||||
@@ -103,25 +106,43 @@ func (b *bolt) stop() error {
|
||||
|
||||
// adds commands to bot command map for use
|
||||
func (b *bolt) AddCommands(cmd ...Command) {
|
||||
//command trigger words are mapped directly to payload functions to prevent
|
||||
//extra loops when searching for the appropriate payload
|
||||
for _, c := range cmd {
|
||||
b.commands[c.Trigger] = c
|
||||
}
|
||||
}
|
||||
|
||||
// handler function that parses message data and executes any command payloads
|
||||
// internal message handler function. Ensure's the message is a message the bot should interpret
|
||||
// and checks to see if it contains a command then executes the proper payload function
|
||||
func (b *bolt) messageHandler(s *dg.Session, msg *dg.MessageCreate) {
|
||||
//get server information
|
||||
server, err := s.Guild(msg.GuildID)
|
||||
if err != nil {
|
||||
log.Printf("failed to get guild: %e\n", err)
|
||||
return
|
||||
}
|
||||
//get channel name to, idk if i like this
|
||||
channel, err := s.Channel(msg.ChannelID)
|
||||
if err != nil {
|
||||
log.Printf("failed to get channel from guild: %e\n", err)
|
||||
return
|
||||
}
|
||||
|
||||
if len(b.channels) > 0 {
|
||||
var check bool
|
||||
for _, c := range b.channels {
|
||||
if channel.Name == c {
|
||||
check = true
|
||||
}
|
||||
}
|
||||
|
||||
//not a channel we care about
|
||||
if !check {
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
server, err := s.Guild(msg.GuildID)
|
||||
if err != nil {
|
||||
log.Printf("failed to get guild: %e\n", err)
|
||||
return
|
||||
}
|
||||
|
||||
//if there is no content it is likely an image, gif, or sticker, updating message content for
|
||||
//better logging and to avoid confusion
|
||||
if len(msg.Content) == 0 {
|
||||
|
41
message.go
41
message.go
@@ -5,15 +5,40 @@ import "fmt"
|
||||
//built-in Discord reactions
|
||||
type Reaction string
|
||||
|
||||
//a few easy-to-use emojis, Discordgo/Discord API requires them to be saved like this.
|
||||
const (
|
||||
ReactionThumbsUp Reaction = ":+1:"
|
||||
ReactionThumbsDown Reaction = ":-1:"
|
||||
ReactionPointUp Reaction = ":point_up:"
|
||||
ReactionPointDown Reaction = ":point_down:"
|
||||
ReactionHotdog Reaction = ":hotdog:"
|
||||
ReactionGiraffe Reaction = ":giraffe:"
|
||||
ReactionWatermelon Reaction = ":watermelon:"
|
||||
ReactionNoPedestrians Reaction = ":no_pedestrian:"
|
||||
ReactionThumbsUp Reaction = "👍"
|
||||
ReactionThumbsDown Reaction = "👎"
|
||||
ReactionHundred Reaction = "💯"
|
||||
ReactionHeart Reaction = "❤️"
|
||||
ReactionPinkHeart Reaction = "🩷"
|
||||
ReactionOrangeHeart Reaction = "🧡"
|
||||
ReactionYellowHeart Reaction = "💛"
|
||||
ReactionGreenHeart Reaction = "💚"
|
||||
ReactionBlueHeart Reaction = "💙"
|
||||
ReactionBlackHeart Reaction = "🖤"
|
||||
ReactionPointUp Reaction = "☝️"
|
||||
ReactionPointDown Reaction = "👇"
|
||||
ReactionHotdog Reaction = "🌭"
|
||||
ReactionDog Reaction = "🐶"
|
||||
ReactionCat Reaction = "🐱"
|
||||
ReactionMonkey Reaction = "🐒"
|
||||
ReactionGiraffe Reaction = "🦒"
|
||||
ReactionDuck Reaction = "🦆"
|
||||
ReactionGoose Reaction = "🪿"
|
||||
ReactionWatermelon Reaction = "🍉"
|
||||
ReactionHoney Reaction = "🍯"
|
||||
ReactionSandwich Reaction = "🥪"
|
||||
ReactionPepper Reaction = "🌶️"
|
||||
ReactionNoPedestrians Reaction = "🚷"
|
||||
ReactionExclamation Reaction = "❗"
|
||||
ReactionDoubleExclamation Reaction = "‼️"
|
||||
ReactionSkull Reaction = "💀"
|
||||
ReactionSpeakingHead Reaction = "🗣️"
|
||||
ReactionGreenCheck Reaction = "✅"
|
||||
ReactionDragon Reaction = "🐉"
|
||||
ReactionLizard Reaction = "🦎"
|
||||
ReactionTakeAKnee Reaction = "🧎♂️➡️"
|
||||
)
|
||||
|
||||
// information about attachments to messages
|
||||
|
Reference in New Issue
Block a user