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

2026-09-11 16:11:09 +08:00
commit 490e85634e
457 changed files with 86862 additions and 0 deletions
@@ -0,0 +1,106 @@
+package resolvers
+
+import (
+	"errors"
+	"fmt"
+
+	"github.com/photoview/photoview/api/graphql/models"
+	"gorm.io/gorm"
+)
+
+const faceGroupIDIsQuestion = "face_group_id = ?"
+const faceGroupIDsInQuestion = "face_group_id IN (?)"
+const mediaAlbumIDInQuestion = "media.album_id IN (?)"
+const imageFacesIDInQuestion = "image_faces.id IN (?)"
+
+var ErrFaceDetectorNotInitialized = errors.New("face detector not initialized")
+
+func userOwnedFaceGroup(db *gorm.DB, user *models.User, faceGroupID int) (*models.FaceGroup, error) {
+	if user.Admin {
+		var faceGroup models.FaceGroup
+		if err := db.Where("id = ?", faceGroupID).Find(&faceGroup).Error; err != nil {
+			return nil, err
+		}
+
+		return &faceGroup, nil
+	}
+
+	if err := user.FillAlbums(db); err != nil {
+		return nil, err
+	}
+
+	userAlbumIDs := make([]int, len(user.Albums))
+	for i, album := range user.Albums {
+		userAlbumIDs[i] = album.ID
+	}
+
+	// Verify that user owns at least one of the images in the face group
+	imageFaceQuery := db.
+		Select("image_faces.id").
+		Table("image_faces").
+		Joins("JOIN media ON media.id = image_faces.media_id").
+		Where(mediaAlbumIDInQuestion, userAlbumIDs)
+
+	faceGroupQuery := db.
+		Model(&models.FaceGroup{}).
+		Joins("JOIN image_faces ON face_groups.id = image_faces.face_group_id").
+		Where("face_groups.id = ?", faceGroupID).
+		Where(imageFacesIDInQuestion, imageFaceQuery)
+
+	var faceGroup models.FaceGroup
+	if err := faceGroupQuery.Find(&faceGroup).Error; err != nil {
+		if err == gorm.ErrRecordNotFound {
+			return nil, fmt.Errorf("face group does not exist or is not owned by the user: %w", err)
+		}
+		return nil, err
+	}
+
+	return &faceGroup, nil
+}
+
+func getUserOwnedImageFaces(tx *gorm.DB, user *models.User, imageFaceIDs []int) ([]*models.ImageFace, error) {
+	if err := user.FillAlbums(tx); err != nil {
+		return nil, err
+	}
+
+	userAlbumIDs := make([]int, len(user.Albums))
+	for i, album := range user.Albums {
+		userAlbumIDs[i] = album.ID
+	}
+
+	var userOwnedImageFaces []*models.ImageFace
+	if err := tx.
+		Joins("JOIN media ON media.id = image_faces.media_id").
+		Where(mediaAlbumIDInQuestion, userAlbumIDs).
+		Where(imageFacesIDInQuestion, imageFaceIDs).
+		Find(&userOwnedImageFaces).Error; err != nil {
+		return nil, err
+	}
+
+	return userOwnedImageFaces, nil
+}
+
+func deleteEmptyFaceGroups(sourceFaceGroups []*models.FaceGroup, tx *gorm.DB) error {
+	for _, faceGroup := range sourceFaceGroups {
+		var count int64
+		if err := tx.Model(&models.ImageFace{}).Where(faceGroupIDIsQuestion, faceGroup.ID).Count(&count).Error; err != nil {
+			return err
+		}
+
+		if count == 0 {
+			if err := tx.Delete(&faceGroup).Error; err != nil {
+				return err
+			}
+		}
+	}
+	return nil
+}
+
+func deleteFaceGroups(sourceFaceGroups []*models.FaceGroup, tx *gorm.DB) error {
+	for _, faceGroup := range sourceFaceGroups {
+		if err := tx.Delete(&faceGroup).Error; err != nil {
+			return fmt.Errorf("Delete FaceGroup(%d) error: %w", faceGroup.ID, err)
+		}
+	}
+	return nil
+}