Implement downloading videos
This commit is contained in:
62
download.go
62
download.go
@@ -1,10 +1,64 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import "log"
|
import (
|
||||||
|
"bufio"
|
||||||
|
"log"
|
||||||
|
"os/exec"
|
||||||
|
)
|
||||||
|
|
||||||
func Download(event PBEvent) (err error) {
|
func Download(event PBEvent) (err error) {
|
||||||
log.Printf("Download event: %+v\n", event)
|
cmd := exec.Command("yt-dlp",
|
||||||
// SetDownloaded(event)
|
"-v",
|
||||||
// ...do something
|
"-o", "C:/Users/Administrator/ytdlpVideos/%(title)s.%(ext)s",
|
||||||
|
"-f", "bestvideo[ext=mp4]+bestaudio[ext=m4a]",
|
||||||
|
event.Record.Link)
|
||||||
|
|
||||||
|
|
||||||
|
// Get the pipes for stdout and stderr
|
||||||
|
stdout, err := cmd.StdoutPipe()
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("Error creating StdoutPipe: %++v for cmd %++v", err, cmd)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
stderr, err := cmd.StderrPipe()
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("Error creating StderrPipe: %++v for cmd %++v", err, cmd)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = cmd.Start()
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("Error starting command: %++v with error %++v", cmd, err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
stdoutScanner := bufio.NewScanner(stdout)
|
||||||
|
go func() {
|
||||||
|
for stdoutScanner.Scan() {
|
||||||
|
log.Printf("%s: %s", event.Record.Id, stdoutScanner.Text())
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
stderrScanner := bufio.NewScanner(stderr)
|
||||||
|
go func() {
|
||||||
|
for stderrScanner.Scan() {
|
||||||
|
log.Printf("%s: %s", event.Record.Id, stderrScanner.Text())
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
err = cmd.Wait()
|
||||||
|
if err != nil {
|
||||||
|
exitError, ok := err.(*exec.ExitError)
|
||||||
|
if ok {
|
||||||
|
exitCode := exitError.ExitCode()
|
||||||
|
log.Printf("Command exited with code %d\n", exitCode)
|
||||||
|
} else {
|
||||||
|
log.Printf("Error running yt-dlp: %++v; %++v", cmd, err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Printf("Downloaded %s (%s)", event.Record.Id, event.Record.Link)
|
||||||
|
SetDownloaded(event)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
5
main.go
5
main.go
@@ -45,13 +45,10 @@ func main() {
|
|||||||
listener.Collections = []string{COLLECTION_NAME}
|
listener.Collections = []string{COLLECTION_NAME}
|
||||||
listener.initialize()
|
listener.initialize()
|
||||||
|
|
||||||
for {
|
for event := range listener.Create {
|
||||||
select {
|
|
||||||
case event := <-listener.Create:
|
|
||||||
log.Printf("Create event: %+v\n", event)
|
log.Printf("Create event: %+v\n", event)
|
||||||
go Download(event)
|
go Download(event)
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
time.Sleep(1 * time.Hour)
|
time.Sleep(1 * time.Hour)
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user