using System; using System.Collections.Generic; using System.Linq; using System.Threading.Tasks; using ErsatzTV.Core; using ErsatzTV.Core.Domain; using ErsatzTV.Core.Interfaces.Metadata; using ErsatzTV.Core.Interfaces.Plex; using ErsatzTV.Core.Plex; using ErsatzTV.Infrastructure.Plex.Models; using LanguageExt; using Refit; using static LanguageExt.Prelude; namespace ErsatzTV.Infrastructure.Plex { public class PlexServerApiClient : IPlexServerApiClient { private readonly IFallbackMetadataProvider _fallbackMetadataProvider; public PlexServerApiClient(IFallbackMetadataProvider fallbackMetadataProvider) => _fallbackMetadataProvider = fallbackMetadataProvider; public async Task>> GetLibraries( PlexConnection connection, PlexServerAuthToken token) { try { IPlexServerApi service = RestService.For(connection.Uri); List directory = await service.GetLibraries(token.AuthToken).Map(r => r.MediaContainer.Directory); return directory .Filter(l => l.Hidden == 0) .Map(Project) .Somes() .ToList(); } catch (Exception ex) { return BaseError.New(ex.Message); } } public async Task>> GetMovieLibraryContents( PlexLibrary library, PlexConnection connection, PlexServerAuthToken token) { try { IPlexServerApi service = RestService.For(connection.Uri); return await service.GetLibrarySectionContents(library.Key, token.AuthToken) .Map(r => r.MediaContainer.Metadata.Filter(m => m.Media.Count > 0 && m.Media[0].Part.Count > 0)) .Map(list => list.Map(metadata => ProjectToMovie(metadata, library.MediaSourceId)).ToList()); } catch (Exception ex) { return BaseError.New(ex.Message); } } public async Task>> GetShowLibraryContents( PlexLibrary library, PlexConnection connection, PlexServerAuthToken token) { try { IPlexServerApi service = RestService.For(connection.Uri); return await service.GetLibrarySectionContents(library.Key, token.AuthToken) .Map(r => r.MediaContainer.Metadata) .Map(list => list.Map(metadata => ProjectToShow(metadata, library.MediaSourceId)).ToList()); } catch (Exception ex) { return BaseError.New(ex.Message); } } public async Task>> GetShowSeasons( PlexLibrary library, PlexShow show, PlexConnection connection, PlexServerAuthToken token) { try { IPlexServerApi service = RestService.For(connection.Uri); return await service.GetChildren(show.Key.Split("/").Reverse().Skip(1).Head(), token.AuthToken) .Map(r => r.MediaContainer.Metadata) .Map(list => list.Map(metadata => ProjectToSeason(metadata, library.MediaSourceId)).ToList()); } catch (Exception ex) { return BaseError.New(ex.Message); } } public async Task>> GetSeasonEpisodes( PlexLibrary library, PlexSeason season, PlexConnection connection, PlexServerAuthToken token) { try { IPlexServerApi service = RestService.For(connection.Uri); return await service.GetChildren(season.Key.Split("/").Reverse().Skip(1).Head(), token.AuthToken) .Map(r => r.MediaContainer.Metadata.Filter(m => m.Media.Count > 0 && m.Media[0].Part.Count > 0)) .Map(list => list.Map(metadata => ProjectToEpisode(metadata, library.MediaSourceId)).ToList()); } catch (Exception ex) { return BaseError.New(ex.Message); } } public async Task> GetStatistics( string key, PlexConnection connection, PlexServerAuthToken token) { try { IPlexServerApi service = RestService.For(connection.Uri); return await service.GetMetadata(key, token.AuthToken) .Map( r => r.MediaContainer.Metadata.Filter(m => m.Media.Count > 0 && m.Media[0].Part.Count > 0) .HeadOrNone()) .BindT(ProjectToMediaVersion) .Map(o => o.ToEither("Unable to locate metadata")); } catch (Exception ex) { return BaseError.New(ex.Message); } } private static Option Project(PlexLibraryResponse response) => response.Type switch { "show" => new PlexLibrary { Key = response.Key, Name = response.Title, MediaKind = LibraryMediaKind.Shows, ShouldSyncItems = false, Paths = new List { new() { Path = $"plex://{response.Uuid}" } } }, "movie" => new PlexLibrary { Key = response.Key, Name = response.Title, MediaKind = LibraryMediaKind.Movies, ShouldSyncItems = false, Paths = new List { new() { Path = $"plex://{response.Uuid}" } } }, // TODO: "artist" for music libraries _ => None }; private PlexMovie ProjectToMovie(PlexMetadataResponse response, int mediaSourceId) { PlexMediaResponse media = response.Media.Head(); PlexPartResponse part = media.Part.Head(); DateTime dateAdded = DateTimeOffset.FromUnixTimeSeconds(response.AddedAt).DateTime; DateTime lastWriteTime = DateTimeOffset.FromUnixTimeSeconds(response.UpdatedAt).DateTime; var metadata = new MovieMetadata { Title = response.Title, SortTitle = _fallbackMetadataProvider.GetSortTitle(response.Title), Plot = response.Summary, Year = response.Year, Tagline = response.Tagline, DateAdded = dateAdded, DateUpdated = lastWriteTime, Genres = Optional(response.Genre).Flatten().Map(g => new Genre { Name = g.Tag }).ToList() }; if (DateTime.TryParse(response.OriginallyAvailableAt, out DateTime releaseDate)) { metadata.ReleaseDate = releaseDate; } if (!string.IsNullOrWhiteSpace(response.Thumb)) { var path = $"plex/{mediaSourceId}{response.Thumb}"; var artwork = new Artwork { ArtworkKind = ArtworkKind.Poster, Path = path, DateAdded = dateAdded, DateUpdated = lastWriteTime }; metadata.Artwork ??= new List(); metadata.Artwork.Add(artwork); } if (!string.IsNullOrWhiteSpace(response.Art)) { var path = $"plex/{mediaSourceId}{response.Art}"; var artwork = new Artwork { ArtworkKind = ArtworkKind.FanArt, Path = path, DateAdded = dateAdded, DateUpdated = lastWriteTime }; metadata.Artwork ??= new List(); metadata.Artwork.Add(artwork); } var version = new MediaVersion { Name = "Main", Duration = TimeSpan.FromMilliseconds(media.Duration), Width = media.Width, Height = media.Height, AudioCodec = media.AudioCodec, VideoCodec = media.VideoCodec, VideoProfile = media.VideoProfile, // specifically omit sample aspect ratio DateUpdated = lastWriteTime, MediaFiles = new List { new PlexMediaFile { PlexId = part.Id, Key = part.Key, Path = part.File } } }; var movie = new PlexMovie { Key = response.Key, MovieMetadata = new List { metadata }, MediaVersions = new List { version } }; return movie; } private Option ProjectToMediaVersion(PlexMetadataResponse response) { Option maybeStream = response.Media.Head().Part.Head().Stream.Find(s => s.StreamType == 1); return maybeStream.Map( stream => new MediaVersion { SampleAspectRatio = stream.PixelAspectRatio, VideoScanKind = stream.ScanType switch { "interlaced" => VideoScanKind.Interlaced, "progressive" => VideoScanKind.Progressive, _ => VideoScanKind.Unknown } }); } private PlexShow ProjectToShow(PlexMetadataResponse response, int mediaSourceId) { DateTime dateAdded = DateTimeOffset.FromUnixTimeSeconds(response.AddedAt).DateTime; DateTime lastWriteTime = DateTimeOffset.FromUnixTimeSeconds(response.UpdatedAt).DateTime; var metadata = new ShowMetadata { Title = response.Title, SortTitle = _fallbackMetadataProvider.GetSortTitle(response.Title), Plot = response.Summary, Year = response.Year, Tagline = response.Tagline, DateAdded = dateAdded, DateUpdated = lastWriteTime, Genres = Optional(response.Genre).Flatten().Map(g => new Genre { Name = g.Tag }).ToList() }; if (DateTime.TryParse(response.OriginallyAvailableAt, out DateTime releaseDate)) { metadata.ReleaseDate = releaseDate; } if (!string.IsNullOrWhiteSpace(response.Thumb)) { var path = $"plex/{mediaSourceId}{response.Thumb}"; var artwork = new Artwork { ArtworkKind = ArtworkKind.Poster, Path = path, DateAdded = dateAdded, DateUpdated = lastWriteTime }; metadata.Artwork ??= new List(); metadata.Artwork.Add(artwork); } if (!string.IsNullOrWhiteSpace(response.Art)) { var path = $"plex/{mediaSourceId}{response.Art}"; var artwork = new Artwork { ArtworkKind = ArtworkKind.FanArt, Path = path, DateAdded = dateAdded, DateUpdated = lastWriteTime }; metadata.Artwork ??= new List(); metadata.Artwork.Add(artwork); } var show = new PlexShow { Key = response.Key, ShowMetadata = new List { metadata } }; return show; } private PlexSeason ProjectToSeason(PlexMetadataResponse response, int mediaSourceId) { DateTime dateAdded = DateTimeOffset.FromUnixTimeSeconds(response.AddedAt).DateTime; DateTime lastWriteTime = DateTimeOffset.FromUnixTimeSeconds(response.UpdatedAt).DateTime; var metadata = new SeasonMetadata { Title = response.Title, SortTitle = _fallbackMetadataProvider.GetSortTitle(response.Title), Year = response.Year, DateAdded = dateAdded, DateUpdated = lastWriteTime }; if (!string.IsNullOrWhiteSpace(response.Thumb)) { var path = $"plex/{mediaSourceId}{response.Thumb}"; var artwork = new Artwork { ArtworkKind = ArtworkKind.Poster, Path = path, DateAdded = dateAdded, DateUpdated = lastWriteTime }; metadata.Artwork ??= new List(); metadata.Artwork.Add(artwork); } if (!string.IsNullOrWhiteSpace(response.Art)) { var path = $"plex/{mediaSourceId}{response.Art}"; var artwork = new Artwork { ArtworkKind = ArtworkKind.FanArt, Path = path, DateAdded = dateAdded, DateUpdated = lastWriteTime }; metadata.Artwork ??= new List(); metadata.Artwork.Add(artwork); } var season = new PlexSeason { Key = response.Key, SeasonNumber = response.Index, SeasonMetadata = new List { metadata } }; return season; } private PlexEpisode ProjectToEpisode(PlexMetadataResponse response, int mediaSourceId) { PlexMediaResponse media = response.Media.Head(); PlexPartResponse part = media.Part.Head(); DateTime dateAdded = DateTimeOffset.FromUnixTimeSeconds(response.AddedAt).DateTime; DateTime lastWriteTime = DateTimeOffset.FromUnixTimeSeconds(response.UpdatedAt).DateTime; var metadata = new EpisodeMetadata { Title = response.Title, SortTitle = _fallbackMetadataProvider.GetSortTitle(response.Title), Plot = response.Summary, Year = response.Year, Tagline = response.Tagline, DateAdded = dateAdded, DateUpdated = lastWriteTime }; if (DateTime.TryParse(response.OriginallyAvailableAt, out DateTime releaseDate)) { metadata.ReleaseDate = releaseDate; } if (!string.IsNullOrWhiteSpace(response.Thumb)) { var path = $"plex/{mediaSourceId}{response.Thumb}"; var artwork = new Artwork { ArtworkKind = ArtworkKind.Thumbnail, Path = path, DateAdded = dateAdded, DateUpdated = lastWriteTime }; metadata.Artwork ??= new List(); metadata.Artwork.Add(artwork); } var version = new MediaVersion { Name = "Main", Duration = TimeSpan.FromMilliseconds(media.Duration), Width = media.Width, Height = media.Height, AudioCodec = media.AudioCodec, VideoCodec = media.VideoCodec, VideoProfile = media.VideoProfile, // specifically omit sample aspect ratio DateUpdated = lastWriteTime, MediaFiles = new List { new PlexMediaFile { PlexId = part.Id, Key = part.Key, Path = part.File } } }; var episode = new PlexEpisode { Key = response.Key, EpisodeNumber = response.Index, EpisodeMetadata = new List { metadata }, MediaVersions = new List { version } }; return episode; } } }