mirror of https://github.com/velour/catbase.git
Up your nerd game, nerd.
This commit is contained in:
parent
bbe0cdd28e
commit
6e39e6ec2b
2
main.go
2
main.go
|
@ -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))
|
||||||
|
|
|
@ -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 }
|
|
@ -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)
|
||||||
|
}
|
Loading…
Reference in New Issue