htmgo/htmgo-site/internal/sitemap/generate.go
2024-10-15 11:15:47 -05:00

64 lines
1.2 KiB
Go

package sitemap
import (
"bytes"
"encoding/xml"
"fmt"
)
type URL struct {
Loc string `xml:"loc"`
ChangeFreq string `xml:"changefreq,omitempty"`
Priority float32 `xml:"priority,omitempty"`
}
type URLSet struct {
XMLName xml.Name `xml:"urlset"`
XmlNS string `xml:"xmlns,attr"`
URLs []URL `xml:"url"`
}
func NewSitemap(urls []URL) *URLSet {
return &URLSet{
XmlNS: "https://www.sitemaps.org/schemas/sitemap/0.9",
URLs: urls,
}
}
func serialize(sitemap *URLSet) ([]byte, error) {
buffer := bytes.Buffer{}
enc := xml.NewEncoder(&buffer)
enc.Indent("", " ")
if err := enc.Encode(sitemap); err != nil {
return make([]byte, 0), fmt.Errorf("could not encode sitemap: %w", err)
}
return buffer.Bytes(), nil
}
func Generate() ([]byte, error) {
urls := []URL{
{
Loc: "/",
Priority: 0.5,
ChangeFreq: "weekly",
},
{
Loc: "/docs",
Priority: 1.0,
ChangeFreq: "daily",
},
{
Loc: "/examples",
Priority: 0.7,
ChangeFreq: "daily",
},
{
Loc: "/html-to-go",
Priority: 0.5,
ChangeFreq: "weekly",
},
}
sitemap := NewSitemap(urls)
return serialize(sitemap)
}