add more urls to sitemap

htmgo-site/internal/sitemap/generate.go

@@ -4,6 +4,8 @@ import (
 	"bytes"
 	"encoding/xml"
 	"fmt"
+	"github.com/go-chi/chi/v5"
+	"strings"
 )
 
 type URL struct {
@@ -35,8 +37,8 @@ func serialize(sitemap *URLSet) ([]byte, error) {
 	return buffer.Bytes(), nil
 }
 
-func Generate() ([]byte, error) {
+func Generate(router *chi.Mux) ([]byte, error) {
-
+	routes := router.Routes()
 	urls := []URL{
 		{
 			Loc:        "/",
@@ -59,6 +61,25 @@ func Generate() ([]byte, error) {
 			ChangeFreq: "weekly",
 		},
 	}
+
+	for _, route := range routes {
+		if strings.HasPrefix(route.Pattern, "/docs/") {
+			urls = append(urls, URL{
+				Loc:        route.Pattern,
+				Priority:   1.0,
+				ChangeFreq: "weekly",
+			})
+		}
+
+		if strings.HasPrefix(route.Pattern, "/examples/") {
+			urls = append(urls, URL{
+				Loc:        route.Pattern,
+				Priority:   0.7,
+				ChangeFreq: "weekly",
+			})
+		}
+	}
+
 	sitemap := NewSitemap(urls)
 	return serialize(sitemap)
 }

htmgo-site/main.go

@@ -37,7 +37,7 @@ func main() {
 			http.FileServerFS(sub)
 
 			app.Router.Handle("/sitemap.xml", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-				s, err := sitemap.Generate()
+				s, err := sitemap.Generate(app.Router)
 				if err != nil {
 					http.Error(w, "failed to generate sitemap", http.StatusInternalServerError)
 					return