精简后的项目：仅保留 api/ui 源码、文档、构建依赖

2026-09-11 16:11:09 +08:00
commit 490e85634e
457 changed files with 86862 additions and 0 deletions
@@ -0,0 +1,126 @@
+package processing_tasks
+
+import (
+	"os"
+	"path"
+
+	"github.com/photoview/photoview/api/graphql/models"
+	"github.com/photoview/photoview/api/log"
+	"github.com/photoview/photoview/api/scanner/media_encoding"
+	"github.com/photoview/photoview/api/scanner/scanner_task"
+	"github.com/pkg/errors"
+)
+
+type ProcessPhotoTask struct {
+	scanner_task.ScannerTaskBase
+}
+
+func (t ProcessPhotoTask) ProcessMedia(ctx scanner_task.TaskContext, mediaData *media_encoding.EncodeMediaData, mediaCachePath string) ([]*models.MediaURL, error) {
+	if mediaData.Media.Type != models.MediaTypePhoto {
+		return []*models.MediaURL{}, nil
+	}
+
+	updatedURLs := make([]*models.MediaURL, 0)
+	photo := mediaData.Media
+
+	log.Info(ctx, "Processing photo", "photo", photo.Path)
+
+	photoURLFromDB := makePhotoURLChecker(ctx.GetDB(), photo.ID)
+
+	// original photo url
+	origURL, err := photoURLFromDB(models.MediaOriginal)
+	if err != nil {
+		return []*models.MediaURL{}, err
+	}
+
+	// Thumbnail
+	thumbURL, err := photoURLFromDB(models.PhotoThumbnail)
+	if err != nil {
+		return []*models.MediaURL{}, errors.Wrap(err, "error processing photo thumbnail")
+	}
+
+	// Highres
+	highResURL, err := photoURLFromDB(models.PhotoHighRes)
+	if err != nil {
+		return []*models.MediaURL{}, errors.Wrap(err, "error processing photo highres")
+	}
+
+	var baseImagePath string = photo.Path
+
+	// Generate high res jpeg
+	if highResURL == nil {
+
+		contentType, err := mediaData.ContentType()
+		if err != nil {
+			return []*models.MediaURL{}, err
+		}
+
+		if !contentType.IsWebCompatible() {
+			highresName := generateUniqueMediaNamePrefixed("highres", photo.Path, ".jpg")
+			baseImagePath = path.Join(mediaCachePath, highresName)
+
+			highRes, err := generateSaveHighResJPEG(ctx.GetDB(), photo, mediaData, highresName, baseImagePath, nil)
+			if err != nil {
+				return []*models.MediaURL{}, err
+			}
+
+			updatedURLs = append(updatedURLs, highRes)
+		}
+	} else {
+		// Verify that highres photo still exists in cache
+		baseImagePath = path.Join(mediaCachePath, highResURL.MediaName)
+
+		if _, err := os.Stat(baseImagePath); os.IsNotExist(err) {
+			log.Info(ctx, "High-res photo found in database but not in cache, re-encoding photo to cache", "media_name", highResURL.MediaName)
+			updatedURLs = append(updatedURLs, highResURL)
+
+			err = mediaData.EncodeHighRes(baseImagePath)
+			if err != nil {
+				return []*models.MediaURL{}, errors.Wrap(err, "creating high-res cached image")
+			}
+		}
+	}
+
+	// Save original photo to database
+	if origURL == nil {
+
+		// Make sure photo dimensions is set
+		photoDimensions, err := media_encoding.GetPhotoDimensions(photo.Path)
+		if err != nil {
+			return []*models.MediaURL{}, err
+		}
+
+		original, err := saveOriginalPhotoToDB(ctx.GetDB(), photo, mediaData, photoDimensions)
+		if err != nil {
+			return []*models.MediaURL{}, errors.Wrap(err, "saving original photo to database")
+		}
+
+		updatedURLs = append(updatedURLs, original)
+	}
+
+	// Save thumbnail to cache
+	if thumbURL == nil {
+		thumbnailName := generateUniqueMediaNamePrefixed("thumbnail", photo.Path, ".jpg")
+		thumbnail, err := generateSaveThumbnailJPEG(ctx.GetDB(), photo, thumbnailName, mediaCachePath, baseImagePath, nil)
+		if err != nil {
+			return []*models.MediaURL{}, err
+		}
+
+		updatedURLs = append(updatedURLs, thumbnail)
+	} else {
+		// Verify that thumbnail photo still exists in cache
+		thumbPath := path.Join(mediaCachePath, thumbURL.MediaName)
+
+		if _, err := os.Stat(thumbPath); os.IsNotExist(err) {
+			updatedURLs = append(updatedURLs, thumbURL)
+			log.Info(ctx, "Thumbnail photo found in database but not in cache, re-encoding photo to cache", "media_name", thumbURL.MediaName)
+
+			_, err := media_encoding.EncodeThumbnail(ctx.GetDB(), baseImagePath, thumbPath)
+			if err != nil {
+				return []*models.MediaURL{}, errors.Wrap(err, "could not create thumbnail cached image")
+			}
+		}
+	}
+
+	return updatedURLs, nil
+}