1
0
mirror of https://github.com/Sonarr/Sonarr.git synced 2024-12-14 11:23:42 +02:00

New: Add Notifiarr connection

This commit is contained in:
bakerboy448 2022-11-06 10:23:51 -06:00 committed by GitHub
parent e5d6e569cf
commit 2fb5655789
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 383 additions and 0 deletions

View File

@ -0,0 +1,225 @@
using System;
using System.Collections.Generic;
using System.Collections.Specialized;
using System.IO;
using System.Linq;
using FluentValidation.Results;
using NzbDrone.Common.Extensions;
using NzbDrone.Core.HealthCheck;
using NzbDrone.Core.MediaFiles;
using NzbDrone.Core.MediaFiles.MediaInfo;
using NzbDrone.Core.Tv;
namespace NzbDrone.Core.Notifications.Notifiarr
{
public class Notifiarr : NotificationBase<NotifiarrSettings>
{
private readonly INotifiarrProxy _proxy;
public Notifiarr(INotifiarrProxy proxy)
{
_proxy = proxy;
}
public override string Link => "https://notifiarr.com";
public override string Name => "Notifiarr";
public override void OnGrab(GrabMessage message)
{
var series = message.Series;
var remoteEpisode = message.Episode;
var releaseGroup = remoteEpisode.ParsedEpisodeInfo.ReleaseGroup;
var variables = new StringDictionary();
variables.Add("Sonarr_EventType", "Grab");
variables.Add("Sonarr_Series_Id", series.Id.ToString());
variables.Add("Sonarr_Series_Title", series.Title);
variables.Add("Sonarr_Series_TvdbId", series.TvdbId.ToString());
variables.Add("Sonarr_Series_TvMazeId", series.TvMazeId.ToString());
variables.Add("Sonarr_Series_ImdbId", series.ImdbId ?? string.Empty);
variables.Add("Sonarr_Series_Type", series.SeriesType.ToString());
variables.Add("Sonarr_Release_EpisodeCount", remoteEpisode.Episodes.Count.ToString());
variables.Add("Sonarr_Release_SeasonNumber", remoteEpisode.Episodes.First().SeasonNumber.ToString());
variables.Add("Sonarr_Release_EpisodeNumbers", string.Join(",", remoteEpisode.Episodes.Select(e => e.EpisodeNumber)));
variables.Add("Sonarr_Release_AbsoluteEpisodeNumbers", string.Join(",", remoteEpisode.Episodes.Select(e => e.AbsoluteEpisodeNumber)));
variables.Add("Sonarr_Release_EpisodeAirDates", string.Join(",", remoteEpisode.Episodes.Select(e => e.AirDate)));
variables.Add("Sonarr_Release_EpisodeAirDatesUtc", string.Join(",", remoteEpisode.Episodes.Select(e => e.AirDateUtc)));
variables.Add("Sonarr_Release_EpisodeTitles", string.Join("|", remoteEpisode.Episodes.Select(e => e.Title)));
variables.Add("Sonarr_Release_Title", remoteEpisode.Release.Title);
variables.Add("Sonarr_Release_Indexer", remoteEpisode.Release.Indexer ?? string.Empty);
variables.Add("Sonarr_Release_Size", remoteEpisode.Release.Size.ToString());
variables.Add("Sonarr_Release_Quality", remoteEpisode.ParsedEpisodeInfo.Quality.Quality.Name);
variables.Add("Sonarr_Release_QualityVersion", remoteEpisode.ParsedEpisodeInfo.Quality.Revision.Version.ToString());
variables.Add("Sonarr_Release_ReleaseGroup", releaseGroup ?? string.Empty);
variables.Add("Sonarr_Download_Client", message.DownloadClientName ?? string.Empty);
variables.Add("Sonarr_Download_Client_Type", message.DownloadClientType ?? string.Empty);
variables.Add("Sonarr_Download_Id", message.DownloadId ?? string.Empty);
variables.Add("Sonarr_Release_CustomFormat", string.Join("|", remoteEpisode.CustomFormats));
variables.Add("Sonarr_Release_CustomFormatScore", remoteEpisode.CustomFormatScore.ToString());
_proxy.SendNotification(variables, Settings);
}
public override void OnDownload(DownloadMessage message)
{
var series = message.Series;
var episodeFile = message.EpisodeFile;
var sourcePath = message.SourcePath;
var variables = new StringDictionary();
variables.Add("Sonarr_EventType", "Download");
variables.Add("Sonarr_IsUpgrade", message.OldFiles.Any().ToString());
variables.Add("Sonarr_Series_Id", series.Id.ToString());
variables.Add("Sonarr_Series_Title", series.Title);
variables.Add("Sonarr_Series_Path", series.Path);
variables.Add("Sonarr_Series_TvdbId", series.TvdbId.ToString());
variables.Add("Sonarr_Series_TvMazeId", series.TvMazeId.ToString());
variables.Add("Sonarr_Series_ImdbId", series.ImdbId ?? string.Empty);
variables.Add("Sonarr_Series_Type", series.SeriesType.ToString());
variables.Add("Sonarr_EpisodeFile_Id", episodeFile.Id.ToString());
variables.Add("Sonarr_EpisodeFile_EpisodeCount", episodeFile.Episodes.Value.Count.ToString());
variables.Add("Sonarr_EpisodeFile_RelativePath", episodeFile.RelativePath);
variables.Add("Sonarr_EpisodeFile_Path", Path.Combine(series.Path, episodeFile.RelativePath));
variables.Add("Sonarr_EpisodeFile_EpisodeIds", string.Join(",", episodeFile.Episodes.Value.Select(e => e.Id)));
variables.Add("Sonarr_EpisodeFile_SeasonNumber", episodeFile.SeasonNumber.ToString());
variables.Add("Sonarr_EpisodeFile_EpisodeNumbers", string.Join(",", episodeFile.Episodes.Value.Select(e => e.EpisodeNumber)));
variables.Add("Sonarr_EpisodeFile_EpisodeAirDates", string.Join(",", episodeFile.Episodes.Value.Select(e => e.AirDate)));
variables.Add("Sonarr_EpisodeFile_EpisodeAirDatesUtc", string.Join(",", episodeFile.Episodes.Value.Select(e => e.AirDateUtc)));
variables.Add("Sonarr_EpisodeFile_EpisodeTitles", string.Join("|", episodeFile.Episodes.Value.Select(e => e.Title)));
variables.Add("Sonarr_EpisodeFile_Quality", episodeFile.Quality.Quality.Name);
variables.Add("Sonarr_EpisodeFile_QualityVersion", episodeFile.Quality.Revision.Version.ToString());
variables.Add("Sonarr_EpisodeFile_ReleaseGroup", episodeFile.ReleaseGroup ?? string.Empty);
variables.Add("Sonarr_EpisodeFile_SceneName", episodeFile.SceneName ?? string.Empty);
variables.Add("Sonarr_EpisodeFile_SourcePath", sourcePath);
variables.Add("Sonarr_EpisodeFile_SourceFolder", Path.GetDirectoryName(sourcePath));
variables.Add("Sonarr_Download_Client", message.DownloadClientInfo?.Name ?? string.Empty);
variables.Add("Sonarr_Download_Client_Type", message.DownloadClientInfo?.Type ?? string.Empty);
variables.Add("Sonarr_Download_Id", message.DownloadId ?? string.Empty);
variables.Add("Sonarr_EpisodeFile_MediaInfo_AudioChannels", MediaInfoFormatter.FormatAudioChannels(episodeFile.MediaInfo).ToString());
variables.Add("Sonarr_EpisodeFile_MediaInfo_AudioCodec", MediaInfoFormatter.FormatAudioCodec(episodeFile.MediaInfo, null));
variables.Add("Sonarr_EpisodeFile_MediaInfo_AudioLanguages", episodeFile.MediaInfo.AudioLanguages.Distinct().ConcatToString(" / "));
variables.Add("Sonarr_EpisodeFile_MediaInfo_Languages", episodeFile.MediaInfo.AudioLanguages.ConcatToString(" / "));
variables.Add("Sonarr_EpisodeFile_MediaInfo_Height", episodeFile.MediaInfo.Height.ToString());
variables.Add("Sonarr_EpisodeFile_MediaInfo_Width", episodeFile.MediaInfo.Width.ToString());
variables.Add("Sonarr_EpisodeFile_MediaInfo_Subtitles", episodeFile.MediaInfo.Subtitles.ConcatToString(" / "));
variables.Add("Sonarr_EpisodeFile_MediaInfo_VideoCodec", MediaInfoFormatter.FormatVideoCodec(episodeFile.MediaInfo, null));
variables.Add("Sonarr_EpisodeFile_MediaInfo_VideoDynamicRangeType", MediaInfoFormatter.FormatVideoDynamicRangeType(episodeFile.MediaInfo));
if (message.OldFiles.Any())
{
variables.Add("Sonarr_DeletedRelativePaths", string.Join("|", message.OldFiles.Select(e => e.RelativePath)));
variables.Add("Sonarr_DeletedPaths", string.Join("|", message.OldFiles.Select(e => Path.Combine(series.Path, e.RelativePath))));
variables.Add("Sonarr_DeletedDateAdded", string.Join("|", message.OldFiles.Select(e => e.DateAdded)));
}
_proxy.SendNotification(variables, Settings);
}
public override void OnRename(Series series, List<RenamedEpisodeFile> renamedFiles)
{
var variables = new StringDictionary();
variables.Add("Sonarr_EventType", "Rename");
variables.Add("Sonarr_Series_Id", series.Id.ToString());
variables.Add("Sonarr_Series_Title", series.Title);
variables.Add("Sonarr_Series_Path", series.Path);
variables.Add("Sonarr_Series_TvdbId", series.TvdbId.ToString());
variables.Add("Sonarr_Series_TvMazeId", series.TvMazeId.ToString());
variables.Add("Sonarr_Series_ImdbId", series.ImdbId ?? string.Empty);
variables.Add("Sonarr_Series_Type", series.SeriesType.ToString());
variables.Add("Sonarr_EpisodeFile_Ids", string.Join(",", renamedFiles.Select(e => e.EpisodeFile.Id)));
variables.Add("Sonarr_EpisodeFile_RelativePaths", string.Join("|", renamedFiles.Select(e => e.EpisodeFile.RelativePath)));
variables.Add("Sonarr_EpisodeFile_Paths", string.Join("|", renamedFiles.Select(e => e.EpisodeFile.Path)));
variables.Add("Sonarr_EpisodeFile_PreviousRelativePaths", string.Join("|", renamedFiles.Select(e => e.PreviousRelativePath)));
variables.Add("Sonarr_EpisodeFile_PreviousPaths", string.Join("|", renamedFiles.Select(e => e.PreviousPath)));
_proxy.SendNotification(variables, Settings);
}
public override void OnEpisodeFileDelete(EpisodeDeleteMessage deleteMessage)
{
var series = deleteMessage.Series;
var episodeFile = deleteMessage.EpisodeFile;
var variables = new StringDictionary();
variables.Add("Sonarr_EventType", "EpisodeFileDelete");
variables.Add("Sonarr_EpisodeFile_DeleteReason", deleteMessage.Reason.ToString());
variables.Add("Sonarr_Series_Id", series.Id.ToString());
variables.Add("Sonarr_Series_Title", series.Title);
variables.Add("Sonarr_Series_Path", series.Path);
variables.Add("Sonarr_Series_TvdbId", series.TvdbId.ToString());
variables.Add("Sonarr_Series_TvMazeId", series.TvMazeId.ToString());
variables.Add("Sonarr_Series_ImdbId", series.ImdbId ?? string.Empty);
variables.Add("Sonarr_Series_Type", series.SeriesType.ToString());
variables.Add("Sonarr_EpisodeFile_Id", episodeFile.Id.ToString());
variables.Add("Sonarr_EpisodeFile_EpisodeCount", episodeFile.Episodes.Value.Count.ToString());
variables.Add("Sonarr_EpisodeFile_RelativePath", episodeFile.RelativePath);
variables.Add("Sonarr_EpisodeFile_Path", Path.Combine(series.Path, episodeFile.RelativePath));
variables.Add("Sonarr_EpisodeFile_EpisodeIds", string.Join(",", episodeFile.Episodes.Value.Select(e => e.Id)));
variables.Add("Sonarr_EpisodeFile_SeasonNumber", episodeFile.SeasonNumber.ToString());
variables.Add("Sonarr_EpisodeFile_EpisodeNumbers", string.Join(",", episodeFile.Episodes.Value.Select(e => e.EpisodeNumber)));
variables.Add("Sonarr_EpisodeFile_EpisodeAirDates", string.Join(",", episodeFile.Episodes.Value.Select(e => e.AirDate)));
variables.Add("Sonarr_EpisodeFile_EpisodeAirDatesUtc", string.Join(",", episodeFile.Episodes.Value.Select(e => e.AirDateUtc)));
variables.Add("Sonarr_EpisodeFile_EpisodeTitles", string.Join("|", episodeFile.Episodes.Value.Select(e => e.Title)));
variables.Add("Sonarr_EpisodeFile_Quality", episodeFile.Quality.Quality.Name);
variables.Add("Sonarr_EpisodeFile_QualityVersion", episodeFile.Quality.Revision.Version.ToString());
variables.Add("Sonarr_EpisodeFile_ReleaseGroup", episodeFile.ReleaseGroup ?? string.Empty);
variables.Add("Sonarr_EpisodeFile_SceneName", episodeFile.SceneName ?? string.Empty);
_proxy.SendNotification(variables, Settings);
}
public override void OnSeriesDelete(SeriesDeleteMessage deleteMessage)
{
var series = deleteMessage.Series;
var variables = new StringDictionary();
variables.Add("Sonarr_EventType", "SeriesDelete");
variables.Add("Sonarr_Series_Id", series.Id.ToString());
variables.Add("Sonarr_Series_Title", series.Title);
variables.Add("Sonarr_Series_Path", series.Path);
variables.Add("Sonarr_Series_TvdbId", series.TvdbId.ToString());
variables.Add("Sonarr_Series_TvMazeId", series.TvMazeId.ToString());
variables.Add("Sonarr_Series_ImdbId", series.ImdbId ?? string.Empty);
variables.Add("Sonarr_Series_Type", series.SeriesType.ToString());
variables.Add("Sonarr_Series_DeletedFiles", deleteMessage.DeletedFiles.ToString());
_proxy.SendNotification(variables, Settings);
}
public override void OnHealthIssue(HealthCheck.HealthCheck healthCheck)
{
var variables = new StringDictionary();
variables.Add("Sonarr_EventType", "HealthIssue");
variables.Add("Sonarr_Health_Issue_Level", Enum.GetName(typeof(HealthCheckResult), healthCheck.Type));
variables.Add("Sonarr_Health_Issue_Message", healthCheck.Message);
variables.Add("Sonarr_Health_Issue_Type", healthCheck.Source.Name);
variables.Add("Sonarr_Health_Issue_Wiki", healthCheck.WikiUrl.ToString() ?? string.Empty);
_proxy.SendNotification(variables, Settings);
}
public override void OnApplicationUpdate(ApplicationUpdateMessage updateMessage)
{
var variables = new StringDictionary();
variables.Add("Sonarr_EventType", "ApplicationUpdate");
variables.Add("Sonarr_Update_Message", updateMessage.Message);
variables.Add("Sonarr_Update_NewVersion", updateMessage.NewVersion.ToString());
variables.Add("Sonarr_Update_PreviousVersion", updateMessage.PreviousVersion.ToString());
_proxy.SendNotification(variables, Settings);
}
public override ValidationResult Test()
{
var failures = new List<ValidationFailure>();
failures.AddIfNotNull(_proxy.Test(Settings));
return new ValidationResult(failures);
}
}
}

View File

@ -0,0 +1,18 @@
using System;
using NzbDrone.Common.Exceptions;
namespace NzbDrone.Core.Notifications.Notifiarr
{
public class NotifiarrException : NzbDroneException
{
public NotifiarrException(string message)
: base(message)
{
}
public NotifiarrException(string message, Exception innerException, params object[] args)
: base(message, innerException, args)
{
}
}
}

View File

@ -0,0 +1,112 @@
using System;
using System.Collections.Specialized;
using FluentValidation.Results;
using NLog;
using NzbDrone.Common.Extensions;
using NzbDrone.Common.Http;
using NzbDrone.Core.Configuration;
namespace NzbDrone.Core.Notifications.Notifiarr
{
public interface INotifiarrProxy
{
void SendNotification(StringDictionary message, NotifiarrSettings settings);
ValidationFailure Test(NotifiarrSettings settings);
}
public class NotifiarrProxy : INotifiarrProxy
{
private const string URL = "https://notifiarr.com";
private readonly IHttpClient _httpClient;
private readonly IConfigFileProvider _configFileProvider;
private readonly Logger _logger;
public NotifiarrProxy(IHttpClient httpClient, IConfigFileProvider configFileProvider, Logger logger)
{
_httpClient = httpClient;
_configFileProvider = configFileProvider;
_logger = logger;
}
public void SendNotification(StringDictionary message, NotifiarrSettings settings)
{
try
{
ProcessNotification(message, settings);
}
catch (NotifiarrException ex)
{
throw ex;
}
}
public ValidationFailure Test(NotifiarrSettings settings)
{
try
{
var variables = new StringDictionary();
variables.Add("Sonarr_EventType", "Test");
SendNotification(variables, settings);
return null;
}
catch (NotifiarrException ex)
{
return new ValidationFailure("ApiKey", ex.Message);
}
catch (Exception ex)
{
_logger.Error(ex, ex.Message);
return new ValidationFailure("", "Unable to send test notification. Check the log for more details.");
}
}
private void ProcessNotification(StringDictionary message, NotifiarrSettings settings)
{
try
{
var instanceName = _configFileProvider.InstanceName;
var requestBuilder = new HttpRequestBuilder(URL + "/api/v1/notification/sonarr").Post();
requestBuilder.AddFormParameter("instanceName", instanceName).Build();
requestBuilder.SetHeader("X-API-Key", settings.ApiKey);
foreach (string key in message.Keys)
{
requestBuilder.AddFormParameter(key, message[key]);
}
var request = requestBuilder.Build();
_httpClient.Post(request);
}
catch (HttpException ex)
{
var responseCode = ex.Response.StatusCode;
switch ((int)responseCode)
{
case 401:
_logger.Error("Unauthorized", "HTTP 401 - API key is invalid");
throw new NotifiarrException("API key is invalid");
case 400:
_logger.Error("Invalid Request", "HTTP 400 - Unable to send notification. Ensure Sonarr Integration is enabled & assigned a channel on Notifiarr");
throw new NotifiarrException("Unable to send notification. Ensure Sonarr Integration is enabled & assigned a channel on Notifiarr");
case 502:
case 503:
case 504:
_logger.Error("Service Unavailable", "Unable to send notification. Service Unavailable");
throw new NotifiarrException("Unable to send notification. Service Unavailable", ex);
case 520:
case 521:
case 522:
case 523:
case 524:
_logger.Error(ex, "Cloudflare Related HTTP Error - Unable to send notification");
throw new NotifiarrException("Cloudflare Related HTTP Error - Unable to send notification", ex);
default:
_logger.Error(ex, "Unknown HTTP Error - Unable to send notification");
throw new NotifiarrException("Unknown HTTP Error - Unable to send notification", ex);
}
}
}
}
}

View File

@ -0,0 +1,28 @@
using FluentValidation;
using NzbDrone.Core.Annotations;
using NzbDrone.Core.ThingiProvider;
using NzbDrone.Core.Validation;
namespace NzbDrone.Core.Notifications.Notifiarr
{
public class NotifiarrSettingsValidator : AbstractValidator<NotifiarrSettings>
{
public NotifiarrSettingsValidator()
{
RuleFor(c => c.ApiKey).NotEmpty();
}
}
public class NotifiarrSettings : IProviderConfig
{
private static readonly NotifiarrSettingsValidator Validator = new NotifiarrSettingsValidator();
[FieldDefinition(0, Label = "API Key", Privacy = PrivacyLevel.ApiKey, HelpText = "Your API key from your profile", HelpLink = "https://notifiarr.com")]
public string ApiKey { get; set; }
public NzbDroneValidationResult Validate()
{
return new NzbDroneValidationResult(Validator.Validate(this));
}
}
}