mirror of
https://github.com/go-task/task.git
synced 2024-12-14 10:52:43 +02:00
44a52359dc
future-proof since the .task directory can be used to save other state in the future
88 lines
1.8 KiB
Go
88 lines
1.8 KiB
Go
package status
|
|
|
|
import (
|
|
"crypto/md5"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"regexp"
|
|
"strings"
|
|
)
|
|
|
|
// Checksum validades if a task is up to date by calculating its source
|
|
// files checksum
|
|
type Checksum struct {
|
|
Dir string
|
|
Task string
|
|
Sources []string
|
|
}
|
|
|
|
// IsUpToDate implements the Checker interface
|
|
func (c *Checksum) IsUpToDate() (bool, error) {
|
|
checksumFile := c.checksumFilePath()
|
|
|
|
data, _ := ioutil.ReadFile(checksumFile)
|
|
oldMd5 := strings.TrimSpace(string(data))
|
|
|
|
sources, err := glob(c.Dir, c.Sources)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
newMd5, err := c.checksum(sources...)
|
|
if err != nil {
|
|
return false, nil
|
|
}
|
|
|
|
_ = os.MkdirAll(filepath.Join(c.Dir, ".task", "checksum"), 0755)
|
|
if err = ioutil.WriteFile(checksumFile, []byte(newMd5+"\n"), 0644); err != nil {
|
|
return false, err
|
|
}
|
|
return oldMd5 == newMd5, nil
|
|
}
|
|
|
|
func (c *Checksum) checksum(files ...string) (string, error) {
|
|
h := md5.New()
|
|
|
|
for _, f := range files {
|
|
f, err := os.Open(f)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
info, err := f.Stat()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if info.IsDir() {
|
|
continue
|
|
}
|
|
// also sum the filename, so checksum changes for renaming a file
|
|
if _, err = io.Copy(h, strings.NewReader(info.Name())); err != nil {
|
|
return "", err
|
|
}
|
|
if _, err = io.Copy(h, f); err != nil {
|
|
return "", err
|
|
}
|
|
}
|
|
|
|
return fmt.Sprintf("%x", h.Sum(nil)), nil
|
|
}
|
|
|
|
// OnError implements the Checker interface
|
|
func (c *Checksum) OnError() error {
|
|
return os.Remove(c.checksumFilePath())
|
|
}
|
|
|
|
func (c *Checksum) checksumFilePath() string {
|
|
return filepath.Join(c.Dir, ".task", "checksum", c.normalizeFilename(c.Task))
|
|
}
|
|
|
|
var checksumFilenameRegexp = regexp.MustCompile("[^A-z0-9]")
|
|
|
|
// replaces invalid caracters on filenames with "-"
|
|
func (*Checksum) normalizeFilename(f string) string {
|
|
return checksumFilenameRegexp.ReplaceAllString(f, "-")
|
|
}
|