youtube-dl to yt-dlp

config/config.go

@@ -46,4 +46,5 @@ func init() {
 		POP3:     as.Get("pop3").GetString(),
 		Password: as.Get("password").GetString(),
 	}
+	Interval = as.GetDuration("interval")
 }

youtubedl/client.go

@@ -3,6 +3,7 @@ package youtubedl
 import (
 	"fmt"
 	"log"
+	"os"
 	"os/exec"
 	"path"
 )
@@ -15,13 +16,17 @@ func New() (*Client, error) {
 }
 
 func (c *Client) Download(video, local string) error {
+	if _, ok := os.LookupEnv("TEST"); ok {
+		log.Println("youtube-dl download", video, local)
+		return nil
+	}
 	errs := []error{}
 	for _, extras := range [][]string{
 		[]string{"--add-metadata", "--metadata-from-title", ".*[0-9]/[0-9][0-9] (?P<title>.+)"},
 		[]string{},
 	} {
 		args := append([]string{
-			"youtube-dl",
+			"/tmp/yt-dlp",
 			"-f",
 			"22",
 			"-o",
@@ -30,6 +35,7 @@ func (c *Client) Download(video, local string) error {
 			"--write-auto-sub",
 			"--sub-format",
 			"srt",
+			"--no-check-certificate",
 		}, extras...)
 		args = append(args, video)
 		cmd := exec.Command(args[0], args[1:]...)
@@ -52,7 +58,8 @@ func (c *Client) Download(video, local string) error {
 		"/bin/sh",
 		"-c",
 		fmt.Sprintf(
-			"true; python3 -m vtt_to_srt %q || python3 /usr/bin/vtt_to_srt.py %q",
+			"true; python3 -m vtt_to_srt %q || python3 /usr/bin/vtt_to_srt.py %q || vtt_to_srt %q",
+			path.Dir(local),
 			path.Dir(local),
 			path.Dir(local),
 		),

youtubedl/install.go

@@ -2,11 +2,16 @@ package youtubedl
 
 import (
 	"errors"
+	"fmt"
 	"log"
+	"os"
 	"os/exec"
 )
 
 func install() error {
+	if _, ok := os.LookupEnv("TEST"); ok {
+		return nil
+	}
 	if err := installPyPip3FFMPEG(); err != nil {
 		return err
 	}
@@ -48,16 +53,30 @@ func installPyPip3FFMPEG() error {
 }
 
 func installYtdl() error {
-	cmd := exec.Command("youtube-dl", "--version")
+	cmd := exec.Command("/tmp/yt-dlp", "--version")
 	if err := cmd.Run(); err == nil {
 		return err
 	}
-	cmd = exec.Command("sudo", "pip3", "install", "youtube-dl")
+	os.Remove("/tmp/yt-dlp")
+	cmd = exec.Command(
+		"wget",
+		"https://github.com/yt-dlp/yt-dlp/releases/latest/download/yt-dlp",
+		"--no-check-certificate",
+		"-O",
+		"/tmp/yt-dlp",
+	)
 	if err := cmd.Run(); err != nil {
-		cmd = exec.Command("pip3", "install", "youtube-dl")
-		if err := cmd.Run(); err != nil {
-			return err
-		}
+		o, _ := cmd.CombinedOutput()
+		return fmt.Errorf("failed wget yt-dlp: %v: %s", err, o)
+	}
+	cmd = exec.Command(
+		"chmod",
+		"a+rx",
+		"/tmp/yt-dlp",
+	)
+	if err := cmd.Run(); err != nil {
+		o, _ := cmd.CombinedOutput()
+		return fmt.Errorf("failed chmod yt-dlp: %v: %s", err, o)
 	}
 	return nil
 }