Compare commits
7 Commits
v0.4.0
...
bfe9601cd3
Author | SHA1 | Date | |
---|---|---|---|
bfe9601cd3
|
|||
c6d877b101
|
|||
3bf763f196
|
|||
8f9205fbf0
|
|||
e1bae3edea
|
|||
34fdf453c1
|
|||
113c6927cb
|
@@ -58,7 +58,10 @@ import (
|
|||||||
func main() {
|
func main() {
|
||||||
//bolt defaults the command indicator to '.' however that can be changed with the options
|
//bolt defaults the command indicator to '.' however that can be changed with the options
|
||||||
//Example: bolt.New(bolt.WithIndicator('!')) would support commands like !ping
|
//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(
|
b.AddCommands(
|
||||||
// basic ping pong command, .ping can be run at anytime by anyone and will reply "pong"
|
// 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()
|
//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 {
|
if err != nil {
|
||||||
panic(err)
|
panic(err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
```
|
```
|
||||||
|
|
||||||
## Development
|
## Development
|
||||||
|
47
bolt.go
47
bolt.go
@@ -20,7 +20,8 @@ const (
|
|||||||
dg.IntentGuildMembers |
|
dg.IntentGuildMembers |
|
||||||
dg.IntentGuildPresences |
|
dg.IntentGuildPresences |
|
||||||
dg.IntentMessageContent |
|
dg.IntentMessageContent |
|
||||||
dg.IntentsGuildMessages
|
dg.IntentsGuildMessages |
|
||||||
|
dg.IntentGuildMessageReactions
|
||||||
)
|
)
|
||||||
|
|
||||||
// basic bot structure containing discordgo connection as well as the command map
|
// basic bot structure containing discordgo connection as well as the command map
|
||||||
@@ -44,20 +45,16 @@ type Bolt interface {
|
|||||||
timeoutCheck(msg *dg.MessageCreate, s *dg.Session, run Command) (bool, error)
|
timeoutCheck(msg *dg.MessageCreate, s *dg.Session, run Command) (bool, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
// setup
|
// create a new bolt interface
|
||||||
func init() {
|
func New(opts ...Option) (Bolt, error) {
|
||||||
//validate environment variables
|
|
||||||
_, check := os.LookupEnv(TOKEN_ENV_VAR)
|
_, check := os.LookupEnv(TOKEN_ENV_VAR)
|
||||||
if !check {
|
if !check {
|
||||||
log.Fatalf("the %s environment variable must be set", TOKEN_ENV_VAR)
|
return nil, fmt.Errorf("environment variable %s must be set", TOKEN_ENV_VAR)
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
// create a new bolt interface
|
|
||||||
func New(opts ...Option) Bolt {
|
|
||||||
bot, err := dg.New(fmt.Sprintf("Bot %s", os.Getenv(TOKEN_ENV_VAR)))
|
bot, err := dg.New(fmt.Sprintf("Bot %s", os.Getenv(TOKEN_ENV_VAR)))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal(err)
|
return nil, fmt.Errorf("failed to create Discord session: %e", err)
|
||||||
}
|
}
|
||||||
bot.Identify.Intents = BOT_INTENTS
|
bot.Identify.Intents = BOT_INTENTS
|
||||||
|
|
||||||
@@ -74,7 +71,7 @@ func New(opts ...Option) Bolt {
|
|||||||
opt(b)
|
opt(b)
|
||||||
}
|
}
|
||||||
|
|
||||||
return b
|
return b, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// starts the bot, commands are added and the connection to Discord is opened, this is a BLOCKING
|
// starts the bot, commands are added and the connection to Discord is opened, this is a BLOCKING
|
||||||
@@ -85,7 +82,7 @@ func (b *bolt) Start() error {
|
|||||||
|
|
||||||
err := b.Open()
|
err := b.Open()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return fmt.Errorf("failed to open websocket connection with Discord: %e", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
//safe shutdown handler
|
//safe shutdown handler
|
||||||
@@ -117,12 +114,12 @@ func (b *bolt) messageHandler(s *dg.Session, msg *dg.MessageCreate) {
|
|||||||
//get server information
|
//get server information
|
||||||
server, err := s.Guild(msg.GuildID)
|
server, err := s.Guild(msg.GuildID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Println(err)
|
log.Printf("failed to get guild: %e\n", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
channel, err := s.Channel(msg.ChannelID)
|
channel, err := s.Channel(msg.ChannelID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Println(err)
|
log.Printf("failed to get channel from guild: %e\n", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -132,20 +129,20 @@ func (b *bolt) messageHandler(s *dg.Session, msg *dg.MessageCreate) {
|
|||||||
msg.Content = "[Embedded Content]"
|
msg.Content = "[Embedded Content]"
|
||||||
}
|
}
|
||||||
|
|
||||||
if b.logLvl == LogLevelAll {
|
|
||||||
//log message
|
|
||||||
log.Printf("< %s | %s | %s > %s\n", server.Name, channel.Name, msg.Author.Username, msg.Content)
|
|
||||||
}
|
|
||||||
|
|
||||||
//the bot will ignore it's own messages to prevent command loops
|
//the bot will ignore it's own messages to prevent command loops
|
||||||
if msg.Author.ID == s.State.User.ID {
|
if msg.Author.ID == s.State.User.ID {
|
||||||
if b.logLvl == LogLevelCmd {
|
if b.logLvl == LogLevelCmd {
|
||||||
//log commands
|
//log command responses
|
||||||
log.Printf("< %s | %s | %s > %s\n", server.Name, channel.Name, msg.Author.Username, msg.Content)
|
log.Printf("< %s | %s | %s > %s\n", server.Name, channel.Name, msg.Author.Username, msg.Content)
|
||||||
}
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if b.logLvl == LogLevelAll {
|
||||||
|
//log message
|
||||||
|
log.Printf("< %s | %s | %s > %s\n", server.Name, channel.Name, msg.Author.Username, msg.Content)
|
||||||
|
}
|
||||||
|
|
||||||
//does the message have the command indicator
|
//does the message have the command indicator
|
||||||
lg := len(b.indicator)
|
lg := len(b.indicator)
|
||||||
if msg.Content[:lg] == b.indicator {
|
if msg.Content[:lg] == b.indicator {
|
||||||
@@ -173,7 +170,7 @@ func (b *bolt) handleCommand(msg *dg.MessageCreate, s *dg.Session, server *dg.Gu
|
|||||||
//has command met its timeout requirements
|
//has command met its timeout requirements
|
||||||
tc, err := b.timeoutCheck(msg, s, run)
|
tc, err := b.timeoutCheck(msg, s, run)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return fmt.Errorf("failed to calculate timeout for %s\n%e", run.Trigger, err)
|
||||||
}
|
}
|
||||||
if !tc {
|
if !tc {
|
||||||
return nil
|
return nil
|
||||||
@@ -183,7 +180,7 @@ func (b *bolt) handleCommand(msg *dg.MessageCreate, s *dg.Session, server *dg.Gu
|
|||||||
if run.Roles != nil {
|
if run.Roles != nil {
|
||||||
check, err := b.roleCheck(msg, s, run)
|
check, err := b.roleCheck(msg, s, run)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return fmt.Errorf("failed to perform permission checks for %s\n%e", run.Trigger, err)
|
||||||
}
|
}
|
||||||
if !check {
|
if !check {
|
||||||
return nil
|
return nil
|
||||||
@@ -226,7 +223,7 @@ func (b *bolt) handleCommand(msg *dg.MessageCreate, s *dg.Session, server *dg.Gu
|
|||||||
//run command payload
|
//run command payload
|
||||||
err = run.Payload(plMsg)
|
err = run.Payload(plMsg)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return fmt.Errorf("failed to execute payload function: %e", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
//update run time
|
//update run time
|
||||||
@@ -279,7 +276,7 @@ func (b *bolt) roleCheck(msg *dg.MessageCreate, s *dg.Session, run Command) (boo
|
|||||||
//get role name from ID
|
//get role name from ID
|
||||||
n, err := s.State.Role(msg.GuildID, r)
|
n, err := s.State.Role(msg.GuildID, r)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return false, err
|
return false, fmt.Errorf("failed to get role from ID %s\n%e", msg.GuildID, err)
|
||||||
}
|
}
|
||||||
//does this role exist in command roles
|
//does this role exist in command roles
|
||||||
check := slices.Contains(run.Roles, n.Name)
|
check := slices.Contains(run.Roles, n.Name)
|
||||||
@@ -294,7 +291,7 @@ func (b *bolt) roleCheck(msg *dg.MessageCreate, s *dg.Session, run Command) (boo
|
|||||||
reply := b.createReply("you do not have permissions to run that command", msg.ID, msg.ChannelID, msg.GuildID)
|
reply := b.createReply("you do not have permissions to run that command", msg.ID, msg.ChannelID, msg.GuildID)
|
||||||
_, err := s.ChannelMessageSendComplex(msg.ChannelID, reply)
|
_, err := s.ChannelMessageSendComplex(msg.ChannelID, reply)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return false, err
|
return false, fmt.Errorf("failed to send permission response: %e", err)
|
||||||
}
|
}
|
||||||
return false, nil
|
return false, nil
|
||||||
}
|
}
|
||||||
@@ -309,7 +306,7 @@ func (b *bolt) timeoutCheck(msg *dg.MessageCreate, s *dg.Session, run Command) (
|
|||||||
reply := b.createReply(fmt.Sprintf("that command cannot be run for another %s", b.getRemainingTimeout(wait)), msg.ID, msg.ChannelID, msg.GuildID)
|
reply := b.createReply(fmt.Sprintf("that command cannot be run for another %s", b.getRemainingTimeout(wait)), msg.ID, msg.ChannelID, msg.GuildID)
|
||||||
_, err := s.ChannelMessageSendComplex(msg.ChannelID, reply)
|
_, err := s.ChannelMessageSendComplex(msg.ChannelID, reply)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return false, err
|
return false, fmt.Errorf("failed to send timeout response: %e", err)
|
||||||
}
|
}
|
||||||
return false, nil
|
return false, nil
|
||||||
}
|
}
|
||||||
|
39
message.go
39
message.go
@@ -5,15 +5,38 @@ import "fmt"
|
|||||||
//built-in Discord reactions
|
//built-in Discord reactions
|
||||||
type Reaction string
|
type Reaction string
|
||||||
|
|
||||||
|
//a few easy-to-use emojis, Discordgo/Discord API requires them to be saved like this.
|
||||||
const (
|
const (
|
||||||
ReactionThumbsUp Reaction = ":+1:"
|
ReactionThumbsUp Reaction = "👍"
|
||||||
ReactionThumbsDown Reaction = ":-1:"
|
ReactionThumbsDown Reaction = "👎"
|
||||||
ReactionPointUp Reaction = ":point_up:"
|
ReactionHundred Reaction = "💯"
|
||||||
ReactionPointDown Reaction = ":point_down:"
|
ReactionHeart Reaction = "❤️"
|
||||||
ReactionHotdog Reaction = ":hotdog:"
|
ReactionPinkHeart Reaction = "🩷"
|
||||||
ReactionGiraffe Reaction = ":giraffe:"
|
ReactionOrangeHeart Reaction = "🧡"
|
||||||
ReactionWatermelon Reaction = ":watermelon:"
|
ReactionYellowHeart Reaction = "💛"
|
||||||
ReactionNoPedestrians Reaction = ":no_pedestrian:"
|
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 = "🐉"
|
||||||
)
|
)
|
||||||
|
|
||||||
// information about attachments to messages
|
// information about attachments to messages
|
||||||
|
Reference in New Issue
Block a user