2017-09-01 01:10:50 +00:00
|
|
|
using System.Collections.Generic;
|
2016-03-15 22:43:21 +00:00
|
|
|
using System.Collections.Specialized;
|
|
|
|
using System.IO;
|
|
|
|
using System.Linq;
|
2015-05-20 23:22:10 +00:00
|
|
|
using FluentValidation.Results;
|
2016-03-15 22:43:21 +00:00
|
|
|
using NLog;
|
|
|
|
using NzbDrone.Common.Disk;
|
|
|
|
using NzbDrone.Common.Processes;
|
2017-09-23 01:59:24 +00:00
|
|
|
using NzbDrone.Core.Music;
|
2016-03-15 22:43:21 +00:00
|
|
|
using NzbDrone.Core.Validation;
|
2015-05-20 23:22:10 +00:00
|
|
|
|
|
|
|
namespace NzbDrone.Core.Notifications.CustomScript
|
|
|
|
{
|
|
|
|
public class CustomScript : NotificationBase<CustomScriptSettings>
|
|
|
|
{
|
2016-03-15 22:43:21 +00:00
|
|
|
private readonly IDiskProvider _diskProvider;
|
|
|
|
private readonly IProcessProvider _processProvider;
|
|
|
|
private readonly Logger _logger;
|
2015-05-20 23:22:10 +00:00
|
|
|
|
2016-03-15 22:43:21 +00:00
|
|
|
public CustomScript(IDiskProvider diskProvider, IProcessProvider processProvider, Logger logger)
|
2015-05-20 23:22:10 +00:00
|
|
|
{
|
2016-03-15 22:43:21 +00:00
|
|
|
_diskProvider = diskProvider;
|
|
|
|
_processProvider = processProvider;
|
|
|
|
_logger = logger;
|
2015-05-20 23:22:10 +00:00
|
|
|
}
|
|
|
|
|
2016-12-28 07:52:20 +00:00
|
|
|
public override string Name => "Custom Script";
|
|
|
|
|
2017-03-30 03:49:38 +00:00
|
|
|
public override string Link => "https://github.com/Lidarr/Lidarr/wiki/Custom-Post-Processing-Scripts";
|
2015-05-20 23:22:10 +00:00
|
|
|
|
2016-03-15 22:43:21 +00:00
|
|
|
public override void OnGrab(GrabMessage message)
|
2015-05-20 23:22:10 +00:00
|
|
|
{
|
2017-09-23 01:59:24 +00:00
|
|
|
var artist = message.Artist;
|
|
|
|
var remoteAlbum = message.Album;
|
|
|
|
var releaseGroup = remoteAlbum.ParsedAlbumInfo.ReleaseGroup;
|
2016-03-15 22:43:21 +00:00
|
|
|
var environmentVariables = new StringDictionary();
|
|
|
|
|
2017-03-30 03:49:38 +00:00
|
|
|
environmentVariables.Add("Lidarr_EventType", "Grab");
|
2017-09-23 01:59:24 +00:00
|
|
|
environmentVariables.Add("Lidarr_Artist_Id", artist.Id.ToString());
|
2018-12-15 00:02:43 +00:00
|
|
|
environmentVariables.Add("Lidarr_Artist_Name", artist.Metadata.Value.Name);
|
|
|
|
environmentVariables.Add("Lidarr_Artist_MBId", artist.Metadata.Value.ForeignArtistId);
|
|
|
|
environmentVariables.Add("Lidarr_Artist_Type", artist.Metadata.Value.Type);
|
2017-09-23 01:59:24 +00:00
|
|
|
environmentVariables.Add("Lidarr_Release_AlbumCount", remoteAlbum.Albums.Count.ToString());
|
|
|
|
environmentVariables.Add("Lidarr_Release_AlbumReleaseDates", string.Join(",", remoteAlbum.Albums.Select(e => e.ReleaseDate)));
|
|
|
|
environmentVariables.Add("Lidarr_Release_AlbumTitles", string.Join("|", remoteAlbum.Albums.Select(e => e.Title)));
|
|
|
|
environmentVariables.Add("Lidarr_Release_Title", remoteAlbum.Release.Title);
|
2018-05-20 02:20:08 +00:00
|
|
|
environmentVariables.Add("Lidarr_Release_Indexer", remoteAlbum.Release.Indexer ?? string.Empty);
|
2017-09-23 01:59:24 +00:00
|
|
|
environmentVariables.Add("Lidarr_Release_Size", remoteAlbum.Release.Size.ToString());
|
|
|
|
environmentVariables.Add("Lidarr_Release_Quality", remoteAlbum.ParsedAlbumInfo.Quality.Quality.Name);
|
|
|
|
environmentVariables.Add("Lidarr_Release_QualityVersion", remoteAlbum.ParsedAlbumInfo.Quality.Revision.Version.ToString());
|
2017-10-30 01:44:03 +00:00
|
|
|
environmentVariables.Add("Lidarr_Release_ReleaseGroup", releaseGroup ?? string.Empty);
|
2017-10-28 20:30:38 +00:00
|
|
|
environmentVariables.Add("Lidarr_Download_Client", message.DownloadClient ?? string.Empty);
|
|
|
|
environmentVariables.Add("Lidarr_Download_Id", message.DownloadId ?? string.Empty);
|
2016-03-15 22:43:21 +00:00
|
|
|
|
|
|
|
ExecuteScript(environmentVariables);
|
2015-05-20 23:22:10 +00:00
|
|
|
}
|
|
|
|
|
2018-04-04 03:45:59 +00:00
|
|
|
public override void OnDownload(TrackDownloadMessage message)
|
2015-05-20 23:22:10 +00:00
|
|
|
{
|
2017-09-23 01:59:24 +00:00
|
|
|
var artist = message.Artist;
|
2018-04-25 03:19:00 +00:00
|
|
|
var album = message.Album;
|
2018-12-27 02:48:03 +00:00
|
|
|
var release = message.Release;
|
2017-09-23 01:59:24 +00:00
|
|
|
var trackFile = message.TrackFile;
|
2016-03-15 22:43:21 +00:00
|
|
|
var sourcePath = message.SourcePath;
|
|
|
|
var environmentVariables = new StringDictionary();
|
|
|
|
|
2017-03-30 03:49:38 +00:00
|
|
|
environmentVariables.Add("Lidarr_EventType", "Download");
|
2017-11-16 02:27:41 +00:00
|
|
|
environmentVariables.Add("Lidarr_IsUpgrade", message.OldFiles.Any().ToString());
|
2017-09-23 01:59:24 +00:00
|
|
|
environmentVariables.Add("Lidarr_Artist_Id", artist.Id.ToString());
|
2018-12-15 00:02:43 +00:00
|
|
|
environmentVariables.Add("Lidarr_Artist_Name", artist.Metadata.Value.Name);
|
2017-09-23 01:59:24 +00:00
|
|
|
environmentVariables.Add("Lidarr_Artist_Path", artist.Path);
|
2018-12-15 00:02:43 +00:00
|
|
|
environmentVariables.Add("Lidarr_Artist_MBId", artist.Metadata.Value.ForeignArtistId);
|
|
|
|
environmentVariables.Add("Lidarr_Artist_Type", artist.Metadata.Value.Type);
|
2018-04-25 03:19:00 +00:00
|
|
|
environmentVariables.Add("Lidarr_Album_Id", album.Id.ToString());
|
|
|
|
environmentVariables.Add("Lidarr_Album_Title", album.Title);
|
|
|
|
environmentVariables.Add("Lidarr_Album_MBId", album.ForeignAlbumId);
|
2018-12-27 02:48:03 +00:00
|
|
|
environmentVariables.Add("Lidarr_AlbumRelease_MBId", release.ForeignReleaseId);
|
2018-04-25 03:19:00 +00:00
|
|
|
environmentVariables.Add("Lidarr_Album_ReleaseDate", album.ReleaseDate.ToString());
|
2017-09-23 01:59:24 +00:00
|
|
|
environmentVariables.Add("Lidarr_TrackFile_Id", trackFile.Id.ToString());
|
2017-10-31 00:51:46 +00:00
|
|
|
environmentVariables.Add("Lidarr_TrackFile_TrackCount", trackFile.Tracks.Value.Count.ToString());
|
2017-09-23 01:59:24 +00:00
|
|
|
environmentVariables.Add("Lidarr_TrackFile_RelativePath", trackFile.RelativePath);
|
|
|
|
environmentVariables.Add("Lidarr_TrackFile_Path", Path.Combine(artist.Path, trackFile.RelativePath));
|
|
|
|
environmentVariables.Add("Lidarr_TrackFile_TrackNumbers", string.Join(",", trackFile.Tracks.Value.Select(e => e.TrackNumber)));
|
|
|
|
environmentVariables.Add("Lidarr_TrackFile_TrackTitles", string.Join("|", trackFile.Tracks.Value.Select(e => e.Title)));
|
|
|
|
environmentVariables.Add("Lidarr_TrackFile_Quality", trackFile.Quality.Quality.Name);
|
|
|
|
environmentVariables.Add("Lidarr_TrackFile_QualityVersion", trackFile.Quality.Revision.Version.ToString());
|
|
|
|
environmentVariables.Add("Lidarr_TrackFile_ReleaseGroup", trackFile.ReleaseGroup ?? string.Empty);
|
|
|
|
environmentVariables.Add("Lidarr_TrackFile_SceneName", trackFile.SceneName ?? string.Empty);
|
|
|
|
environmentVariables.Add("Lidarr_TrackFile_SourcePath", sourcePath);
|
|
|
|
environmentVariables.Add("Lidarr_TrackFile_SourceFolder", Path.GetDirectoryName(sourcePath));
|
2017-10-28 20:30:38 +00:00
|
|
|
environmentVariables.Add("Lidarr_Download_Client", message.DownloadClient ?? string.Empty);
|
|
|
|
environmentVariables.Add("Lidarr_Download_Id", message.DownloadId ?? string.Empty);
|
2016-03-15 22:43:21 +00:00
|
|
|
|
2017-09-01 01:10:50 +00:00
|
|
|
if (message.OldFiles.Any())
|
|
|
|
{
|
|
|
|
environmentVariables.Add("Lidarr_DeletedRelativePaths", string.Join("|", message.OldFiles.Select(e => e.RelativePath)));
|
2017-09-23 01:59:24 +00:00
|
|
|
environmentVariables.Add("Lidarr_DeletedPaths", string.Join("|", message.OldFiles.Select(e => Path.Combine(artist.Path, e.RelativePath))));
|
2017-09-01 01:10:50 +00:00
|
|
|
}
|
|
|
|
|
2016-03-15 22:43:21 +00:00
|
|
|
ExecuteScript(environmentVariables);
|
2015-05-20 23:22:10 +00:00
|
|
|
}
|
|
|
|
|
2018-04-04 03:45:59 +00:00
|
|
|
public override void OnAlbumDownload(AlbumDownloadMessage message)
|
|
|
|
{
|
|
|
|
var artist = message.Artist;
|
|
|
|
var album = message.Album;
|
2018-12-27 02:48:03 +00:00
|
|
|
var release = message.Release;
|
2018-04-04 03:45:59 +00:00
|
|
|
var environmentVariables = new StringDictionary();
|
|
|
|
|
|
|
|
environmentVariables.Add("Lidarr_EventType", "AlbumDownload");
|
|
|
|
environmentVariables.Add("Lidarr_Artist_Id", artist.Id.ToString());
|
2018-12-15 00:02:43 +00:00
|
|
|
environmentVariables.Add("Lidarr_Artist_Name", artist.Metadata.Value.Name);
|
2018-04-04 03:45:59 +00:00
|
|
|
environmentVariables.Add("Lidarr_Artist_Path", artist.Path);
|
2018-12-15 00:02:43 +00:00
|
|
|
environmentVariables.Add("Lidarr_Artist_MBId", artist.Metadata.Value.ForeignArtistId);
|
|
|
|
environmentVariables.Add("Lidarr_Artist_Type", artist.Metadata.Value.Type);
|
2018-04-04 03:45:59 +00:00
|
|
|
environmentVariables.Add("Lidarr_Album_Id", album.Id.ToString());
|
|
|
|
environmentVariables.Add("Lidarr_Album_Title", album.Title);
|
|
|
|
environmentVariables.Add("Lidarr_Album_MBId", album.ForeignAlbumId);
|
2018-12-27 02:48:03 +00:00
|
|
|
environmentVariables.Add("Lidarr_AlbumRelease_MBId", release.ForeignReleaseId);
|
2018-04-25 03:19:00 +00:00
|
|
|
environmentVariables.Add("Lidarr_Album_ReleaseDate", album.ReleaseDate.ToString());
|
2018-04-04 03:45:59 +00:00
|
|
|
environmentVariables.Add("Lidarr_Download_Client", message.DownloadClient ?? string.Empty);
|
|
|
|
environmentVariables.Add("Lidarr_Download_Id", message.DownloadId ?? string.Empty);
|
|
|
|
|
2018-05-27 04:55:43 +00:00
|
|
|
if (message.TrackFiles.Any())
|
|
|
|
{
|
|
|
|
environmentVariables.Add("Lidarr_AddedTrackRelativePaths", string.Join("|", message.TrackFiles.Select(e => e.RelativePath)));
|
|
|
|
environmentVariables.Add("Lidarr_AddedTrackPaths", string.Join("|", message.TrackFiles.Select(e => Path.Combine(artist.Path, e.RelativePath))));
|
|
|
|
}
|
|
|
|
|
|
|
|
if (message.OldFiles.Any())
|
|
|
|
{
|
|
|
|
environmentVariables.Add("Lidarr_DeletedRelativePaths", string.Join("|", message.OldFiles.Select(e => e.RelativePath)));
|
|
|
|
environmentVariables.Add("Lidarr_DeletedPaths", string.Join("|", message.OldFiles.Select(e => Path.Combine(artist.Path, e.RelativePath))));
|
|
|
|
}
|
|
|
|
|
2018-04-04 03:45:59 +00:00
|
|
|
ExecuteScript(environmentVariables);
|
|
|
|
}
|
|
|
|
|
2017-09-23 01:59:24 +00:00
|
|
|
public override void OnRename(Artist artist)
|
2015-05-20 23:22:10 +00:00
|
|
|
{
|
2016-03-15 22:43:21 +00:00
|
|
|
var environmentVariables = new StringDictionary();
|
|
|
|
|
2017-03-30 03:49:38 +00:00
|
|
|
environmentVariables.Add("Lidarr_EventType", "Rename");
|
2017-09-23 01:59:24 +00:00
|
|
|
environmentVariables.Add("Lidarr_Artist_Id", artist.Id.ToString());
|
2018-12-15 00:02:43 +00:00
|
|
|
environmentVariables.Add("Lidarr_Artist_Name", artist.Metadata.Value.Name);
|
2017-09-23 01:59:24 +00:00
|
|
|
environmentVariables.Add("Lidarr_Artist_Path", artist.Path);
|
2018-12-15 00:02:43 +00:00
|
|
|
environmentVariables.Add("Lidarr_Artist_MBId", artist.Metadata.Value.ForeignArtistId);
|
|
|
|
environmentVariables.Add("Lidarr_Artist_Type", artist.Metadata.Value.Type);
|
2016-03-15 22:43:21 +00:00
|
|
|
|
|
|
|
ExecuteScript(environmentVariables);
|
2015-05-20 23:22:10 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2016-03-15 22:43:21 +00:00
|
|
|
public override ValidationResult Test()
|
2015-05-20 23:22:10 +00:00
|
|
|
{
|
2016-03-15 22:43:21 +00:00
|
|
|
var failures = new List<ValidationFailure>();
|
|
|
|
|
|
|
|
if (!_diskProvider.FileExists(Settings.Path))
|
2015-05-20 23:22:10 +00:00
|
|
|
{
|
2016-03-15 22:43:21 +00:00
|
|
|
failures.Add(new NzbDroneValidationFailure("Path", "File does not exist"));
|
2015-05-20 23:22:10 +00:00
|
|
|
}
|
2016-03-15 22:43:21 +00:00
|
|
|
|
|
|
|
return new ValidationResult(failures);
|
2015-05-20 23:22:10 +00:00
|
|
|
}
|
|
|
|
|
2016-03-15 22:43:21 +00:00
|
|
|
private void ExecuteScript(StringDictionary environmentVariables)
|
2015-05-20 23:22:10 +00:00
|
|
|
{
|
2016-03-15 22:43:21 +00:00
|
|
|
_logger.Debug("Executing external script: {0}", Settings.Path);
|
2015-05-20 23:22:10 +00:00
|
|
|
|
2016-03-15 22:43:21 +00:00
|
|
|
var process = _processProvider.StartAndCapture(Settings.Path, Settings.Arguments, environmentVariables);
|
|
|
|
|
|
|
|
_logger.Debug("Executed external script: {0} - Status: {1}", Settings.Path, process.ExitCode);
|
2019-02-16 14:49:24 +00:00
|
|
|
_logger.Debug($"Script Output: {System.Environment.NewLine}{string.Join(System.Environment.NewLine, process.Lines)}");
|
2015-05-20 23:22:10 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|