Adding initial scaffolding to rename the repository

Signed-off-by: Kris Nóva <kris@nivenly.com>

internal/api/zip.go

@@ -0,0 +1,192 @@
+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: zip %s created in %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
+}