mirror of https://github.com/dirtbags/moth.git
41 lines
732 B
Go
41 lines
732 B
Go
package transpile
|
|
|
|
import (
|
|
"log"
|
|
"sort"
|
|
"strings"
|
|
|
|
"github.com/spf13/afero"
|
|
)
|
|
|
|
// Inventory maps category names to lists of point values.
|
|
type Inventory map[string][]int
|
|
|
|
// FsInventory returns a mapping of category names to puzzle point values.
|
|
func FsInventory(fs afero.Fs) (Inventory, error) {
|
|
dirEnts, err := afero.ReadDir(fs, "")
|
|
if err != nil {
|
|
log.Print(err)
|
|
return nil, err
|
|
}
|
|
|
|
inv := make(Inventory)
|
|
for _, ent := range dirEnts {
|
|
if strings.HasPrefix(ent.Name(), ".") {
|
|
continue
|
|
}
|
|
if ent.IsDir() {
|
|
name := ent.Name()
|
|
c := NewFsCategory(fs, name)
|
|
puzzles, err := c.Inventory()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
sort.Ints(puzzles)
|
|
inv[name] = puzzles
|
|
}
|
|
}
|
|
|
|
return inv, nil
|
|
}
|