Exclude style tags from head element search

alileza · Mar 3, 2024 · 951c0a8 · 951c0a8
1 parent c42b59b
commit 951c0a8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/redirector/htmlparser.go b/redirector/htmlparser.go
@@ -21,14 +21,15 @@ func FetchHeadElements(targetURL string) (string, error) {
 		return html.Parse(resp.Body)
 	}
 
-	// Recursive function to find all elements within <head> excluding scripts and stylesheets
+	// Recursive function to find all elements within <head> excluding scripts, stylesheets, and style tags
 	findHeadElements := func(n *html.Node) []*html.Node {
 		var headElements []*html.Node
 		var traverse func(*html.Node)
 		traverse = func(n *html.Node) {
 			if n.Type == html.ElementNode && n.Data == "head" {
 				for child := n.FirstChild; child != nil; child = child.NextSibling {
-					if child.Data != "script" && child.Data != "link" {
+					// Exclude script, link, and style tags
+					if child.Data != "script" && child.Data != "link" && child.Data != "style" {
 						headElements = append(headElements, child)
 					}
 				}

diff --git a/routes.yaml b/routes.yaml
@@ -2,7 +2,7 @@ a: https://a.com
 abc: https://hehe.com
 db: https://distrobe.at
 doner: kebab123
+go: https://google.com
 me: https://alileza.me
 test: https://google.com
 tos: tos
-