Ombi/src/Ombi.Helpers/PlexHelper.cs

#region Copyright
// /************************************************************************
//    Copyright (c) 2017 Jamie Rees
//    File: PlexHelper.cs
//    Created By: Jamie Rees
//   
//    Permission is hereby granted, free of charge, to any person obtaining
//    a copy of this software and associated documentation files (the
//    "Software"), to deal in the Software without restriction, including
//    without limitation the rights to use, copy, modify, merge, publish,
//    distribute, sublicense, and/or sell copies of the Software, and to
//    permit persons to whom the Software is furnished to do so, subject to
//    the following conditions:
//   
//    The above copyright notice and this permission notice shall be
//    included in all copies or substantial portions of the Software.
//   
//    THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
//    EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
//    MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
//    NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
//    LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
//    OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
//    WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
//  ************************************************************************/
#endregion

using System;
using System.Globalization;
using System.Text.RegularExpressions;

namespace Ombi.Helpers
{
    public class PlexHelper
    {
        private const string ImdbMatchExpression = "tt([0-9]{1,10})";
        private const string TvDbIdMatchExpression = "//[0-9]+/?([0-9]{1,3})/?([0-9]{1,3})";

        public static ProviderId GetProviderIdFromPlexGuid(string guid)
        {
            //com.plexapp.agents.thetvdb://269586/2/8?lang=en
            //com.plexapp.agents.themoviedb://390043?lang=en
            //com.plexapp.agents.imdb://tt2543164?lang=en
            //plex://movie/5e1632df2d4d84003e48e54e
            // https://github.com/Ombi-app/Ombi/issues/3277
            if (string.IsNullOrEmpty(guid))
            {
                return new ProviderId();
            }

            var guidSplit = guid.Split(new[] { '/', '?' }, StringSplitOptions.RemoveEmptyEntries);
            if (guidSplit.Length > 1)
            {
                if (guid.Contains("thetvdb", CompareOptions.IgnoreCase))
                {
                    return new ProviderId
                    {
                        TheTvDb = guidSplit[1]
                    };
                }
                if (guid.Contains("themoviedb", CompareOptions.IgnoreCase) || guid.Contains("tmdb", CompareOptions.IgnoreCase))
                {
                    return new ProviderId
                    {
                        TheMovieDb = guidSplit[1]
                    };
                }
                if (guid.Contains("imdb", CompareOptions.IgnoreCase))
                {
                    return new ProviderId
                    {
                        ImdbId = guidSplit[1]
                    };
                }
                if (guid.Contains("plex://", CompareOptions.IgnoreCase))
                {
                    return new ProviderId
                    {
                        Plex = true
                    };
                }

                var imdbRegex = new Regex(ImdbMatchExpression, RegexOptions.Compiled);
                var tvdbRegex = new Regex(TvDbIdMatchExpression, RegexOptions.Compiled);
                var imdbMatch = imdbRegex.IsMatch(guid);
                if (imdbMatch)
                {
                    return new ProviderId
                    {
                        ImdbId = guidSplit[1]
                    };
                }

                // Check if it matches the TvDb pattern
                var tvdbMatch = tvdbRegex.IsMatch(guid);
                if (tvdbMatch)
                {
                    return new ProviderId
                    {
                        TheTvDb = guidSplit[1]
                    };
                }
            }
            return new ProviderId();
        }

        public static string GetPlexMediaUrl(string machineId, string mediaId, string plexHost)
        {
            var url =
                $"web/#!/server/{machineId}/details?key=%2flibrary%2Fmetadata%2F{mediaId}";
            return BuildPlexMediaUrl(url, plexHost);
        }

        public static string BuildPlexMediaUrl(string savedUrl, string plexHost)
        {
            if (savedUrl.Contains("app.plex.tv"))
            {
                var split = savedUrl.Split("https://app.plex.tv/", StringSplitOptions.RemoveEmptyEntries);
                if (split.Length == 1)
                {
                    savedUrl = split[0];
                } 
                else
                {
                    throw new ArgumentException($"Attempt to parse url {savedUrl} and could not");
                }
            }
            if (!plexHost.HasValue())
            {
                plexHost = "https://app.plex.tv/";
            }
            else
            {
                if (plexHost[plexHost.Length - 1] != '/')
                {
                    plexHost += '/';
                }
            }


            return $"{plexHost}{savedUrl}";
        }

        public static ProviderId GetProviderIdsFromMetadata(params string[] guids)
        {
            var providerIds = new ProviderId();
            foreach (var guid in guids)
            {
                var provider = GetProviderIdFromPlexGuid(guid);
                if (provider.Type == ProviderType.Plex)
                {
                    // There are more guids!
                    continue;
                }

                switch (provider.Type)
                {
                    case ProviderType.ImdbId:
                        providerIds.ImdbId = provider.ImdbId;
                        break;
                    case ProviderType.TheMovieDbId:
                        providerIds.TheMovieDb = provider.TheMovieDb;
                        break;
                    case ProviderType.TvDbId:
                        providerIds.TheTvDb = provider.TheTvDb;
                        break;
                    default:
                        throw new ArgumentOutOfRangeException(nameof(provider.Type));
                }
            }

            return providerIds;
        }
    }

    public class ProviderId
    {
        public string TheTvDb { get; set; }
        public string TheMovieDb { get; set; }
        public string ImdbId { get; set; }
        public bool Plex { get; set; }

        public bool Any() => TheTvDb.HasValue() || TheMovieDb.HasValue() || ImdbId.HasValue();

        public ProviderType Type
        {
            get
            {
                if (ImdbId.HasValue())
                {
                    return ProviderType.ImdbId;
                }
                if (TheMovieDb.HasValue())
                {
                    return ProviderType.TheMovieDbId;
                }
                if (TheTvDb.HasValue())
                {
                    return ProviderType.TvDbId;
                }
                if (Plex)
                {
                    return ProviderType.Plex;
                }
                return ProviderType.ImdbId;
            }
        }
    }

    public enum ProviderType
    {
        ImdbId,
        TheMovieDbId,
        TvDbId,
        Plex
    }
}