You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
179 lines
6.4 KiB
179 lines
6.4 KiB
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using NzbDrone.Core.Datastore;
|
|
using NzbDrone.Core.ImportLists;
|
|
using NzbDrone.Core.Indexers;
|
|
using NzbDrone.Core.Messaging.Events;
|
|
using NzbDrone.Core.Music;
|
|
using NzbDrone.Core.Notifications;
|
|
using NzbDrone.Core.Profiles.Delay;
|
|
using NzbDrone.Core.Profiles.Releases;
|
|
using NzbDrone.Core.RootFolders;
|
|
|
|
namespace NzbDrone.Core.Tags
|
|
{
|
|
public interface ITagService
|
|
{
|
|
Tag GetTag(int tagId);
|
|
Tag GetTag(string tag);
|
|
TagDetails Details(int tagId);
|
|
List<TagDetails> Details();
|
|
List<Tag> All();
|
|
Tag Add(Tag tag);
|
|
Tag Update(Tag tag);
|
|
void Delete(int tagId);
|
|
}
|
|
|
|
public class TagService : ITagService
|
|
{
|
|
private readonly ITagRepository _repo;
|
|
private readonly IEventAggregator _eventAggregator;
|
|
private readonly IDelayProfileService _delayProfileService;
|
|
private readonly IImportListFactory _importListFactory;
|
|
private readonly INotificationFactory _notificationFactory;
|
|
private readonly IReleaseProfileService _releaseProfileService;
|
|
private readonly IArtistService _artistService;
|
|
private readonly IRootFolderService _rootFolderService;
|
|
private readonly IIndexerFactory _indexerService;
|
|
|
|
public TagService(ITagRepository repo,
|
|
IEventAggregator eventAggregator,
|
|
IDelayProfileService delayProfileService,
|
|
IImportListFactory importListFactory,
|
|
INotificationFactory notificationFactory,
|
|
IReleaseProfileService releaseProfileService,
|
|
IArtistService artistService,
|
|
IRootFolderService rootFolderService,
|
|
IIndexerFactory indexerService)
|
|
{
|
|
_repo = repo;
|
|
_eventAggregator = eventAggregator;
|
|
_delayProfileService = delayProfileService;
|
|
_importListFactory = importListFactory;
|
|
_notificationFactory = notificationFactory;
|
|
_releaseProfileService = releaseProfileService;
|
|
_artistService = artistService;
|
|
_rootFolderService = rootFolderService;
|
|
_indexerService = indexerService;
|
|
}
|
|
|
|
public Tag GetTag(int tagId)
|
|
{
|
|
return _repo.Get(tagId);
|
|
}
|
|
|
|
public Tag GetTag(string tag)
|
|
{
|
|
if (tag.All(char.IsDigit))
|
|
{
|
|
return _repo.Get(int.Parse(tag));
|
|
}
|
|
else
|
|
{
|
|
return _repo.GetByLabel(tag);
|
|
}
|
|
}
|
|
|
|
public TagDetails Details(int tagId)
|
|
{
|
|
var tag = GetTag(tagId);
|
|
var delayProfiles = _delayProfileService.AllForTag(tagId);
|
|
var importLists = _importListFactory.AllForTag(tagId);
|
|
var notifications = _notificationFactory.AllForTag(tagId);
|
|
var restrictions = _releaseProfileService.AllForTag(tagId);
|
|
var artist = _artistService.AllForTag(tagId);
|
|
var rootFolders = _rootFolderService.AllForTag(tagId);
|
|
var indexers = _indexerService.AllForTag(tagId);
|
|
|
|
return new TagDetails
|
|
{
|
|
Id = tagId,
|
|
Label = tag.Label,
|
|
DelayProfileIds = delayProfiles.Select(c => c.Id).ToList(),
|
|
ImportListIds = importLists.Select(c => c.Id).ToList(),
|
|
NotificationIds = notifications.Select(c => c.Id).ToList(),
|
|
RestrictionIds = restrictions.Select(c => c.Id).ToList(),
|
|
ArtistIds = artist.Select(c => c.Id).ToList(),
|
|
RootFolderIds = rootFolders.Select(c => c.Id).ToList(),
|
|
IndexerIds = indexers.Select(c => c.Id).ToList()
|
|
};
|
|
}
|
|
|
|
public List<TagDetails> Details()
|
|
{
|
|
var tags = All();
|
|
var delayProfiles = _delayProfileService.All();
|
|
var importLists = _importListFactory.All();
|
|
var notifications = _notificationFactory.All();
|
|
var restrictions = _releaseProfileService.All();
|
|
var artists = _artistService.GetAllArtists();
|
|
var rootFolders = _rootFolderService.All();
|
|
var indexers = _indexerService.All();
|
|
|
|
var details = new List<TagDetails>();
|
|
|
|
foreach (var tag in tags)
|
|
{
|
|
details.Add(new TagDetails
|
|
{
|
|
Id = tag.Id,
|
|
Label = tag.Label,
|
|
DelayProfileIds = delayProfiles.Where(c => c.Tags.Contains(tag.Id)).Select(c => c.Id).ToList(),
|
|
ImportListIds = importLists.Where(c => c.Tags.Contains(tag.Id)).Select(c => c.Id).ToList(),
|
|
NotificationIds = notifications.Where(c => c.Tags.Contains(tag.Id)).Select(c => c.Id).ToList(),
|
|
RestrictionIds = restrictions.Where(c => c.Tags.Contains(tag.Id)).Select(c => c.Id).ToList(),
|
|
ArtistIds = artists.Where(c => c.Tags.Contains(tag.Id)).Select(c => c.Id).ToList(),
|
|
RootFolderIds = rootFolders.Where(c => c.DefaultTags.Contains(tag.Id)).Select(c => c.Id).ToList(),
|
|
IndexerIds = indexers.Where(c => c.Tags.Contains(tag.Id)).Select(c => c.Id).ToList()
|
|
});
|
|
}
|
|
|
|
return details;
|
|
}
|
|
|
|
public List<Tag> All()
|
|
{
|
|
return _repo.All().OrderBy(t => t.Label).ToList();
|
|
}
|
|
|
|
public Tag Add(Tag tag)
|
|
{
|
|
var existingTag = _repo.FindByLabel(tag.Label);
|
|
|
|
if (existingTag != null)
|
|
{
|
|
return existingTag;
|
|
}
|
|
|
|
tag.Label = tag.Label.ToLowerInvariant();
|
|
|
|
_repo.Insert(tag);
|
|
_eventAggregator.PublishEvent(new TagsUpdatedEvent());
|
|
|
|
return tag;
|
|
}
|
|
|
|
public Tag Update(Tag tag)
|
|
{
|
|
tag.Label = tag.Label.ToLowerInvariant();
|
|
|
|
_repo.Update(tag);
|
|
_eventAggregator.PublishEvent(new TagsUpdatedEvent());
|
|
|
|
return tag;
|
|
}
|
|
|
|
public void Delete(int tagId)
|
|
{
|
|
var details = Details(tagId);
|
|
if (details.InUse)
|
|
{
|
|
throw new ModelConflictException(typeof(Tag), tagId, $"'{details.Label}' cannot be deleted since it's still in use");
|
|
}
|
|
|
|
_repo.Delete(tagId);
|
|
_eventAggregator.PublishEvent(new TagsUpdatedEvent());
|
|
}
|
|
}
|
|
}
|