moth/maintenance.go

84 lines
1.8 KiB
Go
Raw Normal View History

2018-05-02 15:45:45 -06:00
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()
2018-05-02 17:00:53 -06:00
filepath := mothPath("packages", filename)
2018-05-02 15:45:45 -06:00
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
//
2018-05-02 17:00:53 -06:00
pointsLog = os.OpenFile(statePath("points.log"), os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
2018-05-02 15:45:45 -06:00
for f := range allfiles(statePath("points.new")) {
2018-05-02 17:00:53 -06:00
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)
2018-05-02 15:45:45 -06:00
}
2018-05-02 17:00:53 -06:00
pointsLog.Close()
2018-05-02 15:45:45 -06:00
}
// maintenance is the goroutine that runs a periodic maintenance task
func maintenance() {
for ;; time.Sleep(maintenanceInterval) {
tidy()
}
}