Up your nerd game, nerd.

This commit is contained in:
skkiesel 2019-01-18 19:48:12 -05:00
parent bbe0cdd28e
commit 6e39e6ec2b
3 changed files with 121 additions and 0 deletions

View File

@ -23,6 +23,7 @@ import (
"github.com/velour/catbase/plugins/first" "github.com/velour/catbase/plugins/first"
"github.com/velour/catbase/plugins/inventory" "github.com/velour/catbase/plugins/inventory"
"github.com/velour/catbase/plugins/leftpad" "github.com/velour/catbase/plugins/leftpad"
"github.com/velour/catbase/plugins/nerdepedia"
"github.com/velour/catbase/plugins/picker" "github.com/velour/catbase/plugins/picker"
"github.com/velour/catbase/plugins/reaction" "github.com/velour/catbase/plugins/reaction"
"github.com/velour/catbase/plugins/reminder" "github.com/velour/catbase/plugins/reminder"
@ -80,6 +81,7 @@ func main() {
b.AddHandler("sisyphus", sisyphus.New(b)) b.AddHandler("sisyphus", sisyphus.New(b))
b.AddHandler("tell", tell.New(b)) b.AddHandler("tell", tell.New(b))
b.AddHandler("couldashouldawoulda", couldashouldawoulda.New(b)) b.AddHandler("couldashouldawoulda", couldashouldawoulda.New(b))
b.AddHandler("nedepedia", nerdepedia.New(b))
// catches anything left, will always return true // catches anything left, will always return true
b.AddHandler("factoid", fact.New(b)) b.AddHandler("factoid", fact.New(b))
b.AddHandler("db", db.New(b)) b.AddHandler("db", db.New(b))

View File

@ -0,0 +1,84 @@
// © 2019 the CatBase Authors under the WTFPL. See AUTHORS for the list of authors.
package nerdepedia
import (
"strings"
"net/http"
"bufio"
"github.com/velour/catbase/bot"
"github.com/velour/catbase/bot/msg"
"github.com/velour/catbase/config"
)
const (
prefix = "<meta name=\"description\" content=\""
)
type NerdepediaPlugin struct {
bot bot.Bot
config *config.Config
}
// NewNerdepediaPlugin creates a new NerdepediaPlugin with the Plugin interface
func New(bot bot.Bot) *NerdepediaPlugin {
return &NerdepediaPlugin{
bot: bot,
config: bot.Config(),
}
}
// 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 *NerdepediaPlugin) Message(message msg.Message) bool {
lowerCase := strings.ToLower(message.Body)
query := ""
if lowerCase == "may the force be with you" || lowerCase == "help me obi-wan" {
query = "http://starwars.wikia.com/wiki/Special:Random"
} else if lowerCase == "beam me up scotty" || lowerCase == "live long and prosper" || lowerCase == "make it so" {
query = "http://memory-alpha.wikia.com/wiki/Special:Random"
}
if query != "" {
resp, err := http.Get(query)
if err != nil {
panic(err)
}
defer resp.Body.Close()
scanner := bufio.NewScanner(resp.Body)
for scanner.Scan() {
line := scanner.Text()
index := strings.Index(line, prefix)
if index >= 0 {
p.bot.SendMessage(message.Channel, strings.TrimSuffix(strings.TrimPrefix(line, prefix), "\" />"))
return true
}
}
}
return false
}
// Help responds to help requests. Every plugin must implement a help function.
func (p *NerdepediaPlugin) Help(channel string, parts []string) {
p.bot.SendMessage(channel, "star wars/trek stuff")
}
// Empty event handler because this plugin does not do anything on event recv
func (p *NerdepediaPlugin) Event(kind string, message msg.Message) bool {
return false
}
// Handler for bot's own messages
func (p *NerdepediaPlugin) BotMessage(message msg.Message) bool {
return false
}
// Register any web URLs desired
func (p *NerdepediaPlugin) RegisterWeb() *string {
return nil
}
func (p *NerdepediaPlugin) ReplyMessage(message msg.Message, identifier string) bool { return false }

View File

@ -0,0 +1,35 @@
// © 2013 the CatBase Authors under the WTFPL. See AUTHORS for the list of authors.
package nerdepedia
import (
"strings"
"testing"
"github.com/stretchr/testify/assert"
"github.com/velour/catbase/bot"
"github.com/velour/catbase/bot/msg"
"github.com/velour/catbase/bot/user"
)
func makeMessage(payload string) msg.Message {
isCmd := strings.HasPrefix(payload, "!")
if isCmd {
payload = payload[1:]
}
return msg.Message{
User: &user.User{Name: "tester"},
Channel: "test",
Body: payload,
Command: isCmd,
}
}
func TestObiWan(t *testing.T) {
mb := bot.NewMockBot()
c := New(mb)
assert.NotNil(t, c)
res := c.Message(makeMessage("help me obi-wan"))
assert.Len(t, mb.Messages, 1)
assert.True(t, res)
}