mirror of https://github.com/dirtbags/moth.git
84 lines
1.8 KiB
Go
84 lines
1.8 KiB
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()
|
|
}
|
|
}
|