mirror of
https://github.com/go-gitea/gitea.git
synced 2024-08-25 21:44:20 -04:00
60 lines
1.7 KiB
Go
60 lines
1.7 KiB
Go
|
// Copyright 2020 The Gitea Authors. All rights reserved.
|
||
|
// Use of this source code is governed by a MIT-style
|
||
|
// license that can be found in the LICENSE file.
|
||
|
|
||
|
package httpcache
|
||
|
|
||
|
import (
|
||
|
"encoding/base64"
|
||
|
"fmt"
|
||
|
"net/http"
|
||
|
"os"
|
||
|
"strconv"
|
||
|
"time"
|
||
|
|
||
|
"code.gitea.io/gitea/modules/setting"
|
||
|
)
|
||
|
|
||
|
// GetCacheControl returns a suitable "Cache-Control" header value
|
||
|
func GetCacheControl() string {
|
||
|
if setting.RunMode == "dev" {
|
||
|
return "no-store"
|
||
|
}
|
||
|
return "private, max-age=" + strconv.FormatInt(int64(setting.StaticCacheTime.Seconds()), 10)
|
||
|
}
|
||
|
|
||
|
// generateETag generates an ETag based on size, filename and file modification time
|
||
|
func generateETag(fi os.FileInfo) string {
|
||
|
etag := fmt.Sprint(fi.Size()) + fi.Name() + fi.ModTime().UTC().Format(http.TimeFormat)
|
||
|
return base64.StdEncoding.EncodeToString([]byte(etag))
|
||
|
}
|
||
|
|
||
|
// HandleTimeCache handles time-based caching for a HTTP request
|
||
|
func HandleTimeCache(req *http.Request, w http.ResponseWriter, fi os.FileInfo) (handled bool) {
|
||
|
ifModifiedSince := req.Header.Get("If-Modified-Since")
|
||
|
if ifModifiedSince != "" {
|
||
|
t, err := time.Parse(http.TimeFormat, ifModifiedSince)
|
||
|
if err == nil && fi.ModTime().Unix() <= t.Unix() {
|
||
|
w.WriteHeader(http.StatusNotModified)
|
||
|
return true
|
||
|
}
|
||
|
}
|
||
|
|
||
|
w.Header().Set("Cache-Control", GetCacheControl())
|
||
|
w.Header().Set("Last-Modified", fi.ModTime().Format(http.TimeFormat))
|
||
|
return false
|
||
|
}
|
||
|
|
||
|
// HandleEtagCache handles ETag-based caching for a HTTP request
|
||
|
func HandleEtagCache(req *http.Request, w http.ResponseWriter, fi os.FileInfo) (handled bool) {
|
||
|
etag := generateETag(fi)
|
||
|
if req.Header.Get("If-None-Match") == etag {
|
||
|
w.WriteHeader(http.StatusNotModified)
|
||
|
return true
|
||
|
}
|
||
|
|
||
|
w.Header().Set("Cache-Control", GetCacheControl())
|
||
|
w.Header().Set("ETag", etag)
|
||
|
return false
|
||
|
}
|