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 }