Ombi/Ombi.Api/TvMazeApi.cs

#region Copyright
// /************************************************************************
//    Copyright (c) 2016 Jamie Rees
//    File: TvMazeApi.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.Collections.Generic;
using System.Linq;
using Newtonsoft.Json;
using NLog;
using Ombi.Api.Models.Tv;
using RestSharp;

namespace Ombi.Api
{
    public class TvMazeApi : TvMazeBase
    {
        public TvMazeApi()
        {
            Api = new ApiRequest();
        }
        private ApiRequest Api { get;  }
        private static Logger Log = LogManager.GetCurrentClassLogger();
        public List<TvMazeSearch> Search(string searchTerm)
        {
            var request = new RestRequest
            {
                Method = Method.GET,
                Resource = "search/shows?q={searchTerm}"
            };
            request.AddUrlSegment("searchTerm", searchTerm);
            request.AddHeader("Content-Type", "application/json");

            return Api.Execute<List<TvMazeSearch>>(request, new Uri(Uri));
        }

        public TvMazeShow ShowLookup(int showId)
        {
            var request = new RestRequest
            {
                Method = Method.GET,
                Resource = "shows/{id}"
            };
            request.AddUrlSegment("id", showId.ToString());
            request.AddHeader("Content-Type", "application/json");

            return Api.Execute<TvMazeShow>(request, new Uri(Uri));
        }

        public IEnumerable<TvMazeEpisodes> EpisodeLookup(int showId)
        {
            var request = new RestRequest
            {
                Method = Method.GET,
                Resource = "shows/{id}/episodes"
            };
            request.AddUrlSegment("id", showId.ToString());
            request.AddHeader("Content-Type", "application/json");

            return Api.Execute<List<TvMazeEpisodes>>(request, new Uri(Uri));
        }

        public TvMazeShow ShowLookupByTheTvDbId(int theTvDbId)
        {
            var request = new RestRequest
            {
                Method = Method.GET,
                Resource = "lookup/shows?thetvdb={id}"
            };
            request.AddUrlSegment("id", theTvDbId.ToString());
            request.AddHeader("Content-Type", "application/json");
            try
            {
                var result = Api.Execute(request, new Uri(Uri));
                var obj = JsonConvert.DeserializeObject<TvMazeShow>(result.Content);

                var episodes = EpisodeLookup(obj.id).ToList();

                foreach (var e in episodes)
                {
                    obj.Season.Add(new TvMazeCustomSeason
                    {
                        SeasonNumber = e.season,
                        EpisodeNumber = e.number
                    });
                }

                return obj;
            }
            catch (Exception e)
            {
                Log.Error(e);
                return null;
            }           
        }

        public List<TvMazeSeasons> GetSeasons(int id)
        {
            var request = new RestRequest
            {
                Method = Method.GET,
                Resource = "shows/{id}/seasons"
            };
            request.AddUrlSegment("id", id.ToString());
            request.AddHeader("Content-Type", "application/json");

            return Api.Execute<List<TvMazeSeasons>>(request, new Uri(Uri));
        }

        public int GetSeasonCount(int id)
        {
            var obj = GetSeasons(id);
            var seasons = obj.Select(x => x.number > 0);
            return seasons.Count();
        }

    }
}