awesome/awesome-go
1
0
Fork 0
mirror of https://github.com/avelino/awesome-go.git synced 2024-09-19 06:06:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

change unsafe strings.Replace to goquery.SetAttr

Browse Source
This commit is contained in:
Kirill Zhuravlev 2023-02-14 23:26:55 +01:00 committed by Avelino
parent a2d1d19caf
commit 258c4fb4d6
No known key found for this signature in database
GPG Key ID: B345B4D52E98180A
1 changed files with 16 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
main.go
View File

@ -104,7 +104,10 @@ func renderAll() error {
// FIXME: remove all panics // FIXME: remove all panics
return fmt.Errorf("unable to render categories: %w", err) return fmt.Errorf("unable to render categories: %w", err)
} }
changeLinksInIndex(string(input), query, objs)
if err := rewriteLinksInIndex(query, objs); err != nil {
return fmt.Errorf("unable to rewrite links in index: %w", err)
}
makeSitemap(objs) makeSitemap(objs)
@ -216,7 +219,7 @@ func makeObjByID(selector string, s *goquery.Selection) (*Object, error) {
return &obj, nil return &obj, nil
} }
func changeLinksInIndex(html string, query *goquery.Document, objs map[string]Object) { func rewriteLinksInIndex(query *goquery.Document, objs map[string]Object) error {
query.Find("body #content ul li ul li a").Each(func(_ int, s *goquery.Selection) { query.Find("body #content ul li ul li a").Each(func(_ int, s *goquery.Selection) {
href, hrefExists := s.Attr("href") href, hrefExists := s.Attr("href")
if !hrefExists { if !hrefExists {
@ -234,15 +237,19 @@ func changeLinksInIndex(html string, query *goquery.Document, objs map[string]Ob
// FIXME: parse url // FIXME: parse url
uri := strings.SplitAfter(href, "#") uri := strings.SplitAfter(href, "#")
if len(uri) >= 2 && uri[1] != "contents" { if len(uri) >= 2 && uri[1] != "contents" {
// FIXME: use s.SetAttr s.SetAttr("href", uri[1])
html = strings.ReplaceAll(
html,
fmt.Sprintf(`href="%s"`, href),
fmt.Sprintf(`href="%s"`, uri[1]),
)
} }
}) })
fmt.Printf("Rewrite links in Index file: %s\n", outIndexFile) fmt.Printf("Rewrite links in Index file: %s\n", outIndexFile)
_ = os.WriteFile(outIndexFile, []byte(html), 0644) resultHtml, err := query.Html()
if err != nil {
return fmt.Errorf("unable to render html: %w", err)
}
if err := os.WriteFile(outIndexFile, []byte(resultHtml), 0644); err != nil {
return fmt.Errorf("unable to rewrite index file: %w", err)
}
return nil
} }