mirror of https://github.com/dirtbags/moth.git
120 lines
2.4 KiB
Go
120 lines
2.4 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 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))
|
|
}
|
|
}
|