moth/cmd/mothd/mothballs.go

package main

import (
	"github.com/spf13/afero"
	"log"
	"io"
	"strings"
)

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) Metadata(cat string, points int) (io.ReadCloser, error) {
	f, err := m.Fs.Open("/dev/null")
	return f, err
}

func (m *Mothballs) Open(cat string, points int, filename string) (io.ReadCloser, error) {
	f, err := m.Fs.Open("/dev/null")
	return f, err
}

func (m *Mothballs) Inventory() []Category {
	return []Category{}
}


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
		}
	}
}