diff --git a/src/Lidarr.Api.V1/TrackFiles/TrackFileController.cs b/src/Lidarr.Api.V1/TrackFiles/TrackFileController.cs index f1aa00f9d..3df6d6d91 100644 --- a/src/Lidarr.Api.V1/TrackFiles/TrackFileController.cs +++ b/src/Lidarr.Api.V1/TrackFiles/TrackFileController.cs @@ -11,6 +11,7 @@ using NzbDrone.Core.MediaFiles; using NzbDrone.Core.MediaFiles.Events; using NzbDrone.Core.Messaging.Events; using NzbDrone.Core.Music; +using NzbDrone.Core.Parser; using NzbDrone.SignalR; using BadRequestException = Lidarr.Http.REST.BadRequestException; using HttpStatusCode = System.Net.HttpStatusCode; @@ -111,6 +112,17 @@ namespace Lidarr.Api.V1.TrackFiles { var trackFile = _mediaFileService.Get(trackFileResource.Id); trackFile.Quality = trackFileResource.Quality; + + if (trackFileResource.SceneName != null && SceneChecker.IsSceneTitle(trackFileResource.SceneName)) + { + trackFile.SceneName = trackFileResource.SceneName; + } + + if (trackFileResource.ReleaseGroup != null) + { + trackFile.ReleaseGroup = trackFileResource.ReleaseGroup; + } + _mediaFileService.Update(trackFile); return Accepted(trackFile.Id); } @@ -126,6 +138,16 @@ namespace Lidarr.Api.V1.TrackFiles { trackFile.Quality = resource.Quality; } + + if (resource.SceneName != null && SceneChecker.IsSceneTitle(resource.SceneName)) + { + trackFile.SceneName = resource.SceneName; + } + + if (resource.ReleaseGroup != null) + { + trackFile.ReleaseGroup = resource.ReleaseGroup; + } } _mediaFileService.Update(trackFiles); diff --git a/src/Lidarr.Api.V1/TrackFiles/TrackFileListResource.cs b/src/Lidarr.Api.V1/TrackFiles/TrackFileListResource.cs index 166836d9c..bd78e8a4a 100644 --- a/src/Lidarr.Api.V1/TrackFiles/TrackFileListResource.cs +++ b/src/Lidarr.Api.V1/TrackFiles/TrackFileListResource.cs @@ -7,5 +7,7 @@ namespace Lidarr.Api.V1.TrackFiles { public List TrackFileIds { get; set; } public QualityModel Quality { get; set; } + public string SceneName { get; set; } + public string ReleaseGroup { get; set; } } } diff --git a/src/Lidarr.Api.V1/TrackFiles/TrackFileResource.cs b/src/Lidarr.Api.V1/TrackFiles/TrackFileResource.cs index 5470cb99c..cb22479e2 100644 --- a/src/Lidarr.Api.V1/TrackFiles/TrackFileResource.cs +++ b/src/Lidarr.Api.V1/TrackFiles/TrackFileResource.cs @@ -15,6 +15,8 @@ namespace Lidarr.Api.V1.TrackFiles public string Path { get; set; } public long Size { get; set; } public DateTime DateAdded { get; set; } + public string SceneName { get; set; } + public string ReleaseGroup { get; set; } public QualityModel Quality { get; set; } public int QualityWeight { get; set; } public MediaInfoResource MediaInfo { get; set; } @@ -52,6 +54,8 @@ namespace Lidarr.Api.V1.TrackFiles Path = model.Path, Size = model.Size, DateAdded = model.DateAdded, + SceneName = model.SceneName, + ReleaseGroup = model.ReleaseGroup, Quality = model.Quality, QualityWeight = QualityWeight(model.Quality), MediaInfo = model.MediaInfo.ToResource() @@ -74,6 +78,8 @@ namespace Lidarr.Api.V1.TrackFiles Path = model.Path, Size = model.Size, DateAdded = model.DateAdded, + SceneName = model.SceneName, + ReleaseGroup = model.ReleaseGroup, Quality = model.Quality, QualityWeight = QualityWeight(model.Quality), MediaInfo = model.MediaInfo.ToResource(),