120 lines
2.6 KiB
Go
120 lines
2.6 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"mime"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/patrickmn/go-cache"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var (
|
|
pageCache = cache.New(6 * time.Hour, 1 * time.Hour)
|
|
)
|
|
|
|
type PageContentCache struct {
|
|
Content []byte
|
|
mimeType string
|
|
RequestedAt time.Time
|
|
}
|
|
|
|
func makePageContentCacheEntry(username, path string) string {
|
|
return username + ":" + path
|
|
}
|
|
|
|
func serveFile(username, reponame, path, giteaUrl string, w http.ResponseWriter) {
|
|
// Provide a default
|
|
if path == "" {
|
|
path = "/index.html"
|
|
}
|
|
|
|
// Strip away a starting / as it messes with Gitea
|
|
if path[:1] == "/" {
|
|
path = path[1:]
|
|
}
|
|
|
|
key := makePageContentCacheEntry(username, path)
|
|
entry, found := pageCache.Get(key)
|
|
var content []byte
|
|
var mimeType string
|
|
var err error
|
|
if found {
|
|
log.Debugf("Returning %s from cache", path)
|
|
content = entry.(PageContentCache).Content
|
|
mimeType = entry.(PageContentCache).mimeType
|
|
}
|
|
|
|
// We have to do the raw request manually because the Gitea SDK does not allow
|
|
// passing the If-Modfied-Since header.
|
|
apiUrl := fmt.Sprintf(
|
|
"%s/api/v1/repos/%s/%s/raw/%s?ref=%s",
|
|
giteaUrl,
|
|
username,
|
|
reponame,
|
|
path,
|
|
PagesBranch,
|
|
)
|
|
client := &http.Client{}
|
|
req, err := http.NewRequest("GET", apiUrl, nil)
|
|
if found {
|
|
since := entry.(PageContentCache).RequestedAt.Format(time.RFC1123)
|
|
log.Debugf("Found %s in cache. Adding '%s' as If-Modified-Since", key, since)
|
|
req.Header.Add("If-Modified-Since", since)
|
|
}
|
|
resp, err := client.Do(req)
|
|
if err != nil {
|
|
if !found {
|
|
log.Errorf("Failed to get file %s/%s/%s (%s)", username, reponame, path, err)
|
|
w.WriteHeader(404)
|
|
} else {
|
|
log.Debugf("Request failed but page %s is cached in memory", path)
|
|
w.WriteHeader(200)
|
|
w.Header().Set("Content-Type", mimeType)
|
|
w.Write(content)
|
|
}
|
|
|
|
return
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
log.Debugf("Gitea API request returned %d", resp.StatusCode)
|
|
if found && resp.StatusCode == 302 {
|
|
log.Debugf("Page %s is unchanged and cached in memory", path)
|
|
w.WriteHeader(200)
|
|
w.Header().Set("Content-Type", mimeType)
|
|
w.Write(content)
|
|
return
|
|
}
|
|
|
|
content, err = io.ReadAll(resp.Body)
|
|
if err != nil {
|
|
log.Errorf("Failed to get file %s/%s/%s (%s)", username, reponame, path, err)
|
|
w.WriteHeader(404)
|
|
return
|
|
}
|
|
|
|
pathParts := strings.Split(path, ".")
|
|
ext := pathParts[len(pathParts) - 1]
|
|
mimeType = mime.TypeByExtension("." + ext)
|
|
|
|
now := time.Now()
|
|
pageCache.Set(
|
|
key,
|
|
PageContentCache{
|
|
content,
|
|
mimeType,
|
|
now,
|
|
},
|
|
cache.DefaultExpiration,
|
|
)
|
|
|
|
log.Debugf("Page %s requested from Gitea and cached in memory at %v", path, now)
|
|
w.WriteHeader(200)
|
|
w.Header().Set("Content-Type", mimeType)
|
|
w.Write(content)
|
|
}
|