Start on album download

api/routes/downloads.go

@@ -0,0 +1,84 @@
+package routes
+
+import (
+	"archive/zip"
+	"fmt"
+	"io"
+	"log"
+	"net/http"
+	"os"
+
+	"github.com/gorilla/mux"
+	"github.com/photoview/photoview/api/graphql/models"
+	"gorm.io/gorm"
+)
+
+func RegisterDownloadRoutes(db *gorm.DB, router *mux.Router) {
+	router.HandleFunc("/album/{album_id}/{media_purpose}", func(w http.ResponseWriter, r *http.Request) {
+		albumID := mux.Vars(r)["album_id"]
+		mediaPurpose := mux.Vars(r)["media_purpose"]
+
+		var album models.Album
+		if err := db.Find(&album, albumID).Error; err != nil {
+			w.WriteHeader(http.StatusNotFound)
+			w.Write([]byte("404"))
+			return
+		}
+
+		var mediaURLs []*models.MediaURL
+		if err := db.Joins("Media").Where("media.album_id = ?", album.ID).Where("media_urls.purpose = ?", mediaPurpose).Find(&mediaURLs).Error; err != nil {
+			w.WriteHeader(http.StatusInternalServerError)
+			w.Write([]byte("internal server error"))
+			return
+		}
+
+		w.Header().Set("Content-Type", "application/zip")
+		w.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename=\"%s.zip\"", album.Title))
+
+		zipWriter := zip.NewWriter(w)
+
+		for _, media := range mediaURLs {
+			zipFile, err := zipWriter.Create(media.MediaName)
+			if err != nil {
+				log.Printf("ERROR: Failed to create a file in zip, when downloading album (%d): %v\n", album.ID, err)
+				w.WriteHeader(http.StatusInternalServerError)
+				w.Write([]byte("internal server error"))
+				return
+			}
+
+			filePath, err := media.CachedPath()
+			if err != nil {
+				log.Printf("ERROR: Failed to get mediaURL cache path, when downloading album (%d): %v\n", album.ID, err)
+				w.WriteHeader(http.StatusInternalServerError)
+				w.Write([]byte("internal server error"))
+				return
+			}
+
+			fileData, err := os.Open(filePath)
+			if err != nil {
+				log.Printf("ERROR: Failed to open file to include in zip, when downloading album (%d): %v\n", album.ID, err)
+				w.WriteHeader(http.StatusInternalServerError)
+				w.Write([]byte("internal server error"))
+				return
+			}
+
+			_, err = io.Copy(zipFile, fileData)
+			if err != nil {
+				log.Printf("ERROR: Failed to copy file data, when downloading album (%d): %v\n", album.ID, err)
+				w.WriteHeader(http.StatusInternalServerError)
+				w.Write([]byte("internal server error"))
+				return
+			}
+
+			if err := fileData.Close(); err != nil {
+				log.Printf("ERROR: Failed to close file, when downloading album (%d): %v\n", album.ID, err)
+				w.WriteHeader(http.StatusInternalServerError)
+				w.Write([]byte("internal server error"))
+				return
+			}
+		}
+
+		// close the zip Writer to flush the contents to the ResponseWriter
+		zipWriter.Close()
+	})
+}

api/server.go

@@ -107,6 +107,9 @@ func main() {
 	videoRouter := endpointRouter.PathPrefix("/video").Subrouter()
 	routes.RegisterVideoRoutes(db, videoRouter)
 
+	downloadsRouter := endpointRouter.PathPrefix("/download").Subrouter()
+	routes.RegisterDownloadRoutes(db, downloadsRouter)
+
 	shouldServeUI := utils.ShouldServeUI()
 
 	if shouldServeUI {