package api import ( "fmt" "strconv" "github.com/gin-gonic/gin" "github.com/photoprism/photoprism/internal/entity" "github.com/photoprism/photoprism/internal/get" "github.com/photoprism/photoprism/internal/session" ) // AddCountHeader adds the actual result count to the response. func AddCountHeader(c *gin.Context, count int) { c.Header("X-Count", strconv.Itoa(count)) } // AddLimitHeader adds the max result count to the response. func AddLimitHeader(c *gin.Context, limit int) { c.Header("X-Limit", strconv.Itoa(limit)) } // AddOffsetHeader adds the result offset to the response. func AddOffsetHeader(c *gin.Context, offset int) { c.Header("X-Offset", strconv.Itoa(offset)) } // AddDownloadHeader adds a header indicating the response is expected to be downloaded. func AddDownloadHeader(c *gin.Context, fileName string) { c.Header("Content-Disposition", fmt.Sprintf("attachment; filename=%s", fileName)) } // AddSessionHeader adds a session id header to the response. func AddSessionHeader(c *gin.Context, id string) { c.Header(session.Header, id) } // AddContentTypeHeader adds a content type header to the response. func AddContentTypeHeader(c *gin.Context, contentType string) { c.Header("Content-Type", contentType) } // AddFileCountHeaders adds file and folder counts to the response. func AddFileCountHeaders(c *gin.Context, filesCount, foldersCount int) { c.Header("X-Files", strconv.Itoa(filesCount)) c.Header("X-Folders", strconv.Itoa(foldersCount)) } // AddTokenHeaders adds preview token headers to the response. func AddTokenHeaders(c *gin.Context, s *entity.Session) { if get.Config().Public() { return } if s.PreviewToken != "" { c.Header("X-Preview-Token", s.PreviewToken) } if s.DownloadToken != "" { c.Header("X-Download-Token", s.DownloadToken) } }