OpenFlare/backend/plugins/domain/upload/ingest/remove.go

// Copyright 2026 Arctel.net
// SPDX-License-Identifier: Apache-2.0

package ingest

import (
	"context"

	"gorm.io/gorm"

	uploadcache "Wavelet/plugins/domain/upload/cache"
	"Wavelet/plugins/domain/upload/models"
	"Wavelet/plugins/domain/upload/repository"
	"Wavelet/plugins/domain/upload/shared"
	uploadstats "Wavelet/plugins/domain/upload/stats"
)

// Remove soft-deletes an upload and decrements incremental stats.
func Remove(ctx context.Context, uploadID uint64) (models.Upload, error) {
	upload, err := repository.GetActiveUploadByID(ctx, uploadID)
	if err != nil {
		return models.Upload{}, err
	}
	if err := softDeleteUploadWithStats(ctx, &upload); err != nil {
		return models.Upload{}, err
	}
	upload.Status = models.UploadStatusDeleted
	return upload, nil
}

// RemoveOwned soft-deletes an upload owned by userID and decrements incremental stats.
func RemoveOwned(ctx context.Context, userID, uploadID uint64) (models.Upload, error) {
	upload, err := repository.GetActiveUploadByID(ctx, uploadID)
	if err != nil {
		return models.Upload{}, err
	}
	if upload.UserID != userID {
		return models.Upload{}, ErrForbidden
	}
	if err := softDeleteUploadWithStats(ctx, &upload); err != nil {
		return models.Upload{}, err
	}
	upload.Status = models.UploadStatusDeleted
	return upload, nil
}

func softDeleteUploadWithStats(ctx context.Context, upload *models.Upload) error {
	statsSnapshot := *upload
	db := shared.GetDB(ctx)
	if db != nil {
		if err := db.Transaction(func(tx *gorm.DB) error {
			if err := repository.SoftDeleteUploadTx(tx, upload); err != nil {
				return err
			}
			return uploadstats.ApplyUploadStatsDeltaTx(tx, &statsSnapshot, -1)
		}); err != nil {
			return err
		}
	}
	uploadcache.EvictUploadMeta(ctx, upload.ID)
	return nil
}