bel
/
show-ingestion
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

outd is a template

main
bel 2025-04-05 00:51:30 -06:00
parent d57206357e
commit 57d9b74c31
2 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
main.go
View File

@ -1,6 +1,7 @@
package main package main
import ( import (
"bytes"
"context" "context"
"encoding/json" "encoding/json"
"flag" "flag"
@ -13,6 +14,7 @@ import (
"regexp" "regexp"
"strings" "strings"
"syscall" "syscall"
"text/template"
yaml "gopkg.in/yaml.v3" yaml "gopkg.in/yaml.v3"
) )
@ -95,7 +97,7 @@ func Main(ctx context.Context) error {
flags := flag.NewFlagSet(os.Args[0], flag.ContinueOnError) flags := flag.NewFlagSet(os.Args[0], flag.ContinueOnError)
overridesS := flags.String("c", `{"title":"","season":"","episode":""}`, "overrides") overridesS := flags.String("c", `{"title":"","season":"","episode":""}`, "overrides")
ind := flags.String("i", "/dev/null", "in dir") ind := flags.String("i", "/dev/null", "in dir")
outd := flags.String("o", "/dev/null", "out dir") outd := flags.String("o", "/dev/null", "out dir template accepts overrides format title case")
dry := flags.Bool("d", true, "dry run") dry := flags.Bool("d", true, "dry run")
if err := flags.Parse(os.Args[1:]); err != nil { if err := flags.Parse(os.Args[1:]); err != nil {
panic(err) panic(err)
@ -182,7 +184,16 @@ func one(ctx context.Context, outd, inf string, patterns []string, overrides Fie
} }
func foundOne(ctx context.Context, outd, inf string, fields Fields, mvNLn MvNLn) error { func foundOne(ctx context.Context, outd, inf string, fields Fields, mvNLn MvNLn) error {
outf := path.Join(outd, fmt.Sprintf("%s_S%sE%s%s", fields.Title, fields.Season, fields.Episode, path.Ext(inf))) tmpl, err := template.New(inf).Parse(outd)
if err != nil {
return err
}
buff := bytes.NewBuffer(nil)
if err := tmpl.Execute(buff, fields); err != nil {
return err
}
outf := path.Join(string(buff.Bytes()), fmt.Sprintf("%s_S%sE%s%s", fields.Title, fields.Season, fields.Episode, path.Ext(inf)))
return mvNLn(outf, inf) return mvNLn(outf, inf)
} }

4
main_test.go
View File

@ -170,7 +170,7 @@ func TestRecursive(t *testing.T) {
// parse files // parse files
write("./showB/.show-ingestion.yaml", `{ write("./showB/.show-ingestion.yaml", `{
"o": "`+outd+`/B", "o": "`+outd+`/B_{{.Title}}_{{.Season}}_{{.Episode}}",
"p": [] "p": []
}`) }`)
write("./showB/title S01E02.b") write("./showB/title S01E02.b")
@ -198,7 +198,7 @@ func TestRecursive(t *testing.T) {
} }
exists(t, path.Join(outd, "A", "A_SAEA.a")) exists(t, path.Join(outd, "A", "A_SAEA.a"))
exists(t, path.Join(outd, "B", "title_S01E02.b")) exists(t, path.Join(outd, "B_title_01_02", "title_S01E02.b"))
exists(t, path.Join(outd, "C", "t_SsEe.c")) exists(t, path.Join(outd, "C", "t_SsEe.c"))
notExists(t, path.Join(outd, "D", "title_S02E04.d")) notExists(t, path.Join(outd, "D", "title_S02E04.d"))
notExists(t, path.Join(outd, "title_S03E06.e")) notExists(t, path.Join(outd, "title_S03E06.e"))