Use single Context struct updating Invoked

master v0.0.10
DustyP 6 years ago
parent c2f03b7f61
commit 12b47f1fed

@ -91,14 +91,6 @@ func (c *CommandManager) OnMessage(session *discordgo.Session, m *discordgo.Mess
guild, _ := session.Guild(m.GuildID) guild, _ := session.Guild(m.GuildID)
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 {
fmt.Println(err.Error())
if strings.Contains(err.Error(), "Unterminated") {
cmd = strings.Split(strings.TrimPrefix(content, prefix), " ")
} else {
ctx := Context{ ctx := Context{
Session: session, Session: session,
Channel: channel, Channel: channel,
@ -109,6 +101,15 @@ func (c *CommandManager) OnMessage(session *discordgo.Session, m *discordgo.Mess
Invoked: "", Invoked: "",
ErrorChannel: c.ErrorChannel, 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 {
fmt.Println(err.Error())
if strings.Contains(err.Error(), "Unterminated") {
cmd = strings.Split(strings.TrimPrefix(content, prefix), " ")
} else {
c.ErrorChannel <- CommandError{ c.ErrorChannel <- CommandError{
Context: ctx, Context: ctx,
Message: "", Message: "",
@ -123,8 +124,8 @@ func (c *CommandManager) OnMessage(session *discordgo.Session, m *discordgo.Mess
} }
var command *Command var command *Command
invoked := cmd[0] ctx.Invoked = cmd[0]
if cmnd, ok := c.Commands[invoked]; ok { if cmnd, ok := c.Commands[ctx.Invoked]; ok {
command = cmnd command = cmnd
} else { } else {
fmt.Println("Command Not Found") fmt.Println("Command Not Found")
@ -139,16 +140,6 @@ func (c *CommandManager) OnMessage(session *discordgo.Session, m *discordgo.Mess
} }
if !CheckPermissions(session, m.Author.ID, *channel, command.RequiredPermissions) { 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{ c.ErrorChannel <- CommandError{
Context: ctx, Context: ctx,
Message: "You don't have the correct permissions to run this command.", Message: "You don't have the correct permissions to run this command.",
@ -158,16 +149,6 @@ func (c *CommandManager) OnMessage(session *discordgo.Session, m *discordgo.Mess
} }
if !CheckPermissions(session, session.State.User.ID, *channel, command.RequiredPermissions) { 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{ c.ErrorChannel <- CommandError{
Context: ctx, Context: ctx,
Message: "I don't have the correct permissions to run this command.", Message: "I don't have the correct permissions to run this command.",
@ -178,16 +159,6 @@ func (c *CommandManager) OnMessage(session *discordgo.Session, m *discordgo.Mess
} }
if command.OwnerOnly && !c.IsOwner(m.Author.ID) { 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{ c.ErrorChannel <- CommandError{
Context: ctx, Context: ctx,
Message: "Sorry, only the bot owner(s) can run that command!", Message: "Sorry, only the bot owner(s) can run that command!",
@ -197,16 +168,5 @@ func (c *CommandManager) OnMessage(session *discordgo.Session, m *discordgo.Mess
} }
context := Context{ go command.Invoke(ctx, cmd[1:])
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:])
} }

Loading…
Cancel
Save