diff --git a/main.go b/main.go index 8798da3..da43d33 100644 --- a/main.go +++ b/main.go @@ -3,6 +3,7 @@ package main import ( "context" "encoding/json" + "flag" "fmt" "os" "os/signal" @@ -24,13 +25,21 @@ func main() { ctx, can := signal.NotifyContext(context.Background(), syscall.SIGINT) defer can() + flags := flag.NewFlagSet(os.Args[0], flag.ContinueOnError) + overridesS := flags.String("o", `{"title":"","season":"","episode":""}`, "overrides") + ind := flags.String("i", "/dev/null", "in dir") + outd := flags.String("o", "/dev/null", "out dir") + if err := flags.Parse(os.Args[1:]); err != nil { + panic(err) + } + var overrides Fields - json.Unmarshal([]byte(os.Args[3]), &overrides) + json.Unmarshal([]byte(*overridesS), &overrides) if err := Run(ctx, - os.Args[1], - os.Args[2], - os.Args[4:], + *outd, + *ind, + flags.Args(), overrides, RealMvNLn, ); err != nil {