mirror of https://github.com/velour/catbase.git
Compare commits
2 Commits
a4fd340a53
...
fcfec95c2f
Author | SHA1 | Date |
---|---|---|
Scott Kiesel | fcfec95c2f | |
Scott Kiesel | 3cb54e6262 |
1
go.mod
1
go.mod
|
@ -7,7 +7,6 @@ require (
|
|||
github.com/AlekSi/pointer v1.1.0 // indirect
|
||||
github.com/ChimeraCoder/anaconda v2.0.0+incompatible
|
||||
github.com/ChimeraCoder/tokenbucket v0.0.0-20131201223612-c5a927568de7 // indirect
|
||||
github.com/PaulRosset/go-hacknews v0.0.0-20170815075127-4aad99273a3c
|
||||
github.com/PuerkitoBio/goquery v1.5.0
|
||||
github.com/andybalholm/cascadia v1.1.0 // indirect
|
||||
github.com/antchfx/htmlquery v1.2.0 // indirect
|
||||
|
|
7
go.sum
7
go.sum
|
@ -1,5 +1,3 @@
|
|||
code.chrissexton.org/cws/getaoc v0.0.0-20191201041923-dc696067b57c h1:X5c+asGWxR/Whkgl/YYXDanC4n2RHYeFcC7mT37+uZ4=
|
||||
code.chrissexton.org/cws/getaoc v0.0.0-20191201041923-dc696067b57c/go.mod h1:rEpfJR9MplF2TUj2Oy+u4XAaLve2kwB8I2zlzeIQxl8=
|
||||
code.chrissexton.org/cws/getaoc v0.0.0-20191201043947-d5417d4b618d h1:XS13tP+cMAvXYHQiYqcst64wQ854pueMRZSU4+6puU4=
|
||||
code.chrissexton.org/cws/getaoc v0.0.0-20191201043947-d5417d4b618d/go.mod h1:rEpfJR9MplF2TUj2Oy+u4XAaLve2kwB8I2zlzeIQxl8=
|
||||
dmitri.shuralyov.com/gpu/mtl v0.0.0-20190408044501-666a987793e9/go.mod h1:H6x//7gZCb22OMCxBHrMx7a5I7Hp++hsVxbQ4BYO7hU=
|
||||
|
@ -12,8 +10,6 @@ github.com/ChimeraCoder/anaconda v2.0.0+incompatible h1:F0eD7CHXieZ+VLboCD5UAqCe
|
|||
github.com/ChimeraCoder/anaconda v2.0.0+incompatible/go.mod h1:TCt3MijIq3Qqo9SBtuW/rrM4x7rDfWqYWHj8T7hLcLg=
|
||||
github.com/ChimeraCoder/tokenbucket v0.0.0-20131201223612-c5a927568de7 h1:r+EmXjfPosKO4wfiMLe1XQictsIlhErTufbWUsjOTZs=
|
||||
github.com/ChimeraCoder/tokenbucket v0.0.0-20131201223612-c5a927568de7/go.mod h1:b2EuEMLSG9q3bZ95ql1+8oVqzzrTNSiOQqSXWFBzxeI=
|
||||
github.com/PaulRosset/go-hacknews v0.0.0-20170815075127-4aad99273a3c h1:bJ0HbTMaInVjakxM76G+2gsmbKTdHzpTUGyLGYxdMO0=
|
||||
github.com/PaulRosset/go-hacknews v0.0.0-20170815075127-4aad99273a3c/go.mod h1:8+24kIp7vJsYy0GmQDDNnPwAYEWkl3OcaPxJSDAfe1U=
|
||||
github.com/PuerkitoBio/goquery v1.5.0 h1:uGvmFXOA73IKluu/F84Xd1tt/z07GYm8X49XKHP7EJk=
|
||||
github.com/PuerkitoBio/goquery v1.5.0/go.mod h1:qD2PgZ9lccMbQlc7eEOjaeRlFQON7xY8kdmcsrnKqMg=
|
||||
github.com/ajstarks/svgo v0.0.0-20180226025133-644b8db467af/go.mod h1:K08gAheRH3/J6wwsYMMT4xOr94bZjxIelGM0+d/wbFw=
|
||||
|
@ -51,6 +47,7 @@ github.com/fogleman/gg v1.2.1-0.20190220221249-0403632d5b90/go.mod h1:R/bRT+9gY/
|
|||
github.com/garyburd/go-oauth v0.0.0-20180319155456-bca2e7f09a17 h1:GOfMz6cRgTJ9jWV0qAezv642OhPnKEG7gtUjJSdStHE=
|
||||
github.com/garyburd/go-oauth v0.0.0-20180319155456-bca2e7f09a17/go.mod h1:HfkOCN6fkKKaPSAeNq/er3xObxTW4VLeY6UUK895gLQ=
|
||||
github.com/go-gl/glfw v0.0.0-20190409004039-e6da0acd62b1/go.mod h1:vR7hzQXu2zJy9AVAgeJqvqgH9Q5CA+iKCZ2gyEVpxRU=
|
||||
github.com/go-sql-driver/mysql v1.4.0 h1:7LxgVwFb2hIQtMm87NdgAVfXjnt4OePseqT1tKx+opk=
|
||||
github.com/go-sql-driver/mysql v1.4.0/go.mod h1:zAC/RDZ24gD3HViQzih4MyKcchzm+sOG5ZlKdlhCg5w=
|
||||
github.com/gobwas/glob v0.2.3 h1:A4xDbljILXROh+kObIiy5kIaPYD8e96x1tgBhUI5J+Y=
|
||||
github.com/gobwas/glob v0.2.3/go.mod h1:d3Ez4x06l9bZtSvzIay5+Yzi0fmZzPgnTbPcKjJAkT8=
|
||||
|
@ -78,6 +75,7 @@ github.com/jmoiron/sqlx v1.2.0/go.mod h1:1FEQNm3xlJgrMD+FBdI9+xvCksHtbpVBBw5dYhB
|
|||
github.com/jung-kurt/gofpdf v1.0.3-0.20190309125859-24315acbbda5/go.mod h1:7Id9E/uU8ce6rXgefFLlgrJj/GYY22cpxn+r32jIOes=
|
||||
github.com/kennygrant/sanitize v1.2.4 h1:gN25/otpP5vAsO2djbMhF/LQX6R7+O1TB4yv8NzpJ3o=
|
||||
github.com/kennygrant/sanitize v1.2.4/go.mod h1:LGsjYYtgxbetdg5owWB2mpgUL6e2nfw2eObZ0u0qvak=
|
||||
github.com/lib/pq v1.0.0 h1:X5PMW56eZitiTeO7tKzZxFCSpbFZJtkMMooicw2us9A=
|
||||
github.com/lib/pq v1.0.0/go.mod h1:5WUZQaWbwv1U+lTReE5YruASi9Al49XbQIvNi/34Woo=
|
||||
github.com/mattn/go-sqlite3 v1.9.0/go.mod h1:FPy6KqzDD04eiIsT53CuJW3U88zkxoIYsOqkbpncsNc=
|
||||
github.com/mattn/go-sqlite3 v1.11.0 h1:LDdKkqtYlom37fkvqs8rMPFKAMe8+SgjbwZ6ex1/A/Q=
|
||||
|
@ -161,6 +159,7 @@ gonum.org/v1/gonum v0.6.0/go.mod h1:9mxDZsDKxgMAuccQkewq682L+0eCu4dCN2yonUJTCLU=
|
|||
gonum.org/v1/netlib v0.0.0-20190313105609-8cb42192e0e0 h1:OE9mWmgKkjJyEmDAAtGMPjXu+YNeGvK9VTSHY6+Qihc=
|
||||
gonum.org/v1/netlib v0.0.0-20190313105609-8cb42192e0e0/go.mod h1:wa6Ws7BG/ESfp6dHfk7C6KdzKA7wR7u/rKwOGE66zvw=
|
||||
gonum.org/v1/plot v0.0.0-20190515093506-e2840ee46a6b/go.mod h1:Wt8AAjI+ypCyYX3nZBvf6cAIx93T+c/OS2HFAYskSZc=
|
||||
google.golang.org/appengine v1.6.5 h1:tycE03LOZYQNhDpS27tcQdAzLCVMaj7QT2SXxebnpCM=
|
||||
google.golang.org/appengine v1.6.5/go.mod h1:8WjMMxjGQR8xUklV/ARdw2HLXBOI7O7uCIDZVag1xfc=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
|
|
2
main.go
2
main.go
|
@ -32,6 +32,7 @@ import (
|
|||
"github.com/velour/catbase/plugins/first"
|
||||
"github.com/velour/catbase/plugins/fuck"
|
||||
"github.com/velour/catbase/plugins/git"
|
||||
"github.com/velour/catbase/plugins/impossible"
|
||||
"github.com/velour/catbase/plugins/inventory"
|
||||
"github.com/velour/catbase/plugins/leftpad"
|
||||
"github.com/velour/catbase/plugins/nerdepedia"
|
||||
|
@ -134,6 +135,7 @@ func main() {
|
|||
b.AddPlugin(newsbid.New(b))
|
||||
b.AddPlugin(twitter.New(b))
|
||||
b.AddPlugin(git.New(b))
|
||||
b.AddPlugin(impossible.New(b))
|
||||
b.AddPlugin(cli.New(b))
|
||||
b.AddPlugin(aoc.New(b))
|
||||
// catches anything left, will always return true
|
||||
|
|
|
@ -0,0 +1,149 @@
|
|||
package impossible
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"math/rand"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"regexp"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/rs/zerolog/log"
|
||||
|
||||
"github.com/velour/catbase/bot"
|
||||
"github.com/velour/catbase/bot/msg"
|
||||
"github.com/velour/catbase/config"
|
||||
)
|
||||
|
||||
type Impossible struct {
|
||||
b bot.Bot
|
||||
c *config.Config
|
||||
|
||||
title string
|
||||
content []string
|
||||
updated time.Time
|
||||
testing bool
|
||||
}
|
||||
|
||||
func New(b bot.Bot) *Impossible {
|
||||
i := &Impossible{
|
||||
b: b,
|
||||
c: b.Config(),
|
||||
title: "",
|
||||
content: []string{},
|
||||
updated: getTodaysMidnight().Add(time.Hour * -24),
|
||||
testing: false,
|
||||
}
|
||||
|
||||
b.Register(i, bot.Help, i.help)
|
||||
b.Register(i, bot.Message, i.message)
|
||||
|
||||
return i
|
||||
}
|
||||
|
||||
func newTesting(b bot.Bot) *Impossible {
|
||||
i := &Impossible{
|
||||
b: b,
|
||||
c: b.Config(),
|
||||
title: "",
|
||||
content: []string{},
|
||||
updated: getTodaysMidnight().Add(time.Hour * -24),
|
||||
testing: true,
|
||||
}
|
||||
|
||||
b.Register(i, bot.Help, i.help)
|
||||
b.Register(i, bot.Message, i.message)
|
||||
|
||||
return i
|
||||
}
|
||||
|
||||
func (p *Impossible) help(c bot.Connector, kind bot.Kind, message msg.Message, args ...interface{}) bool {
|
||||
p.b.Send(c, bot.Message, message.Channel, "You don't need to do anything. I'll take care of it. But guess what I'm thinking.")
|
||||
return true
|
||||
}
|
||||
|
||||
func (p *Impossible) message(c bot.Connector, kind bot.Kind, message msg.Message, args ...interface{}) bool {
|
||||
messaged := false
|
||||
if p.updated.Before(time.Now()) {
|
||||
if p.title != "" {
|
||||
p.b.Send(c, bot.Message, message.Channel, fmt.Sprintf("The last impossible wikipedia article was: \"%s\"", p.title))
|
||||
messaged = true
|
||||
}
|
||||
for !p.refreshImpossible() {
|
||||
}
|
||||
|
||||
if p.testing {
|
||||
p.b.Send(c, bot.Message, message.Channel, p.title)
|
||||
messaged = true
|
||||
}
|
||||
}
|
||||
|
||||
lowercase := strings.ToLower(message.Body)
|
||||
if lowercase == "hint" || lowercase == "clue" {
|
||||
messaged = true
|
||||
p.b.Send(c, bot.Message, message.Channel, p.content[rand.Intn(len(p.content))])
|
||||
} else if strings.Contains(lowercase, strings.ToLower(p.title)) {
|
||||
messaged = true
|
||||
p.b.Send(c, bot.Message, message.Channel, fmt.Sprintf("You guessed the last impossible wikipedia article: \"%s\"", p.title))
|
||||
for !p.refreshImpossible() {
|
||||
}
|
||||
}
|
||||
|
||||
return messaged
|
||||
}
|
||||
|
||||
func (p *Impossible) refreshImpossible() bool {
|
||||
p.updated = getTodaysMidnight()
|
||||
resp, err := http.Get("https://en.wikipedia.org/wiki/Special:Random")
|
||||
if err != nil {
|
||||
log.Fatal().Err(err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
body, err := ioutil.ReadAll(resp.Body)
|
||||
|
||||
titleRegex := regexp.MustCompile(`id="firstHeading"[^>]*(?P<Title>[^<]*)`)
|
||||
results := titleRegex.FindStringSubmatch(string(body))
|
||||
title := results[1][1:] //remove the leading <
|
||||
|
||||
if title == "" {
|
||||
return false
|
||||
}
|
||||
|
||||
p.title = title
|
||||
p.content = []string{}
|
||||
|
||||
resp, err = http.Get("https://en.wikipedia.org/w/api.php?format=json&action=query&prop=extracts&explaintext&titles=" + url.PathEscape(title))
|
||||
if err != nil {
|
||||
log.Fatal().Err(err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
body, err = ioutil.ReadAll(resp.Body)
|
||||
|
||||
var object map[string]interface{}
|
||||
json.Unmarshal([]byte(body), &object)
|
||||
|
||||
pages := object["query"].(map[string]interface{})["pages"].(map[string]interface{})
|
||||
for _, page := range pages {
|
||||
descriptionText := page.(map[string]interface{})["extract"].(string)
|
||||
sentences := strings.Split(strings.ReplaceAll(descriptionText, "\n", " "), ". ")
|
||||
for _, sentence := range sentences {
|
||||
trimmed := strings.ToLower(strings.TrimSpace(sentence))
|
||||
if len(trimmed) == 0 || strings.HasPrefix(trimmed, "==") || len(strings.Split(trimmed, " ")) < 5 {
|
||||
continue
|
||||
}
|
||||
|
||||
censored := strings.ReplaceAll(trimmed, strings.ToLower(title), "?????")
|
||||
|
||||
p.content = append(p.content, censored)
|
||||
}
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
||||
func getTodaysMidnight() time.Time {
|
||||
now := time.Now()
|
||||
return time.Date(now.Year(), now.Month(), now.Day(), 24, 0, 0, 0, now.Location())
|
||||
}
|
|
@ -0,0 +1,59 @@
|
|||
package impossible
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"github.com/velour/catbase/plugins/cli"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/velour/catbase/bot"
|
||||
"github.com/velour/catbase/bot/msg"
|
||||
"github.com/velour/catbase/bot/user"
|
||||
"github.com/velour/catbase/plugins/counter"
|
||||
)
|
||||
|
||||
func makeMessage(payload string) (bot.Connector, bot.Kind, msg.Message) {
|
||||
isCmd := strings.HasPrefix(payload, "!")
|
||||
if isCmd {
|
||||
payload = payload[1:]
|
||||
}
|
||||
return &cli.CliPlugin{}, bot.Message, msg.Message{
|
||||
User: &user.User{Name: "tester"},
|
||||
Channel: "test",
|
||||
Body: payload,
|
||||
Command: isCmd,
|
||||
}
|
||||
}
|
||||
|
||||
func makePlugin(t *testing.T) (*Impossible, *bot.MockBot) {
|
||||
mb := bot.NewMockBot()
|
||||
counter.New(mb)
|
||||
p := newTesting(mb)
|
||||
assert.NotNil(t, p)
|
||||
return p, mb
|
||||
}
|
||||
|
||||
func TestNothing(t *testing.T) {
|
||||
p, mb := makePlugin(t)
|
||||
p.message(makeMessage("hi"))
|
||||
p.message(makeMessage("nothing"))
|
||||
assert.Len(t, mb.Messages, 1)
|
||||
}
|
||||
|
||||
func TestHint(t *testing.T) {
|
||||
p, mb := makePlugin(t)
|
||||
p.message(makeMessage("hi"))
|
||||
p.message(makeMessage("!hint"))
|
||||
assert.Len(t, mb.Messages, 2)
|
||||
}
|
||||
|
||||
func TestCorrect(t *testing.T) {
|
||||
p, mb := makePlugin(t)
|
||||
p.message(makeMessage("hi"))
|
||||
p.message(makeMessage(mb.Messages[0]))
|
||||
|
||||
congrats := fmt.Sprintf("You guessed the last impossible wikipedia article: \"%s\"", mb.Messages[0])
|
||||
|
||||
assert.Contains(t, mb.Messages[1], congrats)
|
||||
}
|
Loading…
Reference in New Issue