|
|
|
@ -7,6 +7,7 @@ import (
|
|
|
|
|
"fmt"
|
|
|
|
|
"io"
|
|
|
|
|
"os"
|
|
|
|
|
"os/signal"
|
|
|
|
|
"path/filepath"
|
|
|
|
|
"runtime"
|
|
|
|
|
"runtime/debug"
|
|
|
|
@ -14,6 +15,7 @@ import (
|
|
|
|
|
"strings"
|
|
|
|
|
"time"
|
|
|
|
|
|
|
|
|
|
"github.com/fsnotify/fsnotify"
|
|
|
|
|
"github.com/urfave/cli/v2"
|
|
|
|
|
"go.uber.org/zap"
|
|
|
|
|
|
|
|
|
@ -52,6 +54,7 @@ func main() {
|
|
|
|
|
&cli.BoolFlag{Name: "skip-noop", Aliases: []string{"n"}, Usage: "skip noop decoder", Required: false, Value: true},
|
|
|
|
|
&cli.BoolFlag{Name: "update-metadata", Usage: "update metadata & album art from network", Required: false, Value: false},
|
|
|
|
|
&cli.BoolFlag{Name: "overwrite", Usage: "overwrite output file without asking", Required: false, Value: false},
|
|
|
|
|
&cli.BoolFlag{Name: "watch", Usage: "watch the input dir and process new files", Required: false, Value: false},
|
|
|
|
|
|
|
|
|
|
&cli.BoolFlag{Name: "supported-ext", Usage: "show supported file extensions and exit", Required: false, Value: false},
|
|
|
|
|
},
|
|
|
|
@ -66,6 +69,7 @@ func main() {
|
|
|
|
|
logger.Fatal("run app failed", zap.Error(err))
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func printSupportedExtensions() {
|
|
|
|
|
var exts []string
|
|
|
|
|
for ext := range common.DecoderRegistry {
|
|
|
|
@ -76,6 +80,7 @@ func printSupportedExtensions() {
|
|
|
|
|
fmt.Printf("%s: %d\n", ext, len(common.DecoderRegistry[ext]))
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func appMain(c *cli.Context) (err error) {
|
|
|
|
|
if c.Bool("supported-ext") {
|
|
|
|
|
printSupportedExtensions()
|
|
|
|
@ -134,7 +139,12 @@ func appMain(c *cli.Context) (err error) {
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if inputStat.IsDir() {
|
|
|
|
|
return proc.processDir(input)
|
|
|
|
|
wacthDir := c.Bool("watch")
|
|
|
|
|
if !wacthDir {
|
|
|
|
|
return proc.processDir(input)
|
|
|
|
|
} else {
|
|
|
|
|
return proc.watchDir(input)
|
|
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
|
return proc.processFile(input)
|
|
|
|
|
}
|
|
|
|
@ -150,6 +160,60 @@ type processor struct {
|
|
|
|
|
overwriteOutput bool
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (p *processor) watchDir(inputDir string) error {
|
|
|
|
|
if err := p.processDir(inputDir); err != nil {
|
|
|
|
|
return err
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
watcher, err := fsnotify.NewWatcher()
|
|
|
|
|
if err != nil {
|
|
|
|
|
return fmt.Errorf("failed to create watcher: %w", err)
|
|
|
|
|
}
|
|
|
|
|
defer watcher.Close()
|
|
|
|
|
|
|
|
|
|
go func() {
|
|
|
|
|
for {
|
|
|
|
|
select {
|
|
|
|
|
case event, ok := <-watcher.Events:
|
|
|
|
|
if !ok {
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if event.Has(fsnotify.Create) || event.Has(fsnotify.Write) {
|
|
|
|
|
// try open with exclusive mode, to avoid file is still writing
|
|
|
|
|
f, err := os.OpenFile(event.Name, os.O_RDONLY, os.ModeExclusive)
|
|
|
|
|
if err != nil {
|
|
|
|
|
logger.Debug("failed to open file exclusively", zap.String("path", event.Name), zap.Error(err))
|
|
|
|
|
time.Sleep(1 * time.Second) // wait for file writing complete
|
|
|
|
|
continue
|
|
|
|
|
}
|
|
|
|
|
_ = f.Close()
|
|
|
|
|
|
|
|
|
|
if err := p.processFile(event.Name); err != nil {
|
|
|
|
|
logger.Warn("failed to process file", zap.String("path", event.Name), zap.Error(err))
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
case err, ok := <-watcher.Errors:
|
|
|
|
|
if !ok {
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
logger.Error("file watcher got error", zap.Error(err))
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}()
|
|
|
|
|
|
|
|
|
|
err = watcher.Add(inputDir)
|
|
|
|
|
if err != nil {
|
|
|
|
|
return fmt.Errorf("failed to watch dir %s: %w", inputDir, err)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
signalCtx, stop := signal.NotifyContext(context.Background(), os.Interrupt)
|
|
|
|
|
defer stop()
|
|
|
|
|
|
|
|
|
|
<-signalCtx.Done()
|
|
|
|
|
return nil
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (p *processor) processDir(inputDir string) error {
|
|
|
|
|
items, err := os.ReadDir(inputDir)
|
|
|
|
|
if err != nil {
|
|
|
|
|