files update, added torretbay

files.go

@@ -6,7 +6,9 @@ import (
 	"log"
 	"net/http"
 	"net/url"
+	"regexp"
 	"sort"
+	"strconv"
 	"strings"
 	"time"
 )
@@ -31,7 +33,7 @@ var (
 func initializeTorrentSites() {
 	torrentGalaxy = NewTorrentGalaxy()
 	// nyaa = NewNyaa()
-	// thePirateBay = NewThePirateBay()
+	thePirateBay = NewThePirateBay()
 	// rutor = NewRutor()
 }
 
@@ -136,7 +138,7 @@ func fetchAndCacheFileResults(query, safe, lang string, page int) []TorrentResul
 			allErrors = false
 		}
 		for _, r := range res {
-			r.Magnet = url.QueryEscape(removeMagnetLink(r.Magnet)) // Remove "magnet:" and encode url
+			r.Magnet = removeMagnetLink(r.Magnet) // Remove "magnet:", prehaps usless now?
 			results = append(results, r)
 		}
 	}
@@ -156,7 +158,7 @@ func fetchAndCacheFileResults(query, safe, lang string, page int) []TorrentResul
 
 func removeMagnetLink(magnet string) string {
 	// Remove the magnet: prefix unconditionally
-	return strings.TrimPrefix(magnet, "magnet:?")
+	return strings.TrimPrefix(magnet, "magnet:")
 }
 
 func subtract(a, b int) int {
@@ -166,3 +168,89 @@ func subtract(a, b int) int {
 func add(a, b int) int {
 	return a + b
 }
+
+func parseInt(s string) int {
+	i, err := strconv.Atoi(s)
+	if err != nil {
+		return 0
+	}
+	return i
+}
+
+func parseSize(sizeStr string) int64 {
+	sizeStr = strings.TrimSpace(sizeStr)
+	if sizeStr == "" {
+		return 0
+	}
+
+	// Use regex to extract numeric value and unit separately
+	re := regexp.MustCompile(`(?i)([\d.]+)\s*([KMGT]?B)`)
+	matches := re.FindStringSubmatch(sizeStr)
+	if len(matches) < 3 {
+		log.Printf("Error parsing size: invalid format %s", sizeStr)
+		return 0
+	}
+
+	sizeStr = matches[1]
+	unit := strings.ToUpper(matches[2])
+
+	var multiplier int64 = 1
+	switch unit {
+	case "KB":
+		multiplier = 1024
+	case "MB":
+		multiplier = 1024 * 1024
+	case "GB":
+		multiplier = 1024 * 1024 * 1024
+	case "TB":
+		multiplier = 1024 * 1024 * 1024 * 1024
+	default:
+		log.Printf("Unknown unit: %s", unit)
+		return 0
+	}
+
+	size, err := strconv.ParseFloat(sizeStr, 64)
+	if err != nil {
+		log.Printf("Error parsing size: %v", err)
+		return 0
+	}
+	return int64(size * float64(multiplier))
+}
+
+// apparently this is needed so it can announce that magnet link is being used and people start seeding it, but I dont like the fact that I add trackers purposefully
+func applyTrackers(magnetLink string) string {
+	if magnetLink == "" {
+		return ""
+	}
+	trackers := []string{
+		"udp://tracker.openbittorrent.com:80/announce",
+		"udp://tracker.opentrackr.org:1337/announce",
+		"udp://tracker.coppersurfer.tk:6969/announce",
+		"udp://tracker.leechers-paradise.org:6969/announce",
+	}
+	for _, tracker := range trackers {
+		magnetLink += "&tr=" + url.QueryEscape(tracker)
+	}
+	return magnetLink
+}
+
+func formatSize(size int64) string {
+	if size >= 1024*1024*1024*1024 {
+		return fmt.Sprintf("%.2f TB", float64(size)/(1024*1024*1024*1024))
+	} else if size >= 1024*1024*1024 {
+		return fmt.Sprintf("%.2f GB", float64(size)/(1024*1024*1024))
+	} else if size >= 1024*1024 {
+		return fmt.Sprintf("%.2f MB", float64(size)/(1024*1024))
+	} else if size >= 1024 {
+		return fmt.Sprintf("%.2f KB", float64(size)/1024)
+	}
+	return fmt.Sprintf("%d B", size)
+}
+
+func sanitizeFileName(name string) string {
+	// Replace spaces with dashes
+	sanitized := regexp.MustCompile(`\s+`).ReplaceAllString(name, "-")
+	// Remove any characters that are not alphanumeric, dashes, or parentheses
+	sanitized = regexp.MustCompile(`[^a-zA-Z0-9\-\(\)]`).ReplaceAllString(sanitized, "")
+	return sanitized
+}