mirror of
https://github.com/lyx0/nourybot.git
synced 2024-11-13 19:49:55 +01:00
241 lines
8.1 KiB
Go
241 lines
8.1 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
|
|
"github.com/gempir/go-twitch-irc/v3"
|
|
"github.com/lyx0/nourybot/internal/common"
|
|
"github.com/lyx0/nourybot/internal/data"
|
|
)
|
|
|
|
// AddCommand splits a message into two parts and passes on the
|
|
// name and text to the database handler.
|
|
func (app *Application) AddCommand(name string, message twitch.PrivateMessage) {
|
|
// snipLength is the length we need to "snip" off of the start of `message`.
|
|
// `()addcommand` = +12
|
|
// trailing space = +1
|
|
// zero-based = +1
|
|
// = 14
|
|
snipLength := 14
|
|
|
|
// Split the twitch message at `snipLength` plus length of the name of the
|
|
// command that we want to add.
|
|
// The part of the message we are left over with is then passed on to the database
|
|
// handlers as the `text` part of the command.
|
|
//
|
|
// e.g. `()addcommand CoolSponsors Check out CoolSponsor.com they are the coolest sponsors!
|
|
// | <- snipLength + name -> | <--- command text with however many characters ---> |
|
|
// | <----- 14 + 12 ------> |
|
|
text := message.Message[snipLength+len(name) : len(message.Message)]
|
|
command := &data.Command{
|
|
Name: name,
|
|
Text: text,
|
|
Category: "uncategorized",
|
|
Level: 0,
|
|
Help: "",
|
|
}
|
|
err := app.Models.Commands.Insert(command)
|
|
|
|
if err != nil {
|
|
reply := fmt.Sprintf("Something went wrong FeelsBadMan %s", err)
|
|
common.Send(message.Channel, reply, app.TwitchClient)
|
|
return
|
|
} else {
|
|
reply := fmt.Sprintf("Successfully added command: %s", name)
|
|
common.Send(message.Channel, reply, app.TwitchClient)
|
|
return
|
|
}
|
|
}
|
|
|
|
// GetCommand queries the database for a name. If an entry exists it checks
|
|
// if the Command.Level is 0, if it is the command.Text value is returned.
|
|
//
|
|
// If the Command.Level is not 0 it queries the database for the level of the
|
|
// user who sent the message. If the users level is equal or higher
|
|
// the command.Text field is returned.
|
|
func (app *Application) GetCommand(name, username string) (string, error) {
|
|
// Fetch the command from the database if it exists.
|
|
command, err := app.Models.Commands.Get(name)
|
|
if err != nil {
|
|
// It probably did not exist
|
|
return "", err
|
|
}
|
|
|
|
// If the command has no level set just return the text.
|
|
// Otherwise check if the level is high enough.
|
|
if command.Level == 0 {
|
|
return command.Text, nil
|
|
} else {
|
|
// Get the user from the database to check if the userlevel is equal
|
|
// or higher than the command.Level.
|
|
user, err := app.Models.Users.Get(username)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if user.Level >= command.Level {
|
|
// Userlevel is sufficient so return the command.Text
|
|
return command.Text, nil
|
|
}
|
|
}
|
|
|
|
// Userlevel was not enough so return an empty string and error.
|
|
return "", ErrUserInsufficientLevel
|
|
}
|
|
|
|
// GetCommand queries the database for a name. If an entry exists it checks
|
|
// if the Command.Level is 0, if it is the command.Text value is returned.
|
|
//
|
|
// If the Command.Level is not 0 it queries the database for the level of the
|
|
// user who sent the message. If the users level is equal or higher
|
|
// the command.Text field is returned.
|
|
func (app *Application) GetCommandHelp(name, username string) (string, error) {
|
|
// Fetch the command from the database if it exists.
|
|
command, err := app.Models.Commands.Get(name)
|
|
if err != nil {
|
|
// It probably did not exist
|
|
return "", err
|
|
}
|
|
|
|
// If the command has no level set just return the text.
|
|
// Otherwise check if the level is high enough.
|
|
if command.Level == 0 {
|
|
return command.Help, nil
|
|
} else {
|
|
// Get the user from the database to check if the userlevel is equal
|
|
// or higher than the command.Level.
|
|
user, err := app.Models.Users.Get(username)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if user.Level >= command.Level {
|
|
// Userlevel is sufficient so return the command.Text
|
|
return command.Help, nil
|
|
}
|
|
}
|
|
|
|
// Userlevel was not enough so return an empty string and error.
|
|
return "", ErrUserInsufficientLevel
|
|
}
|
|
|
|
// EditCommandLevel takes in a name and level string and updates the entry with name
|
|
// to the supplied level value.
|
|
func (app *Application) EditCommandLevel(name, lvl string, message twitch.PrivateMessage) {
|
|
level, err := strconv.Atoi(lvl)
|
|
if err != nil {
|
|
app.Logger.Error(err)
|
|
common.Send(message.Channel, fmt.Sprintf("Something went wrong FeelsBadMan %s", ErrCommandLevelNotInteger), app.TwitchClient)
|
|
return
|
|
}
|
|
|
|
err = app.Models.Commands.SetLevel(name, level)
|
|
|
|
if err != nil {
|
|
common.Send(message.Channel, fmt.Sprintf("Something went wrong FeelsBadMan %s", ErrRecordNotFound), app.TwitchClient)
|
|
app.Logger.Error(err)
|
|
return
|
|
} else {
|
|
reply := fmt.Sprintf("Updated command %s to level %v", name, level)
|
|
common.Send(message.Channel, reply, app.TwitchClient)
|
|
return
|
|
}
|
|
}
|
|
|
|
// EditCommandCategory takes in a name and category string and updates the command
|
|
// in the databse with the passed in new category.
|
|
func (app *Application) EditCommandCategory(name, category string, message twitch.PrivateMessage) {
|
|
err := app.Models.Commands.SetCategory(name, category)
|
|
|
|
if err != nil {
|
|
common.Send(message.Channel, fmt.Sprintf("Something went wrong FeelsBadMan %s", ErrRecordNotFound), app.TwitchClient)
|
|
app.Logger.Error(err)
|
|
return
|
|
} else {
|
|
reply := fmt.Sprintf("Updated command %s to category %v", name, category)
|
|
common.Send(message.Channel, reply, app.TwitchClient)
|
|
return
|
|
}
|
|
}
|
|
|
|
// DebugCommand checks if a command with the provided name exists in the database
|
|
// and outputs information about it in the chat.
|
|
func (app *Application) DebugCommand(name string, message twitch.PrivateMessage) {
|
|
// Query the database for a command with the provided name
|
|
cmd, err := app.Models.Commands.Get(name)
|
|
if err != nil {
|
|
reply := fmt.Sprintf("Something went wrong FeelsBadMan %s", err)
|
|
common.Send(message.Channel, reply, app.TwitchClient)
|
|
return
|
|
} else if cmd.Category == "ascii" {
|
|
// If the command is in the ascii category don't post the Text field
|
|
// otherwise it becomes too spammy and won't fit in the max message length.
|
|
reply := fmt.Sprintf("ID %v: Name %v, Level: %v, Category: %v Help: %v",
|
|
cmd.ID,
|
|
cmd.Name,
|
|
cmd.Level,
|
|
cmd.Category,
|
|
cmd.Help,
|
|
)
|
|
|
|
common.Send(message.Channel, reply, app.TwitchClient)
|
|
return
|
|
} else {
|
|
reply := fmt.Sprintf("ID %v: Name %v, Level: %v, Category: %v, Text: %v, Help: %v",
|
|
cmd.ID,
|
|
cmd.Name,
|
|
cmd.Level,
|
|
cmd.Category,
|
|
cmd.Text,
|
|
cmd.Help,
|
|
)
|
|
|
|
common.Send(message.Channel, reply, app.TwitchClient)
|
|
return
|
|
}
|
|
}
|
|
|
|
// SetCommandHelp updates the `help` column of a given commands name in the
|
|
// database to the provided new help text.
|
|
func (app *Application) EditCommandHelp(name string, message twitch.PrivateMessage) {
|
|
// snipLength is the length we need to "snip" off of the start of `message`.
|
|
// `()editcommand` = +13
|
|
// trailing space = +1
|
|
// zero-based = +1
|
|
// `help` = +4
|
|
// = 19
|
|
snipLength := 19
|
|
|
|
// Split the twitch message at `snipLength` plus length of the name of the
|
|
// command that we want to set the help text for so that we get the
|
|
// actual help message left over and then assign this string to `text`.
|
|
//
|
|
// e.g. `()editcommand help FeelsDankMan Returns a FeelsDankMan ascii art. Requires user level 500.`
|
|
// | <---- snipLength + name ----> | <------ help text with however many characters. ----> |
|
|
// | <--------- 19 + 12 --------> |
|
|
text := message.Message[snipLength+len(name) : len(message.Message)]
|
|
err := app.Models.Commands.SetHelp(name, text)
|
|
|
|
if err != nil {
|
|
common.Send(message.Channel, fmt.Sprintf("Something went wrong FeelsBadMan %s", ErrRecordNotFound), app.TwitchClient)
|
|
app.Logger.Error(err)
|
|
return
|
|
} else {
|
|
reply := fmt.Sprintf("Updated help text for command %s to: %v", name, text)
|
|
common.Send(message.Channel, reply, app.TwitchClient)
|
|
return
|
|
}
|
|
}
|
|
|
|
// DeleteCommand takes in a name value and deletes the command from the database if it exists.
|
|
func (app *Application) DeleteCommand(name string, message twitch.PrivateMessage) {
|
|
err := app.Models.Commands.Delete(name)
|
|
if err != nil {
|
|
common.Send(message.Channel, "Something went wrong FeelsBadMan", app.TwitchClient)
|
|
app.Logger.Error(err)
|
|
return
|
|
}
|
|
|
|
reply := fmt.Sprintf("Deleted command %s", name)
|
|
common.Send(message.Channel, reply, app.TwitchClient)
|
|
}
|