99 lines
2.9 KiB
Go
99 lines
2.9 KiB
Go
package processing_tasks
|
|
|
|
import (
|
|
"os"
|
|
"path"
|
|
|
|
"github.com/photoview/photoview/api/graphql/models"
|
|
"github.com/photoview/photoview/api/scanner/media_encoding"
|
|
"github.com/photoview/photoview/api/scanner/media_encoding/media_utils"
|
|
"github.com/pkg/errors"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func generateSaveHighResJPEG(tx *gorm.DB, media *models.Media, imageData *media_encoding.EncodeMediaData, highres_name string, imagePath string, mediaURL *models.MediaURL) (*models.MediaURL, error) {
|
|
|
|
err := imageData.EncodeHighRes(imagePath)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "creating high-res cached image")
|
|
}
|
|
|
|
photoDimensions, err := media_utils.GetPhotoDimensions(imagePath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
fileStats, err := os.Stat(imagePath)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "reading file stats of highres photo")
|
|
}
|
|
|
|
if mediaURL == nil {
|
|
|
|
mediaURL = &models.MediaURL{
|
|
MediaID: media.ID,
|
|
MediaName: highres_name,
|
|
Width: photoDimensions.Width,
|
|
Height: photoDimensions.Height,
|
|
Purpose: models.PhotoHighRes,
|
|
ContentType: "image/jpeg",
|
|
FileSize: fileStats.Size(),
|
|
}
|
|
|
|
if err := tx.Create(&mediaURL).Error; err != nil {
|
|
return nil, errors.Wrapf(err, "could not insert highres media url (%d, %s)", media.ID, highres_name)
|
|
}
|
|
} else {
|
|
mediaURL.Width = photoDimensions.Width
|
|
mediaURL.Height = photoDimensions.Height
|
|
mediaURL.FileSize = fileStats.Size()
|
|
|
|
if err := tx.Save(&mediaURL).Error; err != nil {
|
|
return nil, errors.Wrapf(err, "could not update media url after side car changes (%d, %s)", media.ID, highres_name)
|
|
}
|
|
}
|
|
|
|
return mediaURL, nil
|
|
}
|
|
|
|
func generateSaveThumbnailJPEG(tx *gorm.DB, media *models.Media, thumbnail_name string, photoCachePath string, baseImagePath string, mediaURL *models.MediaURL) (*models.MediaURL, error) {
|
|
thumbOutputPath := path.Join(photoCachePath, thumbnail_name)
|
|
|
|
thumbSize, err := media_encoding.EncodeThumbnail(tx, baseImagePath, thumbOutputPath)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "could not create thumbnail cached image")
|
|
}
|
|
|
|
fileStats, err := os.Stat(thumbOutputPath)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "reading file stats of thumbnail photo")
|
|
}
|
|
|
|
if mediaURL == nil {
|
|
|
|
mediaURL = &models.MediaURL{
|
|
MediaID: media.ID,
|
|
MediaName: thumbnail_name,
|
|
Width: thumbSize.Width,
|
|
Height: thumbSize.Height,
|
|
Purpose: models.PhotoThumbnail,
|
|
ContentType: "image/jpeg",
|
|
FileSize: fileStats.Size(),
|
|
}
|
|
|
|
if err := tx.Create(&mediaURL).Error; err != nil {
|
|
return nil, errors.Wrapf(err, "could not insert thumbnail media url (%d, %s)", media.ID, thumbnail_name)
|
|
}
|
|
} else {
|
|
mediaURL.Width = thumbSize.Width
|
|
mediaURL.Height = thumbSize.Height
|
|
mediaURL.FileSize = fileStats.Size()
|
|
|
|
if err := tx.Save(&mediaURL).Error; err != nil {
|
|
return nil, errors.Wrapf(err, "could not update media url after side car changes (%d, %s)", media.ID, thumbnail_name)
|
|
}
|
|
}
|
|
|
|
return mediaURL, nil
|
|
}
|