From 78455cceaa0d07a1c553fa5d7a54619462a0a77b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?David=20Majdand=C5=BEi=C4=87?=
 <david.majdandzic@hotmail.com>
Date: Mon, 17 Jun 2024 14:20:45 +0200
Subject: [PATCH] Implement downloading videos

---
 download.go | 62 +++++++++++++++++++++++++++++++++++++++++++++++++----
 main.go     |  9 +++-----
 2 files changed, 61 insertions(+), 10 deletions(-)

diff --git a/download.go b/download.go
index c058895..e82d804 100644
--- a/download.go
+++ b/download.go
@@ -1,10 +1,64 @@
 package main
 
-import "log"
+import (
+	"bufio"
+	"log"
+	"os/exec"
+)
 
 func Download(event PBEvent) (err error) {
-	log.Printf("Download event: %+v\n", event)
-	// SetDownloaded(event)
-	// ...do something
+	cmd := exec.Command("yt-dlp",
+		"-v",
+		"-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
 }
diff --git a/main.go b/main.go
index 22e6f49..bc0c839 100644
--- a/main.go
+++ b/main.go
@@ -45,12 +45,9 @@ func main() {
 	listener.Collections = []string{COLLECTION_NAME}
 	listener.initialize()
 
-	for {
-		select {
-		case event := <-listener.Create:
-			log.Printf("Create event: %+v\n", event)
-			go Download(event)
-		}
+	for event := range listener.Create {
+		log.Printf("Create event: %+v\n", event)
+		go Download(event)
 	}
 
 	time.Sleep(1 * time.Hour)