tomato.go

package main

import (
	"fmt"
	"log"
	"os"
	"os/signal"
	"strconv"
	"strings"
	"time"

	"github.com/andrewwatson/goslackbot"
)

var (
	bot            goslackbot.SlackBot
	rtmToken       string
	tomatoDuration int64
)

func main() {

	if len(os.Args) != 3 {
		fmt.Fprintf(os.Stderr, "usage: mybot slack-bot-token tomato-duration\n")
		os.Exit(1)
	}

	rtmToken = os.Args[1]
	tomatoArg := os.Args[2]

	tomatoDuration, err := strconv.Atoi(tomatoArg)
	if err != nil {
		fmt.Errorf("Invalid Number of Minutes for :tomato: %s", err.Error())
		os.Exit(1)
	}

	// token := "xoxb-16191968215-yRDoPV15kCXYldYA5Qx4U2FV" 
	// no, that's not my real token...

	bot, err := goslackbot.NewSlackBot(rtmToken)

	if err != nil {
		log.Fatalf("CAN NOT BOT %s", err)
	}

	// fmt.Println(bot)

	bot.RegisterIncomingFunction("function", func(m goslackbot.SlackMessage) {

		if m.Type == "message" {

			user := bot.GetUser(m.User)
			channel := bot.GetChannel(m.Channel)

			log.Printf("Message from %s on %s", user.Name, channel.Name)

			if strings.HasPrefix(m.Text, ":tomato:") || strings.HasPrefix(m.Text, ":pomodoro:") {

				remindMe := strings.Replace(m.Text, ":tomato: ", "", 1)
				m.Text = fmt.Sprintf("Ok <@%s>, got it! I'll remind you in %d minutes", m.User, tomatoDuration)

				bot.SendMessage(m.Channel, m.Text)

				go func() {

					after := time.After(time.Minute * time.Duration(tomatoDuration))
					<-after

					m.Text = fmt.Sprintf("Hey <@%s>, it's been %d minutes since you started '%s'", m.User, tomatoDuration, remindMe)
					m.ReplyTo = m.Id
					bot.SendMessage(m.Channel, m.Text)

				}()

			}

		}
	})

	err = bot.Connect()

	if err != nil {
		log.Fatalf("ERR: %s", err)
	}

	c := make(chan os.Signal, 1)
	signal.Notify(c, os.Interrupt, os.Kill)
	s := <-c

	fmt.Println("Got signal:", s)
}

func tomato(bot *goslackbot.SlackBot, tomatoDuration int, m goslackbot.SlackMessage) {

}