Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
12b47f1fed | ||
|
|
c2f03b7f61 | ||
|
|
2cac64f91b | ||
|
|
71efaa0d45 | ||
|
|
fc4b370a1e | ||
|
|
30065b313f |
+23
-57
@@ -9,9 +9,10 @@ package disgoman
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"strings"
|
||||
|
||||
"github.com/bwmarrin/discordgo"
|
||||
"github.com/kballard/go-shellquote"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// AddCommand adds the Command at the address passed in to the Commands array on the CommandManager.
|
||||
@@ -90,23 +91,25 @@ func (c *CommandManager) OnMessage(session *discordgo.Session, m *discordgo.Mess
|
||||
|
||||
guild, _ := session.Guild(m.GuildID)
|
||||
|
||||
ctx := Context{
|
||||
Session: session,
|
||||
Channel: channel,
|
||||
Message: m.Message,
|
||||
User: m.Author,
|
||||
Guild: guild,
|
||||
Member: m.Member,
|
||||
Invoked: "",
|
||||
ErrorChannel: c.ErrorChannel,
|
||||
}
|
||||
|
||||
var cmd []string
|
||||
// If we found our prefix then remove it and split the command into pieces
|
||||
cmd, err = shellquote.Split(strings.TrimPrefix(content, prefix))
|
||||
if err != nil {
|
||||
if err.Error() == "Unterminated double-quoted string" || err.Error() == "Unterminated single-quoted string" {
|
||||
fmt.Println(err.Error())
|
||||
if strings.Contains(err.Error(), "Unterminated") {
|
||||
cmd = strings.Split(strings.TrimPrefix(content, prefix), " ")
|
||||
} else {
|
||||
ctx := Context{
|
||||
Session: session,
|
||||
Channel: channel,
|
||||
Message: m.Message,
|
||||
User: m.Author,
|
||||
Guild: guild,
|
||||
Member: m.Member,
|
||||
Invoked: "",
|
||||
ErrorChannel: c.ErrorChannel,
|
||||
}
|
||||
c.ErrorChannel <- CommandError{
|
||||
Context: ctx,
|
||||
Message: "",
|
||||
@@ -116,9 +119,13 @@ func (c *CommandManager) OnMessage(session *discordgo.Session, m *discordgo.Mess
|
||||
}
|
||||
}
|
||||
|
||||
if len(cmd) < 1 {
|
||||
return
|
||||
}
|
||||
|
||||
var command *Command
|
||||
invoked := cmd[0]
|
||||
if cmnd, ok := c.Commands[invoked]; ok {
|
||||
ctx.Invoked = cmd[0]
|
||||
if cmnd, ok := c.Commands[ctx.Invoked]; ok {
|
||||
command = cmnd
|
||||
} else {
|
||||
fmt.Println("Command Not Found")
|
||||
@@ -126,23 +133,13 @@ func (c *CommandManager) OnMessage(session *discordgo.Session, m *discordgo.Mess
|
||||
}
|
||||
|
||||
if command.SanitizeEveryone {
|
||||
for i := 1; i < len(cmd[1:]); i++ {
|
||||
for i := 1; i < len(cmd); i++ {
|
||||
cmd[i] = strings.ReplaceAll(cmd[i], "@everyone", "@\ufff0everyone")
|
||||
cmd[i] = strings.ReplaceAll(cmd[i], "@here", "@\ufff0here")
|
||||
}
|
||||
}
|
||||
|
||||
if !CheckPermissions(session, m.Author.ID, *channel, command.RequiredPermissions) {
|
||||
ctx := Context{
|
||||
Session: session,
|
||||
Channel: channel,
|
||||
Message: m.Message,
|
||||
User: m.Author,
|
||||
Guild: guild,
|
||||
Member: m.Member,
|
||||
Invoked: cmd[0],
|
||||
ErrorChannel: c.ErrorChannel,
|
||||
}
|
||||
c.ErrorChannel <- CommandError{
|
||||
Context: ctx,
|
||||
Message: "You don't have the correct permissions to run this command.",
|
||||
@@ -152,16 +149,6 @@ func (c *CommandManager) OnMessage(session *discordgo.Session, m *discordgo.Mess
|
||||
}
|
||||
|
||||
if !CheckPermissions(session, session.State.User.ID, *channel, command.RequiredPermissions) {
|
||||
ctx := Context{
|
||||
Session: session,
|
||||
Channel: channel,
|
||||
Message: m.Message,
|
||||
User: m.Author,
|
||||
Guild: guild,
|
||||
Member: m.Member,
|
||||
Invoked: cmd[0],
|
||||
ErrorChannel: c.ErrorChannel,
|
||||
}
|
||||
c.ErrorChannel <- CommandError{
|
||||
Context: ctx,
|
||||
Message: "I don't have the correct permissions to run this command.",
|
||||
@@ -172,16 +159,6 @@ func (c *CommandManager) OnMessage(session *discordgo.Session, m *discordgo.Mess
|
||||
}
|
||||
|
||||
if command.OwnerOnly && !c.IsOwner(m.Author.ID) {
|
||||
ctx := Context{
|
||||
Session: session,
|
||||
Channel: channel,
|
||||
Message: m.Message,
|
||||
User: m.Author,
|
||||
Guild: guild,
|
||||
Member: m.Member,
|
||||
Invoked: cmd[0],
|
||||
ErrorChannel: c.ErrorChannel,
|
||||
}
|
||||
c.ErrorChannel <- CommandError{
|
||||
Context: ctx,
|
||||
Message: "Sorry, only the bot owner(s) can run that command!",
|
||||
@@ -191,16 +168,5 @@ func (c *CommandManager) OnMessage(session *discordgo.Session, m *discordgo.Mess
|
||||
|
||||
}
|
||||
|
||||
context := Context{
|
||||
Session: session,
|
||||
Channel: channel,
|
||||
Message: m.Message,
|
||||
User: m.Author,
|
||||
Guild: guild,
|
||||
Member: m.Member,
|
||||
Invoked: invoked,
|
||||
ErrorChannel: c.ErrorChannel,
|
||||
}
|
||||
|
||||
go command.Invoke(context, cmd[1:])
|
||||
go command.Invoke(ctx, cmd[1:])
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user