vyvanse/vendor/github.com/justinian/dice/dice.go

56 lines
1.1 KiB
Go

package dice
import (
"errors"
"fmt"
"regexp"
"strings"
)
type RollResult interface {
fmt.Stringer
Description() string
Int() int
}
type roller interface {
Pattern() *regexp.Regexp
Roll([]string) (RollResult, error)
}
type basicRollResult struct {
desc string
}
func (r basicRollResult) Description() string { return strings.Trim(r.desc, " \t\r\n") }
var rollHandlers []roller
func addRollHandler(handler roller) {
rollHandlers = append(rollHandlers, handler)
}
/*
open - regexp.MustCompile(`([0-9]+)d([0-9]+)(e)?o$`)
sil - regexp.MustCompile(`([0-9]+)d([0-9]+)(e)?s$`)
*/
func Roll(desc string) (RollResult, string, error) {
for _, rollHandler := range rollHandlers {
rollHandler.Pattern().Longest()
if r := rollHandler.Pattern().FindStringSubmatch(desc); r != nil {
result, err := rollHandler.Roll(r)
if err != nil {
return nil, "", err
}
indexes := rollHandler.Pattern().FindStringSubmatchIndex(desc)
reason := strings.Trim(desc[indexes[0]+len(r[0]):], " \t\r\n")
return result, reason, nil
}
}
return nil, "", errors.New("Bad roll format: " + desc)
}