package middleware

import (
	"fmt"
	"github.com/fthvgb1/wp-go/internal/pkg/config"
	"github.com/gin-gonic/gin"
	"strings"
)

var path = map[string]struct{}{
	"wp-includes": {},
	"wp-content":  {},
	"favicon.ico": {},
}

func SetStaticFileCache(c *gin.Context) {
	f := strings.Split(strings.TrimLeft(c.FullPath(), "/"), "/")
	if _, ok := path[f[0]]; ok {
		t := config.GetConfig().CacheTime.CacheControl
		if t > 0 {
			c.Header("Cache-Control", fmt.Sprintf("private, max-age=%d", int(t.Seconds())))
		}
	}

	c.Next()
}