Handle // scheme prefix (#75)

* Handle /// scheme prefix * Update to two /'s * chore: minor formatting fix * Update hugo_gen_setup.go --------- Co-authored-by: Ashish Bhatia <[email protected]>
ashishb · Oct 29, 2024 · cb78882 · cb78882
1 parent c4d47a3
commit cb78882
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/wp2hugo/internal/hugogenerator/hugo_gen_setup.go b/src/wp2hugo/internal/hugogenerator/hugo_gen_setup.go
@@ -424,6 +424,9 @@ func (g Generator) downloadPageMedia(outputMediaDirPath string, p *hugopage.Page
 	prefixes = append(prefixes, fmt.Sprintf("http://www.%s", hostname))
 
 	for _, link := range links {
+		if strings.HasPrefix(link, "//") {
+			link = strings.Replace(link, "//", fmt.Sprintf("%s://", pageURL.Scheme), 1)
+		}
 		for _, prefix := range prefixes {
 			link = strings.TrimPrefix(link, prefix)
 		}