1
0
mirror of https://github.com/Radarr/Radarr.git synced 2024-07-07 04:19:25 +02:00

Refactor movie tags for CustomScript, Webhook and Notifiarr events

(cherry picked from commit cc0a284660f139d5f47b27a2c389973e5e888587)

Closes #10003
This commit is contained in:
Mark McDowall 2024-05-07 17:45:28 -07:00 committed by Bogdan
parent 190c4c5893
commit 38e7e37d57
3 changed files with 34 additions and 44 deletions

View File

@ -69,7 +69,7 @@ public override void OnGrab(GrabMessage message)
environmentVariables.Add("Radarr_Movie_Year", movie.MovieMetadata.Value.Year.ToString()); environmentVariables.Add("Radarr_Movie_Year", movie.MovieMetadata.Value.Year.ToString());
environmentVariables.Add("Radarr_Movie_OriginalLanguage", IsoLanguages.Get(movie.MovieMetadata.Value.OriginalLanguage).ThreeLetterCode); environmentVariables.Add("Radarr_Movie_OriginalLanguage", IsoLanguages.Get(movie.MovieMetadata.Value.OriginalLanguage).ThreeLetterCode);
environmentVariables.Add("Radarr_Movie_Genres", string.Join("|", movie.MovieMetadata.Value.Genres)); environmentVariables.Add("Radarr_Movie_Genres", string.Join("|", movie.MovieMetadata.Value.Genres));
environmentVariables.Add("Radarr_Movie_Tags", string.Join("|", movie.Tags.Select(t => _tagRepository.Find(t)?.Label).Where(l => l.IsNotNullOrWhiteSpace()))); environmentVariables.Add("Radarr_Movie_Tags", string.Join("|", GetTagLabels(movie)));
environmentVariables.Add("Radarr_Movie_ImdbId", movie.MovieMetadata.Value.ImdbId ?? string.Empty); environmentVariables.Add("Radarr_Movie_ImdbId", movie.MovieMetadata.Value.ImdbId ?? string.Empty);
environmentVariables.Add("Radarr_Movie_TmdbId", movie.MovieMetadata.Value.TmdbId.ToString()); environmentVariables.Add("Radarr_Movie_TmdbId", movie.MovieMetadata.Value.TmdbId.ToString());
environmentVariables.Add("Radarr_Movie_In_Cinemas_Date", movie.MovieMetadata.Value.InCinemas.ToString() ?? string.Empty); environmentVariables.Add("Radarr_Movie_In_Cinemas_Date", movie.MovieMetadata.Value.InCinemas.ToString() ?? string.Empty);
@ -107,7 +107,7 @@ public override void OnDownload(DownloadMessage message)
environmentVariables.Add("Radarr_Movie_Year", movie.MovieMetadata.Value.Year.ToString()); environmentVariables.Add("Radarr_Movie_Year", movie.MovieMetadata.Value.Year.ToString());
environmentVariables.Add("Radarr_Movie_OriginalLanguage", IsoLanguages.Get(movie.MovieMetadata.Value.OriginalLanguage).ThreeLetterCode); environmentVariables.Add("Radarr_Movie_OriginalLanguage", IsoLanguages.Get(movie.MovieMetadata.Value.OriginalLanguage).ThreeLetterCode);
environmentVariables.Add("Radarr_Movie_Genres", string.Join("|", movie.MovieMetadata.Value.Genres)); environmentVariables.Add("Radarr_Movie_Genres", string.Join("|", movie.MovieMetadata.Value.Genres));
environmentVariables.Add("Radarr_Movie_Tags", string.Join("|", movie.Tags.Select(t => _tagRepository.Get(t).Label))); environmentVariables.Add("Radarr_Movie_Tags", string.Join("|", GetTagLabels(movie)));
environmentVariables.Add("Radarr_Movie_Path", movie.Path); environmentVariables.Add("Radarr_Movie_Path", movie.Path);
environmentVariables.Add("Radarr_Movie_ImdbId", movie.MovieMetadata.Value.ImdbId ?? string.Empty); environmentVariables.Add("Radarr_Movie_ImdbId", movie.MovieMetadata.Value.ImdbId ?? string.Empty);
environmentVariables.Add("Radarr_Movie_TmdbId", movie.MovieMetadata.Value.TmdbId.ToString()); environmentVariables.Add("Radarr_Movie_TmdbId", movie.MovieMetadata.Value.TmdbId.ToString());
@ -164,7 +164,7 @@ public override void OnMovieRename(Movie movie, List<RenamedMovieFile> renamedFi
environmentVariables.Add("Radarr_Movie_Year", movie.MovieMetadata.Value.Year.ToString()); environmentVariables.Add("Radarr_Movie_Year", movie.MovieMetadata.Value.Year.ToString());
environmentVariables.Add("Radarr_Movie_OriginalLanguage", IsoLanguages.Get(movie.MovieMetadata.Value.OriginalLanguage).ThreeLetterCode); environmentVariables.Add("Radarr_Movie_OriginalLanguage", IsoLanguages.Get(movie.MovieMetadata.Value.OriginalLanguage).ThreeLetterCode);
environmentVariables.Add("Radarr_Movie_Genres", string.Join("|", movie.MovieMetadata.Value.Genres)); environmentVariables.Add("Radarr_Movie_Genres", string.Join("|", movie.MovieMetadata.Value.Genres));
environmentVariables.Add("Radarr_Movie_Tags", string.Join("|", movie.Tags.Select(t => _tagRepository.Get(t).Label))); environmentVariables.Add("Radarr_Movie_Tags", string.Join("|", GetTagLabels(movie)));
environmentVariables.Add("Radarr_Movie_Path", movie.Path); environmentVariables.Add("Radarr_Movie_Path", movie.Path);
environmentVariables.Add("Radarr_Movie_ImdbId", movie.MovieMetadata.Value.ImdbId ?? string.Empty); environmentVariables.Add("Radarr_Movie_ImdbId", movie.MovieMetadata.Value.ImdbId ?? string.Empty);
environmentVariables.Add("Radarr_Movie_TmdbId", movie.MovieMetadata.Value.TmdbId.ToString()); environmentVariables.Add("Radarr_Movie_TmdbId", movie.MovieMetadata.Value.TmdbId.ToString());
@ -191,7 +191,7 @@ public override void OnMovieAdded(Movie movie)
environmentVariables.Add("Radarr_Movie_Year", movie.MovieMetadata.Value.Year.ToString()); environmentVariables.Add("Radarr_Movie_Year", movie.MovieMetadata.Value.Year.ToString());
environmentVariables.Add("Radarr_Movie_OriginalLanguage", IsoLanguages.Get(movie.MovieMetadata.Value.OriginalLanguage).ThreeLetterCode); environmentVariables.Add("Radarr_Movie_OriginalLanguage", IsoLanguages.Get(movie.MovieMetadata.Value.OriginalLanguage).ThreeLetterCode);
environmentVariables.Add("Radarr_Movie_Genres", string.Join("|", movie.MovieMetadata.Value.Genres)); environmentVariables.Add("Radarr_Movie_Genres", string.Join("|", movie.MovieMetadata.Value.Genres));
environmentVariables.Add("Radarr_Movie_Tags", string.Join("|", movie.Tags.Select(t => _tagRepository.Get(t).Label))); environmentVariables.Add("Radarr_Movie_Tags", string.Join("|", GetTagLabels(movie)));
environmentVariables.Add("Radarr_Movie_Path", movie.Path); environmentVariables.Add("Radarr_Movie_Path", movie.Path);
environmentVariables.Add("Radarr_Movie_ImdbId", movie.MovieMetadata.Value.ImdbId ?? string.Empty); environmentVariables.Add("Radarr_Movie_ImdbId", movie.MovieMetadata.Value.ImdbId ?? string.Empty);
environmentVariables.Add("Radarr_Movie_TmdbId", movie.MovieMetadata.Value.TmdbId.ToString()); environmentVariables.Add("Radarr_Movie_TmdbId", movie.MovieMetadata.Value.TmdbId.ToString());
@ -216,7 +216,7 @@ public override void OnMovieFileDelete(MovieFileDeleteMessage deleteMessage)
environmentVariables.Add("Radarr_Movie_Year", movie.Year.ToString()); environmentVariables.Add("Radarr_Movie_Year", movie.Year.ToString());
environmentVariables.Add("Radarr_Movie_OriginalLanguage", IsoLanguages.Get(movie.MovieMetadata.Value.OriginalLanguage).ThreeLetterCode); environmentVariables.Add("Radarr_Movie_OriginalLanguage", IsoLanguages.Get(movie.MovieMetadata.Value.OriginalLanguage).ThreeLetterCode);
environmentVariables.Add("Radarr_Movie_Genres", string.Join("|", movie.MovieMetadata.Value.Genres)); environmentVariables.Add("Radarr_Movie_Genres", string.Join("|", movie.MovieMetadata.Value.Genres));
environmentVariables.Add("Radarr_Movie_Tags", string.Join("|", movie.Tags.Select(t => _tagRepository.Get(t).Label))); environmentVariables.Add("Radarr_Movie_Tags", string.Join("|", GetTagLabels(movie)));
environmentVariables.Add("Radarr_Movie_Path", movie.Path); environmentVariables.Add("Radarr_Movie_Path", movie.Path);
environmentVariables.Add("Radarr_Movie_ImdbId", movie.MovieMetadata.Value.ImdbId ?? string.Empty); environmentVariables.Add("Radarr_Movie_ImdbId", movie.MovieMetadata.Value.ImdbId ?? string.Empty);
environmentVariables.Add("Radarr_Movie_TmdbId", movie.MovieMetadata.Value.TmdbId.ToString()); environmentVariables.Add("Radarr_Movie_TmdbId", movie.MovieMetadata.Value.TmdbId.ToString());
@ -246,7 +246,7 @@ public override void OnMovieDelete(MovieDeleteMessage deleteMessage)
environmentVariables.Add("Radarr_Movie_Year", movie.MovieMetadata.Value.Year.ToString()); environmentVariables.Add("Radarr_Movie_Year", movie.MovieMetadata.Value.Year.ToString());
environmentVariables.Add("Radarr_Movie_OriginalLanguage", IsoLanguages.Get(movie.MovieMetadata.Value.OriginalLanguage).ThreeLetterCode); environmentVariables.Add("Radarr_Movie_OriginalLanguage", IsoLanguages.Get(movie.MovieMetadata.Value.OriginalLanguage).ThreeLetterCode);
environmentVariables.Add("Radarr_Movie_Genres", string.Join("|", movie.MovieMetadata.Value.Genres)); environmentVariables.Add("Radarr_Movie_Genres", string.Join("|", movie.MovieMetadata.Value.Genres));
environmentVariables.Add("Radarr_Movie_Tags", string.Join("|", movie.Tags.Select(t => _tagRepository.Get(t).Label))); environmentVariables.Add("Radarr_Movie_Tags", string.Join("|", GetTagLabels(movie)));
environmentVariables.Add("Radarr_Movie_Path", movie.Path); environmentVariables.Add("Radarr_Movie_Path", movie.Path);
environmentVariables.Add("Radarr_Movie_ImdbId", movie.MovieMetadata.Value.ImdbId ?? string.Empty); environmentVariables.Add("Radarr_Movie_ImdbId", movie.MovieMetadata.Value.ImdbId ?? string.Empty);
environmentVariables.Add("Radarr_Movie_TmdbId", movie.MovieMetadata.Value.TmdbId.ToString()); environmentVariables.Add("Radarr_Movie_TmdbId", movie.MovieMetadata.Value.TmdbId.ToString());
@ -318,7 +318,7 @@ public override void OnManualInteractionRequired(ManualInteractionRequiredMessag
environmentVariables.Add("Radarr_Movie_Year", movie.MovieMetadata.Value.Year.ToString()); environmentVariables.Add("Radarr_Movie_Year", movie.MovieMetadata.Value.Year.ToString());
environmentVariables.Add("Radarr_Movie_OriginalLanguage", IsoLanguages.Get(movie.MovieMetadata.Value.OriginalLanguage).ThreeLetterCode); environmentVariables.Add("Radarr_Movie_OriginalLanguage", IsoLanguages.Get(movie.MovieMetadata.Value.OriginalLanguage).ThreeLetterCode);
environmentVariables.Add("Radarr_Movie_Genres", string.Join("|", movie.MovieMetadata.Value.Genres)); environmentVariables.Add("Radarr_Movie_Genres", string.Join("|", movie.MovieMetadata.Value.Genres));
environmentVariables.Add("Radarr_Movie_Tags", string.Join("|", movie.Tags.Select(t => _tagRepository.Get(t).Label))); environmentVariables.Add("Radarr_Movie_Tags", string.Join("|", GetTagLabels(movie)));
environmentVariables.Add("Radarr_Movie_Path", movie.Path); environmentVariables.Add("Radarr_Movie_Path", movie.Path);
environmentVariables.Add("Radarr_Movie_ImdbId", movie.MovieMetadata.Value.ImdbId ?? string.Empty); environmentVariables.Add("Radarr_Movie_ImdbId", movie.MovieMetadata.Value.ImdbId ?? string.Empty);
environmentVariables.Add("Radarr_Movie_TmdbId", movie.MovieMetadata.Value.TmdbId.ToString()); environmentVariables.Add("Radarr_Movie_TmdbId", movie.MovieMetadata.Value.TmdbId.ToString());
@ -385,5 +385,14 @@ private bool ValidatePathParent(string possibleParent, string path)
{ {
return possibleParent.IsParentPath(path); return possibleParent.IsParentPath(path);
} }
private List<string> GetTagLabels(Movie movie)
{
return _tagRepository.GetTags(movie.Tags)
.Select(t => t.Label)
.Where(l => l.IsNotNullOrWhiteSpace())
.OrderBy(l => l)
.ToList();
}
} }
} }

View File

@ -37,10 +37,7 @@ protected WebhookGrabPayload BuildOnGrabPayload(GrabMessage message)
EventType = WebhookEventType.Grab, EventType = WebhookEventType.Grab,
InstanceName = _configFileProvider.InstanceName, InstanceName = _configFileProvider.InstanceName,
ApplicationUrl = _configService.ApplicationUrl, ApplicationUrl = _configService.ApplicationUrl,
Movie = new WebhookMovie(message.Movie) Movie = new WebhookMovie(message.Movie, GetTagLabels(message.Movie)),
{
Tags = GetTagLabels(message.Movie)
},
RemoteMovie = new WebhookRemoteMovie(remoteMovie), RemoteMovie = new WebhookRemoteMovie(remoteMovie),
Release = new WebhookRelease(quality, remoteMovie), Release = new WebhookRelease(quality, remoteMovie),
DownloadClient = message.DownloadClientName, DownloadClient = message.DownloadClientName,
@ -59,10 +56,7 @@ protected WebhookImportPayload BuildOnDownloadPayload(DownloadMessage message)
EventType = WebhookEventType.Download, EventType = WebhookEventType.Download,
InstanceName = _configFileProvider.InstanceName, InstanceName = _configFileProvider.InstanceName,
ApplicationUrl = _configService.ApplicationUrl, ApplicationUrl = _configService.ApplicationUrl,
Movie = new WebhookMovie(message.Movie) Movie = new WebhookMovie(message.Movie, GetTagLabels(message.Movie)),
{
Tags = GetTagLabels(message.Movie)
},
RemoteMovie = new WebhookRemoteMovie(message.Movie), RemoteMovie = new WebhookRemoteMovie(message.Movie),
MovieFile = new WebhookMovieFile(movieFile), MovieFile = new WebhookMovieFile(movieFile),
Release = new WebhookGrabbedRelease(message.Release), Release = new WebhookGrabbedRelease(message.Release),
@ -93,10 +87,7 @@ protected WebhookAddedPayload BuildOnMovieAdded(Movie movie)
EventType = WebhookEventType.MovieAdded, EventType = WebhookEventType.MovieAdded,
InstanceName = _configFileProvider.InstanceName, InstanceName = _configFileProvider.InstanceName,
ApplicationUrl = _configService.ApplicationUrl, ApplicationUrl = _configService.ApplicationUrl,
Movie = new WebhookMovie(movie) Movie = new WebhookMovie(movie, GetTagLabels(movie)),
{
Tags = GetTagLabels(movie)
},
AddMethod = movie.AddOptions.AddMethod AddMethod = movie.AddOptions.AddMethod
}; };
} }
@ -108,10 +99,7 @@ protected WebhookMovieFileDeletePayload BuildOnMovieFileDelete(MovieFileDeleteMe
EventType = WebhookEventType.MovieFileDelete, EventType = WebhookEventType.MovieFileDelete,
InstanceName = _configFileProvider.InstanceName, InstanceName = _configFileProvider.InstanceName,
ApplicationUrl = _configService.ApplicationUrl, ApplicationUrl = _configService.ApplicationUrl,
Movie = new WebhookMovie(deleteMessage.Movie) Movie = new WebhookMovie(deleteMessage.Movie, GetTagLabels(deleteMessage.Movie)),
{
Tags = GetTagLabels(deleteMessage.Movie)
},
MovieFile = new WebhookMovieFile(deleteMessage.MovieFile), MovieFile = new WebhookMovieFile(deleteMessage.MovieFile),
DeleteReason = deleteMessage.Reason DeleteReason = deleteMessage.Reason
}; };
@ -124,10 +112,7 @@ protected WebhookMovieDeletePayload BuildOnMovieDelete(MovieDeleteMessage delete
EventType = WebhookEventType.MovieDelete, EventType = WebhookEventType.MovieDelete,
InstanceName = _configFileProvider.InstanceName, InstanceName = _configFileProvider.InstanceName,
ApplicationUrl = _configService.ApplicationUrl, ApplicationUrl = _configService.ApplicationUrl,
Movie = new WebhookMovie(deleteMessage.Movie) Movie = new WebhookMovie(deleteMessage.Movie, GetTagLabels(deleteMessage.Movie)),
{
Tags = GetTagLabels(deleteMessage.Movie)
},
DeletedFiles = deleteMessage.DeletedFiles DeletedFiles = deleteMessage.DeletedFiles
}; };
@ -146,10 +131,7 @@ protected WebhookRenamePayload BuildOnRenamePayload(Movie movie, List<RenamedMov
EventType = WebhookEventType.Rename, EventType = WebhookEventType.Rename,
InstanceName = _configFileProvider.InstanceName, InstanceName = _configFileProvider.InstanceName,
ApplicationUrl = _configService.ApplicationUrl, ApplicationUrl = _configService.ApplicationUrl,
Movie = new WebhookMovie(movie) Movie = new WebhookMovie(movie, GetTagLabels(movie)),
{
Tags = GetTagLabels(movie)
},
RenamedMovieFiles = renamedFiles.ConvertAll(x => new WebhookRenamedMovieFile(x)) RenamedMovieFiles = renamedFiles.ConvertAll(x => new WebhookRenamedMovieFile(x))
}; };
} }
@ -204,10 +186,7 @@ protected WebhookManualInteractionPayload BuildManualInteractionRequiredPayload(
EventType = WebhookEventType.ManualInteractionRequired, EventType = WebhookEventType.ManualInteractionRequired,
InstanceName = _configFileProvider.InstanceName, InstanceName = _configFileProvider.InstanceName,
ApplicationUrl = _configService.ApplicationUrl, ApplicationUrl = _configService.ApplicationUrl,
Movie = new WebhookMovie(message.Movie) Movie = new WebhookMovie(message.Movie, GetTagLabels(message.Movie)),
{
Tags = GetTagLabels(message.Movie)
},
DownloadInfo = new WebhookDownloadClientItem(quality, message.TrackedDownload.DownloadItem), DownloadInfo = new WebhookDownloadClientItem(quality, message.TrackedDownload.DownloadItem),
DownloadClient = message.DownloadClientInfo?.Name, DownloadClient = message.DownloadClientInfo?.Name,
DownloadClientType = message.DownloadClientInfo?.Type, DownloadClientType = message.DownloadClientInfo?.Type,
@ -233,7 +212,7 @@ protected WebhookPayload BuildTestPayload()
Year = 1970, Year = 1970,
FolderPath = "C:\\testpath", FolderPath = "C:\\testpath",
ReleaseDate = "1970-01-01", ReleaseDate = "1970-01-01",
Tags = new[] { "test-tag" } Tags = new List<string> { "test-tag" }
}, },
RemoteMovie = new WebhookRemoteMovie RemoteMovie = new WebhookRemoteMovie
{ {
@ -254,12 +233,13 @@ protected WebhookPayload BuildTestPayload()
}; };
} }
private IEnumerable<string> GetTagLabels(Movie movie) private List<string> GetTagLabels(Movie movie)
{ {
return movie.Tags? return _tagRepository.GetTags(movie.Tags)
.Select(t => _tagRepository.Find(t)?.Label) .Select(t => t.Label)
.Where(l => l.IsNotNullOrWhiteSpace()) .Where(l => l.IsNotNullOrWhiteSpace())
.OrderBy(l => l); .OrderBy(l => l)
.ToList();
} }
} }
} }

View File

@ -16,13 +16,13 @@ public class WebhookMovie
public int TmdbId { get; set; } public int TmdbId { get; set; }
public string ImdbId { get; set; } public string ImdbId { get; set; }
public string Overview { get; set; } public string Overview { get; set; }
public IEnumerable<string> Tags { get; set; } public List<string> Tags { get; set; }
public WebhookMovie() public WebhookMovie()
{ {
} }
public WebhookMovie(Movie movie) public WebhookMovie(Movie movie, List<string> tags)
{ {
Id = movie.Id; Id = movie.Id;
Title = movie.Title; Title = movie.Title;
@ -32,10 +32,11 @@ public WebhookMovie(Movie movie)
TmdbId = movie.TmdbId; TmdbId = movie.TmdbId;
ImdbId = movie.ImdbId; ImdbId = movie.ImdbId;
Overview = movie.MovieMetadata.Value.Overview; Overview = movie.MovieMetadata.Value.Overview;
Tags = tags;
} }
public WebhookMovie(Movie movie, MovieFile movieFile) public WebhookMovie(Movie movie, MovieFile movieFile, List<string> tags)
: this(movie) : this(movie, tags)
{ {
FilePath = Path.Combine(movie.Path, movieFile.RelativePath); FilePath = Path.Combine(movie.Path, movieFile.RelativePath);
} }