mirror of https://github.com/velour/catbase.git
77 lines
2.1 KiB
Go
77 lines
2.1 KiB
Go
// © 2013 the CatBase Authors under the WTFPL. See AUTHORS for the list of authors.
|
|
|
|
package your
|
|
|
|
import (
|
|
"log"
|
|
"math/rand"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/velour/catbase/bot"
|
|
)
|
|
|
|
type YourPlugin struct {
|
|
bot bot.Bot
|
|
}
|
|
|
|
// NewYourPlugin creates a new YourPlugin with the Plugin interface
|
|
func New(bot bot.Bot) *YourPlugin {
|
|
rand.Seed(time.Now().Unix())
|
|
return &YourPlugin{
|
|
bot: bot,
|
|
}
|
|
}
|
|
|
|
// Message responds to the bot hook on recieving messages.
|
|
// This function returns true if the plugin responds in a meaningful way to the users message.
|
|
// Otherwise, the function returns false and the bot continues execution of other plugins.
|
|
func (p *YourPlugin) Message(message bot.Message) bool {
|
|
lower := strings.ToLower(message.Body)
|
|
config := p.bot.Config().Your
|
|
if len(message.Body) > config.MaxLength {
|
|
return false
|
|
}
|
|
|
|
if strings.Contains(message.Body, "the fucking") { // let's not mess with case
|
|
log.Println("Found a fucking")
|
|
if rand.Float64() < config.FuckingChance {
|
|
log.Println("Replacing a fucking")
|
|
r := strings.NewReplacer("the fucking", "fucking the")
|
|
msg := r.Replace(message.Body)
|
|
p.bot.SendMessage(message.Channel, msg)
|
|
return true
|
|
}
|
|
}
|
|
if strings.Contains(lower, "your") || strings.Contains(lower, "you're") {
|
|
if rand.Float64() < config.YourChance {
|
|
r := strings.NewReplacer("Your", "You're", "your", "you're", "You're",
|
|
"Your", "you're", "your", "Youre", "Your", "youre", "your")
|
|
msg := r.Replace(message.Body)
|
|
p.bot.SendMessage(message.Channel, msg)
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
// Help responds to help requests. Every plugin must implement a help function.
|
|
func (p *YourPlugin) Help(channel string, parts []string) {
|
|
p.bot.SendMessage(channel, "Your corrects people's grammar.")
|
|
}
|
|
|
|
// Empty event handler because this plugin does not do anything on event recv
|
|
func (p *YourPlugin) Event(kind string, message bot.Message) bool {
|
|
return false
|
|
}
|
|
|
|
// Handler for bot's own messages
|
|
func (p *YourPlugin) BotMessage(message bot.Message) bool {
|
|
return false
|
|
}
|
|
|
|
// Register any web URLs desired
|
|
func (p *YourPlugin) RegisterWeb() *string {
|
|
return nil
|
|
}
|