package main import ( "github.com/spf13/afero" "log" "strings" "bufio" "strconv" "time" "fmt" ) type Mothballs struct { categories map[string]*Zipfs afero.Fs } func NewMothballs(fs afero.Fs) *Mothballs { return &Mothballs{ Fs: fs, categories: make(map[string]*Zipfs), } } func (m *Mothballs) Open(cat string, points int, filename string) (ReadSeekCloser, time.Time, error) { mb, ok := m.categories[cat] if ! ok { return nil, time.Time{}, fmt.Errorf("No such category: %s", cat) } f, err := mb.Open(fmt.Sprintf("content/%d/%s", points, filename)) return f, mb.ModTime(), err } func (m *Mothballs) Inventory() []Category { categories := make([]Category, 0, 20) for cat, zfs := range m.categories { pointsList := make([]int, 0, 20) pf, err := zfs.Open("puzzles.txt") if err != nil { // No puzzles = no category continue } scanner := bufio.NewScanner(pf) for scanner.Scan() { line := scanner.Text() if pointval, err := strconv.Atoi(line); err != nil { log.Printf("Reading points for %s: %s", cat, err.Error()) } else { pointsList = append(pointsList, pointval) } } categories = append(categories, Category{cat, pointsList}) } return categories } func (m *Mothballs) CheckAnswer(cat string, points int, answer string) error { zfs, ok := m.categories[cat] if ! ok { return fmt.Errorf("No such category: %s", cat) } af, err := zfs.Open("answers.txt") if err != nil { return fmt.Errorf("No answers.txt file") } defer af.Close() needle := fmt.Sprintf("%d %s", points, answer) scanner := bufio.NewScanner(af) for scanner.Scan() { if scanner.Text() == needle { return nil } } return fmt.Errorf("Invalid answer") } func (m *Mothballs) Update() { // Any new categories? files, err := afero.ReadDir(m.Fs, "/") if err != nil { log.Print("Error listing mothballs: ", err) return } for _, f := range files { filename := f.Name() if !strings.HasSuffix(filename, ".mb") { continue } categoryName := strings.TrimSuffix(filename, ".mb") if _, ok := m.categories[categoryName]; !ok { zfs, err := OpenZipfs(m.Fs, filename) if err != nil { log.Print("Error opening ", filename, ": ", err) continue } log.Print("New mothball: ", filename) m.categories[categoryName] = zfs } } }