photoprism/internal/api/download_album.go

102 lines
2.4 KiB
Go
Raw Normal View History

package api
import (
"archive/zip"
"net/http"
"strings"
"time"
"github.com/gin-gonic/gin"
"github.com/photoprism/photoprism/internal/i18n"
"github.com/photoprism/photoprism/internal/photoprism"
"github.com/photoprism/photoprism/internal/query"
"github.com/photoprism/photoprism/internal/search"
"github.com/photoprism/photoprism/internal/service"
"github.com/photoprism/photoprism/pkg/clean"
"github.com/photoprism/photoprism/pkg/fs"
)
// DownloadAlbum streams the album contents as zip archive.
//
// GET /api/v1/albums/:uid/dl
func DownloadAlbum(router *gin.RouterGroup) {
router.GET("/albums/:uid/dl", func(c *gin.Context) {
if InvalidDownloadToken(c) {
AbortForbidden(c)
return
}
conf := service.Config()
if !conf.Settings().Features.Download {
AbortFeatureDisabled(c)
return
}
start := time.Now()
a, err := query.AlbumByUID(clean.UID(c.Param("uid")))
if err != nil {
AbortAlbumNotFound(c)
return
}
files, err := search.AlbumPhotos(a, 10000, true)
if err != nil {
AbortEntityNotFound(c)
return
}
zipFileName := a.ZipName()
AddDownloadHeader(c, zipFileName)
zipWriter := zip.NewWriter(c.Writer)
defer zipWriter.Close()
var aliases = make(map[string]int)
for _, file := range files {
if file.FileHash == "" {
log.Warnf("download: empty file hash, skipped %s", clean.Log(file.FileName))
continue
} else if file.FileName == "" {
log.Warnf("download: empty file name, skipped %s", clean.Log(file.FileUID))
continue
}
if file.FileSidecar {
log.Debugf("download: skipped sidecar %s", clean.Log(file.FileName))
continue
}
fileName := photoprism.FileName(file.FileRoot, file.FileName)
alias := file.ShareBase(0)
key := strings.ToLower(alias)
if seq := aliases[key]; seq > 0 {
alias = file.ShareBase(seq)
}
aliases[key] += 1
if fs.FileExists(fileName) {
if err := addFileToZip(zipWriter, fileName, alias); err != nil {
log.Errorf("download: failed adding %s to album zip (%s)", clean.Log(file.FileName), err)
Abort(c, http.StatusInternalServerError, i18n.ErrZipFailed)
return
}
log.Infof("download: added %s as %s", clean.Log(file.FileName), clean.Log(alias))
} else {
log.Warnf("download: album file %s is missing", clean.Log(file.FileName))
}
}
log.Infof("download: created %s [%s]", clean.Log(zipFileName), time.Since(start))
})
}