package api import ( "archive/zip" "net/http" "strings" "time" "github.com/gin-gonic/gin" "github.com/photoprism/photoprism/internal/get" "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/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 := get.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)) }) }