package api import ( "archive/zip" "fmt" "io" "net/http" "os" "path" "path/filepath" "strings" "time" "github.com/photoprism/photoprism/pkg/rnd" "github.com/photoprism/photoprism/internal/acl" "github.com/photoprism/photoprism/internal/form" "github.com/photoprism/photoprism/internal/i18n" "github.com/photoprism/photoprism/internal/photoprism" "github.com/photoprism/photoprism/internal/query" "github.com/photoprism/photoprism/internal/service" "github.com/photoprism/photoprism/pkg/fs" "github.com/photoprism/photoprism/pkg/txt" "github.com/gin-gonic/gin" ) // POST /api/v1/zip func CreateZip(router *gin.RouterGroup) { router.POST("/zip", func(c *gin.Context) { s := Auth(SessionID(c), acl.ResourcePhotos, acl.ActionDownload) if s.Invalid() { AbortUnauthorized(c) return } conf := service.Config() if !conf.Settings().Features.Download { AbortFeatureDisabled(c) return } var f form.Selection start := time.Now() if err := c.BindJSON(&f); err != nil { AbortBadRequest(c) return } if f.Empty() { Abort(c, http.StatusBadRequest, i18n.ErrNoItemsSelected) return } files, err := query.FileSelection(f) if err != nil { Error(c, http.StatusBadRequest, err, i18n.ErrZipFailed) return } else if len(files) == 0 { Abort(c, http.StatusNotFound, i18n.ErrNoFilesForDownload) return } zipPath := path.Join(conf.TempPath(), "zip") zipToken := rnd.Token(8) zipBaseName := fmt.Sprintf("photoprism-download-%s-%s.zip", time.Now().Format("20060102-150405"), zipToken) zipFileName := path.Join(zipPath, zipBaseName) if err := os.MkdirAll(zipPath, 0700); err != nil { Error(c, http.StatusInternalServerError, err, i18n.ErrZipFailed) return } newZipFile, err := os.Create(zipFileName) if err != nil { Error(c, http.StatusInternalServerError, err, i18n.ErrZipFailed) return } defer newZipFile.Close() zipWriter := zip.NewWriter(newZipFile) defer zipWriter.Close() dlName := DownloadName(c) var aliases = make(map[string]int) for _, file := range files { if file.FileHash == "" { log.Warnf("download: empty file hash, skipped %s", txt.Quote(file.FileName)) continue } if file.FileSidecar { log.Debugf("download: skipped sidecar %s", txt.Quote(file.FileName)) continue } fileName := photoprism.FileName(file.FileRoot, file.FileName) alias := file.DownloadName(dlName, 0) key := strings.ToLower(alias) if seq := aliases[key]; seq > 0 { alias = file.DownloadName(dlName, seq) } aliases[key] += 1 if fs.FileExists(fileName) { if err := addFileToZip(zipWriter, fileName, alias); err != nil { Error(c, http.StatusInternalServerError, err, i18n.ErrZipFailed) return } log.Infof("download: added %s as %s", txt.Quote(file.FileName), txt.Quote(alias)) } else { log.Warnf("download: file %s is missing", txt.Quote(file.FileName)) logError("download", file.Update("FileMissing", true)) } } elapsed := int(time.Since(start).Seconds()) log.Infof("download: created %s [%s]", txt.Quote(zipBaseName), time.Since(start)) c.JSON(http.StatusOK, gin.H{"code": http.StatusOK, "message": i18n.Msg(i18n.MsgZipCreatedIn, elapsed), "filename": zipBaseName}) }) } // GET /api/v1/zip/:filename func DownloadZip(router *gin.RouterGroup) { router.GET("/zip/:filename", func(c *gin.Context) { if InvalidDownloadToken(c) { c.Data(http.StatusForbidden, "image/svg+xml", brokenIconSvg) return } conf := service.Config() zipBaseName := filepath.Base(c.Param("filename")) zipPath := path.Join(conf.TempPath(), "zip") zipFileName := path.Join(zipPath, zipBaseName) if !fs.FileExists(zipFileName) { log.Errorf("could not find zip file: %s", zipFileName) c.Data(404, "image/svg+xml", photoIconSvg) return } c.FileAttachment(zipFileName, zipBaseName) if err := os.Remove(zipFileName); err != nil { log.Errorf("download: failed removing %s (%s)", txt.Quote(zipFileName), err.Error()) } }) } func addFileToZip(zipWriter *zip.Writer, fileName, fileAlias string) error { fileToZip, err := os.Open(fileName) if err != nil { return err } defer fileToZip.Close() // Get the file information info, err := fileToZip.Stat() if err != nil { return err } header, err := zip.FileInfoHeader(info) if err != nil { return err } header.Name = fileAlias // Change to deflate to gain better compression // see http://golang.org/pkg/archive/zip/#pkg-constants header.Method = zip.Deflate writer, err := zipWriter.CreateHeader(header) if err != nil { return err } _, err = io.Copy(writer, fileToZip) return err }