MeBox/internal/service/duplicate.go

// Package service — duplicate-file finder.
//
// DuplicateService computes a sparse-sample MD5 (head + middle + tail,
// 1 MiB each, plus the file size to break collisions) for every media
// file and groups identical hashes into "duplicate sets". The first row
// (preferring scraped + larger files) is kept as the primary; the rest
// get is_duplicate = true and duplicate_of pointing at the primary.
//
// Why sparse: a full hash on a 50 GB Blu-ray remux takes minutes; the
// 3-window 3 MiB sample is enough to differentiate real-world copies
// while finishing per-file in well under a second.
package service

import (
	"context"
	"crypto/md5"
	"encoding/hex"
	"errors"
	"fmt"
	"io"
	"os"
	"sort"

	"go.uber.org/zap"

	"github.com/ShukeBta/MediaStationGo/internal/model"
	"github.com/ShukeBta/MediaStationGo/internal/repository"
)

const sampleSize = 1 << 20 // 1 MiB per sample window

// DuplicateService is the entry point for the duplicate finder.
type DuplicateService struct {
	log  *zap.Logger
	repo *repository.Container
	hub  *Hub
}

// NewDuplicateService is the constructor.
func NewDuplicateService(log *zap.Logger, repo *repository.Container, hub *Hub) *DuplicateService {
	return &DuplicateService{log: log, repo: repo, hub: hub}
}

// Group describes one set of duplicates returned by Detect.
type Group struct {
	Hash       string        `json:"hash"`
	Primary    model.Media   `json:"primary"`
	Duplicates []model.Media `json:"duplicates"`
}

// Report is the summary the React UI displays.
type Report struct {
	TotalScanned int     `json:"total_scanned"`
	GroupsFound  int     `json:"groups_found"`
	ItemsMarked  int     `json:"items_marked"`
	Groups       []Group `json:"groups"`
}

// Detect walks every media row in the given library (or all libraries
// when libraryID is empty), computes a hash for the ones missing it,
// then groups by hash and marks duplicates in the DB.
func (d *DuplicateService) Detect(ctx context.Context, libraryID string) (*Report, error) {
	var rows []model.Media
	q := d.repo.DB.WithContext(ctx).Model(&model.Media{})
	if libraryID != "" {
		q = q.Where("library_id = ?", libraryID)
	}
	if err := q.Find(&rows).Error; err != nil {
		return nil, err
	}

	rep := &Report{TotalScanned: len(rows)}
	totalToHash := 0
	for i := range rows {
		if rows[i].FileHash == "" && rows[i].Path != "" {
			totalToHash++
		}
	}

	hashed := 0
	for i := range rows {
		select {
		case <-ctx.Done():
			return rep, ctx.Err()
		default:
		}
		if rows[i].FileHash != "" || rows[i].Path == "" {
			continue
		}
		h, err := SparseFileHash(rows[i].Path)
		if err != nil {
			d.log.Debug("hash failed", zap.String("path", rows[i].Path), zap.Error(err))
			continue
		}
		rows[i].FileHash = h
		if err := d.repo.DB.WithContext(ctx).
			Model(&model.Media{}).
			Where("id = ?", rows[i].ID).
			Update("file_hash", h).Error; err != nil {
			d.log.Warn("hash persist failed", zap.Error(err))
		}
		hashed++
		if d.hub != nil && totalToHash > 0 {
			d.hub.Publish("duplicate", map[string]any{
				"hashed":  hashed,
				"total":   totalToHash,
				"current": rows[i].Title,
			})
		}
	}

	// Group rows by file_hash.
	groups := make(map[string][]model.Media)
	for _, r := range rows {
		if r.FileHash == "" {
			continue
		}
		groups[r.FileHash] = append(groups[r.FileHash], r)
	}

	for hash, group := range groups {
		if len(group) < 2 {
			continue
		}
		primary := pickPrimary(group)
		dupes := make([]model.Media, 0, len(group)-1)
		for _, m := range group {
			if m.ID == primary.ID {
				continue
			}
			dupes = append(dupes, m)
			if err := d.repo.DB.WithContext(ctx).
				Model(&model.Media{}).
				Where("id = ?", m.ID).
				Updates(map[string]any{
					"is_duplicate": true,
					"duplicate_of": primary.ID,
				}).Error; err != nil {
				d.log.Warn("dup mark failed", zap.Error(err))
				continue
			}
			rep.ItemsMarked++
		}
		rep.Groups = append(rep.Groups, Group{
			Hash:       hash,
			Primary:    primary,
			Duplicates: dupes,
		})
	}
	rep.GroupsFound = len(rep.Groups)
	if d.hub != nil {
		d.hub.Publish("duplicate", map[string]any{
			"finished": true,
			"groups":   rep.GroupsFound,
			"marked":   rep.ItemsMarked,
		})
	}
	return rep, nil
}

// Unmark clears the is_duplicate flag for every row in the given library
// (or all when libraryID is empty). Useful when the operator deletes the
// physical duplicates manually.
func (d *DuplicateService) Unmark(ctx context.Context, libraryID string) (int64, error) {
	q := d.repo.DB.WithContext(ctx).Model(&model.Media{}).Where("is_duplicate = ?", true)
	if libraryID != "" {
		q = q.Where("library_id = ?", libraryID)
	}
	res := q.Updates(map[string]any{"is_duplicate": false, "duplicate_of": ""})
	return res.RowsAffected, res.Error
}

// pickPrimary picks the "best" media row to keep: prefer scraped > size > id.
func pickPrimary(group []model.Media) model.Media {
	sort.SliceStable(group, func(i, j int) bool {
		ai, aj := group[i].ScrapeStatus == "matched", group[j].ScrapeStatus == "matched"
		if ai != aj {
			return ai
		}
		if group[i].SizeBytes != group[j].SizeBytes {
			return group[i].SizeBytes > group[j].SizeBytes
		}
		return group[i].ID < group[j].ID
	})
	return group[0]
}

// SparseFileHash computes the head+mid+tail MD5 of a file, suffixed with
// the file size so two files that happen to collide on the sample window
// but differ in length are still distinguishable.
func SparseFileHash(path string) (string, error) {
	if path == "" {
		return "", errors.New("empty path")
	}
	f, err := os.Open(path)
	if err != nil {
		return "", err
	}
	defer f.Close()
	st, err := f.Stat()
	if err != nil {
		return "", err
	}
	size := st.Size()
	h := md5.New()
	if size <= int64(sampleSize)*3 {
		if _, err := io.Copy(h, f); err != nil {
			return "", err
		}
		return fmt.Sprintf("%s-%d", hex.EncodeToString(h.Sum(nil)), size), nil
	}
	buf := make([]byte, sampleSize)
	// head
	if _, err := io.ReadFull(f, buf); err != nil {
		return "", err
	}
	h.Write(buf)
	// middle
	if _, err := f.Seek(size/2-int64(sampleSize)/2, io.SeekStart); err != nil {
		return "", err
	}
	if _, err := io.ReadFull(f, buf); err != nil {
		return "", err
	}
	h.Write(buf)
	// tail
	if _, err := f.Seek(size-int64(sampleSize), io.SeekStart); err != nil {
		return "", err
	}
	if _, err := io.ReadFull(f, buf); err != nil {
		return "", err
	}
	h.Write(buf)
	return fmt.Sprintf("%s-%d", hex.EncodeToString(h.Sum(nil)), size), nil
}