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

2026-09-11 16:11:09 +08:00
commit 490e85634e
457 changed files with 86862 additions and 0 deletions
@@ -0,0 +1,100 @@
+package scanner_tasks
+
+import (
+	"fmt"
+	"log"
+	"strconv"
+	"strings"
+
+	"github.com/photoview/photoview/api/graphql/models"
+	"github.com/photoview/photoview/api/scanner/scanner_task"
+	"github.com/photoview/photoview/api/scanner/scanner_tasks/processing_tasks"
+	"github.com/pkg/errors"
+	"gopkg.in/vansante/go-ffprobe.v2"
+	"gorm.io/gorm"
+)
+
+type VideoMetadataTask struct {
+	scanner_task.ScannerTaskBase
+}
+
+func (t VideoMetadataTask) AfterMediaFound(ctx scanner_task.TaskContext, media *models.Media, newMedia bool) error {
+
+	if !newMedia || media.Type != models.MediaTypeVideo {
+		return nil
+	}
+
+	err := ScanVideoMetadata(ctx.GetDB(), media)
+	if err != nil {
+		log.Printf("WARN: ScanVideoMetadata for %s failed: %s\n", media.Title, err)
+	}
+
+	return nil
+}
+
+func ScanVideoMetadata(tx *gorm.DB, video *models.Media) error {
+
+	data, err := processing_tasks.ReadVideoMetadata(video.Path)
+	if err != nil {
+		return errors.Wrapf(err, "scan video metadata failed (%s)", video.Title)
+	}
+
+	stream := data.FirstVideoStream()
+	if stream == nil {
+		return errors.New(fmt.Sprintf("could not get video stream from metadata (%s)", video.Path))
+	}
+
+	audio := data.FirstAudioStream()
+	var audioText string
+	if audio == nil {
+		audioText = "No audio"
+	} else {
+		switch audio.Channels {
+		case 0:
+			audioText = "No audio"
+		case 1:
+			audioText = "Mono audio"
+		case 2:
+			audioText = "Stereo audio"
+		default:
+			audioText = fmt.Sprintf("Audio (%d channels)", audio.Channels)
+		}
+	}
+
+	framerate := getFrameRate(stream)
+
+	videoMetadata := models.VideoMetadata{
+		Width:        stream.Width,
+		Height:       stream.Height,
+		Duration:     data.Format.DurationSeconds,
+		Codec:        &stream.CodecLongName,
+		Framerate:    framerate,
+		Bitrate:      &stream.BitRate,
+		ColorProfile: &stream.Profile,
+		Audio:        &audioText,
+	}
+
+	video.VideoMetadata = &videoMetadata
+
+	if err := tx.Save(video).Error; err != nil {
+		return errors.Wrapf(err, "failed to add video metadata to database (%s)", video.Title)
+	}
+
+	return nil
+}
+
+func getFrameRate(stream *ffprobe.Stream) *float64 {
+	var framerate *float64 = nil
+	if stream.AvgFrameRate != "" {
+		parts := strings.Split(stream.AvgFrameRate, "/")
+		if len(parts) == 2 {
+			if numerator, err := strconv.ParseInt(parts[0], 10, 64); err == nil {
+				if denominator, err := strconv.ParseInt(parts[1], 10, 64); err == nil {
+					result := float64(numerator) / float64(denominator)
+					framerate = &result
+				}
+			}
+		}
+	}
+	return framerate
+}