package dl import ( "bufio" "log" "net/url" "os/exec" "path/filepath" "strconv" "strings" "sync" "git.meatbag.se/varl/subsyt/internal/config" ) type Download struct { Url string OutDir string Name string DryRun bool } func Get(d Download, p config.Provider) { output := filepath.Join("%(upload_date>%Y)s/%(upload_date)s-%(title)s-%(id)s.%(ext)s") archive := filepath.Join(d.OutDir, d.Name, "archive.txt") outdir := filepath.Join(d.OutDir, d.Name) curl := strings.TrimPrefix(d.Url, "/feed/") furl, err := url.JoinPath(p.Url, curl, "videos") if err != nil { panic(err) } fullUrl, err := url.Parse(furl) if err != nil { panic(err) } var simulate string if d.DryRun == true { simulate = "--simulate" log.Printf("/!\\ DRY RUN ENABLED /!\\") } else { simulate = "--no-simulate" } cmd := exec.Command(p.Cmd, fullUrl.String(), simulate, "--no-progress", "--sleep-interval", strconv.Itoa(p.Throttle), "--sleep-subtitles", strconv.Itoa(p.Throttle), "--sleep-requests", strconv.Itoa(p.Throttle), "--prefer-free-formats", "--write-subs", "--no-write-automatic-subs", "--sub-langs", "en", "--paths", outdir, "--output", output, "--download-archive", archive, "--break-on-existing", "--playlist-items", p.Range, "--restrict-filenames", "--embed-metadata", "--cookies", p.Cookies, ) stdout, err := cmd.StdoutPipe() if err != nil { log.Fatal(err) } stderr, err := cmd.StderrPipe() if err != nil { log.Fatal(err) } log.Printf("[%s] running yt-dlp for: %s", d.Name, d.Url) var wg sync.WaitGroup wg.Add(2) go func() { defer wg.Done() scanner := bufio.NewScanner(stdout) for scanner.Scan() { log.Printf("[%s] %s\n", d.Name, scanner.Text()) } }() go func() { defer wg.Done() scanner := bufio.NewScanner(stderr) for scanner.Scan() { log.Printf("[%s] %s\n", d.Name, scanner.Text()) } }() err = cmd.Start() if err != nil { log.Fatal(err) } wg.Wait() err = cmd.Wait() if err != nil { log.Printf("Error: %s\n", err) } }