qbtmud/Lantean.QBitTorrentClient/ApiClientExtensions.cs

using System.Linq;
using Lantean.QBitTorrentClient.Models;

namespace Lantean.QBitTorrentClient
{
    public static class ApiClientExtensions
    {
        public static Task StopTorrent(this IApiClient apiClient, string hash)
        {
            return apiClient.StopTorrents(null, hash);
        }

        public static Task StopTorrents(this IApiClient apiClient, IEnumerable<string> hashes)
        {
            return apiClient.StopTorrents(null, hashes.ToArray());
        }

        public static Task StopAllTorrents(this IApiClient apiClient)
        {
            return apiClient.StopTorrents(true);
        }

        public static Task StartTorrent(this IApiClient apiClient, string hash)
        {
            return apiClient.StartTorrents(null, hash);
        }

        public static Task StartTorrents(this IApiClient apiClient, IEnumerable<string> hashes)
        {
            return apiClient.StartTorrents(null, hashes.ToArray());
        }

        public static Task StartAllTorrents(this IApiClient apiClient)
        {
            return apiClient.StartTorrents(true);
        }

        public static Task DeleteTorrent(this IApiClient apiClient, string hash, bool deleteFiles)
        {
            return apiClient.DeleteTorrents(null, deleteFiles, hash);
        }

        public static Task DeleteTorrents(this IApiClient apiClient, IEnumerable<string> hashes, bool deleteFiles)
        {
            return apiClient.DeleteTorrents(null, deleteFiles, hashes.ToArray());
        }

        public static Task DeleteAllTorrents(this IApiClient apiClient, bool deleteFiles)
        {
            return apiClient.DeleteTorrents(true, deleteFiles);
        }

        public static async Task<Torrent?> GetTorrent(this IApiClient apiClient, string hash)
        {
            var torrents = await apiClient.GetTorrentList(hashes: hash);

            if (torrents.Count == 0)
            {
                return null;
            }

            return torrents[0];
        }

        public static Task SetTorrentCategory(this IApiClient apiClient, string category, string hash)
        {
            return apiClient.SetTorrentCategory(category, null, hash);
        }

        public static Task SetTorrentCategory(this IApiClient apiClient, string category, IEnumerable<string> hashes)
        {
            return apiClient.SetTorrentCategory(category, null, hashes.ToArray());
        }

        public static Task RemoveTorrentCategory(this IApiClient apiClient, string hash)
        {
            return apiClient.SetTorrentCategory(string.Empty, null, hash);
        }

        public static Task RemoveTorrentCategory(this IApiClient apiClient, IEnumerable<string> hashes)
        {
            return apiClient.SetTorrentCategory(string.Empty, null, hashes.ToArray());
        }

        public static Task RemoveTorrentTags(this IApiClient apiClient, IEnumerable<string> tags, string hash)
        {
            return apiClient.RemoveTorrentTags(tags, null, hash);
        }

        public static Task RemoveTorrentTags(this IApiClient apiClient, IEnumerable<string> tags, IEnumerable<string> hashes)
        {
            return apiClient.RemoveTorrentTags(tags, null, hashes.ToArray());
        }

        public static Task RemoveTorrentTag(this IApiClient apiClient, string tag, string hash)
        {
            return apiClient.RemoveTorrentTags([tag], hash);
        }

        public static Task RemoveTorrentTag(this IApiClient apiClient, string tag, IEnumerable<string> hashes)
        {
            return apiClient.RemoveTorrentTags([tag], null, hashes.ToArray());
        }

        public static Task AddTorrentTags(this IApiClient apiClient, IEnumerable<string> tags, string hash)
        {
            return apiClient.AddTorrentTags(tags, null, hash);
        }

        public static Task AddTorrentTags(this IApiClient apiClient, IEnumerable<string> tags, IEnumerable<string> hashes)
        {
            return apiClient.AddTorrentTags(tags, null, hashes.ToArray());
        }

        public static Task AddTorrentTag(this IApiClient apiClient, string tag, string hash)
        {
            return apiClient.AddTorrentTags([tag], hash);
        }

        public static Task AddTorrentTag(this IApiClient apiClient, string tag, IEnumerable<string> hashes)
        {
            return apiClient.AddTorrentTags([tag], null, hashes.ToArray());
        }

        public static Task RecheckTorrent(this IApiClient apiClient, string hash)
        {
            return apiClient.RecheckTorrents(null, hash);
        }

        public static Task ReannounceTorrent(this IApiClient apiClient, string hash)
        {
            return apiClient.ReannounceTorrents(null, null, hash);
        }

        public static async Task<IEnumerable<string>> RemoveUnusedCategories(this IApiClient apiClient)
        {
            var torrents = await apiClient.GetTorrentList();
            var categories = await apiClient.GetAllCategories();

            var selectedCategories = torrents.Select(t => t.Category).Distinct().ToList();

            var unusedCategories = categories.Values.Select(v => v.Name).Except(selectedCategories).Where(v => v is not null).Select(v => v!).ToArray();

            await apiClient.RemoveCategories(unusedCategories);

            return unusedCategories;
        }

        public static async Task<IEnumerable<string>> RemoveUnusedTags(this IApiClient apiClient)
        {
            var torrents = await apiClient.GetTorrentList();
            var tags = await apiClient.GetAllTags();

            var selectedTags = torrents.Where(t => t.Tags is not null).SelectMany(t => t.Tags!).Distinct().ToList();

            var unusedTags = tags.Except(selectedTags).ToArray();

            await apiClient.DeleteTags(unusedTags);

            return unusedTags;
        }
    }
}