moth/maintenance.go

package main

import (
	"log"
	"io/ioutil"
	"time"
	"strings"
)

func allfiles(dirpath) []string {
	files, err := ioutil.ReadDir(dirpath)
	if (err != nil) {
		log.Printf("Error reading directory %s: %s", dirpath, err)
		return []
	}
	return files
}

// maintenance runs
func tidy() {
	// Skip if we've been disabled
	if exists(statePath("disabled")) {
		log.Print("disabled file found, suspending maintenance")
		return
	}
	
	// Skip if we've expired
	untilspec, _ := ioutil.ReadFile(statePath("until"))
	until, err := time.Parse(time.RFC3339, string(untilspec))
	if err == nil {
		if until.Before(time.Now()) {
			log.Print("until file time reached, suspending maintenance")
			return
		}
	}
	
	log.Print("Hello, I'm maintaining!")
	
	//
	// Get current list of categories
	//
	newCategories := []string{}
	for f := range(allfiles(mothPath("packages"))) {
		filename := f.Name()
		filepath := mothPath("packages", filename)
		if ! strings.HasSuffix(filename, ".mb") {
			continue
		}
		
		categoryName := strings.TrimSuffix(filename, ".mb")
		newCategories = append(newCategories, categoryName)
		
		// Uncompress into cache directory
		cacheMothball(filepath, categoryName)
	}
	categories = newCategories
	
	//
	// Collect new points
	//
	pointsLog = os.OpenFile(statePath("points.log"), os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
	for f := range allfiles(statePath("points.new")) {
		filename := statePath("points.new", f.Name())
		s := ioutil.ReadFile(filename)
		award, err := ParseAward(s)
		if (err != nil) {
			log.Printf("Can't parse award file %s: %s", filename, err)
			continue
		}
		fmt.Fprintf(pointsLog, "%s\n", award.String())
		log.Print(award.String())
		pointsLog.Sync()
		os.Remove(filename)
	}
	pointsLog.Close()
}

// maintenance is the goroutine that runs a periodic maintenance task
func maintenance() {
	for ;; time.Sleep(maintenanceInterval) {
		tidy()
	}
}