Go 言語でダイスロール IRC ボット

go-dicebot.go

package main

import (
	"fmt"
	"github.com/thoj/go-ircevent"
	"log"
	"math/rand"
	"regexp"
	"strconv"
	"strings"
	"time"
)

func main() {
	r := rand.New(rand.NewSource(time.Now().UnixNano()))
	ircobj := irc.IRC("go-dicebot", "go-dicebot")

	if err := ircobj.Connect("irc.cre.ne.jp:6667"); err != nil {
		log.Fatal(err)
	}

	const chName string = "#irc_test"
	ircobj.AddCallback("001", func(e *irc.Event) {
		ircobj.Join(chName)
	})

	ircobj.AddCallback("PRIVMSG", func(e *irc.Event) {
		message := e.Message()
		nick := e.Nick

		rollRe := regexp.MustCompile("^.goroll ([1-9]\\d*)d([1-9]\\d*)$")
		if m := rollRe.FindStringSubmatch(message); m != nil {
			nDice, _ := strconv.Atoi(m[1])
			max, _ := strconv.Atoi(m[2])

			values := make([]int, nDice)
			valuesStrs := make([]string, nDice)
			sum := 0
			for i := 0; i < nDice; i++ {
				values[i] = r.Intn(max) + 1
				valuesStrs[i] = fmt.Sprint(values[i])
				sum += values[i]
			}

			ircobj.Notice(chName, nick+" -> "+m[1]+"d"+m[2]+" = ["+strings.Join(valuesStrs, ", ")+"] = "+fmt.Sprint(sum))
		}
	})

	ircobj.Loop()
}