1 Commits

Author SHA1 Message Date
b47c979e3b pushing code for channel lockdowns
- honestly not sure if its worth adding
2025-08-13 15:42:06 -04:00
7 changed files with 139 additions and 215 deletions

View File

@@ -44,10 +44,6 @@ The React method will react to the command message by adding the requested emoji
func (m *Message) Respond(res string) error func (m *Message) Respond(res string) error
``` ```
The Respond method will send the value of <b>res</b> in response to the command message. The Respond method will send the value of <b>res</b> in response to the command message.
```go
func Delete() error
```
The Delete method will delete the message from the text channel
### Example ### Example
```go ```go

170
bolt.go
View File

@@ -10,12 +10,13 @@ import (
"syscall" "syscall"
"time" "time"
"code.jakeyoungdev.com/go/lite"
dg "github.com/bwmarrin/discordgo" dg "github.com/bwmarrin/discordgo"
) )
const ( 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 | BOT_INTENTS = dg.IntentGuilds |
dg.IntentGuildMembers | dg.IntentGuildMembers |
@@ -23,42 +24,30 @@ const (
dg.IntentMessageContent | dg.IntentMessageContent |
dg.IntentsGuildMessages | dg.IntentsGuildMessages |
dg.IntentGuildMessageReactions dg.IntentGuildMessageReactions
DEFAULT_DB_NAME = "boltdb"
DB_PATH = "/var/lib/bolt"
) )
type Data struct {
//user, which user saved this
//data, does this just go interface?
//date saved?
//expires?
}
// basic bot structure containing discordgo connection as well as the command map
type bolt struct { type bolt struct {
*dg.Session //holds discordgo internals *dg.Session //holds discordgo internals
commands map[string]Command //maps trigger phrase to command struct for fast lookup 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 indicator string //the indicator used to detect whether a message is a command
logLvl LogLevel //determines how much the bot logs logLvl LogLevel //determines how much the bot logs
db lite.Database channels []string //optional list of channels that the bot listens in, all channels are used if list is empty
} }
type Bolt interface { type Bolt interface {
Start() error Start() error
AddCommands(cmd ...Command) AddCommands(cmd ...Command)
Database(file string) error
//filtered methods //filtered methods
stop() error stop() error
messageHandler(s *dg.Session, msg *dg.MessageCreate) messageHandler(s *dg.Session, msg *dg.MessageCreate)
handleCommand(msgEvent *MessageCreateEvent, s *dg.Session, lg int) error handleCommand(msg *dg.MessageCreate, s *dg.Session, server *dg.Guild, channel *dg.Channel, lg int) error
createReply(content, message, channel, guild string) *dg.MessageSend createReply(content, message, channel, guild string) *dg.MessageSend
getRemainingTimeout(timeout time.Time) string getRemainingTimeout(timeout time.Time) string
roleCheck(guild string, roles []string, s *dg.Session, run Command) (bool, error) roleCheck(msg *dg.MessageCreate, s *dg.Session, run Command) (bool, error)
timeoutCheck(msgID, channelID, guildID string, s *dg.Session, run Command) (bool, error) 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) { func New(opts ...Option) (Bolt, error) {
_, check := os.LookupEnv(TOKEN_ENV_VAR) _, check := os.LookupEnv(TOKEN_ENV_VAR)
if !check { if !check {
@@ -79,7 +68,7 @@ func New(opts ...Option) (Bolt, error) {
//set default command indicator //set default command indicator
b.indicator = "." b.indicator = "."
//apply options //apply options to bolt
for _, opt := range opts { for _, opt := range opts {
opt(b) opt(b)
} }
@@ -87,10 +76,10 @@ func New(opts ...Option) (Bolt, error) {
return b, nil return b, nil
} }
// starts the bot, commands are added and the connection to Discord is opened, this is a BLOCKING call // starts the bot, commands are added and the connection to Discord is opened, this is a BLOCKING
// that handles safe shutdown of the bot // call that handles safe shutdown
func (b *bolt) Start() error { func (b *bolt) Start() error {
//register commands and open connection //registers the message handler used internally to detect commands
b.AddHandler(b.messageHandler) b.AddHandler(b.messageHandler)
err := b.Open() err := b.Open()
@@ -110,51 +99,50 @@ func (b *bolt) Start() error {
return nil return nil
} }
func (b *bolt) Database(file string) error {
if file == "" {
file = DEFAULT_DB_NAME
}
db, err := lite.New(lite.WithFile(file), lite.WithPath(DB_PATH))
if err != nil {
return err
}
b.db = db
return nil
}
// stops the bot // stops the bot
func (b *bolt) stop() error { func (b *bolt) stop() error {
if b.db != nil {
b.db.Close()
}
return b.Close() return b.Close()
} }
// adds commands to bot command map for use // adds commands to bot command map for use
func (b *bolt) AddCommands(cmd ...Command) { 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 { for _, c := range cmd {
b.commands[c.Trigger] = c b.commands[c.Trigger] = c
} }
} }
// handler function that parses message data, handles logging the message based on logLevel, and executes // internal message handler function. Ensure's the message is a message the bot should interpret
// the payload function in a goroutine // 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) { func (b *bolt) messageHandler(s *dg.Session, msg *dg.MessageCreate) {
//get server information //get channel name to, idk if i like this
server, err := s.Guild(msg.GuildID)
if err != nil {
log.Printf("failed to get guild: %e\n", err)
return
}
channel, err := s.Channel(msg.ChannelID) channel, err := s.Channel(msg.ChannelID)
if err != nil { if err != nil {
log.Printf("failed to get channel from guild: %e\n", err) log.Printf("failed to get channel from guild: %e\n", err)
return 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 //if there is no content it is likely an image, gif, or sticker, updating message content for
//better logging and to avoid confusion //better logging and to avoid confusion
if len(msg.Content) == 0 { if len(msg.Content) == 0 {
@@ -163,7 +151,7 @@ func (b *bolt) messageHandler(s *dg.Session, msg *dg.MessageCreate) {
//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 || b.logLvl == LogLevelAll { if b.logLvl == LogLevelCmd {
//log command responses //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)
} }
@@ -178,42 +166,29 @@ func (b *bolt) messageHandler(s *dg.Session, msg *dg.MessageCreate) {
//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 {
mCreate := &MessageCreateEvent{ err := b.handleCommand(msg, s, server, channel, lg)
AuthorUsername: msg.Author.Username,
AuthorID: msg.Author.ID,
AuthorRoles: msg.Member.Roles,
MsgID: msg.ID,
Msg: msg.Content,
MsgChanID: msg.ChannelID,
MsgGuildID: msg.GuildID,
MsgAttachments: msg.Attachments,
}
if b.logLvl == LogLevelCmd {
//log commands
log.Printf("< %s | %s | %s > %s\n", mCreate.MsgGuildName, mCreate.MsgChanName, mCreate.AuthorUsername, mCreate.Msg)
}
//handled in its own goroutine to allow for async commands
go func() {
err := b.handleCommand(mCreate, s, lg)
if err != nil { if err != nil {
log.Println(err) log.Println(err)
return
} }
}()
} }
} }
// parses command from message and handles timeout checks, role checks, and command execution. All command responses are sent back to Discord // parses command from message and handles timeout checks, role checks, and command execution. All command responses are sent back to Discord
func (b *bolt) handleCommand(msgEvent *MessageCreateEvent, s *dg.Session, lg int) error { func (b *bolt) handleCommand(msg *dg.MessageCreate, s *dg.Session, server *dg.Guild, channel *dg.Channel, lg int) error {
words := strings.Split(msgEvent.Msg, " ") if b.logLvl == LogLevelCmd {
//log commands
log.Printf("< %s | %s | %s > %s\n", server.Name, channel.Name, msg.Author.Username, msg.Content)
}
words := strings.Split(msg.Content, " ")
run, ok := b.commands[words[0][lg:]] run, ok := b.commands[words[0][lg:]]
if !ok { if !ok {
return nil //command doesn't exist, maybe log or respond to author return nil //command doesn't exist, maybe log or respond to author
} }
//has command met its timeout requirements //has command met its timeout requirements
tc, err := b.timeoutCheck(msgEvent.MsgID, msgEvent.MsgChanID, msgEvent.MsgGuildID, s, run) tc, err := b.timeoutCheck(msg, s, run)
if err != nil { if err != nil {
return fmt.Errorf("failed to calculate timeout for %s\n%e", run.Trigger, err) return fmt.Errorf("failed to calculate timeout for %s\n%e", run.Trigger, err)
} }
@@ -223,38 +198,32 @@ func (b *bolt) handleCommand(msgEvent *MessageCreateEvent, s *dg.Session, lg int
//does user have correct permissions //does user have correct permissions
if run.Roles != nil { if run.Roles != nil {
check, err := b.roleCheck(msgEvent.MsgGuildID, msgEvent.AuthorRoles, s, run) check, err := b.roleCheck(msg, s, run)
if err != nil { if err != nil {
return fmt.Errorf("failed to perform permission checks for %s\n%e", run.Trigger, err) return fmt.Errorf("failed to perform permission checks for %s\n%e", run.Trigger, err)
} }
if !check { if !check {
reply := b.createReply("you do not have permissions to run that command", msgEvent.MsgID, msgEvent.MsgChanID, msgEvent.MsgGuildID)
_, err := s.ChannelMessageSendComplex(msgEvent.MsgChanID, reply)
if err != nil {
return err
}
return nil return nil
} }
} }
//populate message struct exposed to client
plMsg := Message{ plMsg := Message{
Author: msgEvent.AuthorUsername, Author: msg.Author.Username,
ID: msgEvent.AuthorID, ID: msg.Author.ID,
msgID: msgEvent.MsgID, msgID: msg.ID,
Words: words, Words: words,
Content: msgEvent.Msg, Content: msg.Content,
Channel: msgEvent.MsgChanName, Channel: channel.Name,
channelID: msgEvent.MsgChanID, channelID: msg.ChannelID,
Server: msgEvent.MsgGuildName, Server: server.Name,
serverID: msgEvent.MsgGuildID, serverID: server.ID,
sesh: b, sesh: b,
} }
//check for file attachments //check for file attachments
if len(msgEvent.MsgAttachments) > 0 { if len(msg.Attachments) > 0 {
var att []MessageAttachment var att []MessageAttachment
for _, a := range msgEvent.MsgAttachments { for _, a := range msg.Attachments {
att = append(att, MessageAttachment{ att = append(att, MessageAttachment{
ID: a.ID, ID: a.ID,
URL: a.URL, URL: a.URL,
@@ -319,15 +288,15 @@ func (b *bolt) getRemainingTimeout(timeout time.Time) string {
} }
// checks if the author of msg has the correct role to run the requested command // checks if the author of msg has the correct role to run the requested command
func (b *bolt) roleCheck(guild string, roles []string, s *dg.Session, run Command) (bool, error) { func (b *bolt) roleCheck(msg *dg.MessageCreate, s *dg.Session, run Command) (bool, error) {
var found bool var found bool
//loop thru author roles, there may be a better way to check for this UNION //loop thru author roles, there may be a better way to check for this UNION
//TODO: improve role search performance to support bigger lists //TODO: improve role search performance to support bigger lists
for _, r := range roles { for _, r := range msg.Member.Roles {
//get role name from ID //get role name from ID
n, err := s.State.Role(guild, r) n, err := s.State.Role(msg.GuildID, r)
if err != nil { if err != nil {
return false, fmt.Errorf("failed to get role from ID %s\n%e", guild, 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)
@@ -337,8 +306,13 @@ func (b *bolt) roleCheck(guild string, roles []string, s *dg.Session, run Comman
} }
} }
//can't find role, don't run command //can't find role, don't run command, alert user of missing permissions
if !found { if !found {
reply := b.createReply("you do not have permissions to run that command", msg.ID, msg.ChannelID, msg.GuildID)
_, err := s.ChannelMessageSendComplex(msg.ChannelID, reply)
if err != nil {
return false, fmt.Errorf("failed to send permission response: %e", err)
}
return false, nil return false, nil
} }
@@ -346,11 +320,11 @@ func (b *bolt) roleCheck(guild string, roles []string, s *dg.Session, run Comman
} }
// check if the command timeout has been met, responding with remaining time if timeout has not been met yet. // check if the command timeout has been met, responding with remaining time if timeout has not been met yet.
func (b *bolt) timeoutCheck(msgID, channelID, guildID string, s *dg.Session, run Command) (bool, error) { func (b *bolt) timeoutCheck(msg *dg.MessageCreate, s *dg.Session, run Command) (bool, error) {
wait := run.lastRun.Add(run.Timeout) wait := run.lastRun.Add(run.Timeout)
if !time.Now().After(wait) { if !time.Now().After(wait) {
reply := b.createReply(fmt.Sprintf("that command cannot be run for another %s", b.getRemainingTimeout(wait)), msgID, channelID, 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(channelID, reply) _, err := s.ChannelMessageSendComplex(msg.ChannelID, reply)
if err != nil { if err != nil {
return false, fmt.Errorf("failed to send timeout response: %e", err) return false, fmt.Errorf("failed to send timeout response: %e", err)
} }

13
go.mod
View File

@@ -2,17 +2,10 @@ module code.jakeyoungdev.com/jake/bolt
go 1.24.0 go 1.24.0
require ( require github.com/bwmarrin/discordgo v0.29.0
code.jakeyoungdev.com/go/lite v0.3.0
github.com/bwmarrin/discordgo v0.29.0
)
require ( require (
github.com/gorilla/websocket v1.4.2 // indirect github.com/gorilla/websocket v1.4.2 // indirect
github.com/jmoiron/sqlx v1.4.0 // indirect golang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b // indirect
github.com/ncruces/go-sqlite3 v0.26.1 // indirect golang.org/x/sys v0.0.0-20201119102817-f84b799fce68 // indirect
github.com/ncruces/julianday v1.0.0 // indirect
github.com/tetratelabs/wazero v1.9.0 // indirect
golang.org/x/crypto v0.39.0 // indirect
golang.org/x/sys v0.33.0 // indirect
) )

26
go.sum
View File

@@ -1,34 +1,12 @@
code.jakeyoungdev.com/go/lite v0.3.0 h1:QJuinHNtb3h+f0Cht1UNzOmUTT7YyaFYCaYAmCHAxMg=
code.jakeyoungdev.com/go/lite v0.3.0/go.mod h1:NadLydJU2G+RsyokYjA/+EnwyNoMujswVwfZ1SQx1+E=
filippo.io/edwards25519 v1.1.0 h1:FNf4tywRC1HmFuKW5xopWpigGjJKiJSV0Cqo0cJWDaA=
filippo.io/edwards25519 v1.1.0/go.mod h1:BxyFTGdWcka3PhytdK4V28tE5sGfRvvvRV7EaN4VDT4=
github.com/bwmarrin/discordgo v0.29.0 h1:FmWeXFaKUwrcL3Cx65c20bTRW+vOb6k8AnaP+EgjDno= github.com/bwmarrin/discordgo v0.29.0 h1:FmWeXFaKUwrcL3Cx65c20bTRW+vOb6k8AnaP+EgjDno=
github.com/bwmarrin/discordgo v0.29.0/go.mod h1:NJZpH+1AfhIcyQsPeuBKsUtYrRnjkyu0kIVMCHkZtRY= github.com/bwmarrin/discordgo v0.29.0/go.mod h1:NJZpH+1AfhIcyQsPeuBKsUtYrRnjkyu0kIVMCHkZtRY=
github.com/go-sql-driver/mysql v1.8.1 h1:LedoTUt/eveggdHS9qUFC1EFSa8bU2+1pZjSRpvNJ1Y=
github.com/go-sql-driver/mysql v1.8.1/go.mod h1:wEBSXgmK//2ZFJyE+qWnIsVGmvmEKlqwuVSjsCm7DZg=
github.com/gorilla/websocket v1.4.2 h1:+/TMaTYc4QFitKJxsQ7Yye35DkWvkdLcvGKqM+x0Ufc= github.com/gorilla/websocket v1.4.2 h1:+/TMaTYc4QFitKJxsQ7Yye35DkWvkdLcvGKqM+x0Ufc=
github.com/gorilla/websocket v1.4.2/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE= github.com/gorilla/websocket v1.4.2/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
github.com/jmoiron/sqlx v1.4.0 h1:1PLqN7S1UYp5t4SrVVnt4nUVNemrDAtxlulVe+Qgm3o= golang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b h1:7mWr3k41Qtv8XlltBkDkl8LoP3mpSgBW8BUoxtEdbXg=
github.com/jmoiron/sqlx v1.4.0/go.mod h1:ZrZ7UsYB/weZdl2Bxg6jCRO9c3YHl8r3ahlKmRT4JLY=
github.com/lib/pq v1.10.9 h1:YXG7RB+JIjhP29X+OtkiDnYaXQwpS4JEWq7dtCCRUEw=
github.com/lib/pq v1.10.9/go.mod h1:AlVN5x4E4T544tWzH6hKfbfQvm3HdbOxrmggDNAPY9o=
github.com/mattn/go-sqlite3 v1.14.22 h1:2gZY6PC6kBnID23Tichd1K+Z0oS6nE/XwU+Vz/5o4kU=
github.com/mattn/go-sqlite3 v1.14.22/go.mod h1:Uh1q+B4BYcTPb+yiD3kU8Ct7aC0hY9fxUwlHK0RXw+Y=
github.com/ncruces/go-sqlite3 v0.26.1 h1:lBXmbmucH1Bsj57NUQR6T84UoMN7jnNImhF+ibEITJU=
github.com/ncruces/go-sqlite3 v0.26.1/go.mod h1:XFTPtFIo1DmGCh+XVP8KGn9b/o2f+z0WZuT09x2N6eo=
github.com/ncruces/julianday v1.0.0 h1:fH0OKwa7NWvniGQtxdJRxAgkBMolni2BjDHaWTxqt7M=
github.com/ncruces/julianday v1.0.0/go.mod h1:Dusn2KvZrrovOMJuOt0TNXL6tB7U2E8kvza5fFc9G7g=
github.com/tetratelabs/wazero v1.9.0 h1:IcZ56OuxrtaEz8UYNRHBrUa9bYeX9oVY93KspZZBf/I=
github.com/tetratelabs/wazero v1.9.0/go.mod h1:TSbcXCfFP0L2FGkRPxHphadXPjo1T6W+CseNNY7EkjM=
golang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b/go.mod h1:T9bdIzuCu7OtxOm1hfPfRQxPLYneinmdGuTeoZ9dtd4= golang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b/go.mod h1:T9bdIzuCu7OtxOm1hfPfRQxPLYneinmdGuTeoZ9dtd4=
golang.org/x/crypto v0.39.0 h1:SHs+kF4LP+f+p14esP5jAoDpHU8Gu/v9lFRK6IT5imM=
golang.org/x/crypto v0.39.0/go.mod h1:L+Xg3Wf6HoL4Bn4238Z6ft6KfEpN0tJGo53AAPC632U=
golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg= golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68 h1:nxC68pudNYkKU6jWhgrqdreuFiOQWj1Fs7T3VrH4Pjw=
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
golang.org/x/sys v0.33.0 h1:q3i8TbbEz+JRD9ywIRlyRAQbM0qF7hu24q3teo2hbuw=
golang.org/x/sys v0.33.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
golang.org/x/text v0.26.0 h1:P42AVeLghgTYr4+xUnTRKDMqpar+PtX7KWuNQL21L8M=
golang.org/x/text v0.26.0/go.mod h1:QK15LZJUUQVJxhz7wXgxSy/CJaTFjd0G+YLonydOVQA=
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=

View File

@@ -1,13 +1,60 @@
package bolt package bolt
import ( import "fmt"
"fmt"
dg "github.com/bwmarrin/discordgo" //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 = "👍"
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 = "🧎‍♂️‍➡️"
) )
// the message struct is passed to the command payload providing basic // information about attachments to messages
// message information and needed methods type MessageAttachment struct {
ID string
URL string
ProxyURL string
Filename string
ContentType string
Width int
Height int
Size int
DurationSecs float64
}
// represents a Discord message
type Message struct { type Message struct {
Author string //username of message author Author string //username of message author
ID string //discord ID of message author ID string //discord ID of message author
@@ -27,42 +74,9 @@ func (m *Message) React(emoji Reaction) error {
return m.sesh.MessageReactionAdd(m.channelID, m.msgID, fmt.Sprint(emoji)) return m.sesh.MessageReactionAdd(m.channelID, m.msgID, fmt.Sprint(emoji))
} }
// sends response to message // sends the value of res in response to the message
func (m *Message) Respond(res string) error { func (m *Message) Respond(res string) error {
rep := m.sesh.createReply(res, m.msgID, m.channelID, m.serverID) rep := m.sesh.createReply(res, m.msgID, m.channelID, m.serverID)
_, err := m.sesh.ChannelMessageSendComplex(m.channelID, rep) _, err := m.sesh.ChannelMessageSendComplex(m.channelID, rep)
return err return err
} }
// deletes the message from the channel
func (m *Message) Delete() error {
return m.sesh.ChannelMessageDelete(m.channelID, m.msgID, nil)
}
// this struct has all of the needed information from the messageCreate event so that
// commands can be run asynchronously. Passing the messageCreate to payloads can block routines
type MessageCreateEvent struct {
AuthorUsername string
AuthorID string
AuthorRoles []string
MsgID string
Msg string
MsgChanID string
MsgChanName string
MsgGuildID string
MsgGuildName string
MsgAttachments []*dg.MessageAttachment
}
// message attachment details
type MessageAttachment struct {
ID string
URL string
ProxyURL string
Filename string
ContentType string
Width int
Height int
Size int
DurationSecs float64
}

View File

@@ -23,3 +23,10 @@ func WithLogLevel(lvl LogLevel) Option {
b.logLvl = lvl b.logLvl = lvl
} }
} }
// restrict the channels the bot listens in and responds to, must be a list of channel names
func WithListenChannels(channels []string) Option {
return func(b *bolt) {
b.channels = channels
}
}

View File

@@ -1,38 +0,0 @@
package bolt
// 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 = "👍"
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 = "🐉"
)