sisyphus: add plugin

main.go

@@ -23,6 +23,7 @@ import (
 	"github.com/velour/catbase/plugins/reminder"
 	"github.com/velour/catbase/plugins/rpgORdie"
 	"github.com/velour/catbase/plugins/rss"
+	"github.com/velour/catbase/plugins/sisyphus"
 	"github.com/velour/catbase/plugins/stats"
 	"github.com/velour/catbase/plugins/talker"
 	"github.com/velour/catbase/plugins/twitch"
@@ -71,6 +72,7 @@ func main() {
 	b.AddHandler("twitch", twitch.New(b))
 	b.AddHandler("inventory", inventory.New(b))
 	b.AddHandler("rpgORdie", rpgORdie.New(b))
+	b.AddHandler("sisyphus", sisyphus.New(b))
 	// catches anything left, will always return true
 	b.AddHandler("factoid", fact.New(b))
 

plugins/sisyphus/sisyphus.go

@@ -0,0 +1,229 @@
+package sisyphus
+
+import (
+	"fmt"
+	"log"
+	"math/rand"
+	"strconv"
+	"strings"
+	"time"
+
+	"github.com/velour/catbase/bot"
+	"github.com/velour/catbase/bot/msg"
+)
+
+const (
+	BOULDER  = ":full_moon:"
+	MOUNTAIN = ":new_moon:"
+)
+
+type SisyphusPlugin struct {
+	Bot       bot.Bot
+	listenFor map[string]*game
+}
+
+type game struct {
+	id       string
+	channel  string
+	bot      bot.Bot
+	who      string
+	start    time.Time
+	size     int
+	hardness int
+	current  int
+	nextPush time.Time
+	nextDec  time.Time
+	timers   [2]*time.Timer
+	ended    bool
+	nextAns  int
+}
+
+func NewRandomGame(bot bot.Bot, channel, who string) *game {
+	size := rand.Intn(9) + 2
+	g := game{
+		channel:  channel,
+		bot:      bot,
+		who:      who,
+		start:    time.Now(),
+		size:     size,
+		hardness: rand.Intn(9) + 1,
+		current:  size / 2,
+	}
+	g.id = bot.SendMessage(channel, g.toMessageString())
+
+	g.schedulePush()
+	g.scheduleDecrement()
+
+	return &g
+}
+
+func (g *game) scheduleDecrement() {
+	if g.timers[0] != nil {
+		g.timers[0].Stop()
+	}
+	g.nextDec = time.Now().Add(time.Duration((60 + rand.Intn(60*5))) * time.Second)
+	go func() {
+		t := time.NewTimer(g.nextDec.Sub(time.Now()))
+		g.timers[0] = t
+		select {
+		case <-t.C:
+			g.handleDecrement()
+		}
+	}()
+}
+
+func (g *game) schedulePush() {
+	if g.timers[1] != nil {
+		g.timers[1].Stop()
+	}
+	g.nextPush = time.Now().Add(time.Duration(rand.Intn(60)+1) * time.Minute)
+	go func() {
+		t := time.NewTimer(g.nextPush.Sub(time.Now()))
+		g.timers[1] = t
+		select {
+		case <-t.C:
+			g.handleNotify()
+		}
+	}()
+}
+
+func (g *game) endGame() {
+	for _, t := range g.timers {
+		t.Stop()
+	}
+	g.ended = true
+}
+
+func (g *game) handleDecrement() {
+	g.current++
+	g.bot.Edit(g.channel, g.toMessageString(), g.id)
+	if g.current > g.size-2 {
+		g.bot.ReplyToMessageIdentifier(g.channel, "you lose", g.id)
+		msg := fmt.Sprintf("%s just lost the game after %s", g.who, time.Now().Sub(g.start))
+		g.bot.SendMessage(g.channel, msg)
+		g.endGame()
+	} else {
+		g.scheduleDecrement()
+	}
+}
+
+func (g *game) handleNotify() {
+	g.bot.ReplyToMessageIdentifier(g.channel, "You can push now.\n"+g.generateQuestion(), g.id)
+}
+
+func (g *game) generateQuestion() string {
+	n1 := rand.Intn(99) + (rand.Intn(9)+1)*100
+	n2 := rand.Intn(99) + (rand.Intn(9)+1)*100
+	var op string
+	switch i := rand.Intn(3); i {
+	case 0:
+		// times
+		g.nextAns = n1 * n2
+		op = "*"
+	case 1:
+		// plus
+		g.nextAns = n1 + n2
+		op = "+"
+	case 2:
+		// minus
+		g.nextAns = n1 - n2
+		op = "-"
+	}
+	return fmt.Sprintf("What is %d %s %d?", n1, op, n2)
+}
+
+func (g *game) checkAnswer(ans string) bool {
+	if strings.Contains(ans, strconv.Itoa(g.nextAns)) {
+		g.current--
+		if g.current < 0 {
+			g.current = 0
+		}
+		return true
+	}
+	return false
+}
+
+func (g *game) toMessageString() string {
+	out := ""
+	for i := 0; i < g.size; i++ {
+		for j := 0; j < i; j++ {
+			out = out + MOUNTAIN
+		}
+		if i == g.current {
+			out = out + BOULDER
+		} else if i == g.current+1 {
+			out = out + ":" + g.who + ":"
+		}
+		out = out + "\n"
+	}
+	return out
+}
+
+func New(b bot.Bot) *SisyphusPlugin {
+	return &SisyphusPlugin{
+		Bot:       b,
+		listenFor: map[string]*game{},
+	}
+}
+
+func (p *SisyphusPlugin) Message(message msg.Message) bool {
+	if strings.ToLower(message.Body) == "start sisyphus" {
+		b := NewRandomGame(p.Bot, message.Channel, message.User.Name)
+		p.listenFor[b.id] = b
+		p.Bot.ReplyToMessageIdentifier(message.Channel, "Over here.", b.id)
+		return true
+	}
+	return false
+}
+
+func (p *SisyphusPlugin) Help(channel string, parts []string) {
+	p.Bot.SendMessage(channel, "https://en.wikipedia.org/wiki/Sisyphus")
+}
+
+func (p *SisyphusPlugin) Event(kind string, message msg.Message) bool {
+	return false
+}
+
+func (p *SisyphusPlugin) BotMessage(message msg.Message) bool {
+	return false
+}
+
+func (p *SisyphusPlugin) RegisterWeb() *string {
+	return nil
+}
+
+func (p *SisyphusPlugin) ReplyMessage(message msg.Message, identifier string) bool {
+	if strings.ToLower(message.User.Name) != strings.ToLower(p.Bot.Config().Nick) {
+		if g, ok := p.listenFor[identifier]; ok {
+
+			log.Printf("got message on %s: %+v", identifier, message)
+
+			if g.ended {
+				return false
+			}
+
+			if strings.ToLower(message.Body) == "end game" {
+				g.endGame()
+				return true
+			}
+
+			if time.Now().After(g.nextPush) {
+				if g.checkAnswer(message.Body) {
+					p.Bot.Edit(message.Channel, g.toMessageString(), identifier)
+					g.schedulePush()
+					msg := fmt.Sprintf("Ok. You can push again in %s", g.nextPush.Sub(time.Now()))
+					p.Bot.ReplyToMessageIdentifier(message.Channel, msg, identifier)
+				} else {
+					p.Bot.ReplyToMessageIdentifier(message.Channel, "you lose", identifier)
+					msg := fmt.Sprintf("%s just lost the sisyphus game after %s", g.who, time.Now().Sub(g.start))
+					p.Bot.SendMessage(message.Channel, msg)
+					g.endGame()
+				}
+			} else {
+				p.Bot.ReplyToMessageIdentifier(message.Channel, "you cannot push yet", identifier)
+			}
+			return true
+		}
+	}
+	return false
+}

plugins/sisyphus/sisyphus_test.go

@@ -0,0 +1 @@
+package sisyphus

slack/slack.go

@@ -39,8 +39,8 @@ type Slack struct {
 
 	emoji map[string]string
 
-	eventReceived   func(msg.Message)
+	eventReceived        func(msg.Message)
-	messageReceived func(msg.Message)
+	messageReceived      func(msg.Message)
 	replyMessageReceived func(msg.Message, string)
 }
 
@@ -135,7 +135,7 @@ type slackMessage struct {
 	User     string `json:"user"`
 	Username string `json:"username"`
 	Ts       string `json:"ts"`
-	ThreadTs       string `json:"thread_ts"`
+	ThreadTs string `json:"thread_ts"`
 	Error    struct {
 		Code uint64 `json:"code"`
 		Msg  string `json:"msg"`
@@ -207,6 +207,7 @@ func (s *Slack) SendMessageType(channel, message string, meMessage bool) (string
 
 	resp, err := http.PostForm(postUrl,
 		url.Values{"token": {s.config.Slack.Token},
+			"as_user": {"true"},
 			"channel": {channel},
 			"text":    {message},
 		})
@@ -256,6 +257,7 @@ func (s *Slack) SendAction(channel, message string) string {
 func (s *Slack) ReplyToMessageIdentifier(channel, message, identifier string) (string, bool) {
 	resp, err := http.PostForm("https://slack.com/api/chat.postMessage",
 		url.Values{"token": {s.config.Slack.Token},
+			"as_user":   {"true"},
 			"channel":   {channel},
 			"text":      {message},
 			"thread_ts": {identifier},