using MediaBrowser.Common.Net; using MediaBrowser.Controller.LiveTv; using MediaBrowser.Model.Dto; using MediaBrowser.Model.LiveTv; using MediaBrowser.Model.Logging; using MediaBrowser.Model.Serialization; using System; using System.Collections.Concurrent; using System.Collections.Generic; using System.Globalization; using System.IO; using System.Linq; using System.Threading; using System.Threading.Tasks; namespace MediaBrowser.Server.Implementations.LiveTv.Listings { public class SchedulesDirect : IListingsProvider { private readonly ILogger _logger; private readonly IJsonSerializer _jsonSerializer; private readonly IHttpClient _httpClient; private const string UserAgent = "EmbyTV"; private readonly SemaphoreSlim _tokenSemaphore = new SemaphoreSlim(1, 1); private const string ApiUrl = "https://json.schedulesdirect.org/20141201"; private readonly ConcurrentDictionary _channelPair = new ConcurrentDictionary(); public SchedulesDirect(ILogger logger, IJsonSerializer jsonSerializer, IHttpClient httpClient) { _logger = logger; _jsonSerializer = jsonSerializer; _httpClient = httpClient; } public async Task> GetProgramsAsync(ListingsProviderInfo info, string channelNumber, DateTime startDateUtc, DateTime endDateUtc, CancellationToken cancellationToken) { List programsInfo = new List(); var token = await GetToken(info, cancellationToken); if (string.IsNullOrWhiteSpace(token)) { return programsInfo; } if (string.IsNullOrWhiteSpace(info.ListingsId)) { return programsInfo; } var httpOptions = new HttpRequestOptions() { Url = ApiUrl + "/schedules", UserAgent = UserAgent, CancellationToken = cancellationToken }; httpOptions.RequestHeaders["token"] = token; List dates = new List(); int numberOfDay = 0; DateTime lastEntry = startDateUtc; while (lastEntry != endDateUtc) { lastEntry = startDateUtc.AddDays(numberOfDay); dates.Add(lastEntry.ToString("yyyy-MM-dd")); numberOfDay++; } ScheduleDirect.Station station = null; if (!_channelPair.TryGetValue(channelNumber, out station)) { return programsInfo; } string stationID = station.stationID; _logger.Info("Channel Station ID is: " + stationID); List requestList = new List() { new ScheduleDirect.RequestScheduleForChannel() { stationID = stationID, date = dates } }; var requestString = _jsonSerializer.SerializeToString(requestList); _logger.Debug("Request string for schedules is: " + requestString); httpOptions.RequestContent = requestString; using (var response = await _httpClient.Post(httpOptions)) { StreamReader reader = new StreamReader(response.Content); string responseString = reader.ReadToEnd(); var dailySchedules = _jsonSerializer.DeserializeFromString>(responseString); _logger.Debug("Found " + dailySchedules.Count() + " programs on " + channelNumber + " ScheduleDirect"); httpOptions = new HttpRequestOptions() { Url = ApiUrl + "/programs", UserAgent = UserAgent, CancellationToken = cancellationToken }; httpOptions.RequestHeaders["token"] = token; List programsID = new List(); programsID = dailySchedules.SelectMany(d => d.programs.Select(s => s.programID)).Distinct().ToList(); var requestBody = "[\"" + string.Join("\", \"", programsID) + "\"]"; httpOptions.RequestContent = requestBody; using (var innerResponse = await _httpClient.Post(httpOptions)) { StreamReader innerReader = new StreamReader(innerResponse.Content); responseString = innerReader.ReadToEnd(); var programDetails = _jsonSerializer.DeserializeFromString>( responseString); var programDict = programDetails.ToDictionary(p => p.programID, y => y); var images = await GetImageForPrograms(programDetails.Where(p => p.hasImageArtwork).Select(p => p.programID).ToList(), cancellationToken); var schedules = dailySchedules.SelectMany(d => d.programs); foreach (ScheduleDirect.Program schedule in schedules) { _logger.Debug("Proccesing Schedule for statio ID " + stationID + " which corresponds to channel " + channelNumber + " and program id " + schedule.programID + " which says it has images? " + programDict[schedule.programID].hasImageArtwork); var imageIndex = images.FindIndex(i => i.programID == schedule.programID.Substring(0, 10)); if (imageIndex > -1) { programDict[schedule.programID].images = GetProgramLogo(ApiUrl, images[imageIndex]); } programsInfo.Add(GetProgram(channelNumber, schedule, programDict[schedule.programID])); } _logger.Info("Finished with EPGData"); } } return programsInfo; } public async Task AddMetadata(ListingsProviderInfo info, List channels, CancellationToken cancellationToken) { var token = await GetToken(info, cancellationToken); _channelPair.Clear(); if (!String.IsNullOrWhiteSpace(token) && !String.IsNullOrWhiteSpace(info.ListingsId)) { var httpOptions = new HttpRequestOptions() { Url = ApiUrl + "/lineups/" + info.ListingsId, UserAgent = UserAgent, CancellationToken = cancellationToken }; httpOptions.RequestHeaders["token"] = token; using (var response = await _httpClient.Get(httpOptions)) { var root = _jsonSerializer.DeserializeFromStream(response); _logger.Info("Found " + root.map.Count() + " channels on the lineup on ScheduleDirect"); _logger.Info("Mapping Stations to Channel"); foreach (ScheduleDirect.Map map in root.map) { var channel = (map.channel ?? (map.atscMajor + "." + map.atscMinor)).TrimStart('0'); _logger.Debug("Found channel: " + channel + " in Schedules Direct"); var schChannel = root.stations.FirstOrDefault(item => item.stationID == map.stationID); if (!_channelPair.ContainsKey(channel) && channel != "0.0" && schChannel != null) { _channelPair.TryAdd(channel, schChannel); } } _logger.Info("Added " + _channelPair.Count() + " channels to the dictionary"); foreach (ChannelInfo channel in channels) { // Helper.logger.Info("Modifyin channel " + channel.Number); if (_channelPair.ContainsKey(channel.Number)) { string channelName; if (_channelPair[channel.Number].logo != null) { channel.ImageUrl = _channelPair[channel.Number].logo.URL; channel.HasImage = true; } if (_channelPair[channel.Number].affiliate != null) { channelName = _channelPair[channel.Number].affiliate; } else { channelName = _channelPair[channel.Number].name; } channel.Name = channelName; } else { _logger.Info("Schedules Direct doesnt have data for channel: " + channel.Number + " " + channel.Name); } } } } } private ProgramInfo GetProgram(string channel, ScheduleDirect.Program programInfo, ScheduleDirect.ProgramDetails details) { //_logger.Debug("Show type is: " + (details.showType ?? "No ShowType")); DateTime startAt = DateTime.ParseExact(programInfo.airDateTime, "yyyy'-'MM'-'dd'T'HH':'mm':'ss'Z'", CultureInfo.InvariantCulture); DateTime endAt = startAt.AddSeconds(programInfo.duration); ProgramAudio audioType = ProgramAudio.Stereo; bool hdtv = false; bool repeat = (programInfo.@new == null); string newID = programInfo.programID + "T" + startAt.Ticks + "C" + channel; if (programInfo.audioProperties != null) { if (programInfo.audioProperties.Exists(item => item == "stereo")) { audioType = ProgramAudio.Stereo; } else { audioType = ProgramAudio.Mono; } } if ((programInfo.videoProperties != null)) { hdtv = programInfo.videoProperties.Exists(item => item == "hdtv"); } string desc = ""; if (details.descriptions != null) { if (details.descriptions.description1000 != null) { desc = details.descriptions.description1000[0].description; } else if (details.descriptions.description100 != null) { desc = details.descriptions.description100[0].description; } } ScheduleDirect.Gracenote gracenote; string EpisodeTitle = ""; if (details.metadata != null) { gracenote = details.metadata.Find(x => x.Gracenote != null).Gracenote; if ((details.showType ?? "No ShowType") == "Series") { EpisodeTitle = "Season: " + gracenote.season + " Episode: " + gracenote.episode; } } if (details.episodeTitle150 != null) { EpisodeTitle = EpisodeTitle + " " + details.episodeTitle150; } bool hasImage = false; var imageLink = ""; if (details.hasImageArtwork) { imageLink = details.images; } var info = new ProgramInfo { ChannelId = channel, Id = newID, Overview = desc, StartDate = startAt, EndDate = endAt, Genres = new List() { "N/A" }, Name = details.titles[0].title120 ?? "Unkown", OfficialRating = "0", CommunityRating = null, EpisodeTitle = EpisodeTitle, Audio = audioType, IsHD = hdtv, IsRepeat = repeat, IsSeries = ((details.showType ?? "No ShowType") == "Series") || (details.showType ?? "No ShowType") == "Miniseries", ImageUrl = imageLink, HasImage = details.hasImageArtwork, IsNews = false, IsKids = false, IsSports = ((details.showType ?? "No ShowType") == "Sports non-event") || (details.showType ?? "No ShowType") == "Sports event", IsLive = false, IsMovie = (details.showType ?? "No ShowType") == "Feature Film" || (details.showType ?? "No ShowType") == "TV Movie" || (details.showType ?? "No ShowType") == "Short Film", IsPremiere = false, }; //logger.Info("Done init"); if (null != details.originalAirDate) { info.OriginalAirDate = DateTime.Parse(details.originalAirDate); } if (details.genres != null) { info.Genres = details.genres.Where(g => !string.IsNullOrWhiteSpace(g)).ToList(); info.IsNews = details.genres.Contains("news", StringComparer.OrdinalIgnoreCase); info.IsKids = false; } return info; } private string GetProgramLogo(string apiUrl, ScheduleDirect.ShowImages images) { string url = ""; if (images.data != null) { var smallImages = images.data.Where(i => i.size == "Sm").ToList(); if (smallImages.Any()) { images.data = smallImages; } var logoIndex = images.data.FindIndex(i => i.category == "Logo"); if (logoIndex == -1) { logoIndex = 0; } if (images.data[logoIndex].uri.Contains("http")) { url = images.data[logoIndex].uri; } else { url = apiUrl + "/image/" + images.data[logoIndex].uri; } //_logger.Debug("URL for image is : " + url); } return url; } private async Task> GetImageForPrograms(List programIds, CancellationToken cancellationToken) { var imageIdString = "["; programIds.ForEach(i => { if (!imageIdString.Contains(i.Substring(0, 10))) { imageIdString += "\"" + i.Substring(0, 10) + "\","; } ; }); imageIdString = imageIdString.TrimEnd(',') + "]"; _logger.Debug("Json for show images = " + imageIdString); var httpOptions = new HttpRequestOptions() { Url = ApiUrl + "/metadata/programs", UserAgent = UserAgent, CancellationToken = cancellationToken, RequestContent = imageIdString }; List images; using (var innerResponse2 = await _httpClient.Post(httpOptions)) { images = _jsonSerializer.DeserializeFromStream>( innerResponse2.Content); } return images; } public async Task> GetHeadends(ListingsProviderInfo info, string country, string location, CancellationToken cancellationToken) { var token = await GetToken(info, cancellationToken); var lineups = new List(); if (string.IsNullOrWhiteSpace(token)) { return lineups; } _logger.Info("Headends on account "); var countryParam = string.Equals("ca", country, StringComparison.OrdinalIgnoreCase) ? "can" : "USA"; var options = new HttpRequestOptions() { Url = ApiUrl + "/headends?country=" + countryParam + "&postalcode=" + location, UserAgent = UserAgent, CancellationToken = cancellationToken }; options.RequestHeaders["token"] = token; try { using (Stream responce = await _httpClient.Get(options).ConfigureAwait(false)) { var root = _jsonSerializer.DeserializeFromStream>(responce); _logger.Info("Lineups on account "); if (root != null) { foreach (ScheduleDirect.Headends headend in root) { _logger.Info("Headend: " + headend.headend); foreach (ScheduleDirect.Lineup lineup in headend.lineups) { _logger.Info("Headend: " + lineup.uri.Substring(18)); lineups.Add(new NameIdPair() { Name = string.IsNullOrWhiteSpace(lineup.name) ? lineup.lineup : lineup.name, Id = lineup.uri.Substring(18) }); } } } else { _logger.Info("No lineups on account"); } } } catch (Exception ex) { _logger.Error("Error getting headends", ex); } return lineups; } private readonly ConcurrentDictionary _tokens = new ConcurrentDictionary(); private async Task GetToken(ListingsProviderInfo info, CancellationToken cancellationToken) { var username = info.Username; // Reset the token if there's no username if (string.IsNullOrWhiteSpace(username)) { return null; } var password = info.Password; if (string.IsNullOrWhiteSpace(password)) { return null; } NameValuePair savedToken = null; if (!_tokens.TryGetValue(username, out savedToken)) { savedToken = new NameValuePair(); _tokens.TryAdd(username, savedToken); } if (!string.IsNullOrWhiteSpace(savedToken.Name) && !string.IsNullOrWhiteSpace(savedToken.Value)) { long ticks; if (long.TryParse(savedToken.Value, NumberStyles.Any, CultureInfo.InvariantCulture, out ticks)) { // If it's under 24 hours old we can still use it if ((DateTime.UtcNow.Ticks - ticks) < TimeSpan.FromHours(24).Ticks) { return savedToken.Name; } } } await _tokenSemaphore.WaitAsync(cancellationToken).ConfigureAwait(false); try { var result = await GetTokenInternal(username, password, cancellationToken).ConfigureAwait(false); savedToken.Name = result; savedToken.Value = DateTime.UtcNow.Ticks.ToString(CultureInfo.InvariantCulture); return result; } finally { _tokenSemaphore.Release(); } } private async Task GetTokenInternal(string username, string password, CancellationToken cancellationToken) { var httpOptions = new HttpRequestOptions() { Url = ApiUrl + "/token", UserAgent = UserAgent, RequestContent = "{\"username\":\"" + username + "\",\"password\":\"" + password + "\"}", CancellationToken = cancellationToken }; //_logger.Info("Obtaining token from Schedules Direct from addres: " + httpOptions.Url + " with body " + // httpOptions.RequestContent); using (var responce = await _httpClient.Post(httpOptions)) { var root = _jsonSerializer.DeserializeFromStream(responce.Content); if (root.message == "OK") { _logger.Info("Authenticated with Schedules Direct token: " + root.token); return root.token; } throw new ApplicationException("Could not authenticate with Schedules Direct Error: " + root.message); } } private async Task AddLineupToAccount(ListingsProviderInfo info, CancellationToken cancellationToken) { var token = await GetToken(info, cancellationToken); if (string.IsNullOrWhiteSpace(token)) { throw new ArgumentException("Authentication required."); } if (string.IsNullOrWhiteSpace(info.ListingsId)) { throw new ArgumentException("Listings Id required"); } _logger.Info("Adding new LineUp "); var httpOptions = new HttpRequestOptions() { Url = ApiUrl + "/lineups/" + info.ListingsId, UserAgent = UserAgent, CancellationToken = cancellationToken }; httpOptions.RequestHeaders["token"] = token; using (var response = await _httpClient.SendAsync(httpOptions, "PUT")) { } } public string Name { get { return "Schedules Direct"; } } public string Type { get { return "SchedulesDirect"; } } private async Task HasLineup(ListingsProviderInfo info, CancellationToken cancellationToken) { if (string.IsNullOrWhiteSpace(info.ListingsId)) { throw new ArgumentException("Listings Id required"); } var token = await GetToken(info, cancellationToken); _logger.Info("Headends on account "); var options = new HttpRequestOptions() { Url = ApiUrl + "/lineups", UserAgent = UserAgent, CancellationToken = cancellationToken }; options.RequestHeaders["token"] = token; using (var response = await _httpClient.Get(options).ConfigureAwait(false)) { var root = _jsonSerializer.DeserializeFromStream(response); return root.lineups.Any(i => string.Equals(info.ListingsId, i.lineup, StringComparison.OrdinalIgnoreCase)); } } public async Task Validate(ListingsProviderInfo info) { if (string.IsNullOrWhiteSpace(info.ListingsId)) { throw new ArgumentException("Listings Id required"); } var hasLineup = await HasLineup(info, CancellationToken.None).ConfigureAwait(false); if (!hasLineup) { await AddLineupToAccount(info, CancellationToken.None).ConfigureAwait(false); } } public Task> GetLineups(ListingsProviderInfo info, string country, string location) { return GetHeadends(info, country, location, CancellationToken.None); } public class ScheduleDirect { public class Token { public int code { get; set; } public string message { get; set; } public string serverID { get; set; } public string token { get; set; } } public class Lineup { public string lineup { get; set; } public string name { get; set; } public string transport { get; set; } public string location { get; set; } public string uri { get; set; } } public class Lineups { public int code { get; set; } public string serverID { get; set; } public string datetime { get; set; } public List lineups { get; set; } } public class Headends { public string headend { get; set; } public string transport { get; set; } public string location { get; set; } public List lineups { get; set; } } public class Map { public string stationID { get; set; } public string channel { get; set; } public int uhfVhf { get; set; } public int atscMajor { get; set; } public int atscMinor { get; set; } } public class Broadcaster { public string city { get; set; } public string state { get; set; } public string postalcode { get; set; } public string country { get; set; } } public class Logo { public string URL { get; set; } public int height { get; set; } public int width { get; set; } public string md5 { get; set; } } public class Station { public string stationID { get; set; } public string name { get; set; } public string callsign { get; set; } public List broadcastLanguage { get; set; } public List descriptionLanguage { get; set; } public Broadcaster broadcaster { get; set; } public string affiliate { get; set; } public Logo logo { get; set; } public bool? isCommercialFree { get; set; } } public class Metadata { public string lineup { get; set; } public string modified { get; set; } public string transport { get; set; } } public class Channel { public List map { get; set; } public List stations { get; set; } public Metadata metadata { get; set; } } public class RequestScheduleForChannel { public string stationID { get; set; } public List date { get; set; } } public class Rating { public string body { get; set; } public string code { get; set; } } public class Multipart { public int partNumber { get; set; } public int totalParts { get; set; } } public class Program { public string programID { get; set; } public string airDateTime { get; set; } public int duration { get; set; } public string md5 { get; set; } public List audioProperties { get; set; } public List videoProperties { get; set; } public List ratings { get; set; } public bool? @new { get; set; } public Multipart multipart { get; set; } } public class MetadataSchedule { public string modified { get; set; } public string md5 { get; set; } public string startDate { get; set; } public string endDate { get; set; } public int days { get; set; } } public class Day { public string stationID { get; set; } public List programs { get; set; } public MetadataSchedule metadata { get; set; } } // public class Title { public string title120 { get; set; } } public class EventDetails { public string subType { get; set; } } public class Description100 { public string descriptionLanguage { get; set; } public string description { get; set; } } public class Description1000 { public string descriptionLanguage { get; set; } public string description { get; set; } } public class DescriptionsProgram { public List description100 { get; set; } public List description1000 { get; set; } } public class Gracenote { public int season { get; set; } public int episode { get; set; } } public class MetadataPrograms { public Gracenote Gracenote { get; set; } } public class ContentRating { public string body { get; set; } public string code { get; set; } } public class Cast { public string billingOrder { get; set; } public string role { get; set; } public string nameId { get; set; } public string personId { get; set; } public string name { get; set; } public string characterName { get; set; } } public class Crew { public string billingOrder { get; set; } public string role { get; set; } public string nameId { get; set; } public string personId { get; set; } public string name { get; set; } } public class QualityRating { public string ratingsBody { get; set; } public string rating { get; set; } public string minRating { get; set; } public string maxRating { get; set; } public string increment { get; set; } } public class Movie { public string year { get; set; } public int duration { get; set; } public List qualityRating { get; set; } } public class Recommendation { public string programID { get; set; } public string title120 { get; set; } } public class ProgramDetails { public string programID { get; set; } public List titles { get; set; } public EventDetails eventDetails { get; set; } public DescriptionsProgram descriptions { get; set; } public string originalAirDate { get; set; } public List<string> genres { get; set; } public string episodeTitle150 { get; set; } public List<MetadataPrograms> metadata { get; set; } public List<ContentRating> contentRating { get; set; } public List<Cast> cast { get; set; } public List<Crew> crew { get; set; } public string showType { get; set; } public bool hasImageArtwork { get; set; } public string images { get; set; } public string imageID { get; set; } public string md5 { get; set; } public List<string> contentAdvisory { get; set; } public Movie movie { get; set; } public List<Recommendation> recommendations { get; set; } } public class Caption { public string content { get; set; } public string lang { get; set; } } public class ImageData { public string width { get; set; } public string height { get; set; } public string uri { get; set; } public string size { get; set; } public string aspect { get; set; } public string category { get; set; } public string text { get; set; } public string primary { get; set; } public string tier { get; set; } public Caption caption { get; set; } } public class ShowImages { public string programID { get; set; } public List<ImageData> data { get; set; } } } } }