using System; using System.Collections.Generic; using FluentValidation.Results; using NzbDrone.Core.ThingiProvider; using NzbDrone.Core.Music; namespace NzbDrone.Core.Notifications { public abstract class NotificationBase : INotification where TSettings : IProviderConfig, new() { protected const string ALBUM_GRABBED_TITLE = "Album Grabbed"; protected const string ALBUM_DOWNLOADED_TITLE = "Album Downloaded"; protected const string HEALTH_ISSUE_TITLE = "Health Check Failure"; protected const string DOWNLOAD_FAILURE_TITLE = "Download Failed"; protected const string IMPORT_FAILURE_TITLE = "Import Failed"; protected const string TRACK_RETAGGED_TITLE = "Track File Tags Updated"; protected const string ALBUM_GRABBED_TITLE_BRANDED = "Lidarr - " + ALBUM_GRABBED_TITLE; protected const string ALBUM_DOWNLOADED_TITLE_BRANDED = "Lidarr - " + ALBUM_DOWNLOADED_TITLE; protected const string HEALTH_ISSUE_TITLE_BRANDED = "Lidarr - " + HEALTH_ISSUE_TITLE; protected const string DOWNLOAD_FAILURE_TITLE_BRANDED = "Lidarr - " + DOWNLOAD_FAILURE_TITLE; protected const string IMPORT_FAILURE_TITLE_BRANDED = "Lidarr - " + IMPORT_FAILURE_TITLE; protected const string TRACK_RETAGGED_TITLE_BRANDED = "Lidarr - " + TRACK_RETAGGED_TITLE; public abstract string Name { get; } public Type ConfigContract => typeof(TSettings); public virtual ProviderMessage Message => null; public IEnumerable DefaultDefinitions => new List(); public ProviderDefinition Definition { get; set; } public abstract ValidationResult Test(); public abstract string Link { get; } public virtual void OnGrab(GrabMessage grabMessage) { } public virtual void OnReleaseImport(AlbumDownloadMessage message) { } public virtual void OnRename(Artist artist) { } public virtual void OnHealthIssue(HealthCheck.HealthCheck healthCheck) { } public virtual void OnDownloadFailure(DownloadFailedMessage message) { } public virtual void OnImportFailure(AlbumDownloadMessage message) { } public virtual void OnTrackRetag(TrackRetagMessage message) { } public bool SupportsOnGrab => HasConcreteImplementation("OnGrab"); public bool SupportsOnRename => HasConcreteImplementation("OnRename"); public bool SupportsOnReleaseImport => HasConcreteImplementation("OnReleaseImport"); public bool SupportsOnUpgrade => SupportsOnReleaseImport; public bool SupportsOnHealthIssue => HasConcreteImplementation("OnHealthIssue"); public bool SupportsOnDownloadFailure => HasConcreteImplementation("OnDownloadFailure"); public bool SupportsOnImportFailure => HasConcreteImplementation("OnImportFailure"); public bool SupportsOnTrackRetag => HasConcreteImplementation("OnTrackRetag"); protected TSettings Settings => (TSettings)Definition.Settings; public override string ToString() { return GetType().Name; } public virtual object RequestAction(string action, IDictionary query) { return null; } private bool HasConcreteImplementation(string methodName) { var method = GetType().GetMethod(methodName); if (method == null) { throw new MissingMethodException(GetType().Name, Name); } return !method.DeclaringType.IsAbstract; } } }