49 lines
813 B
Go
49 lines
813 B
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"flag"
|
||
|
|
"fmt"
|
||
|
|
"math/rand"
|
||
|
|
"os"
|
||
|
|
"regexp"
|
||
|
|
"strconv"
|
||
|
|
|
||
|
|
"git.wntrmute.dev/kyle/goutils/die"
|
||
|
|
)
|
||
|
|
|
||
|
|
var dieRollFormat = regexp.MustCompile(`^(\d+)[dD](\d+)$`)
|
||
|
|
|
||
|
|
func rollDie(count, sides int) []int {
|
||
|
|
sum := 0
|
||
|
|
var rolls []int
|
||
|
|
|
||
|
|
for i := 0; i < count; i++ {
|
||
|
|
roll := rand.Intn(sides) + 1
|
||
|
|
sum += roll
|
||
|
|
rolls = append(rolls, roll)
|
||
|
|
}
|
||
|
|
|
||
|
|
rolls = append(rolls, sum)
|
||
|
|
return rolls
|
||
|
|
}
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
flag.Parse()
|
||
|
|
|
||
|
|
for _, arg := range flag.Args() {
|
||
|
|
if !dieRollFormat.MatchString(arg) {
|
||
|
|
fmt.Fprintf(os.Stderr, "invalid die format %s: should be XdY\n", arg)
|
||
|
|
os.Exit(1)
|
||
|
|
}
|
||
|
|
|
||
|
|
dieRoll := dieRollFormat.FindAllStringSubmatch(arg, -1)
|
||
|
|
count, err := strconv.Atoi(dieRoll[0][1])
|
||
|
|
die.If(err)
|
||
|
|
|
||
|
|
sides, err := strconv.Atoi(dieRoll[0][2])
|
||
|
|
die.If(err)
|
||
|
|
|
||
|
|
fmt.Println(rollDie(count, sides))
|
||
|
|
}
|
||
|
|
}
|