2
2
mirror of https://github.com/octoleo/restic.git synced 2024-11-24 21:57:41 +00:00

Add dirdiff command

This commit is contained in:
Alexander Neumann 2014-08-06 20:09:24 +02:00
parent ab602b88de
commit 16d7974b3f

139
cmd/dirdiff/main.go Normal file
View File

@ -0,0 +1,139 @@
package main
import (
"fmt"
"os"
"path/filepath"
"syscall"
)
const backlog = 100
type entry struct {
path string
fi os.FileInfo
}
func walk(dir string) <-chan *entry {
ch := make(chan *entry, 100)
go func() {
err := filepath.Walk(dir, func(path string, info os.FileInfo, err error) error {
if err != nil {
fmt.Fprintf(os.Stderr, "error: %v\n", err)
return nil
}
name, err := filepath.Rel(dir, path)
if err != nil {
fmt.Fprintf(os.Stderr, "error: %v\n", err)
return nil
}
ch <- &entry{
path: name,
fi: info,
}
return nil
})
if err != nil {
fmt.Fprintf(os.Stderr, "Walk() error: %v\n", err)
}
close(ch)
}()
// first element is root
_ = <-ch
return ch
}
func (e *entry) equals(other *entry) bool {
if e.path != other.path {
return false
}
if e.fi.Mode() != other.fi.Mode() {
return false
}
if e.fi.ModTime() != other.fi.ModTime() {
return false
}
stat, _ := e.fi.Sys().(*syscall.Stat_t)
stat2, _ := other.fi.Sys().(*syscall.Stat_t)
if stat.Uid != stat2.Uid || stat2.Gid != stat2.Gid {
return false
}
return true
}
func main() {
if len(os.Args) != 3 {
fmt.Fprintf(os.Stderr, "USAGE: %s DIR1 DIR2\n", os.Args[0])
os.Exit(1)
}
ch1 := walk(os.Args[1])
ch2 := walk(os.Args[2])
changes := false
var a, b *entry
for {
var ok bool
if ch1 != nil && a == nil {
a, ok = <-ch1
if !ok {
ch1 = nil
}
}
if ch2 != nil && b == nil {
b, ok = <-ch2
if !ok {
ch2 = nil
}
}
if ch1 == nil && ch2 == nil {
break
}
if ch1 == nil {
fmt.Printf("+%v\n", b.path)
changes = true
} else if ch2 == nil {
fmt.Printf("-%v\n", a.path)
changes = true
} else if !a.equals(b) {
if a.path < b.path {
fmt.Printf("-%v\n", a.path)
changes = true
a = nil
continue
} else if a.path > b.path {
fmt.Printf("+%v\n", b.path)
changes = true
b = nil
continue
} else {
fmt.Printf("%%%v\n", a.path)
changes = true
}
}
a, b = nil, nil
}
if changes {
os.Exit(1)
}
}