Configurable interval

.gitignore

@@ -1,3 +1,4 @@
 ytubeSpdrun
 **/*.sw*
+exec-*
 youtuber

Dockerfile

@@ -6,6 +6,8 @@ WORKDIR /main
 
 COPY . .
 
+RUN test -e /main/exec-youtuber
+
 ENV GOPATH=""
 ENV MNT="/mnt/"
 ENTRYPOINT ["/main/exec-youtuber"]

config/config.go

@@ -4,14 +4,16 @@ import (
 	"local/args"
 	"local/sandbox/contact/contact"
 	"os"
+	"time"
 )
 
 var (
-	Feed    string
-	Root    string
-	To      string
-	Subject string
-	Emailer *contact.Emailer
+	Feed     string
+	Root     string
+	To       string
+	Subject  string
+	Emailer  *contact.Emailer
+	Interval time.Duration
 )
 
 func init() {
@@ -28,6 +30,7 @@ func init() {
 
 	as.Append(args.STRING, "feed", "feed URL", "https://www.youtube.com/feeds/videos.xml?channel_id=UCwX0AEx-qIhQ9kgtlNhyIXw")
 	as.Append(args.STRING, "root", "root to save videos", "/tmp")
+	as.Append(args.DURATION, "interval", "check interval", time.Hour)
 
 	if err := as.Parse(); err != nil {
 		panic(err)

main.go

@@ -18,7 +18,7 @@ func main() {
 	if err != nil {
 		panic(err)
 	}
-	interval := time.Hour * 4
+	interval := config.Interval
 	for true {
 		err := do(client)
 		if err != nil {
@@ -45,18 +45,17 @@ func do(client *youtubedl.Client) error {
 	}
 
 	for _, result := range results {
-		target := fmt.Sprintf(
-			"s%02d%02de%02d_%s.%%(ext)s",
-			result.Date.Year()-2000,
-			result.Date.Month(),
-			result.Date.Day(),
-			strings.Join(strings.Split(result.Title, " ")[2:], "_"),
-		)
+		seasonNumber := fmt.Sprintf("%02d%02d", result.Date.Year()-2000, result.Date.Month())
+		episodeNumber := fmt.Sprintf("%02d", result.Date.Day())
+		season := fmt.Sprintf("Season_%s", seasonNumber)
+		episode := fmt.Sprintf("Episode_%s_-_%s", episodeNumber, strings.Join(strings.Split(result.Title, " ")[2:], "_"))
+		target := fmt.Sprintf("%s/%s.%%(ext)s", season, episode)
 		target = path.Join(config.Root, target)
 		if _, err := os.Stat(strings.ReplaceAll(target, "%(ext)s", "mp4")); !os.IsNotExist(err) {
 			log.Printf("already exists: %s", target)
 			continue
 		}
+		os.MkdirAll(path.Dir(target), os.ModePerm)
 		if err := client.Download(result.Link, target); err != nil {
 			errs += ", " + err.Error()
 		}

youtubedl/client.go

@@ -17,7 +17,7 @@ func New() (*Client, error) {
 func (c *Client) Download(video, local string) error {
 	errs := []error{}
 	for _, extras := range [][]string{
-		[]string{"--metadata-from-title", ".*[0-9]/[0-9][0-9] (?P<title>.+)"},
+		[]string{"--add-metadata", "--metadata-from-title", ".*[0-9]/[0-9][0-9] (?P<title>.+)"},
 		[]string{},
 	} {
 		args := append([]string{
@@ -48,7 +48,15 @@ func (c *Client) Download(video, local string) error {
 		err := fmt.Errorf("%v", errs)
 		return err
 	}
-	cmd := exec.Command("python3", "-m", "vtt_to_srt", path.Dir(local))
+	cmd := exec.Command(
+		"/bin/sh",
+		"-c",
+		fmt.Sprintf(
+			"true; python3 -m vtt_to_srt %q || python3 /usr/bin/vtt_to_srt.py %q",
+			path.Dir(local),
+			path.Dir(local),
+		),
+	)
 	out, err := cmd.CombinedOutput()
 	if err != nil {
 		err = fmt.Errorf("%v: %s", err, out)

youtubedl/install.go

@@ -7,7 +7,7 @@ import (
 )
 
 func install() error {
-	if err := installPyPip3(); err != nil {
+	if err := installPyPip3FFMPEG(); err != nil {
 		return err
 	}
 	if err := installYtdl(); err != nil {
@@ -19,30 +19,32 @@ func install() error {
 	return nil
 }
 
-func installPyPip3() error {
+func installPyPip3FFMPEG() error {
 	hasPy := exec.Command("python3", "--version")
 	hasPip := exec.Command("pip3", "--version")
+	hasFFMPEG := exec.Command("which", "ffmpeg")
 	if err := hasPy.Run(); err != nil {
 	} else if err := hasPip.Run(); err != nil {
+	} else if err := hasFFMPEG.Run(); err != nil {
 	} else {
 		return nil
 	}
 	for _, combo := range [][]string{
-		[]string{"sudo", "apt", "install", "python3", "pip3"},
-		[]string{"apt", "install", "python3", "pip3"},
-		[]string{"apk", "add", "python3", "py3-pip"},
-		[]string{"sudo", "apk", "add", "python3", "py3-pip"},
-		[]string{"/sbin/apk", "add", "python3", "py3-pip"},
-		[]string{"sudo", "/sbin/apk", "add", "python3", "py3-pip"},
+		[]string{"sudo", "apt", "install", "python3", "pip3", "ffmpeg"},
+		[]string{"apt", "install", "python3", "pip3", "ffmpeg"},
+		[]string{"apk", "add", "python3", "py3-pip", "ffmpeg"},
+		[]string{"sudo", "apk", "add", "python3", "py3-pip", "ffmpeg"},
+		[]string{"/sbin/apk", "add", "python3", "py3-pip", "ffmpeg"},
+		[]string{"sudo", "/sbin/apk", "add", "python3", "py3-pip", "ffmpeg"},
 	} {
 		cmd := exec.Command(combo[0], combo[1:]...)
 		err := cmd.Run()
 		if err == nil {
 			return nil
 		}
-		log.Println("%v: %v", err, combo)
+		log.Printf("%v: %v", err, combo)
 	}
-	return errors.New("cannot get python3 and pip3")
+	return errors.New("cannot get python3 and pip3 and ffmpeg")
 }
 
 func installYtdl() error {