func (rs *RombaService) diffdat(cmd *commander.Command, args []string) error {
oldDatPath := cmd.Flag.Lookup("old").Value.Get().(string)
newDatPath := cmd.Flag.Lookup("new").Value.Get().(string)
outPath := cmd.Flag.Lookup("out").Value.Get().(string)
givenName := cmd.Flag.Lookup("name").Value.Get().(string)
givenDescription := cmd.Flag.Lookup("description").Value.Get().(string)
if oldDatPath == "" {
fmt.Fprintf(cmd.Stdout, "-old argument required")
return errors.New("missing old argument")
}
if newDatPath == "" {
fmt.Fprintf(cmd.Stdout, "-new argument required")
return errors.New("missing new argument")
}
if outPath == "" {
fmt.Fprintf(cmd.Stdout, "-out argument required")
return errors.New("missing out argument")
}
glog.Infof("diffdat new dat %s and old dat %s into %s", newDatPath, oldDatPath, outPath)
oldDat, _, err := parser.Parse(oldDatPath)
if err != nil {
return err
}
newDat, _, err := parser.Parse(newDatPath)
if err != nil {
return err
}
if givenName == "" {
givenName = strings.TrimSuffix(filepath.Base(outPath), filepath.Ext(outPath))
}
if givenDescription == "" {
givenDescription = givenName
}
dd, err := dedup.NewLevelDBDeduper()
if err != nil {
return err
}
defer dd.Close()
err = dedup.Declare(oldDat, dd)
if err != nil {
return err
}
diffDat, err := dedup.Dedup(newDat, dd)
if err != nil {
return err
}
diffDat = diffDat.FilterRoms(func(r *types.Rom) bool {
return r.Size > 0
})
var endMsg string
if diffDat != nil {
diffDat.Name = givenName
diffDat.Description = givenDescription
diffDat.Path = outPath
diffFile, err := os.Create(outPath)
if err != nil {
return err
}
defer diffFile.Close()
diffWriter := bufio.NewWriter(diffFile)
defer diffWriter.Flush()
err = types.ComposeCompliantDat(diffDat, diffWriter)
if err != nil {
return err
}
endMsg = fmt.Sprintf("diffdat finished, %d games with diffs found, written diffdat file %s",
len(diffDat.Games), outPath)
} else {
endMsg = "diffdat finished, no diffs found, no diffdat file written"
}
glog.Infof(endMsg)
fmt.Fprintf(cmd.Stdout, endMsg)
rs.broadCastProgress(time.Now(), false, true, endMsg)
return nil
}
func (rs *RombaService) ediffdatWork(cmd *commander.Command, args []string) error {
oldDatPath := cmd.Flag.Lookup("old").Value.Get().(string)
newDatPath := cmd.Flag.Lookup("new").Value.Get().(string)
outPath := cmd.Flag.Lookup("out").Value.Get().(string)
if oldDatPath == "" {
fmt.Fprintf(cmd.Stdout, "-old argument required")
return errors.New("missing old argument")
}
if newDatPath == "" {
fmt.Fprintf(cmd.Stdout, "-new argument required")
return errors.New("missing new argument")
}
if outPath == "" {
fmt.Fprintf(cmd.Stdout, "-out argument required")
return errors.New("missing out argument")
}
err := os.MkdirAll(outPath, 0777)
if err != nil {
return err
}
glog.Infof("ediffdat new dat %s and old dat %s into %s", newDatPath, oldDatPath, outPath)
dd, err := dedup.NewLevelDBDeduper()
if err != nil {
return err
}
defer dd.Close()
err = filepath.Walk(oldDatPath, func(path string, info os.FileInfo, err error) error {
if info.IsDir() {
return nil
}
ext := filepath.Ext(path)
if ext == ".dat" || ext == ".xml" {
rs.pt.DeclareFile(path)
oldDat, _, err := parser.Parse(path)
if err != nil {
return err
}
err = dedup.Declare(oldDat, dd)
if err != nil {
return err
}
rs.pt.AddBytesFromFile(info.Size(), false)
}
return nil
})
if err != nil {
return err
}
err = filepath.Walk(newDatPath, func(path string, info os.FileInfo, err error) error {
if info.IsDir() {
return nil
}
ext := filepath.Ext(path)
if ext == ".dat" || ext == ".xml" {
rs.pt.DeclareFile(path)
newDat, _, err := parser.Parse(path)
if err != nil {
return err
}
oneDiffDat, err := dedup.Dedup(newDat, dd)
if err != nil {
return err
}
if oneDiffDat != nil {
oneDiffDat = oneDiffDat.FilterRoms(func(r *types.Rom) bool {
return r.Size > 0
})
}
if oneDiffDat != nil {
err = writeDiffDat(oneDiffDat, filepath.Join(outPath, oneDiffDat.Name+".dat"))
}
rs.pt.AddBytesFromFile(info.Size(), err != nil)
return err
}
return nil
})
if err != nil {
return err
}
return nil
}