95 lines
2.3 KiB
Go
95 lines
2.3 KiB
Go
|
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"
|
||
|
"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)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
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
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
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
|
||
|
}
|