moth/cmd/transpile/category.go

130 lines
2.5 KiB
Go

package main
import (
"os"
"flag"
"fmt"
"log"
"path/filepath"
"hash/fnv"
"encoding/binary"
"encoding/json"
"encoding/hex"
"strconv"
"math/rand"
)
type PuzzleEntry struct {
Id string
Points int
Puzzle Puzzle
}
func HashHex(input ...string) (string) {
hasher := fnv.New64()
for _, s := range input {
fmt.Fprintln(hasher, s)
}
return hex.EncodeToString(hasher.Sum(nil))
}
func PrngOfStrings(input ...string) (*rand.Rand) {
hasher := fnv.New64()
for _, s := range input {
fmt.Fprint(hasher, s, "\n")
}
seed := binary.BigEndian.Uint64(hasher.Sum(nil))
source := rand.NewSource(int64(seed))
return rand.New(source)
}
func ParsePuzzle(puzzlePath string, seed string) (*Puzzle, error) {
puzzleFd, err := os.Open(puzzlePath)
if err != nil {
return nil, err
}
defer puzzleFd.Close()
puzzle, err := ParseMoth(puzzleFd)
if err != nil {
return nil, err
}
return puzzle, nil
}
func ParseCategory(categoryPath string, seed string) ([]PuzzleEntry, error) {
categoryFd, err := os.Open(categoryPath)
if err != nil {
return nil, err
}
defer categoryFd.Close()
puzzleDirs, err := categoryFd.Readdirnames(0)
if err != nil {
return nil, err
}
puzzleEntries := make([]PuzzleEntry, 0, len(puzzleDirs))
for _, puzzleDir := range puzzleDirs {
puzzlePath := filepath.Join(categoryPath, puzzleDir, "puzzle.moth")
puzzleSeed := fmt.Sprintf("%s/%s", seed, puzzleDir)
points, err := strconv.Atoi(puzzleDir)
if err != nil {
log.Printf("Skipping %s: %v", puzzlePath, err)
continue
}
puzzle, err := ParsePuzzle(puzzlePath, puzzleSeed)
if err != nil {
log.Printf("Skipping %s: %v", puzzlePath, err)
continue
}
prng := PrngOfStrings(puzzleSeed)
idBytes := make([]byte, 16)
prng.Read(idBytes)
id := hex.EncodeToString(idBytes)
puzzleEntry := PuzzleEntry{
Id: id,
Puzzle: *puzzle,
Points: points,
}
puzzleEntries = append(puzzleEntries, puzzleEntry)
}
return puzzleEntries, nil
}
func main() {
// XXX: We need a way to pass in "only run this one point value puzzle"
// XXX: Convert puzzle.py to standalone thingies
flag.Parse()
baseSeedString := os.Getenv("SEED")
for _, dirname := range flag.Args() {
categoryName := filepath.Base(dirname)
categorySeed := fmt.Sprintf("%s/%s", baseSeedString, categoryName)
puzzles, err := ParseCategory(dirname, categorySeed)
if err != nil {
log.Print(err)
continue
}
jpuzzles, err := json.MarshalIndent(puzzles, "", " ")
if err != nil {
log.Print(err)
continue
}
fmt.Println(string(jpuzzles))
}
}