124 lines
3.3 KiB
Go
124 lines
3.3 KiB
Go
package api
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"github.com/photoprism/photoprism/internal/get"
|
|
"github.com/photoprism/photoprism/internal/query"
|
|
"github.com/photoprism/photoprism/internal/thumb"
|
|
"github.com/photoprism/photoprism/internal/ttl"
|
|
"github.com/photoprism/photoprism/pkg/fs"
|
|
"github.com/photoprism/photoprism/pkg/rnd"
|
|
)
|
|
|
|
type ThumbCache struct {
|
|
FileName string
|
|
ShareName string
|
|
}
|
|
|
|
type ByteCache struct {
|
|
Data []byte
|
|
}
|
|
|
|
// CacheKey returns a cache key string based on namespace, uid and name.
|
|
func CacheKey(ns, uid, name string) string {
|
|
return fmt.Sprintf("%s:%s:%s", ns, uid, name)
|
|
}
|
|
|
|
// RemoveFromFolderCache removes an item from the folder cache e.g. after indexing.
|
|
func RemoveFromFolderCache(rootName string) {
|
|
cache := get.FolderCache()
|
|
|
|
cacheKey := fmt.Sprintf("folder:%s:%t:%t", rootName, true, false)
|
|
|
|
cache.Delete(cacheKey)
|
|
|
|
if err := query.UpdateAlbumFolderCovers(); err != nil {
|
|
log.Error(err)
|
|
}
|
|
|
|
log.Debugf("removed %s from cache", cacheKey)
|
|
}
|
|
|
|
// RemoveFromAlbumCoverCache removes covers by album UID e.g. after adding or removing photos.
|
|
func RemoveFromAlbumCoverCache(uid string) {
|
|
if !rnd.IsAlnum(uid) {
|
|
return
|
|
}
|
|
|
|
cache := get.CoverCache()
|
|
|
|
// Flush album cover cache.
|
|
for thumbName := range thumb.Sizes {
|
|
cacheKey := CacheKey(albumCover, uid, string(thumbName))
|
|
|
|
cache.Delete(cacheKey)
|
|
|
|
log.Debugf("removed %s from cache", cacheKey)
|
|
}
|
|
|
|
// Delete share preview, if exists.
|
|
if sharePreview := path.Join(get.Config().ThumbCachePath(), "share", uid+fs.ExtJPEG); fs.FileExists(sharePreview) {
|
|
_ = os.Remove(sharePreview)
|
|
}
|
|
|
|
// Update album cover images.
|
|
if err := query.UpdateAlbumCovers(); err != nil {
|
|
log.Error(err)
|
|
}
|
|
}
|
|
|
|
// FlushCoverCache clears the complete cover cache.
|
|
func FlushCoverCache() {
|
|
get.CoverCache().Flush()
|
|
|
|
if err := query.UpdateCovers(); err != nil {
|
|
log.Error(err)
|
|
}
|
|
|
|
log.Debugf("albums: flushed cover cache")
|
|
}
|
|
|
|
// AddCacheHeader adds a cache control header to the response.
|
|
func AddCacheHeader(c *gin.Context, maxAge ttl.Duration, public bool) {
|
|
if c == nil {
|
|
return
|
|
} else if maxAge <= 0 {
|
|
c.Header("Cache-Control", "no-cache")
|
|
} else if public {
|
|
c.Header("Cache-Control", fmt.Sprintf("public, max-age=%s", maxAge.String()))
|
|
} else {
|
|
c.Header("Cache-Control", fmt.Sprintf("private, max-age=%s", maxAge.String()))
|
|
}
|
|
}
|
|
|
|
// AddCoverCacheHeader adds cover image cache control headers to the response.
|
|
func AddCoverCacheHeader(c *gin.Context) {
|
|
AddCacheHeader(c, ttl.Cover, thumb.CachePublic)
|
|
}
|
|
|
|
// AddImmutableCacheHeader adds cache control headers to the response for immutable content like thumbnails.
|
|
func AddImmutableCacheHeader(c *gin.Context) {
|
|
if c == nil {
|
|
return
|
|
} else if thumb.CachePublic {
|
|
c.Header("Cache-Control", fmt.Sprintf("public, max-age=%s, immutable", ttl.Default.String()))
|
|
} else {
|
|
c.Header("Cache-Control", fmt.Sprintf("private, max-age=%s, immutable", ttl.Default.String()))
|
|
}
|
|
}
|
|
|
|
// AddVideoCacheHeader adds video cache control headers to the response.
|
|
func AddVideoCacheHeader(c *gin.Context, cdn bool) {
|
|
if c == nil {
|
|
return
|
|
} else if cdn || thumb.CachePublic {
|
|
c.Header("Cache-Control", fmt.Sprintf("public, max-age=%s, immutable", ttl.Video.String()))
|
|
} else {
|
|
c.Header("Cache-Control", fmt.Sprintf("private, max-age=%s, immutable", ttl.Video.String()))
|
|
}
|
|
}
|