Cleanuparr/code/backend/Cleanuparr.Persistence/Models/Configuration/QueueCleaner/QueueRule.cs

using System.ComponentModel.DataAnnotations;
using System.ComponentModel.DataAnnotations.Schema;
using Cleanuparr.Domain.Entities;
using Cleanuparr.Domain.Enums;

namespace Cleanuparr.Persistence.Models.Configuration.QueueCleaner;

public abstract record QueueRule : IConfig, IQueueRule
{
    [Key]
    [DatabaseGenerated(DatabaseGeneratedOption.Identity)]
    public Guid Id { get; init; } = Guid.NewGuid();
    
    public Guid QueueCleanerConfigId { get; set; }
    
    public QueueCleanerConfig QueueCleanerConfig { get; set; } = null!;
    
    public required string Name { get; init; }
    
    public bool Enabled { get; init; } = true;
    
    public int MaxStrikes { get; init; } = 3;
    
    public TorrentPrivacyType PrivacyType { get; init; } = TorrentPrivacyType.Public;
    
    public ushort MinCompletionPercentage { get; init; } = 0;
    
    public ushort MaxCompletionPercentage { get; init; }
    
    public bool DeletePrivateTorrentsFromClient { get; init; } = false;
    
    public abstract bool MatchesTorrent(ITorrentItemWrapper torrent);
    
    public virtual void Validate()
    {
        if (string.IsNullOrWhiteSpace(Name))
        {
            throw new Cleanuparr.Domain.Exceptions.ValidationException("Rule name cannot be empty");
        }

        if (MaxStrikes < 3)
        {
            throw new Cleanuparr.Domain.Exceptions.ValidationException("Max strikes must be at least 3");
        }

        if (MinCompletionPercentage > 100)
        {
            throw new Cleanuparr.Domain.Exceptions.ValidationException("Minimum completion percentage must be between 0 and 100");
        }

        if (MaxCompletionPercentage > 100)
        {
            throw new Cleanuparr.Domain.Exceptions.ValidationException("Maximum completion percentage must be between 0 and 100");
        }

        if (MaxCompletionPercentage < MinCompletionPercentage)
        {
            throw new Cleanuparr.Domain.Exceptions.ValidationException("Maximum completion percentage must be greater than or equal to the minimum completion percentage");
        }
    }
    
    protected bool MatchesPrivacyType(bool isPrivate)
    {
        return PrivacyType switch
        {
            TorrentPrivacyType.Public => !isPrivate,
            TorrentPrivacyType.Private => isPrivate,
            TorrentPrivacyType.Both => true,
            _ => true
        };
    }
    
    protected bool MatchesCompletionPercentage(double completionPercentage)
    {
        if (MaxCompletionPercentage < MinCompletionPercentage)
        {
            return false;
        }

        bool meetsLowerBound = MinCompletionPercentage == 0
            ? completionPercentage >= 0
            : completionPercentage > MinCompletionPercentage;

        bool meetsUpperBound = completionPercentage <= MaxCompletionPercentage;

        return meetsLowerBound && meetsUpperBound;
    }
}