From a75d50c2a3e66bad70cff8c4f78412c6660e6495 Mon Sep 17 00:00:00 2001
From: gkiviv <georgkivivali7@gmail.com>
Date: Fri, 1 Nov 2024 15:23:43 +0200
Subject: [PATCH] improved directory logic

---
 downloader/downloader.go | 90 ++++++++++++++++++++--------------------
 main.go                  |  6 +--
 2 files changed, 48 insertions(+), 48 deletions(-)

diff --git a/downloader/downloader.go b/downloader/downloader.go
index 2ac2e14..7451b9d 100644
--- a/downloader/downloader.go
+++ b/downloader/downloader.go
@@ -6,6 +6,7 @@ import (
 	"io"
 	"net/http"
 	"os"
+	"path/filepath"
 	"regexp"
 )
 
@@ -54,15 +55,12 @@ type Source struct {
 	File string `json:"file"`
 }
 
-func DownloadSingle(url string, filename string, subtitleLang string) {
+func DownloadSingle(url string, subtitleLang string, parentDirectory string) {
 	id := ExtractContentId(url)
 	data := GetContentPageData(id)
 	downloadUrl := GetDownloadUrl(data)
-	dirCreated := false
 
-	if filename == "" {
-		filename = fmt.Sprintf("%s_%d_%d", data.Data.MainContent.Title, data.Data.MainContent.Season, data.Data.MainContent.Episode)
-	}
+	name := fmt.Sprintf("%s_%d_%d", data.Data.MainContent.Title, data.Data.MainContent.Season, data.Data.MainContent.Episode)
 
 	if subtitleLang != "" {
 		subtitles := data.Data.MainContent.Medias[0].Subtitles
@@ -76,29 +74,31 @@ func DownloadSingle(url string, filename string, subtitleLang string) {
 
 		for _, subtitle := range subtitles {
 			if subtitle.SrcLang == subtitleLang {
-				_ = os.Mkdir(filename, os.ModePerm) // dont care if directory fails to create
-				subitleFileName := fmt.Sprintf("%s/%s_%s", filename, filename, subtitle.FileName)
-				downloadFile(subtitle.Src, subitleFileName)
+				parentDirectory = filepath.Join(parentDirectory, name)
+				_ = os.Mkdir(parentDirectory, os.ModePerm) // dont care if directory fails to create
+
+				subitleFileName := fmt.Sprintf("%s_%s", name, subtitle.FileName)
+				subitleFilePath := filepath.Join(parentDirectory, subitleFileName)
+
+				downloadFile(subtitle.Src, subitleFilePath)
 				fmt.Println("Subtitles downloaded successfully")
-				dirCreated = true
 				break
 			}
 		}
 	}
-	filepath := fmt.Sprintf("%s.mp4", filename)
 
-	if dirCreated {
-		filepath = fmt.Sprintf("%s/%s.mp4", filename, filename)
-	}
+	filenameExt := fmt.Sprintf("%s.mp4", name)
 
-	fmt.Printf("Downloading %s to %s\n", url, filename)
-	downloadFile(downloadUrl, filepath)
-	fmt.Printf("Finished Downloading %s to %s\n", url, filename)
+	path := filepath.Join(parentDirectory, filenameExt)
+
+	fmt.Printf("Downloading %s to %s\n", url, path)
+	downloadFile(downloadUrl, path)
+	fmt.Println("Finished downloading")
+	fmt.Println()
 }
 
 func DownloadSeason(url string, seasonName string, subtitleLang string) {
 	id := ExtractContentId(url)
-	// fmt.Println(id)
 	data := GetContentPageData(id)
 
 	title := data.Data.MainContent.Title
@@ -110,39 +110,14 @@ func DownloadSeason(url string, seasonName string, subtitleLang string) {
 	for _, season := range seasonList.Seasons {
 		if season.Name == seasonName {
 			for _, seasonContent := range season.Contents {
-				_ = os.Mkdir(title, os.ModePerm) // dont care if directory fails to create
-				// fileName := fmt.Sprintf("%s/episood_%s", title, strconv.Itoa(i+1))
-				DownloadSingle(seasonContent.Url, "", subtitleLang)
+				parentDirName := title
+				_ = os.Mkdir(parentDirName, os.ModePerm) // dont care if directory fails to create
+				DownloadSingle(seasonContent.Url, subtitleLang, parentDirName)
 			}
 		}
 	}
 }
 
-func downloadFile(url string, filepath string) {
-	out, err := os.Create(filepath)
-	if err != nil {
-		panic(err)
-	}
-	defer out.Close()
-
-	resp, err := http.Get(url)
-
-	if err != nil {
-		panic(err)
-	}
-
-	defer resp.Body.Close()
-	if resp.StatusCode != http.StatusOK {
-		panic("Bad status")
-	}
-
-	_, err = io.Copy(out, resp.Body)
-
-	if err != nil {
-		panic(err)
-	}
-}
-
 func GetContentPageData(contentId string) *ContentPageData {
 	var data ContentPageData
 
@@ -176,3 +151,28 @@ func ExtractContentId(url string) string {
 func GetDownloadUrl(data *ContentPageData) string {
 	return fmt.Sprintf("https:%s", data.Data.MainContent.Medias[0].Src.File)
 }
+
+func downloadFile(url string, filepath string) {
+	out, err := os.Create(filepath)
+	if err != nil {
+		panic(err)
+	}
+	defer out.Close()
+
+	resp, err := http.Get(url)
+
+	if err != nil {
+		panic(err)
+	}
+
+	defer resp.Body.Close()
+	if resp.StatusCode != http.StatusOK {
+		panic("Bad status")
+	}
+
+	_, err = io.Copy(out, resp.Body)
+
+	if err != nil {
+		panic(err)
+	}
+}
diff --git a/main.go b/main.go
index 73a3ed0..74bc401 100644
--- a/main.go
+++ b/main.go
@@ -6,9 +6,9 @@ import (
 
 func main() {
     // downloader.Download("1038278")
-    // downloader.Download("https://jupiter.err.ee/1038278/aktuaalne-kaamera", "")
-    // downloader.DownloadSingle("https://jupiter.err.ee/1609406782/babulon-berliin", "", "ET")
-    downloader.DownloadSeason("https://jupiter.err.ee/1235599/babulon-berliin", "4", "ET") 
+    downloader.DownloadSingle("https://jupiter.err.ee/1038278/aktuaalne-kaamera", "", "")
+    // downloader.DownloadSingle("https://jupiter.err.ee/1609406782/babulon-berliin", "ET", "")
+    // downloader.DownloadSeason("https://jupiter.err.ee/1235599/babulon-berliin", "4", "ET") 
     // fs := http.FileServer(http.Dir("static/"))
     //
     // http.Handle("/", fs)