package scanner

import (
	"context"
	"strings"
	"time"

	"github.com/Silo-Server/silo-server/internal/models"
)

// NeedsCriticalProbeRepair reports whether playback-critical probe metadata is
// missing and the file should be reprobed before making playback decisions.
func NeedsCriticalProbeRepair(file *models.MediaFile) bool {
	if file == nil {
		return true
	}
	if strings.TrimSpace(file.ProbeSource) == "" || file.ProbeUpdatedAt == nil {
		return true
	}
	if file.Duration <= 0 {
		return true
	}
	if strings.TrimSpace(file.Container) == "" {
		return true
	}
	if strings.TrimSpace(file.CodecVideo) == "" {
		return true
	}
	if strings.TrimSpace(file.CodecAudio) == "" {
		return true
	}
	if strings.TrimSpace(file.Resolution) == "" {
		return true
	}
	if len(file.VideoTracks) == 0 {
		return true
	}
	if len(file.AudioTracks) == 0 {
		return true
	}
	if file.Chapters == nil {
		return true
	}
	return false
}

// PlaybackProbeEnsurer repairs missing playback-critical probe metadata on
// demand by running a local ffprobe and persisting the result.
type PlaybackProbeEnsurer struct {
	fileRepo    *FileRepository
	ffprobePath string
	timeout     time.Duration
}

func NewPlaybackProbeEnsurer(fileRepo *FileRepository, ffprobePath string, timeout time.Duration) *PlaybackProbeEnsurer {
	return &PlaybackProbeEnsurer{
		fileRepo:    fileRepo,
		ffprobePath: ffprobePath,
		timeout:     timeout,
	}
}

func (e *PlaybackProbeEnsurer) Ensure(ctx context.Context, file *models.MediaFile) (*models.MediaFile, error) {
	if file == nil || !NeedsCriticalProbeRepair(file) {
		return file, nil
	}
	if e == nil || e.fileRepo == nil || strings.TrimSpace(e.ffprobePath) == "" {
		return file, nil
	}

	timeout := e.timeout
	if timeout <= 0 {
		timeout = 5 * time.Second
	}
	probeCtx, cancel := context.WithTimeout(ctx, timeout)
	defer cancel()

	probe, err := ProbeFile(probeCtx, e.ffprobePath, file.FilePath)
	if err != nil || probe == nil {
		return file, err
	}

	updated := *file
	applyProbeData(&updated, probe, "local")
	return e.fileRepo.Upsert(ctx, updated)
}