diff --git a/frontend/src/Components/Form/FormInputGroup.js b/frontend/src/Components/Form/FormInputGroup.js
index f7b2ce75e..7a3191cdc 100644
--- a/frontend/src/Components/Form/FormInputGroup.js
+++ b/frontend/src/Components/Form/FormInputGroup.js
@@ -22,6 +22,7 @@ import PasswordInput from './PasswordInput';
import PathInputConnector from './PathInputConnector';
import QualityProfileSelectInputConnector from './QualityProfileSelectInputConnector';
import RootFolderSelectInputConnector from './RootFolderSelectInputConnector';
+import SeriesTagInput from './SeriesTagInput';
import SeriesTypeSelectInput from './SeriesTypeSelectInput';
import TagInputConnector from './TagInputConnector';
import TagSelectInputConnector from './TagSelectInputConnector';
@@ -87,6 +88,9 @@ function getComponent(type) {
case inputTypes.DYNAMIC_SELECT:
return EnhancedSelectInputConnector;
+ case inputTypes.SERIES_TAG:
+ return SeriesTagInput;
+
case inputTypes.SERIES_TYPE_SELECT:
return SeriesTypeSelectInput;
diff --git a/frontend/src/Components/Form/ProviderFieldFormGroup.js b/frontend/src/Components/Form/ProviderFieldFormGroup.js
index a184aa1ec..4fcf99cc0 100644
--- a/frontend/src/Components/Form/ProviderFieldFormGroup.js
+++ b/frontend/src/Components/Form/ProviderFieldFormGroup.js
@@ -27,6 +27,8 @@ function getType({ type, selectOptionsProviderAction }) {
return inputTypes.DYNAMIC_SELECT;
}
return inputTypes.SELECT;
+ case 'seriesTag':
+ return inputTypes.SERIES_TAG;
case 'tag':
return inputTypes.TEXT_TAG;
case 'tagSelect':
diff --git a/frontend/src/Components/Form/SeriesTagInput.tsx b/frontend/src/Components/Form/SeriesTagInput.tsx
new file mode 100644
index 000000000..3d8279aa6
--- /dev/null
+++ b/frontend/src/Components/Form/SeriesTagInput.tsx
@@ -0,0 +1,53 @@
+import React, { useCallback } from 'react';
+import TagInputConnector from './TagInputConnector';
+
+interface SeriesTageInputProps {
+ name: string;
+ value: number | number[];
+ onChange: ({
+ name,
+ value,
+ }: {
+ name: string;
+ value: number | number[];
+ }) => void;
+}
+
+export default function SeriesTagInput(props: SeriesTageInputProps) {
+ const { value, onChange, ...otherProps } = props;
+ const isArray = Array.isArray(value);
+
+ const handleChange = useCallback(
+ ({ name, value: newValue }: { name: string; value: number[] }) => {
+ if (isArray) {
+ onChange({ name, value: newValue });
+ } else {
+ onChange({
+ name,
+ value: newValue.length ? newValue[newValue.length - 1] : 0,
+ });
+ }
+ },
+ [isArray, onChange]
+ );
+
+ let finalValue: number[] = [];
+
+ if (isArray) {
+ finalValue = value;
+ } else if (value === 0) {
+ finalValue = [];
+ } else {
+ finalValue = [value];
+ }
+
+ return (
+ // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+ // @ts-ignore 2786 'TagInputConnector' isn't typed yet
+
+ );
+}
diff --git a/frontend/src/Helpers/Props/inputTypes.js b/frontend/src/Helpers/Props/inputTypes.js
index dcf4b539c..a71c28d8c 100644
--- a/frontend/src/Helpers/Props/inputTypes.js
+++ b/frontend/src/Helpers/Props/inputTypes.js
@@ -17,6 +17,7 @@ export const LANGUAGE_SELECT = 'languageSelect';
export const DOWNLOAD_CLIENT_SELECT = 'downloadClientSelect';
export const ROOT_FOLDER_SELECT = 'rootFolderSelect';
export const SELECT = 'select';
+export const SERIES_TAG = 'seriesTag';
export const DYNAMIC_SELECT = 'dynamicSelect';
export const SERIES_TYPE_SELECT = 'seriesTypeSelect';
export const TAG = 'tag';
@@ -45,6 +46,7 @@ export const all = [
ROOT_FOLDER_SELECT,
LANGUAGE_SELECT,
SELECT,
+ SERIES_TAG,
DYNAMIC_SELECT,
SERIES_TYPE_SELECT,
TAG,
diff --git a/frontend/src/Settings/Tags/TagInUse.js b/frontend/src/Settings/Tags/TagInUse.js
index 9fb57d230..27228fa2e 100644
--- a/frontend/src/Settings/Tags/TagInUse.js
+++ b/frontend/src/Settings/Tags/TagInUse.js
@@ -12,7 +12,7 @@ export default function TagInUse(props) {
return null;
}
- if (count > 1 && labelPlural ) {
+ if (count > 1 && labelPlural) {
return (
{count} {labelPlural.toLowerCase()}
diff --git a/src/NzbDrone.Core.Test/Housekeeping/Housekeepers/CleanupUnusedTagsFixture.cs b/src/NzbDrone.Core.Test/Housekeeping/Housekeepers/CleanupUnusedTagsFixture.cs
index a26f08a67..99b6676be 100644
--- a/src/NzbDrone.Core.Test/Housekeeping/Housekeepers/CleanupUnusedTagsFixture.cs
+++ b/src/NzbDrone.Core.Test/Housekeeping/Housekeepers/CleanupUnusedTagsFixture.cs
@@ -1,6 +1,9 @@
+using System.Collections.Generic;
using FizzWare.NBuilder;
using FluentAssertions;
using NUnit.Framework;
+using NzbDrone.Core.AutoTagging;
+using NzbDrone.Core.AutoTagging.Specifications;
using NzbDrone.Core.Housekeeping.Housekeepers;
using NzbDrone.Core.Profiles.Releases;
using NzbDrone.Core.Tags;
@@ -45,5 +48,35 @@ namespace NzbDrone.Core.Test.Housekeeping.Housekeepers
Subject.Clean();
AllStoredModels.Should().HaveCount(1);
}
+
+ [Test]
+ public void should_not_delete_used_auto_tagging_tag_specification_tags()
+ {
+ var tags = Builder
+ .CreateListOfSize(2)
+ .All()
+ .With(x => x.Id = 0)
+ .BuildList();
+ Db.InsertMany(tags);
+
+ var autoTags = Builder.CreateListOfSize(1)
+ .All()
+ .With(x => x.Id = 0)
+ .With(x => x.Specifications = new List
+ {
+ new TagSpecification
+ {
+ Name = "Test",
+ Value = tags[0].Id
+ }
+ })
+ .BuildList();
+
+ Mocker.GetMock().Setup(s => s.All())
+ .Returns(autoTags);
+
+ Subject.Clean();
+ AllStoredModels.Should().HaveCount(1);
+ }
}
}
diff --git a/src/NzbDrone.Core/Annotations/FieldDefinitionAttribute.cs b/src/NzbDrone.Core/Annotations/FieldDefinitionAttribute.cs
index a0a1896e0..398376117 100644
--- a/src/NzbDrone.Core/Annotations/FieldDefinitionAttribute.cs
+++ b/src/NzbDrone.Core/Annotations/FieldDefinitionAttribute.cs
@@ -85,7 +85,8 @@ namespace NzbDrone.Core.Annotations
Device,
TagSelect,
RootFolder,
- QualityProfile
+ QualityProfile,
+ SeriesTag
}
public enum HiddenType
diff --git a/src/NzbDrone.Core/AutoTagging/Specifications/TagSpecification.cs b/src/NzbDrone.Core/AutoTagging/Specifications/TagSpecification.cs
new file mode 100644
index 000000000..c736f8899
--- /dev/null
+++ b/src/NzbDrone.Core/AutoTagging/Specifications/TagSpecification.cs
@@ -0,0 +1,36 @@
+using FluentValidation;
+using NzbDrone.Core.Annotations;
+using NzbDrone.Core.Tv;
+using NzbDrone.Core.Validation;
+
+namespace NzbDrone.Core.AutoTagging.Specifications
+{
+ public class TagSpecificationValidator : AbstractValidator
+ {
+ public TagSpecificationValidator()
+ {
+ RuleFor(c => c.Value).GreaterThan(0);
+ }
+ }
+
+ public class TagSpecification : AutoTaggingSpecificationBase
+ {
+ private static readonly TagSpecificationValidator Validator = new ();
+
+ public override int Order => 1;
+ public override string ImplementationName => "Tag";
+
+ [FieldDefinition(1, Label = "AutoTaggingSpecificationTag", Type = FieldType.SeriesTag)]
+ public int Value { get; set; }
+
+ protected override bool IsSatisfiedByWithoutNegate(Series series)
+ {
+ return series.Tags.Contains(Value);
+ }
+
+ public override NzbDroneValidationResult Validate()
+ {
+ return new NzbDroneValidationResult(Validator.Validate(this));
+ }
+ }
+}
diff --git a/src/NzbDrone.Core/Housekeeping/Housekeepers/CleanupUnusedTags.cs b/src/NzbDrone.Core/Housekeeping/Housekeepers/CleanupUnusedTags.cs
index 9bd726ba3..c68d053f9 100644
--- a/src/NzbDrone.Core/Housekeeping/Housekeepers/CleanupUnusedTags.cs
+++ b/src/NzbDrone.Core/Housekeeping/Housekeepers/CleanupUnusedTags.cs
@@ -2,6 +2,8 @@ using System.Collections.Generic;
using System.Data;
using System.Linq;
using Dapper;
+using NzbDrone.Core.AutoTagging;
+using NzbDrone.Core.AutoTagging.Specifications;
using NzbDrone.Core.Datastore;
namespace NzbDrone.Core.Housekeeping.Housekeepers
@@ -9,17 +11,24 @@ namespace NzbDrone.Core.Housekeeping.Housekeepers
public class CleanupUnusedTags : IHousekeepingTask
{
private readonly IMainDatabase _database;
+ private readonly IAutoTaggingRepository _autoTaggingRepository;
- public CleanupUnusedTags(IMainDatabase database)
+ public CleanupUnusedTags(IMainDatabase database, IAutoTaggingRepository autoTaggingRepository)
{
_database = database;
+ _autoTaggingRepository = autoTaggingRepository;
}
public void Clean()
{
using var mapper = _database.OpenConnection();
- var usedTags = new[] { "Series", "Notifications", "DelayProfiles", "ReleaseProfiles", "ImportLists", "Indexers", "AutoTagging", "DownloadClients" }
+ var usedTags = new[]
+ {
+ "Series", "Notifications", "DelayProfiles", "ReleaseProfiles", "ImportLists", "Indexers",
+ "AutoTagging", "DownloadClients"
+ }
.SelectMany(v => GetUsedTags(v, mapper))
+ .Concat(GetAutoTaggingTagSpecificationTags(mapper))
.Distinct()
.ToArray();
@@ -37,10 +46,31 @@ namespace NzbDrone.Core.Housekeeping.Housekeepers
private int[] GetUsedTags(string table, IDbConnection mapper)
{
- return mapper.Query>($"SELECT DISTINCT \"Tags\" FROM \"{table}\" WHERE NOT \"Tags\" = '[]' AND NOT \"Tags\" IS NULL")
+ return mapper
+ .Query>(
+ $"SELECT DISTINCT \"Tags\" FROM \"{table}\" WHERE NOT \"Tags\" = '[]' AND NOT \"Tags\" IS NULL")
.SelectMany(x => x)
.Distinct()
.ToArray();
}
+
+ private List GetAutoTaggingTagSpecificationTags(IDbConnection mapper)
+ {
+ var tags = new List();
+ var autoTags = _autoTaggingRepository.All();
+
+ foreach (var autoTag in autoTags)
+ {
+ foreach (var specification in autoTag.Specifications)
+ {
+ if (specification is TagSpecification tagSpec)
+ {
+ tags.Add(tagSpec.Value);
+ }
+ }
+ }
+
+ return tags;
+ }
}
}
diff --git a/src/NzbDrone.Core/Localization/Core/en.json b/src/NzbDrone.Core/Localization/Core/en.json
index c751eefe9..f6b312928 100644
--- a/src/NzbDrone.Core/Localization/Core/en.json
+++ b/src/NzbDrone.Core/Localization/Core/en.json
@@ -136,6 +136,7 @@
"AutoTaggingSpecificationRootFolder": "Root Folder",
"AutoTaggingSpecificationSeriesType": "Series Type",
"AutoTaggingSpecificationStatus": "Status",
+ "AutoTaggingSpecificationTag": "Tag",
"Automatic": "Automatic",
"AutomaticAdd": "Automatic Add",
"AutomaticSearch": "Automatic Search",
diff --git a/src/NzbDrone.Core/Tags/TagService.cs b/src/NzbDrone.Core/Tags/TagService.cs
index da67f0705..b97b279a5 100644
--- a/src/NzbDrone.Core/Tags/TagService.cs
+++ b/src/NzbDrone.Core/Tags/TagService.cs
@@ -1,6 +1,7 @@
using System.Collections.Generic;
using System.Linq;
using NzbDrone.Core.AutoTagging;
+using NzbDrone.Core.AutoTagging.Specifications;
using NzbDrone.Core.Datastore;
using NzbDrone.Core.Download;
using NzbDrone.Core.ImportLists;
@@ -120,7 +121,7 @@ namespace NzbDrone.Core.Tags
var restrictions = _releaseProfileService.All();
var series = _seriesService.GetAllSeriesTags();
var indexers = _indexerService.All();
- var autotags = _autoTaggingService.All();
+ var autoTags = _autoTaggingService.All();
var downloadClients = _downloadClientFactory.All();
var details = new List();
@@ -137,7 +138,7 @@ namespace NzbDrone.Core.Tags
RestrictionIds = restrictions.Where(c => c.Tags.Contains(tag.Id)).Select(c => c.Id).ToList(),
SeriesIds = series.Where(c => c.Value.Contains(tag.Id)).Select(c => c.Key).ToList(),
IndexerIds = indexers.Where(c => c.Tags.Contains(tag.Id)).Select(c => c.Id).ToList(),
- AutoTagIds = autotags.Where(c => c.Tags.Contains(tag.Id)).Select(c => c.Id).ToList(),
+ AutoTagIds = GetAutoTagIds(tag, autoTags),
DownloadClientIds = downloadClients.Where(c => c.Tags.Contains(tag.Id)).Select(c => c.Id).ToList(),
});
}
@@ -188,5 +189,23 @@ namespace NzbDrone.Core.Tags
_repo.Delete(tagId);
_eventAggregator.PublishEvent(new TagsUpdatedEvent());
}
+
+ private List GetAutoTagIds(Tag tag, List autoTags)
+ {
+ var autoTagIds = autoTags.Where(c => c.Tags.Contains(tag.Id)).Select(c => c.Id).ToList();
+
+ foreach (var autoTag in autoTags)
+ {
+ foreach (var specification in autoTag.Specifications)
+ {
+ if (specification is TagSpecification)
+ {
+ autoTagIds.Add(autoTag.Id);
+ }
+ }
+ }
+
+ return autoTagIds.Distinct().ToList();
+ }
}
}