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