80 lines
1.8 KiB
Go
80 lines
1.8 KiB
Go
package mtree
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"sort"
|
|
)
|
|
|
|
// Result of a Check
|
|
type Result struct {
|
|
Failures []Failure // list of any failures in the Check
|
|
}
|
|
|
|
// Failure of a particular keyword for a path
|
|
type Failure struct {
|
|
Path string
|
|
Keyword string
|
|
Expected string
|
|
Got string
|
|
}
|
|
|
|
// String returns a "pretty" formatting for a Failure
|
|
func (f Failure) String() string {
|
|
return fmt.Sprintf("%q: keyword %q: expected %s; got %s", f.Path, f.Keyword, f.Expected, f.Got)
|
|
}
|
|
|
|
// Check a root directory path for a DirectoryHierarchy
|
|
func Check(root string, dh *DirectoryHierarchy) (*Result, error) {
|
|
creator := dhCreator{DH: dh}
|
|
curDir, err := os.Getwd()
|
|
if err == nil {
|
|
defer os.Chdir(curDir)
|
|
}
|
|
|
|
if err := os.Chdir(root); err != nil {
|
|
return nil, err
|
|
}
|
|
sort.Sort(byPos(creator.DH.Entries))
|
|
|
|
var result Result
|
|
for _, e := range creator.DH.Entries {
|
|
switch e.Type {
|
|
case SpecialType:
|
|
if e.Name == "/set" {
|
|
creator.curSet = &e
|
|
} else if e.Name == "/unset" {
|
|
creator.curSet = nil
|
|
}
|
|
case RelativeType, FullType:
|
|
info, err := os.Lstat(filepath.Join(root, e.Path()))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var kvs KeyVals
|
|
if creator.curSet != nil {
|
|
kvs = MergeSet(creator.curSet.Keywords, e.Keywords)
|
|
} else {
|
|
kvs = NewKeyVals(e.Keywords)
|
|
}
|
|
|
|
for _, kv := range kvs {
|
|
keywordFunc, ok := KeywordFuncs[kv.Keyword()]
|
|
if !ok {
|
|
return nil, fmt.Errorf("Unknown keyword %q for file %q", kv.Keyword(), e.Path())
|
|
}
|
|
curKeyVal, err := keywordFunc(filepath.Join(root, e.Path()), info)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if string(kv) != curKeyVal {
|
|
failure := Failure{Path: e.Path(), Keyword: kv.Keyword(), Expected: kv.Value(), Got: KeyVal(curKeyVal).Value()}
|
|
result.Failures = append(result.Failures, failure)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return &result, nil
|
|
}
|