fixed some bugs and more

Author: tester2024

cmd/favicon.ico

@@ -5,6 +5,8 @@ import (
 	"log"
 	"net"
 	"net/http"
+
+	_ "embed"
 	"os"
 
 	"github.com/SyNdicateFoundation/SteamDownloaderAPI/internal/handler"
@@ -29,6 +31,9 @@ func init() {
 	flag.Parse()
 }
 
+//go:embed favicon.ico
+var favicon []byte
+
 func main() {
 	s, err := steamcmd.New(steamCmdPath, steamUser, steamPassword)
 	if err != nil {
@@ -45,14 +50,14 @@ func main() {
 		}
 	}
 
-	router := gin.Default()
-
 	gin.SetMode(gin.ReleaseMode)
 
 	if debugMode {
 		gin.SetMode(gin.DebugMode)
 	}
 
+	router := gin.Default()
+
 	h := handler.New(s)
 	defer h.Cleanup()
 
@@ -65,8 +70,13 @@ func main() {
 
 	router.Any("/workshop/*path", h.SteamProxyHandler)
 	router.Any("/app/*path", h.SteamProxyHandler)
+	router.Any("/public/*path", h.SteamProxyHandler)
 	router.Any("/sharedfiles/*path", h.SteamProxyHandler)
 
+	router.GET("favicon.ico", func(c *gin.Context) {
+		c.Data(http.StatusOK, "image/x-icon", favicon)
+	})
+
 	unsupportedRoutes := []string{
 		"/login/home/", "/market/", "/discussions/", "/my/",
 		"/id/", "/account/", "/profiles/",

cmd/main.go

@@ -53,22 +53,33 @@ func (h *SteamDownloaderAPI) SteamProxyHandler(c *gin.Context) {
 	}
 
 	proxy.ModifyResponse = func(res *http.Response) error {
-		if !strings.Contains(res.Header.Get("Content-Type"), "text/html") {
-			return nil
-		}
-
 		body, err := io.ReadAll(res.Body)
 		if err != nil {
 			return fmt.Errorf("failed to read response body: %w", err)
 		}
 
 		defer res.Body.Close()
 
+		if !strings.Contains(res.Header.Get("Content-Type"), "text/html") {
+			body = bytes.ReplaceAll(body,
+				[]byte("https://steamcommunity.com/workshop/ajaxfindworkshops"),
+				[]byte("/workshop/ajaxfindworkshops"),
+			)
+
+			res.Body = io.NopCloser(bytes.NewReader(body))
+			return nil
+		}
+
 		doc, err := goquery.NewDocumentFromReader(bytes.NewReader(body))
 		if err != nil {
 			return fmt.Errorf("failed to parse HTML: %w", err)
 		}
 
+		titleElm := doc.Find("title")
+		title := titleElm.Text()
+		title = strings.Replace(title, "Steam Community", "SteamDownloaderAPI", 1)
+		titleElm.SetText(title)
+
 		replacedCollection := true
 
 		doc.Find("a[onclick*='SubscribeItem']").Each(func(i int, s *goquery.Selection) {
@@ -80,6 +91,10 @@ func (h *SteamDownloaderAPI) SteamProxyHandler(c *gin.Context) {
 			}
 		})
 
+		doc.Find(`span[class="valve_links"]`).Each(func(i int, s *goquery.Selection) {
+			s.AppendHtml(` |  <a style="color: #1497cb;font-weight: bold;font-size: medium;" href="https://github.com/SyNdicateFoundation/SteamDownloaderAPI" target="_blank">SteamDownloaderAPI GitHub</a>`)
+		})
+
 		doc.Find(".subscribe[onclick*='SubscribeCollection']").Each(func(i int, s *goquery.Selection) {
 			onclick, _ := s.Attr("onclick")
 			if matches := infoRegex.FindStringSubmatch(onclick); len(matches) == 3 {