From b2a34ab235b7ed5e82df0bb034150e1b7b1ee34c Mon Sep 17 00:00:00 2001
From: Kevin Ruffin <kevin.github.com>
Date: Fri, 18 Feb 2022 17:50:08 -0500
Subject: [PATCH] Adds in logging messages to the generated script. Has it
 store files in OS and language folders.

---
 main.go | 75 ++++++++++++++++++++++++++++++---------------------------
 1 file changed, 40 insertions(+), 35 deletions(-)

diff --git a/main.go b/main.go
index 55ad1d9..a8ade4b 100644
--- a/main.go
+++ b/main.go
@@ -178,6 +178,8 @@ type DownloadLinks struct {
 	InstallerName string
 	Filename string
 	Md5 string
+	Os string
+	Language string
 }
 
 func (dl *DownloadLinks) String() string {
@@ -207,6 +209,8 @@ func getDownloadLinks(p *Product) []*DownloadLinks {
 			}
 
 			dl.InstallerName = sanitizeName(i.Name)
+			dl.Os = i.Os
+			dl.Language = i.Language
 			links = append(links, dl)
 		}
 	}
@@ -218,39 +222,14 @@ func sanitizeName(name string) string {
 	// s := strings.Replace(name, " ", "_", -1)
 	// s = strings.Replace(s, "'", "", -1)
 	// s = strings.Replace(s, "™", "", -1)
-	re := regexp.MustCompile("[^0-1a-zA-Z.-_]+")
+	re := regexp.MustCompile("[^0-9a-zA-Z_.-]+")
 	s := name
-	s = re.ReplaceAllString(s, "_")
+	s = re.ReplaceAllString(s, "")
 	return s
 }
 
-// func writeChecksumIfNeeded(dl *DownloadLinks, folder string) {
-// 	// TODO: reuse an existing checksum file somehow
-// 	res, err := client.Get(dl.Checksum)
-// 	if err != nil {
-// 		log.Fatalf("Failed to get file checksum info (%s): %v", dl.Checksum, err)
-// 	}
-// 	defer res.Body.Close()
-// 	body, err := ioutil.ReadAll(res.Body)
-// 	if err != nil {
-// 		log.Fatalf("Failed to parse file checksum body (%s): %v", dl.Checksum, err)
-// 	}
 
-// 	re := regexp.MustCompile(`<file name="([^"]+)".*md5="([^"]+)`)
-// 	m := re.FindSubmatch(body)
-// 	if m != nil {
-// 		dl.Filename = sanitizeName(string(m[1]))
-// 		dl.Md5 = string(m[2])
-
-// 		// err = ioutil.WriteFile(folder + "/" + sanitizeName(dl.InstallerName) + "/" + dl.Filename + ".md5", m[2], 0664)
-// 		// if err != nil {
-// 		// 	log.Fatalf("Failed to write the checksum file: %v", err)
-// 		// }
-// 	}
-// }
-
-
-func updateBashScriptProductFolder(p *Product, folder string, scriptName string) {
+func updateBashScriptProductFolder(p *Product, folder string, scriptName string, productNumber int, totalProducts int) {
 
 	f, err := os.OpenFile("./" + scriptName, os.O_APPEND|os.O_WRONLY, 0664)
 	if err != nil {
@@ -258,14 +237,39 @@ func updateBashScriptProductFolder(p *Product, folder string, scriptName string)
 	}
 	defer f.Close()
 
-	_, err = f.WriteString("mkdir -p " + folder + "/" + sanitizeName(p.Title) + "\n")
+	// Get a list of the OS's
+	var osList []string
+	for _, i := range p.Downloads.Installers {
+		exists := false
+		for _, o := range osList {
+			if o == sanitizeName(i.Os) + "/" + sanitizeName(i.Language) {
+				exists = true
+				break
+			}
+		}
+		if !exists {
+			osList = append(osList, sanitizeName(i.Os) + "/" + sanitizeName(i.Language))
+		}
+	}
+
+	// Print out the product
+	_, err = f.WriteString("echo \"[" + strconv.Itoa(productNumber) + " of " + strconv.Itoa(totalProducts) + "]Downloading files for: " + sanitizeName(p.Title) + "\"\n")
 	if err != nil {
 		log.Fatalf("Failed to update the script: %v", err)
 	}
+
+	// Create a folder for each os installers
+	for _, o := range osList {
+		_, err = f.WriteString("mkdir -p " + folder + "/" + sanitizeName(p.Title) + "/" + o + "\n")
+		if err != nil {
+			log.Fatalf("Failed to update the script: %v", err)
+		}
+	}
+	
 }
 
 
-func writeBashScript(dl *DownloadLinks, folder string, scriptName string) {
+func writeBashScript(dl *DownloadLinks, folder string, scriptName string, productNumber int, totalProducts int, fileNumber int, totalFiles int) {
 	// Find out file name and size; have to follow redirects
 	res, err := client.Head(dl.Downlink)
 	if err != nil {
@@ -278,8 +282,9 @@ func writeBashScript(dl *DownloadLinks, folder string, scriptName string) {
 	dl.Filename = sanitizeName(pp[len(pp) - 1])
 
 	// Add the lines to the script
-	line := "wget --no-clobber --continue --quiet --show-progress -O \"" + folder + "/" + sanitizeName(dl.InstallerName) + "/" + dl.Filename + ".xml\" \"" + dl.Checksum + "\"\n"
-	line += "wget --no-clobber --continue --quiet --show-progress -O \"" + folder + "/" + sanitizeName(dl.InstallerName) + "/" + dl.Filename + "\" \"" + dl.Downlink + "\"\n"
+	line := "echo \"[" + strconv.Itoa(productNumber) + " of " + strconv.Itoa(totalProducts) + "]{" + strconv.Itoa(fileNumber) + "/" + strconv.Itoa(totalFiles) + "}Getting installers for (" + dl.InstallerName + ")" + "\"\n"
+	line += "wget --no-clobber --continue --quiet --show-progress -O \"" + folder + "/" + sanitizeName(dl.InstallerName) + "/" + sanitizeName(dl.Os) + "/" + sanitizeName(dl.Language) + "/" + dl.Filename + ".xml\" \"" + dl.Checksum + "\"\n"
+	line += "wget --no-clobber --continue --quiet --show-progress -O \"" + folder + "/" + sanitizeName(dl.InstallerName) + "/" + sanitizeName(dl.Os) + "/" + sanitizeName(dl.Language) + "/" +  dl.Filename + "\" \"" + dl.Downlink + "\"\n"
 
 	f, err := os.OpenFile("./" + scriptName, os.O_APPEND|os.O_WRONLY, 0664)
 	if err != nil {
@@ -348,11 +353,11 @@ func main() {
 		log.Printf("Inspecting product: %d of %d - %d", index, len(gl.Owned), id)
 		p := getProduct(id)
 		dl := getDownloadLinks(p)
-		updateBashScriptProductFolder(p, productsFolder, scriptName)
+		updateBashScriptProductFolder(p, productsFolder, scriptName, index+1, len(gl.Owned))
 
-		for _, d := range dl {
+		for idx, d := range dl {
 			//writeChecksumIfNeeded(d, productsFolder + "/" + sanitizeName(p.Title))
-			writeBashScript(d, productsFolder, scriptName)
+			writeBashScript(d, productsFolder, scriptName, index+1, len(gl.Owned), idx+1, len(dl))
 		}
 	}