refactor: extract download to package dl

This commit is contained in:
Viktor Varland 2025-03-27 19:41:58 +01:00
parent 3fa382a41f
commit 69ae19e512
Signed by: varl
GPG key ID: 7459F0B410115EE8
4 changed files with 101 additions and 155 deletions

3
.gitignore vendored Normal file
View file

@ -0,0 +1,3 @@
.direnv
config.toml
subs-opml.xml

84
internal/dl/dl.go Normal file
View file

@ -0,0 +1,84 @@
package dl
import (
"bufio"
"log"
"net/url"
"os/exec"
"path/filepath"
"sync"
)
type Download struct {
Url *url.URL
OutDir string
AfterDate string
Name string
}
func Get(d Download) {
output := filepath.Join(d.Name, "%(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)
cmd := exec.Command("yt-dlp",
d.Url.String(),
"--no-simulate",
"--no-progress",
"--sleep-interval", "1",
"--sleep-subtitles", "3",
"--sleep-requests", "1",
"--prefer-free-formats",
"--write-subs",
"--no-write-automatic-subs",
"--sub-langs", "en",
"--dateafter", d.AfterDate,
"--paths", outdir,
"--output", output,
"--download-archive", archive,
"--break-on-existing",
"--restrict-filenames",
)
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)
}
}

View file

@ -1,59 +0,0 @@
package dl
import (
"bufio"
"log"
"os/exec"
"sync"
)
func List(channelId string) {
cmd := exec.Command("yt-dlp",
"--skip-download",
"--force-write-archive",
"--download-archive", "test.txt",
channelId,
)
stdout, err := cmd.StdoutPipe()
if err != nil {
log.Fatal(err)
}
stderr, err := cmd.StderrPipe()
if err != nil {
log.Fatal(err)
}
log.Printf("Running yt-dlp for: %s", d.Url)
var wg sync.WaitGroup
wg.Add(2)
go func() {
defer wg.Done()
scanner := bufio.NewScanner(stdout)
for scanner.Scan() {
log.Printf("%s\n", scanner.Text())
}
}()
go func() {
defer wg.Done()
scanner := bufio.NewScanner(stderr)
for scanner.Scan() {
log.Printf("%s\n", 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)
}
}

110
main.go
View file

@ -1,123 +1,37 @@
package main package main
import ( import (
"bufio"
"fmt" "fmt"
"log"
"net/url" "net/url"
"os" "os"
"os/exec"
"path/filepath"
"strings" "strings"
"sync"
"git.meatbag.se/varl/subsyt/internal/dl"
"git.meatbag.se/varl/subsyt/internal/opml" "git.meatbag.se/varl/subsyt/internal/opml"
toml "github.com/pelletier/go-toml/v2" toml "github.com/pelletier/go-toml/v2"
) )
func check(e error) {
if e != nil {
panic(e)
}
}
type Provider struct { type Provider struct {
Url string Url string
} }
type Channel struct {
Name string
Id string
Provider string
}
type Config struct { type Config struct {
Out_dir string Out_dir string
Provider map[string]Provider Provider map[string]Provider
} }
type Download struct {
Url *url.URL
OutDir string
AfterDate string
Name string
}
func dl(d Download) {
output := filepath.Join(d.OutDir, d.Name, "%(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)
cmd := exec.Command("yt-dlp",
d.Url.String(),
"--no-simulate",
"--no-progress",
"--sleep-interval", "1",
"--sleep-subtitles", "1",
"--sleep-requests", "0",
"--prefer-free-formats",
"--write-subs",
"--no-write-automatic-subs",
"--sub-langs", "en",
"--dateafter", d.AfterDate,
"--paths", outdir,
"--output", output,
"--download-archive", archive,
"--break-on-existing",
)
stdout, err := cmd.StdoutPipe()
if err != nil {
log.Fatal(err)
}
stderr, err := cmd.StderrPipe()
if err != nil {
log.Fatal(err)
}
log.Printf("Running yt-dlp for: %s", 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)
}
}
func main() { func main() {
data, err := os.ReadFile("./config.toml") data, err := os.ReadFile("./config.toml")
check(err) if err != nil {
panic(err)
}
cfg := Config{} cfg := Config{}
err = toml.Unmarshal(data, &cfg) err = toml.Unmarshal(data, &cfg)
check(err) if err != nil {
panic(err)
}
pUrl := cfg.Provider["youtube"].Url pUrl := cfg.Provider["youtube"].Url
@ -133,13 +47,17 @@ func main() {
for _, outline := range outlines.Outlines { for _, outline := range outlines.Outlines {
curl := strings.TrimPrefix(outline.XmlUrl, "/feed/") curl := strings.TrimPrefix(outline.XmlUrl, "/feed/")
furl, err := url.JoinPath(pUrl, curl) furl, err := url.JoinPath(pUrl, curl)
check(err) if err != nil {
panic(err)
}
fmt.Printf("%s - %s\n", outline.Text, furl) fmt.Printf("%s - %s\n", outline.Text, furl)
fullUrl, err := url.Parse(furl) fullUrl, err := url.Parse(furl)
check(err) if err != nil {
panic(err)
}
dl(Download{ dl.Get(dl.Download{
Name: outline.Title, Name: outline.Title,
Url: fullUrl, Url: fullUrl,
OutDir: cfg.Out_dir, OutDir: cfg.Out_dir,