owl-blogs/web/sitemap_handler.go

package web

import (
	"bytes"
	"encoding/xml"
	"net/url"
	"owl-blogs/app"

	"github.com/gofiber/fiber/v2"
)

type SiteMapHandler struct {
	entryService      *app.EntryService
	siteConfigService *app.SiteConfigService
}

type Sitemap struct {
	XMLName xml.Name     `xml:"urlset"`
	Xmlns   string       `xml:"xmlns,attr"`
	Url     []SitemapUrl `xml:"url"`
}

type SitemapUrl struct {
	Loc string `xml:"loc"`
}

func NewSiteMapHandler(entryService *app.EntryService, siteConfigService *app.SiteConfigService) *SiteMapHandler {
	return &SiteMapHandler{entryService: entryService, siteConfigService: siteConfigService}
}

// Handle handles GET /sitemap.xml
func (h *SiteMapHandler) Handle(c *fiber.Ctx) error {
	c.Set(fiber.HeaderContentType, fiber.MIMEApplicationXML)

	siteConfig, err := h.siteConfigService.GetSiteConfig()
	if err != nil {
		return err
	}
	entries, err := h.entryService.FindAllByType(nil, true, false)
	if err != nil {
		return err
	}

	sitemap := Sitemap{
		Xmlns: "http://www.sitemaps.org/schemas/sitemap/0.9",
		Url:   make([]SitemapUrl, 0),
	}

	for _, entry := range entries {
		entryUrl, _ := url.JoinPath(siteConfig.FullUrl, "/posts/", url.PathEscape(entry.ID()), "/")
		sitemap.Url = append(sitemap.Url, SitemapUrl{
			Loc: entryUrl,
		})
	}

	buf := new(bytes.Buffer)
	encoder := xml.NewEncoder(buf)
	encoder.Indent("", "  ")
	err = encoder.Encode(sitemap)
	if err != nil {
		return err
	}

	return c.SendString(xml.Header + buf.String())
}