summaryrefslogtreecommitdiff
path: root/internal/router/handlers/sitemap.xml.go
blob: bb1df932de900fd7dda000ecde442fe513f13cc1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
package handlers

import (
	"log/slog"
	"time"

	"github.com/SayaAndy/saya-today-web/internal/router"
	"github.com/gofiber/fiber/v2"
)

type SitemapXmlHandler struct {
	router.BasicHandler
}

func init() {
	router.Routes = append(router.Routes, &SitemapXmlHandler{})
}

func (r *SitemapXmlHandler) Filter() (method string, path string) {
	return "GET", "/sitemap.xml"
}

func (r *SitemapXmlHandler) IsTemplated() bool {
	return false
}

func (r *SitemapXmlHandler) TemplatesToInject() []string {
	return []string{"views/pages/sitemap.xml"}
}

func (r *SitemapXmlHandler) ToCache() router.CacheSetting {
	return router.ByUrlOnly
}

func (r *SitemapXmlHandler) CacheDuration() time.Duration {
	return time.Hour
}

func (r *SitemapXmlHandler) ToValidateLang() router.LangSetting {
	return router.NotRequired
}

func (r *SitemapXmlHandler) ContentType() string {
	return fiber.MIMEApplicationXMLCharsetUTF8
}

func (r *SitemapXmlHandler) Render(c *fiber.Ctx, supplements *router.Supplements, lang string, templateMap fiber.Map) (statusCode int, err error) {
	urls := []router.SitemapInfo{}
	for _, route := range router.Routes {
		sitemapInfo := route.SitemapInfo(supplements)
		if sitemapInfo == nil {
			continue
		}

		method, match := route.Filter()
		lastMod, err := supplements.TemplateManager.GetLastModified(method + " " + match)
		if err != nil {
			slog.Warn("failed to get last modified for a route",
				slog.String("error", err.Error()),
				slog.String("method", method),
				slog.String("match", match))
			lastMod = time.Now()
		}

		for i := range sitemapInfo {
			if sitemapInfo[i].LastModified.Before(lastMod) {
				sitemapInfo[i].LastModified = lastMod
			}
			if sitemapInfo[i].ChangeFreq == "" {
				sitemapInfo[i].ChangeFreq = "monthly"
			}
			urls = append(urls, sitemapInfo[i])
		}
	}

	templateMap["URLs"] = urls

	return fiber.StatusOK, nil
}