LiveTvManager.cs 91 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496
  1. using MediaBrowser.Common.Configuration;
  2. using MediaBrowser.Common.Extensions;
  3. using MediaBrowser.Common.Progress;
  4. using MediaBrowser.Controller.Configuration;
  5. using MediaBrowser.Controller.Drawing;
  6. using MediaBrowser.Controller.Dto;
  7. using MediaBrowser.Controller.Entities;
  8. using MediaBrowser.Controller.Library;
  9. using MediaBrowser.Controller.LiveTv;
  10. using MediaBrowser.Controller.Persistence;
  11. using MediaBrowser.Controller.Providers;
  12. using MediaBrowser.Controller.Sorting;
  13. using MediaBrowser.Model.Dto;
  14. using MediaBrowser.Model.Entities;
  15. using MediaBrowser.Model.LiveTv;
  16. using Microsoft.Extensions.Logging;
  17. using MediaBrowser.Model.Querying;
  18. using MediaBrowser.Model.Serialization;
  19. using System;
  20. using System.Collections.Generic;
  21. using System.Linq;
  22. using System.Threading;
  23. using System.Threading.Tasks;
  24. using MediaBrowser.Model.IO;
  25. using MediaBrowser.Controller.Entities.Movies;
  26. using MediaBrowser.Controller.Entities.TV;
  27. using MediaBrowser.Model.Events;
  28. using MediaBrowser.Model.Extensions;
  29. using MediaBrowser.Model.Globalization;
  30. using MediaBrowser.Model.Tasks;
  31. using Emby.Server.Implementations.LiveTv.Listings;
  32. using MediaBrowser.Controller.Channels;
  33. using Emby.Server.Implementations.Library;
  34. using MediaBrowser.Controller;
  35. using MediaBrowser.Common.Net;
  36. namespace Emby.Server.Implementations.LiveTv
  37. {
  38. /// <summary>
  39. /// Class LiveTvManager
  40. /// </summary>
  41. public class LiveTvManager : ILiveTvManager, IDisposable
  42. {
  43. private readonly IServerConfigurationManager _config;
  44. private readonly ILogger _logger;
  45. private readonly IItemRepository _itemRepo;
  46. private readonly IUserManager _userManager;
  47. private readonly IUserDataManager _userDataManager;
  48. private readonly ILibraryManager _libraryManager;
  49. private readonly ITaskManager _taskManager;
  50. private readonly IJsonSerializer _jsonSerializer;
  51. private readonly IProviderManager _providerManager;
  52. private readonly Func<IChannelManager> _channelManager;
  53. private readonly IDtoService _dtoService;
  54. private readonly ILocalizationManager _localization;
  55. private readonly LiveTvDtoService _tvDtoService;
  56. private ILiveTvService[] _services = new ILiveTvService[] { };
  57. private ITunerHost[] _tunerHosts = Array.Empty<ITunerHost>();
  58. private IListingsProvider[] _listingProviders = Array.Empty<IListingsProvider>();
  59. private readonly IFileSystem _fileSystem;
  60. public event EventHandler<GenericEventArgs<TimerEventInfo>> SeriesTimerCancelled;
  61. public event EventHandler<GenericEventArgs<TimerEventInfo>> TimerCancelled;
  62. public event EventHandler<GenericEventArgs<TimerEventInfo>> TimerCreated;
  63. public event EventHandler<GenericEventArgs<TimerEventInfo>> SeriesTimerCreated;
  64. public string GetEmbyTvActiveRecordingPath(string id)
  65. {
  66. return EmbyTV.EmbyTV.Current.GetActiveRecordingPath(id);
  67. }
  68. private IServerApplicationHost _appHost;
  69. private IHttpClient _httpClient;
  70. public LiveTvManager(
  71. IServerApplicationHost appHost,
  72. IHttpClient httpClient,
  73. IServerConfigurationManager config,
  74. ILogger logger,
  75. IItemRepository itemRepo,
  76. IImageProcessor imageProcessor,
  77. IUserDataManager userDataManager,
  78. IDtoService dtoService,
  79. IUserManager userManager,
  80. ILibraryManager libraryManager,
  81. ITaskManager taskManager,
  82. ILocalizationManager localization,
  83. IJsonSerializer jsonSerializer,
  84. IProviderManager providerManager,
  85. IFileSystem fileSystem,
  86. Func<IChannelManager> channelManager)
  87. {
  88. _appHost = appHost;
  89. _config = config;
  90. _logger = logger;
  91. _itemRepo = itemRepo;
  92. _userManager = userManager;
  93. _libraryManager = libraryManager;
  94. _taskManager = taskManager;
  95. _localization = localization;
  96. _jsonSerializer = jsonSerializer;
  97. _providerManager = providerManager;
  98. _fileSystem = fileSystem;
  99. _dtoService = dtoService;
  100. _userDataManager = userDataManager;
  101. _channelManager = channelManager;
  102. _httpClient = httpClient;
  103. _tvDtoService = new LiveTvDtoService(dtoService, imageProcessor, logger, appHost, _libraryManager);
  104. }
  105. /// <summary>
  106. /// Gets the services.
  107. /// </summary>
  108. /// <value>The services.</value>
  109. public IReadOnlyList<ILiveTvService> Services => _services;
  110. private LiveTvOptions GetConfiguration()
  111. {
  112. return _config.GetConfiguration<LiveTvOptions>("livetv");
  113. }
  114. /// <summary>
  115. /// Adds the parts.
  116. /// </summary>
  117. /// <param name="services">The services.</param>
  118. /// <param name="tunerHosts">The tuner hosts.</param>
  119. /// <param name="listingProviders">The listing providers.</param>
  120. public void AddParts(IEnumerable<ILiveTvService> services, IEnumerable<ITunerHost> tunerHosts, IEnumerable<IListingsProvider> listingProviders)
  121. {
  122. _services = services.ToArray();
  123. _tunerHosts = tunerHosts.Where(i => i.IsSupported).ToArray();
  124. _listingProviders = listingProviders.ToArray();
  125. foreach (var service in _services)
  126. {
  127. service.DataSourceChanged += service_DataSourceChanged;
  128. if (service is EmbyTV.EmbyTV embyTv)
  129. {
  130. embyTv.TimerCreated += EmbyTv_TimerCreated;
  131. embyTv.TimerCancelled += EmbyTv_TimerCancelled;
  132. }
  133. }
  134. }
  135. private void EmbyTv_TimerCancelled(object sender, GenericEventArgs<string> e)
  136. {
  137. var timerId = e.Argument;
  138. TimerCancelled?.Invoke(this, new GenericEventArgs<TimerEventInfo>
  139. {
  140. Argument = new TimerEventInfo
  141. {
  142. Id = timerId
  143. }
  144. });
  145. }
  146. private void EmbyTv_TimerCreated(object sender, GenericEventArgs<TimerInfo> e)
  147. {
  148. var timer = e.Argument;
  149. var service = sender as ILiveTvService;
  150. TimerCreated?.Invoke(this, new GenericEventArgs<TimerEventInfo>
  151. {
  152. Argument = new TimerEventInfo
  153. {
  154. ProgramId = _tvDtoService.GetInternalProgramId(timer.ProgramId),
  155. Id = timer.Id
  156. }
  157. });
  158. }
  159. public ITunerHost[] TunerHosts => _tunerHosts;
  160. public IListingsProvider[] ListingProviders => _listingProviders;
  161. public List<NameIdPair> GetTunerHostTypes()
  162. {
  163. return _tunerHosts.OrderBy(i => i.Name).Select(i => new NameIdPair
  164. {
  165. Name = i.Name,
  166. Id = i.Type
  167. }).ToList();
  168. }
  169. public Task<List<TunerHostInfo>> DiscoverTuners(bool newDevicesOnly, CancellationToken cancellationToken)
  170. {
  171. return EmbyTV.EmbyTV.Current.DiscoverTuners(newDevicesOnly, cancellationToken);
  172. }
  173. void service_DataSourceChanged(object sender, EventArgs e)
  174. {
  175. if (!_isDisposed)
  176. {
  177. _taskManager.CancelIfRunningAndQueue<RefreshChannelsScheduledTask>();
  178. }
  179. }
  180. public QueryResult<BaseItem> GetInternalChannels(LiveTvChannelQuery query, DtoOptions dtoOptions, CancellationToken cancellationToken)
  181. {
  182. var user = query.UserId.Equals(Guid.Empty) ? null : _userManager.GetUserById(query.UserId);
  183. var topFolder = GetInternalLiveTvFolder(cancellationToken);
  184. var internalQuery = new InternalItemsQuery(user)
  185. {
  186. IsMovie = query.IsMovie,
  187. IsNews = query.IsNews,
  188. IsKids = query.IsKids,
  189. IsSports = query.IsSports,
  190. IsSeries = query.IsSeries,
  191. IncludeItemTypes = new[] { typeof(LiveTvChannel).Name },
  192. TopParentIds = new[] { topFolder.Id },
  193. IsFavorite = query.IsFavorite,
  194. IsLiked = query.IsLiked,
  195. StartIndex = query.StartIndex,
  196. Limit = query.Limit,
  197. DtoOptions = dtoOptions
  198. };
  199. var orderBy = internalQuery.OrderBy.ToList();
  200. orderBy.AddRange(query.SortBy.Select(i => new ValueTuple<string, SortOrder>(i, query.SortOrder ?? SortOrder.Ascending)));
  201. if (query.EnableFavoriteSorting)
  202. {
  203. orderBy.Insert(0, new ValueTuple<string, SortOrder>(ItemSortBy.IsFavoriteOrLiked, SortOrder.Descending));
  204. }
  205. if (!internalQuery.OrderBy.Any(i => string.Equals(i.Item1, ItemSortBy.SortName, StringComparison.OrdinalIgnoreCase)))
  206. {
  207. orderBy.Add(new ValueTuple<string, SortOrder>(ItemSortBy.SortName, SortOrder.Ascending));
  208. }
  209. internalQuery.OrderBy = orderBy.ToArray();
  210. return _libraryManager.GetItemsResult(internalQuery);
  211. }
  212. public async Task<Tuple<MediaSourceInfo, ILiveStream>> GetChannelStream(string id, string mediaSourceId, List<ILiveStream> currentLiveStreams, CancellationToken cancellationToken)
  213. {
  214. if (string.Equals(id, mediaSourceId, StringComparison.OrdinalIgnoreCase))
  215. {
  216. mediaSourceId = null;
  217. }
  218. var channel = (LiveTvChannel)_libraryManager.GetItemById(id);
  219. bool isVideo = channel.ChannelType == ChannelType.TV;
  220. ILiveTvService service = GetService(channel);
  221. _logger.LogInformation("Opening channel stream from {0}, external channel Id: {1}", service.Name, channel.ExternalId);
  222. MediaSourceInfo info;
  223. ILiveStream liveStream;
  224. if (service is ISupportsDirectStreamProvider supportsManagedStream)
  225. {
  226. liveStream = await supportsManagedStream.GetChannelStreamWithDirectStreamProvider(channel.ExternalId, mediaSourceId, currentLiveStreams, cancellationToken).ConfigureAwait(false);
  227. info = liveStream.MediaSource;
  228. }
  229. else
  230. {
  231. info = await service.GetChannelStream(channel.ExternalId, mediaSourceId, cancellationToken).ConfigureAwait(false);
  232. var openedId = info.Id;
  233. Func<Task> closeFn = () => service.CloseLiveStream(openedId, CancellationToken.None);
  234. liveStream = new ExclusiveLiveStream(info, closeFn);
  235. var startTime = DateTime.UtcNow;
  236. await liveStream.Open(cancellationToken).ConfigureAwait(false);
  237. var endTime = DateTime.UtcNow;
  238. _logger.LogInformation("Live stream opened after {0}ms", (endTime - startTime).TotalMilliseconds);
  239. }
  240. info.RequiresClosing = true;
  241. var idPrefix = service.GetType().FullName.GetMD5().ToString("N") + "_";
  242. info.LiveStreamId = idPrefix + info.Id;
  243. Normalize(info, service, isVideo);
  244. return new Tuple<MediaSourceInfo, ILiveStream>(info, liveStream);
  245. }
  246. public async Task<IEnumerable<MediaSourceInfo>> GetChannelMediaSources(BaseItem item, CancellationToken cancellationToken)
  247. {
  248. var baseItem = (LiveTvChannel)item;
  249. var service = GetService(baseItem);
  250. var sources = await service.GetChannelStreamMediaSources(baseItem.ExternalId, cancellationToken).ConfigureAwait(false);
  251. if (sources.Count == 0)
  252. {
  253. throw new NotImplementedException();
  254. }
  255. foreach (var source in sources)
  256. {
  257. Normalize(source, service, baseItem.ChannelType == ChannelType.TV);
  258. }
  259. return sources;
  260. }
  261. private ILiveTvService GetService(LiveTvChannel item)
  262. {
  263. var name = item.ServiceName;
  264. return GetService(name);
  265. }
  266. private ILiveTvService GetService(LiveTvProgram item)
  267. {
  268. var channel = _libraryManager.GetItemById(item.ChannelId) as LiveTvChannel;
  269. return GetService(channel);
  270. }
  271. private ILiveTvService GetService(string name)
  272. {
  273. return _services.FirstOrDefault(i => string.Equals(i.Name, name, StringComparison.OrdinalIgnoreCase));
  274. }
  275. private static void Normalize(MediaSourceInfo mediaSource, ILiveTvService service, bool isVideo)
  276. {
  277. // Not all of the plugins are setting this
  278. mediaSource.IsInfiniteStream = true;
  279. if (mediaSource.MediaStreams.Count == 0)
  280. {
  281. if (isVideo)
  282. {
  283. mediaSource.MediaStreams.AddRange(new List<MediaStream>
  284. {
  285. new MediaStream
  286. {
  287. Type = MediaStreamType.Video,
  288. // Set the index to -1 because we don't know the exact index of the video stream within the container
  289. Index = -1,
  290. // Set to true if unknown to enable deinterlacing
  291. IsInterlaced = true
  292. },
  293. new MediaStream
  294. {
  295. Type = MediaStreamType.Audio,
  296. // Set the index to -1 because we don't know the exact index of the audio stream within the container
  297. Index = -1
  298. }
  299. });
  300. }
  301. else
  302. {
  303. mediaSource.MediaStreams.AddRange(new List<MediaStream>
  304. {
  305. new MediaStream
  306. {
  307. Type = MediaStreamType.Audio,
  308. // Set the index to -1 because we don't know the exact index of the audio stream within the container
  309. Index = -1
  310. }
  311. });
  312. }
  313. }
  314. // Clean some bad data coming from providers
  315. foreach (var stream in mediaSource.MediaStreams)
  316. {
  317. if (stream.BitRate.HasValue && stream.BitRate <= 0)
  318. {
  319. stream.BitRate = null;
  320. }
  321. if (stream.Channels.HasValue && stream.Channels <= 0)
  322. {
  323. stream.Channels = null;
  324. }
  325. if (stream.AverageFrameRate.HasValue && stream.AverageFrameRate <= 0)
  326. {
  327. stream.AverageFrameRate = null;
  328. }
  329. if (stream.RealFrameRate.HasValue && stream.RealFrameRate <= 0)
  330. {
  331. stream.RealFrameRate = null;
  332. }
  333. if (stream.Width.HasValue && stream.Width <= 0)
  334. {
  335. stream.Width = null;
  336. }
  337. if (stream.Height.HasValue && stream.Height <= 0)
  338. {
  339. stream.Height = null;
  340. }
  341. if (stream.SampleRate.HasValue && stream.SampleRate <= 0)
  342. {
  343. stream.SampleRate = null;
  344. }
  345. if (stream.Level.HasValue && stream.Level <= 0)
  346. {
  347. stream.Level = null;
  348. }
  349. }
  350. var indexes = mediaSource.MediaStreams.Select(i => i.Index).Distinct().ToList();
  351. // If there are duplicate stream indexes, set them all to unknown
  352. if (indexes.Count != mediaSource.MediaStreams.Count)
  353. {
  354. foreach (var stream in mediaSource.MediaStreams)
  355. {
  356. stream.Index = -1;
  357. }
  358. }
  359. // Set the total bitrate if not already supplied
  360. mediaSource.InferTotalBitrate();
  361. if (!(service is EmbyTV.EmbyTV))
  362. {
  363. // We can't trust that we'll be able to direct stream it through emby server, no matter what the provider says
  364. //mediaSource.SupportsDirectPlay = false;
  365. //mediaSource.SupportsDirectStream = false;
  366. mediaSource.SupportsTranscoding = true;
  367. foreach (var stream in mediaSource.MediaStreams)
  368. {
  369. if (stream.Type == MediaStreamType.Video && string.IsNullOrWhiteSpace(stream.NalLengthSize))
  370. {
  371. stream.NalLengthSize = "0";
  372. }
  373. if (stream.Type == MediaStreamType.Video)
  374. {
  375. stream.IsInterlaced = true;
  376. }
  377. }
  378. }
  379. }
  380. private const string ExternalServiceTag = "ExternalServiceId";
  381. private LiveTvChannel GetChannel(ChannelInfo channelInfo, string serviceName, BaseItem parentFolder, CancellationToken cancellationToken)
  382. {
  383. var parentFolderId = parentFolder.Id;
  384. var isNew = false;
  385. var forceUpdate = false;
  386. var id = _tvDtoService.GetInternalChannelId(serviceName, channelInfo.Id);
  387. var item = _libraryManager.GetItemById(id) as LiveTvChannel;
  388. if (item == null)
  389. {
  390. item = new LiveTvChannel
  391. {
  392. Name = channelInfo.Name,
  393. Id = id,
  394. DateCreated = DateTime.UtcNow
  395. };
  396. isNew = true;
  397. }
  398. if (channelInfo.Tags != null)
  399. {
  400. if (!channelInfo.Tags.SequenceEqual(item.Tags, StringComparer.OrdinalIgnoreCase))
  401. {
  402. isNew = true;
  403. }
  404. item.Tags = channelInfo.Tags;
  405. }
  406. if (!item.ParentId.Equals(parentFolderId))
  407. {
  408. isNew = true;
  409. }
  410. item.ParentId = parentFolderId;
  411. item.ChannelType = channelInfo.ChannelType;
  412. item.ServiceName = serviceName;
  413. if (!string.Equals(item.GetProviderId(ExternalServiceTag), serviceName, StringComparison.OrdinalIgnoreCase))
  414. {
  415. forceUpdate = true;
  416. }
  417. item.SetProviderId(ExternalServiceTag, serviceName);
  418. if (!string.Equals(channelInfo.Id, item.ExternalId, StringComparison.Ordinal))
  419. {
  420. forceUpdate = true;
  421. }
  422. item.ExternalId = channelInfo.Id;
  423. if (!string.Equals(channelInfo.Number, item.Number, StringComparison.Ordinal))
  424. {
  425. forceUpdate = true;
  426. }
  427. item.Number = channelInfo.Number;
  428. if (!string.Equals(channelInfo.Name, item.Name, StringComparison.Ordinal))
  429. {
  430. forceUpdate = true;
  431. }
  432. item.Name = channelInfo.Name;
  433. if (!item.HasImage(ImageType.Primary))
  434. {
  435. if (!string.IsNullOrWhiteSpace(channelInfo.ImagePath))
  436. {
  437. item.SetImagePath(ImageType.Primary, channelInfo.ImagePath);
  438. forceUpdate = true;
  439. }
  440. else if (!string.IsNullOrWhiteSpace(channelInfo.ImageUrl))
  441. {
  442. item.SetImagePath(ImageType.Primary, channelInfo.ImageUrl);
  443. forceUpdate = true;
  444. }
  445. }
  446. if (isNew)
  447. {
  448. _libraryManager.CreateItem(item, parentFolder);
  449. }
  450. else if (forceUpdate)
  451. {
  452. _libraryManager.UpdateItem(item, parentFolder, ItemUpdateType.MetadataImport, cancellationToken);
  453. }
  454. return item;
  455. }
  456. private const string EtagKey = "ProgramEtag";
  457. private Tuple<LiveTvProgram, bool, bool> GetProgram(ProgramInfo info, Dictionary<Guid, LiveTvProgram> allExistingPrograms, LiveTvChannel channel, ChannelType channelType, string serviceName, CancellationToken cancellationToken)
  458. {
  459. var id = _tvDtoService.GetInternalProgramId(info.Id);
  460. var isNew = false;
  461. var forceUpdate = false;
  462. LiveTvProgram item;
  463. if (!allExistingPrograms.TryGetValue(id, out item))
  464. {
  465. isNew = true;
  466. item = new LiveTvProgram
  467. {
  468. Name = info.Name,
  469. Id = id,
  470. DateCreated = DateTime.UtcNow,
  471. DateModified = DateTime.UtcNow
  472. };
  473. if (!string.IsNullOrEmpty(info.Etag))
  474. {
  475. item.SetProviderId(EtagKey, info.Etag);
  476. }
  477. }
  478. if (!string.Equals(info.ShowId, item.ShowId, StringComparison.OrdinalIgnoreCase))
  479. {
  480. item.ShowId = info.ShowId;
  481. forceUpdate = true;
  482. }
  483. var seriesId = info.SeriesId;
  484. if (!item.ParentId.Equals(channel.Id))
  485. {
  486. forceUpdate = true;
  487. }
  488. item.ParentId = channel.Id;
  489. //item.ChannelType = channelType;
  490. item.Audio = info.Audio;
  491. item.ChannelId = channel.Id;
  492. item.CommunityRating = item.CommunityRating ?? info.CommunityRating;
  493. if ((item.CommunityRating ?? 0).Equals(0))
  494. {
  495. item.CommunityRating = null;
  496. }
  497. item.EpisodeTitle = info.EpisodeTitle;
  498. item.ExternalId = info.Id;
  499. if (!string.IsNullOrWhiteSpace(seriesId) && !string.Equals(item.ExternalSeriesId, seriesId, StringComparison.Ordinal))
  500. {
  501. forceUpdate = true;
  502. }
  503. item.ExternalSeriesId = seriesId;
  504. var isSeries = info.IsSeries || !string.IsNullOrEmpty(info.EpisodeTitle);
  505. if (isSeries || !string.IsNullOrEmpty(info.EpisodeTitle))
  506. {
  507. item.SeriesName = info.Name;
  508. }
  509. var tags = new List<string>();
  510. if (info.IsLive)
  511. {
  512. tags.Add("Live");
  513. }
  514. if (info.IsPremiere)
  515. {
  516. tags.Add("Premiere");
  517. }
  518. if (info.IsNews)
  519. {
  520. tags.Add("News");
  521. }
  522. if (info.IsSports)
  523. {
  524. tags.Add("Sports");
  525. }
  526. if (info.IsKids)
  527. {
  528. tags.Add("Kids");
  529. }
  530. if (info.IsRepeat)
  531. {
  532. tags.Add("Repeat");
  533. }
  534. if (info.IsMovie)
  535. {
  536. tags.Add("Movie");
  537. }
  538. if (isSeries)
  539. {
  540. tags.Add("Series");
  541. }
  542. item.Tags = tags.ToArray();
  543. item.Genres = info.Genres.ToArray();
  544. if (info.IsHD ?? false)
  545. {
  546. item.Width = 1280;
  547. item.Height = 720;
  548. }
  549. item.IsMovie = info.IsMovie;
  550. item.IsRepeat = info.IsRepeat;
  551. if (item.IsSeries != isSeries)
  552. {
  553. forceUpdate = true;
  554. }
  555. item.IsSeries = isSeries;
  556. item.Name = info.Name;
  557. item.OfficialRating = item.OfficialRating ?? info.OfficialRating;
  558. item.Overview = item.Overview ?? info.Overview;
  559. item.RunTimeTicks = (info.EndDate - info.StartDate).Ticks;
  560. item.ProviderIds = info.ProviderIds;
  561. foreach (var providerId in info.SeriesProviderIds)
  562. {
  563. info.ProviderIds["Series" + providerId.Key] = providerId.Value;
  564. }
  565. if (item.StartDate != info.StartDate)
  566. {
  567. forceUpdate = true;
  568. }
  569. item.StartDate = info.StartDate;
  570. if (item.EndDate != info.EndDate)
  571. {
  572. forceUpdate = true;
  573. }
  574. item.EndDate = info.EndDate;
  575. item.ProductionYear = info.ProductionYear;
  576. if (!isSeries || info.IsRepeat)
  577. {
  578. item.PremiereDate = info.OriginalAirDate;
  579. }
  580. item.IndexNumber = info.EpisodeNumber;
  581. item.ParentIndexNumber = info.SeasonNumber;
  582. if (!item.HasImage(ImageType.Primary))
  583. {
  584. if (!string.IsNullOrWhiteSpace(info.ImagePath))
  585. {
  586. item.SetImage(new ItemImageInfo
  587. {
  588. Path = info.ImagePath,
  589. Type = ImageType.Primary
  590. }, 0);
  591. }
  592. else if (!string.IsNullOrWhiteSpace(info.ImageUrl))
  593. {
  594. item.SetImage(new ItemImageInfo
  595. {
  596. Path = info.ImageUrl,
  597. Type = ImageType.Primary
  598. }, 0);
  599. }
  600. }
  601. if (!item.HasImage(ImageType.Thumb))
  602. {
  603. if (!string.IsNullOrWhiteSpace(info.ThumbImageUrl))
  604. {
  605. item.SetImage(new ItemImageInfo
  606. {
  607. Path = info.ThumbImageUrl,
  608. Type = ImageType.Thumb
  609. }, 0);
  610. }
  611. }
  612. if (!item.HasImage(ImageType.Logo))
  613. {
  614. if (!string.IsNullOrWhiteSpace(info.LogoImageUrl))
  615. {
  616. item.SetImage(new ItemImageInfo
  617. {
  618. Path = info.LogoImageUrl,
  619. Type = ImageType.Logo
  620. }, 0);
  621. }
  622. }
  623. if (!item.HasImage(ImageType.Backdrop))
  624. {
  625. if (!string.IsNullOrWhiteSpace(info.BackdropImageUrl))
  626. {
  627. item.SetImage(new ItemImageInfo
  628. {
  629. Path = info.BackdropImageUrl,
  630. Type = ImageType.Backdrop
  631. }, 0);
  632. }
  633. }
  634. var isUpdated = false;
  635. if (isNew)
  636. {
  637. }
  638. else if (forceUpdate || string.IsNullOrWhiteSpace(info.Etag))
  639. {
  640. isUpdated = true;
  641. }
  642. else
  643. {
  644. var etag = info.Etag;
  645. if (!string.Equals(etag, item.GetProviderId(EtagKey), StringComparison.OrdinalIgnoreCase))
  646. {
  647. item.SetProviderId(EtagKey, etag);
  648. isUpdated = true;
  649. }
  650. }
  651. if (isNew || isUpdated)
  652. {
  653. item.OnMetadataChanged();
  654. }
  655. return new Tuple<LiveTvProgram, bool, bool>(item, isNew, isUpdated);
  656. }
  657. public async Task<BaseItemDto> GetProgram(string id, CancellationToken cancellationToken, User user = null)
  658. {
  659. var program = _libraryManager.GetItemById(id);
  660. var dto = _dtoService.GetBaseItemDto(program, new DtoOptions(), user);
  661. var list = new List<Tuple<BaseItemDto, string, string>>() {
  662. new Tuple<BaseItemDto, string, string>(dto, program.ExternalId, program.ExternalSeriesId)
  663. };
  664. await AddRecordingInfo(list, cancellationToken).ConfigureAwait(false);
  665. return dto;
  666. }
  667. public async Task<QueryResult<BaseItemDto>> GetPrograms(InternalItemsQuery query, DtoOptions options, CancellationToken cancellationToken)
  668. {
  669. var user = query.User;
  670. var topFolder = GetInternalLiveTvFolder(cancellationToken);
  671. if (query.OrderBy.Length == 0)
  672. {
  673. if (query.IsAiring ?? false)
  674. {
  675. // Unless something else was specified, order by start date to take advantage of a specialized index
  676. query.OrderBy = new ValueTuple<string, SortOrder>[]
  677. {
  678. new ValueTuple<string, SortOrder>(ItemSortBy.StartDate, SortOrder.Ascending)
  679. };
  680. }
  681. else
  682. {
  683. // Unless something else was specified, order by start date to take advantage of a specialized index
  684. query.OrderBy = new ValueTuple<string, SortOrder>[]
  685. {
  686. new ValueTuple<string, SortOrder>(ItemSortBy.StartDate, SortOrder.Ascending)
  687. };
  688. }
  689. }
  690. RemoveFields(options);
  691. var internalQuery = new InternalItemsQuery(user)
  692. {
  693. IncludeItemTypes = new[] { typeof(LiveTvProgram).Name },
  694. MinEndDate = query.MinEndDate,
  695. MinStartDate = query.MinStartDate,
  696. MaxEndDate = query.MaxEndDate,
  697. MaxStartDate = query.MaxStartDate,
  698. ChannelIds = query.ChannelIds,
  699. IsMovie = query.IsMovie,
  700. IsSeries = query.IsSeries,
  701. IsSports = query.IsSports,
  702. IsKids = query.IsKids,
  703. IsNews = query.IsNews,
  704. Genres = query.Genres,
  705. GenreIds = query.GenreIds,
  706. StartIndex = query.StartIndex,
  707. Limit = query.Limit,
  708. OrderBy = query.OrderBy,
  709. EnableTotalRecordCount = query.EnableTotalRecordCount,
  710. TopParentIds = new[] { topFolder.Id },
  711. Name = query.Name,
  712. DtoOptions = options,
  713. HasAired = query.HasAired,
  714. IsAiring = query.IsAiring
  715. };
  716. if (!string.IsNullOrWhiteSpace(query.SeriesTimerId))
  717. {
  718. var seriesTimers = await GetSeriesTimersInternal(new SeriesTimerQuery { }, cancellationToken).ConfigureAwait(false);
  719. var seriesTimer = seriesTimers.Items.FirstOrDefault(i => string.Equals(_tvDtoService.GetInternalSeriesTimerId(i.Id).ToString("N"), query.SeriesTimerId, StringComparison.OrdinalIgnoreCase));
  720. if (seriesTimer != null)
  721. {
  722. internalQuery.ExternalSeriesId = seriesTimer.SeriesId;
  723. if (string.IsNullOrWhiteSpace(seriesTimer.SeriesId))
  724. {
  725. // Better to return nothing than every program in the database
  726. return new QueryResult<BaseItemDto>();
  727. }
  728. }
  729. else
  730. {
  731. // Better to return nothing than every program in the database
  732. return new QueryResult<BaseItemDto>();
  733. }
  734. }
  735. var queryResult = _libraryManager.QueryItems(internalQuery);
  736. var returnArray = _dtoService.GetBaseItemDtos(queryResult.Items, options, user);
  737. var result = new QueryResult<BaseItemDto>
  738. {
  739. Items = returnArray,
  740. TotalRecordCount = queryResult.TotalRecordCount
  741. };
  742. return result;
  743. }
  744. public QueryResult<BaseItem> GetRecommendedProgramsInternal(InternalItemsQuery query, DtoOptions options, CancellationToken cancellationToken)
  745. {
  746. var user = query.User;
  747. var topFolder = GetInternalLiveTvFolder(cancellationToken);
  748. var internalQuery = new InternalItemsQuery(user)
  749. {
  750. IncludeItemTypes = new[] { typeof(LiveTvProgram).Name },
  751. IsAiring = query.IsAiring,
  752. HasAired = query.HasAired,
  753. IsNews = query.IsNews,
  754. IsMovie = query.IsMovie,
  755. IsSeries = query.IsSeries,
  756. IsSports = query.IsSports,
  757. IsKids = query.IsKids,
  758. EnableTotalRecordCount = query.EnableTotalRecordCount,
  759. OrderBy = new[] { new ValueTuple<string, SortOrder>(ItemSortBy.StartDate, SortOrder.Ascending) },
  760. TopParentIds = new[] { topFolder.Id },
  761. DtoOptions = options,
  762. GenreIds = query.GenreIds
  763. };
  764. if (query.Limit.HasValue)
  765. {
  766. internalQuery.Limit = Math.Max(query.Limit.Value * 4, 200);
  767. }
  768. var programList = _libraryManager.QueryItems(internalQuery).Items;
  769. var totalCount = programList.Length;
  770. IOrderedEnumerable<LiveTvProgram> orderedPrograms = programList.Cast<LiveTvProgram>().OrderBy(i => i.StartDate.Date);
  771. if (query.IsAiring ?? false)
  772. {
  773. orderedPrograms = orderedPrograms
  774. .ThenByDescending(i => GetRecommendationScore(i, user, true));
  775. }
  776. IEnumerable<BaseItem> programs = orderedPrograms;
  777. if (query.Limit.HasValue)
  778. {
  779. programs = programs.Take(query.Limit.Value);
  780. }
  781. return new QueryResult<BaseItem>
  782. {
  783. Items = programs.ToArray(),
  784. TotalRecordCount = totalCount
  785. };
  786. }
  787. public QueryResult<BaseItemDto> GetRecommendedPrograms(InternalItemsQuery query, DtoOptions options, CancellationToken cancellationToken)
  788. {
  789. if (!(query.IsAiring ?? false))
  790. {
  791. return GetPrograms(query, options, cancellationToken).Result;
  792. }
  793. RemoveFields(options);
  794. var internalResult = GetRecommendedProgramsInternal(query, options, cancellationToken);
  795. return new QueryResult<BaseItemDto>
  796. {
  797. Items = _dtoService.GetBaseItemDtos(internalResult.Items, options, query.User),
  798. TotalRecordCount = internalResult.TotalRecordCount
  799. };
  800. }
  801. private int GetRecommendationScore(LiveTvProgram program, User user, bool factorChannelWatchCount)
  802. {
  803. var score = 0;
  804. if (program.IsLive)
  805. {
  806. score++;
  807. }
  808. if (program.IsSeries && !program.IsRepeat)
  809. {
  810. score++;
  811. }
  812. var channel = _libraryManager.GetItemById(program.ChannelId);
  813. if (channel == null)
  814. {
  815. return score;
  816. }
  817. var channelUserdata = _userDataManager.GetUserData(user, channel);
  818. if (channelUserdata.Likes.HasValue)
  819. {
  820. score += channelUserdata.Likes.Value ? 2 : -2;
  821. }
  822. if (channelUserdata.IsFavorite)
  823. {
  824. score += 3;
  825. }
  826. if (factorChannelWatchCount)
  827. {
  828. score += channelUserdata.PlayCount;
  829. }
  830. return score;
  831. }
  832. private async Task AddRecordingInfo(IEnumerable<Tuple<BaseItemDto, string, string>> programs, CancellationToken cancellationToken)
  833. {
  834. var timers = new Dictionary<string, List<TimerInfo>>();
  835. var seriesTimers = new Dictionary<string, List<SeriesTimerInfo>>();
  836. TimerInfo[] timerList = null;
  837. SeriesTimerInfo[] seriesTimerList = null;
  838. foreach (var programTuple in programs)
  839. {
  840. var program = programTuple.Item1;
  841. var externalProgramId = programTuple.Item2;
  842. string externalSeriesId = programTuple.Item3;
  843. if (timerList == null)
  844. {
  845. timerList = (await GetTimersInternal(new TimerQuery(), cancellationToken).ConfigureAwait(false)).Items;
  846. }
  847. var timer = timerList.FirstOrDefault(i => string.Equals(i.ProgramId, externalProgramId, StringComparison.OrdinalIgnoreCase));
  848. var foundSeriesTimer = false;
  849. if (timer != null)
  850. {
  851. if (timer.Status != RecordingStatus.Cancelled && timer.Status != RecordingStatus.Error)
  852. {
  853. program.TimerId = _tvDtoService.GetInternalTimerId(timer.Id);
  854. program.Status = timer.Status.ToString();
  855. }
  856. if (!string.IsNullOrEmpty(timer.SeriesTimerId))
  857. {
  858. program.SeriesTimerId = _tvDtoService.GetInternalSeriesTimerId(timer.SeriesTimerId)
  859. .ToString("N");
  860. foundSeriesTimer = true;
  861. }
  862. }
  863. if (foundSeriesTimer || string.IsNullOrWhiteSpace(externalSeriesId))
  864. {
  865. continue;
  866. }
  867. if (seriesTimerList == null)
  868. {
  869. seriesTimerList = (await GetSeriesTimersInternal(new SeriesTimerQuery(), cancellationToken).ConfigureAwait(false)).Items;
  870. }
  871. var seriesTimer = seriesTimerList.FirstOrDefault(i => string.Equals(i.SeriesId, externalSeriesId, StringComparison.OrdinalIgnoreCase));
  872. if (seriesTimer != null)
  873. {
  874. program.SeriesTimerId = _tvDtoService.GetInternalSeriesTimerId(seriesTimer.Id)
  875. .ToString("N");
  876. }
  877. }
  878. }
  879. internal Task RefreshChannels(IProgress<double> progress, CancellationToken cancellationToken)
  880. {
  881. return RefreshChannelsInternal(progress, cancellationToken);
  882. }
  883. private async Task RefreshChannelsInternal(IProgress<double> progress, CancellationToken cancellationToken)
  884. {
  885. await EmbyTV.EmbyTV.Current.CreateRecordingFolders().ConfigureAwait(false);
  886. await EmbyTV.EmbyTV.Current.ScanForTunerDeviceChanges(cancellationToken).ConfigureAwait(false);
  887. var numComplete = 0;
  888. double progressPerService = _services.Length == 0
  889. ? 0
  890. : 1.0 / _services.Length;
  891. var newChannelIdList = new List<Guid>();
  892. var newProgramIdList = new List<Guid>();
  893. var cleanDatabase = true;
  894. foreach (var service in _services)
  895. {
  896. cancellationToken.ThrowIfCancellationRequested();
  897. _logger.LogDebug("Refreshing guide from {name}", service.Name);
  898. try
  899. {
  900. var innerProgress = new ActionableProgress<double>();
  901. innerProgress.RegisterAction(p => progress.Report(p * progressPerService));
  902. var idList = await RefreshChannelsInternal(service, innerProgress, cancellationToken).ConfigureAwait(false);
  903. newChannelIdList.AddRange(idList.Item1);
  904. newProgramIdList.AddRange(idList.Item2);
  905. }
  906. catch (OperationCanceledException)
  907. {
  908. throw;
  909. }
  910. catch (Exception ex)
  911. {
  912. cleanDatabase = false;
  913. _logger.LogError(ex, "Error refreshing channels for service");
  914. }
  915. numComplete++;
  916. double percent = numComplete;
  917. percent /= _services.Length;
  918. progress.Report(100 * percent);
  919. }
  920. if (cleanDatabase)
  921. {
  922. CleanDatabaseInternal(newChannelIdList.ToArray(), new[] { typeof(LiveTvChannel).Name }, progress, cancellationToken);
  923. CleanDatabaseInternal(newProgramIdList.ToArray(), new[] { typeof(LiveTvProgram).Name }, progress, cancellationToken);
  924. }
  925. var coreService = _services.OfType<EmbyTV.EmbyTV>().FirstOrDefault();
  926. if (coreService != null)
  927. {
  928. await coreService.RefreshSeriesTimers(cancellationToken, new SimpleProgress<double>()).ConfigureAwait(false);
  929. await coreService.RefreshTimers(cancellationToken, new SimpleProgress<double>()).ConfigureAwait(false);
  930. }
  931. // Load these now which will prefetch metadata
  932. var dtoOptions = new DtoOptions();
  933. var fields = dtoOptions.Fields.ToList();
  934. fields.Remove(ItemFields.BasicSyncInfo);
  935. dtoOptions.Fields = fields.ToArray();
  936. progress.Report(100);
  937. }
  938. private async Task<Tuple<List<Guid>, List<Guid>>> RefreshChannelsInternal(ILiveTvService service, IProgress<double> progress, CancellationToken cancellationToken)
  939. {
  940. progress.Report(10);
  941. var allChannelsList = (await service.GetChannelsAsync(cancellationToken).ConfigureAwait(false))
  942. .Select(i => new Tuple<string, ChannelInfo>(service.Name, i))
  943. .ToList();
  944. var list = new List<LiveTvChannel>();
  945. var numComplete = 0;
  946. var parentFolder = GetInternalLiveTvFolder(cancellationToken);
  947. foreach (var channelInfo in allChannelsList)
  948. {
  949. cancellationToken.ThrowIfCancellationRequested();
  950. try
  951. {
  952. var item = GetChannel(channelInfo.Item2, channelInfo.Item1, parentFolder, cancellationToken);
  953. list.Add(item);
  954. }
  955. catch (OperationCanceledException)
  956. {
  957. throw;
  958. }
  959. catch (Exception ex)
  960. {
  961. _logger.LogError(ex, "Error getting channel information for {name}", channelInfo.Item2.Name);
  962. }
  963. numComplete++;
  964. double percent = numComplete;
  965. percent /= allChannelsList.Count;
  966. progress.Report(5 * percent + 10);
  967. }
  968. progress.Report(15);
  969. numComplete = 0;
  970. var programs = new List<Guid>();
  971. var channels = new List<Guid>();
  972. var guideDays = GetGuideDays();
  973. _logger.LogInformation("Refreshing guide with {0} days of guide data", guideDays);
  974. cancellationToken.ThrowIfCancellationRequested();
  975. foreach (var currentChannel in list)
  976. {
  977. channels.Add(currentChannel.Id);
  978. cancellationToken.ThrowIfCancellationRequested();
  979. try
  980. {
  981. var start = DateTime.UtcNow.AddHours(-1);
  982. var end = start.AddDays(guideDays);
  983. var isMovie = false;
  984. var isSports = false;
  985. var isNews = false;
  986. var isKids = false;
  987. var iSSeries = false;
  988. var channelPrograms = (await service.GetProgramsAsync(currentChannel.ExternalId, start, end, cancellationToken).ConfigureAwait(false)).ToList();
  989. var existingPrograms = _libraryManager.GetItemList(new InternalItemsQuery
  990. {
  991. IncludeItemTypes = new string[] { typeof(LiveTvProgram).Name },
  992. ChannelIds = new Guid[] { currentChannel.Id },
  993. DtoOptions = new DtoOptions(true)
  994. }).Cast<LiveTvProgram>().ToDictionary(i => i.Id);
  995. var newPrograms = new List<LiveTvProgram>();
  996. var updatedPrograms = new List<BaseItem>();
  997. foreach (var program in channelPrograms)
  998. {
  999. var programTuple = GetProgram(program, existingPrograms, currentChannel, currentChannel.ChannelType, service.Name, cancellationToken);
  1000. var programItem = programTuple.Item1;
  1001. if (programTuple.Item2)
  1002. {
  1003. newPrograms.Add(programItem);
  1004. }
  1005. else if (programTuple.Item3)
  1006. {
  1007. updatedPrograms.Add(programItem);
  1008. }
  1009. programs.Add(programItem.Id);
  1010. isMovie |= program.IsMovie;
  1011. iSSeries |= program.IsSeries;
  1012. isSports |= program.IsSports;
  1013. isNews |= program.IsNews;
  1014. isKids |= program.IsKids;
  1015. }
  1016. _logger.LogDebug("Channel {0} has {1} new programs and {2} updated programs", currentChannel.Name, newPrograms.Count, updatedPrograms.Count);
  1017. if (newPrograms.Count > 0)
  1018. {
  1019. _libraryManager.CreateItems(newPrograms, null, cancellationToken);
  1020. }
  1021. if (updatedPrograms.Count > 0)
  1022. {
  1023. _libraryManager.UpdateItems(updatedPrograms, currentChannel, ItemUpdateType.MetadataImport, cancellationToken);
  1024. }
  1025. currentChannel.IsMovie = isMovie;
  1026. currentChannel.IsNews = isNews;
  1027. currentChannel.IsSports = isSports;
  1028. currentChannel.IsSeries = iSSeries;
  1029. if (isKids)
  1030. {
  1031. currentChannel.AddTag("Kids");
  1032. }
  1033. //currentChannel.UpdateToRepository(ItemUpdateType.MetadataImport, cancellationToken);
  1034. await currentChannel.RefreshMetadata(new MetadataRefreshOptions(new DirectoryService(_logger, _fileSystem))
  1035. {
  1036. ForceSave = true
  1037. }, cancellationToken).ConfigureAwait(false);
  1038. }
  1039. catch (OperationCanceledException)
  1040. {
  1041. throw;
  1042. }
  1043. catch (Exception ex)
  1044. {
  1045. _logger.LogError(ex, "Error getting programs for channel {name}", currentChannel.Name);
  1046. }
  1047. numComplete++;
  1048. double percent = numComplete / (double) allChannelsList.Count;
  1049. progress.Report(85 * percent + 15);
  1050. }
  1051. progress.Report(100);
  1052. return new Tuple<List<Guid>, List<Guid>>(channels, programs);
  1053. }
  1054. private void CleanDatabaseInternal(Guid[] currentIdList, string[] validTypes, IProgress<double> progress, CancellationToken cancellationToken)
  1055. {
  1056. var list = _itemRepo.GetItemIdsList(new InternalItemsQuery
  1057. {
  1058. IncludeItemTypes = validTypes,
  1059. DtoOptions = new DtoOptions(false)
  1060. });
  1061. var numComplete = 0;
  1062. foreach (var itemId in list)
  1063. {
  1064. cancellationToken.ThrowIfCancellationRequested();
  1065. if (itemId.Equals(Guid.Empty))
  1066. {
  1067. // Somehow some invalid data got into the db. It probably predates the boundary checking
  1068. continue;
  1069. }
  1070. if (!currentIdList.Contains(itemId))
  1071. {
  1072. var item = _libraryManager.GetItemById(itemId);
  1073. if (item != null)
  1074. {
  1075. _libraryManager.DeleteItem(item, new DeleteOptions
  1076. {
  1077. DeleteFileLocation = false,
  1078. DeleteFromExternalProvider = false
  1079. }, false);
  1080. }
  1081. }
  1082. numComplete++;
  1083. double percent = numComplete / (double) list.Count;
  1084. progress.Report(100 * percent);
  1085. }
  1086. }
  1087. private const int MaxGuideDays = 14;
  1088. private double GetGuideDays()
  1089. {
  1090. var config = GetConfiguration();
  1091. if (config.GuideDays.HasValue)
  1092. {
  1093. return Math.Max(1, Math.Min(config.GuideDays.Value, MaxGuideDays));
  1094. }
  1095. return 7;
  1096. }
  1097. private QueryResult<BaseItem> GetEmbyRecordings(RecordingQuery query, DtoOptions dtoOptions, User user)
  1098. {
  1099. if (user == null)
  1100. {
  1101. return new QueryResult<BaseItem>();
  1102. }
  1103. var folderIds = GetRecordingFolders(user, true)
  1104. .Select(i => i.Id)
  1105. .ToList();
  1106. var excludeItemTypes = new List<string>();
  1107. if (folderIds.Count == 0)
  1108. {
  1109. return new QueryResult<BaseItem>();
  1110. }
  1111. var includeItemTypes = new List<string>();
  1112. var genres = new List<string>();
  1113. if (query.IsMovie.HasValue)
  1114. {
  1115. if (query.IsMovie.Value)
  1116. {
  1117. includeItemTypes.Add(typeof(Movie).Name);
  1118. }
  1119. else
  1120. {
  1121. excludeItemTypes.Add(typeof(Movie).Name);
  1122. }
  1123. }
  1124. if (query.IsSeries.HasValue)
  1125. {
  1126. if (query.IsSeries.Value)
  1127. {
  1128. includeItemTypes.Add(typeof(Episode).Name);
  1129. }
  1130. else
  1131. {
  1132. excludeItemTypes.Add(typeof(Episode).Name);
  1133. }
  1134. }
  1135. if (query.IsSports ?? false)
  1136. {
  1137. genres.Add("Sports");
  1138. }
  1139. if (query.IsKids ?? false)
  1140. {
  1141. genres.Add("Kids");
  1142. genres.Add("Children");
  1143. genres.Add("Family");
  1144. }
  1145. var limit = query.Limit;
  1146. if (query.IsInProgress ?? false)
  1147. {
  1148. // limit = (query.Limit ?? 10) * 2;
  1149. limit = null;
  1150. //var allActivePaths = EmbyTV.EmbyTV.Current.GetAllActiveRecordings().Select(i => i.Path).ToArray();
  1151. //var items = allActivePaths.Select(i => _libraryManager.FindByPath(i, false)).Where(i => i != null).ToArray();
  1152. //return new QueryResult<BaseItem>
  1153. //{
  1154. // Items = items,
  1155. // TotalRecordCount = items.Length
  1156. //};
  1157. dtoOptions.Fields = dtoOptions.Fields.Concat(new[] { ItemFields.Tags }).Distinct().ToArray();
  1158. }
  1159. var result = _libraryManager.GetItemsResult(new InternalItemsQuery(user)
  1160. {
  1161. MediaTypes = new[] { MediaType.Video },
  1162. Recursive = true,
  1163. AncestorIds = folderIds.ToArray(),
  1164. IsFolder = false,
  1165. IsVirtualItem = false,
  1166. Limit = limit,
  1167. StartIndex = query.StartIndex,
  1168. OrderBy = new[] { new ValueTuple<string, SortOrder>(ItemSortBy.DateCreated, SortOrder.Descending) },
  1169. EnableTotalRecordCount = query.EnableTotalRecordCount,
  1170. IncludeItemTypes = includeItemTypes.ToArray(),
  1171. ExcludeItemTypes = excludeItemTypes.ToArray(),
  1172. Genres = genres.ToArray(),
  1173. DtoOptions = dtoOptions
  1174. });
  1175. if (query.IsInProgress ?? false)
  1176. {
  1177. //TODO Fix The co-variant conversion between Video[] and BaseItem[], this can generate runtime issues.
  1178. result.Items = result
  1179. .Items
  1180. .OfType<Video>()
  1181. .Where(i => !i.IsCompleteMedia)
  1182. .ToArray();
  1183. result.TotalRecordCount = result.Items.Length;
  1184. }
  1185. return result;
  1186. }
  1187. public Task AddInfoToProgramDto(List<Tuple<BaseItem, BaseItemDto>> tuples, ItemFields[] fields, User user = null)
  1188. {
  1189. var programTuples = new List<Tuple<BaseItemDto, string, string>>();
  1190. var hasChannelImage = fields.Contains(ItemFields.ChannelImage);
  1191. var hasChannelInfo = fields.Contains(ItemFields.ChannelInfo);
  1192. foreach (var tuple in tuples)
  1193. {
  1194. var program = (LiveTvProgram)tuple.Item1;
  1195. var dto = tuple.Item2;
  1196. dto.StartDate = program.StartDate;
  1197. dto.EpisodeTitle = program.EpisodeTitle;
  1198. dto.IsRepeat |= program.IsRepeat;
  1199. dto.IsMovie |= program.IsMovie;
  1200. dto.IsSeries |= program.IsSeries;
  1201. dto.IsSports |= program.IsSports;
  1202. dto.IsLive |= program.IsLive;
  1203. dto.IsNews |= program.IsNews;
  1204. dto.IsKids |= program.IsKids;
  1205. dto.IsPremiere |= program.IsPremiere;
  1206. if (hasChannelInfo || hasChannelImage)
  1207. {
  1208. var channel = _libraryManager.GetItemById(program.ChannelId);
  1209. if (channel is LiveTvChannel liveChannel)
  1210. {
  1211. dto.ChannelName = liveChannel.Name;
  1212. dto.MediaType = liveChannel.MediaType;
  1213. dto.ChannelNumber = liveChannel.Number;
  1214. if (hasChannelImage && liveChannel.HasImage(ImageType.Primary))
  1215. {
  1216. dto.ChannelPrimaryImageTag = _tvDtoService.GetImageTag(liveChannel);
  1217. }
  1218. }
  1219. }
  1220. programTuples.Add(new Tuple<BaseItemDto, string, string>(dto, program.ExternalId, program.ExternalSeriesId));
  1221. }
  1222. return AddRecordingInfo(programTuples, CancellationToken.None);
  1223. }
  1224. public ActiveRecordingInfo GetActiveRecordingInfo(string path)
  1225. {
  1226. return EmbyTV.EmbyTV.Current.GetActiveRecordingInfo(path);
  1227. }
  1228. public void AddInfoToRecordingDto(BaseItem item, BaseItemDto dto, ActiveRecordingInfo activeRecordingInfo, User user = null)
  1229. {
  1230. var service = EmbyTV.EmbyTV.Current;
  1231. var info = activeRecordingInfo.Timer;
  1232. var channel = string.IsNullOrWhiteSpace(info.ChannelId) ? null : _libraryManager.GetItemById(_tvDtoService.GetInternalChannelId(service.Name, info.ChannelId));
  1233. dto.SeriesTimerId = string.IsNullOrEmpty(info.SeriesTimerId)
  1234. ? null
  1235. : _tvDtoService.GetInternalSeriesTimerId(info.SeriesTimerId).ToString("N");
  1236. dto.TimerId = string.IsNullOrEmpty(info.Id)
  1237. ? null
  1238. : _tvDtoService.GetInternalTimerId(info.Id);
  1239. var startDate = info.StartDate;
  1240. var endDate = info.EndDate;
  1241. dto.StartDate = startDate;
  1242. dto.EndDate = endDate;
  1243. dto.Status = info.Status.ToString();
  1244. dto.IsRepeat = info.IsRepeat;
  1245. dto.EpisodeTitle = info.EpisodeTitle;
  1246. dto.IsMovie = info.IsMovie;
  1247. dto.IsSeries = info.IsSeries;
  1248. dto.IsSports = info.IsSports;
  1249. dto.IsLive = info.IsLive;
  1250. dto.IsNews = info.IsNews;
  1251. dto.IsKids = info.IsKids;
  1252. dto.IsPremiere = info.IsPremiere;
  1253. if (info.Status == RecordingStatus.InProgress)
  1254. {
  1255. startDate = info.StartDate.AddSeconds(0 - info.PrePaddingSeconds);
  1256. endDate = info.EndDate.AddSeconds(info.PostPaddingSeconds);
  1257. var now = DateTime.UtcNow.Ticks;
  1258. var start = startDate.Ticks;
  1259. var end = endDate.Ticks;
  1260. var pct = now - start;
  1261. pct /= end;
  1262. pct *= 100;
  1263. dto.CompletionPercentage = pct;
  1264. }
  1265. if (channel != null)
  1266. {
  1267. dto.ChannelName = channel.Name;
  1268. if (channel.HasImage(ImageType.Primary))
  1269. {
  1270. dto.ChannelPrimaryImageTag = _tvDtoService.GetImageTag(channel);
  1271. }
  1272. }
  1273. }
  1274. public QueryResult<BaseItemDto> GetRecordings(RecordingQuery query, DtoOptions options)
  1275. {
  1276. var user = query.UserId.Equals(Guid.Empty) ? null : _userManager.GetUserById(query.UserId);
  1277. RemoveFields(options);
  1278. var internalResult = GetEmbyRecordings(query, options, user);
  1279. var returnArray = _dtoService.GetBaseItemDtos(internalResult.Items, options, user);
  1280. return new QueryResult<BaseItemDto>
  1281. {
  1282. Items = returnArray,
  1283. TotalRecordCount = internalResult.TotalRecordCount
  1284. };
  1285. }
  1286. private async Task<QueryResult<TimerInfo>> GetTimersInternal(TimerQuery query, CancellationToken cancellationToken)
  1287. {
  1288. var tasks = _services.Select(async i =>
  1289. {
  1290. try
  1291. {
  1292. var recs = await i.GetTimersAsync(cancellationToken).ConfigureAwait(false);
  1293. return recs.Select(r => new Tuple<TimerInfo, ILiveTvService>(r, i));
  1294. }
  1295. catch (Exception ex)
  1296. {
  1297. _logger.LogError(ex, "Error getting recordings");
  1298. return new List<Tuple<TimerInfo, ILiveTvService>>();
  1299. }
  1300. });
  1301. var results = await Task.WhenAll(tasks).ConfigureAwait(false);
  1302. var timers = results.SelectMany(i => i.ToList());
  1303. if (query.IsActive.HasValue)
  1304. {
  1305. if (query.IsActive.Value)
  1306. {
  1307. timers = timers.Where(i => i.Item1.Status == RecordingStatus.InProgress);
  1308. }
  1309. else
  1310. {
  1311. timers = timers.Where(i => i.Item1.Status != RecordingStatus.InProgress);
  1312. }
  1313. }
  1314. if (query.IsScheduled.HasValue)
  1315. {
  1316. if (query.IsScheduled.Value)
  1317. {
  1318. timers = timers.Where(i => i.Item1.Status == RecordingStatus.New);
  1319. }
  1320. else
  1321. {
  1322. timers = timers.Where(i => i.Item1.Status != RecordingStatus.New);
  1323. }
  1324. }
  1325. if (!string.IsNullOrEmpty(query.ChannelId))
  1326. {
  1327. var guid = new Guid(query.ChannelId);
  1328. timers = timers.Where(i => guid == _tvDtoService.GetInternalChannelId(i.Item2.Name, i.Item1.ChannelId));
  1329. }
  1330. if (!string.IsNullOrEmpty(query.SeriesTimerId))
  1331. {
  1332. var guid = new Guid(query.SeriesTimerId);
  1333. timers = timers
  1334. .Where(i => _tvDtoService.GetInternalSeriesTimerId(i.Item1.SeriesTimerId) == guid);
  1335. }
  1336. if (!string.IsNullOrEmpty(query.Id))
  1337. {
  1338. var guid = new Guid(query.Id);
  1339. timers = timers
  1340. .Where(i => string.Equals(_tvDtoService.GetInternalTimerId(i.Item1.Id), query.Id, StringComparison.OrdinalIgnoreCase));
  1341. }
  1342. var returnArray = timers
  1343. .Select(i => i.Item1)
  1344. .OrderBy(i => i.StartDate)
  1345. .ToArray();
  1346. return new QueryResult<TimerInfo>
  1347. {
  1348. Items = returnArray,
  1349. TotalRecordCount = returnArray.Length
  1350. };
  1351. }
  1352. public async Task<QueryResult<TimerInfoDto>> GetTimers(TimerQuery query, CancellationToken cancellationToken)
  1353. {
  1354. var tasks = _services.Select(async i =>
  1355. {
  1356. try
  1357. {
  1358. var recs = await i.GetTimersAsync(cancellationToken).ConfigureAwait(false);
  1359. return recs.Select(r => new Tuple<TimerInfo, ILiveTvService>(r, i));
  1360. }
  1361. catch (Exception ex)
  1362. {
  1363. _logger.LogError(ex, "Error getting recordings");
  1364. return new List<Tuple<TimerInfo, ILiveTvService>>();
  1365. }
  1366. });
  1367. var results = await Task.WhenAll(tasks).ConfigureAwait(false);
  1368. var timers = results.SelectMany(i => i.ToList());
  1369. if (query.IsActive.HasValue)
  1370. {
  1371. if (query.IsActive.Value)
  1372. {
  1373. timers = timers.Where(i => i.Item1.Status == RecordingStatus.InProgress);
  1374. }
  1375. else
  1376. {
  1377. timers = timers.Where(i => i.Item1.Status != RecordingStatus.InProgress);
  1378. }
  1379. }
  1380. if (query.IsScheduled.HasValue)
  1381. {
  1382. if (query.IsScheduled.Value)
  1383. {
  1384. timers = timers.Where(i => i.Item1.Status == RecordingStatus.New);
  1385. }
  1386. else
  1387. {
  1388. timers = timers.Where(i => i.Item1.Status != RecordingStatus.New);
  1389. }
  1390. }
  1391. if (!string.IsNullOrEmpty(query.ChannelId))
  1392. {
  1393. var guid = new Guid(query.ChannelId);
  1394. timers = timers.Where(i => guid == _tvDtoService.GetInternalChannelId(i.Item2.Name, i.Item1.ChannelId));
  1395. }
  1396. if (!string.IsNullOrEmpty(query.SeriesTimerId))
  1397. {
  1398. var guid = new Guid(query.SeriesTimerId);
  1399. timers = timers
  1400. .Where(i => _tvDtoService.GetInternalSeriesTimerId(i.Item1.SeriesTimerId) == guid);
  1401. }
  1402. if (!string.IsNullOrEmpty(query.Id))
  1403. {
  1404. timers = timers
  1405. .Where(i => string.Equals(_tvDtoService.GetInternalTimerId(i.Item1.Id), query.Id, StringComparison.OrdinalIgnoreCase));
  1406. }
  1407. var returnList = new List<TimerInfoDto>();
  1408. foreach (var i in timers)
  1409. {
  1410. var program = string.IsNullOrEmpty(i.Item1.ProgramId) ?
  1411. null :
  1412. _libraryManager.GetItemById(_tvDtoService.GetInternalProgramId(i.Item1.ProgramId)) as LiveTvProgram;
  1413. var channel = string.IsNullOrEmpty(i.Item1.ChannelId) ? null : _libraryManager.GetItemById(_tvDtoService.GetInternalChannelId(i.Item2.Name, i.Item1.ChannelId));
  1414. returnList.Add(_tvDtoService.GetTimerInfoDto(i.Item1, i.Item2, program, channel));
  1415. }
  1416. var returnArray = returnList
  1417. .OrderBy(i => i.StartDate)
  1418. .ToArray();
  1419. return new QueryResult<TimerInfoDto>
  1420. {
  1421. Items = returnArray,
  1422. TotalRecordCount = returnArray.Length
  1423. };
  1424. }
  1425. public async Task CancelTimer(string id)
  1426. {
  1427. var timer = await GetTimer(id, CancellationToken.None).ConfigureAwait(false);
  1428. if (timer == null)
  1429. {
  1430. throw new ResourceNotFoundException(string.Format("Timer with Id {0} not found", id));
  1431. }
  1432. var service = GetService(timer.ServiceName);
  1433. await service.CancelTimerAsync(timer.ExternalId, CancellationToken.None).ConfigureAwait(false);
  1434. if (!(service is EmbyTV.EmbyTV))
  1435. {
  1436. TimerCancelled?.Invoke(this, new GenericEventArgs<TimerEventInfo>
  1437. {
  1438. Argument = new TimerEventInfo
  1439. {
  1440. Id = id
  1441. }
  1442. });
  1443. }
  1444. }
  1445. public async Task CancelSeriesTimer(string id)
  1446. {
  1447. var timer = await GetSeriesTimer(id, CancellationToken.None).ConfigureAwait(false);
  1448. if (timer == null)
  1449. {
  1450. throw new ResourceNotFoundException(string.Format("SeriesTimer with Id {0} not found", id));
  1451. }
  1452. var service = GetService(timer.ServiceName);
  1453. await service.CancelSeriesTimerAsync(timer.ExternalId, CancellationToken.None).ConfigureAwait(false);
  1454. SeriesTimerCancelled?.Invoke(this, new GenericEventArgs<TimerEventInfo>
  1455. {
  1456. Argument = new TimerEventInfo
  1457. {
  1458. Id = id
  1459. }
  1460. });
  1461. }
  1462. public async Task<TimerInfoDto> GetTimer(string id, CancellationToken cancellationToken)
  1463. {
  1464. var results = await GetTimers(new TimerQuery
  1465. {
  1466. Id = id
  1467. }, cancellationToken).ConfigureAwait(false);
  1468. return results.Items.FirstOrDefault(i => string.Equals(i.Id, id, StringComparison.OrdinalIgnoreCase));
  1469. }
  1470. public async Task<SeriesTimerInfoDto> GetSeriesTimer(string id, CancellationToken cancellationToken)
  1471. {
  1472. var results = await GetSeriesTimers(new SeriesTimerQuery(), cancellationToken).ConfigureAwait(false);
  1473. return results.Items.FirstOrDefault(i => string.Equals(i.Id, id, StringComparison.OrdinalIgnoreCase));
  1474. }
  1475. private async Task<QueryResult<SeriesTimerInfo>> GetSeriesTimersInternal(SeriesTimerQuery query, CancellationToken cancellationToken)
  1476. {
  1477. var tasks = _services.Select(async i =>
  1478. {
  1479. try
  1480. {
  1481. var recs = await i.GetSeriesTimersAsync(cancellationToken).ConfigureAwait(false);
  1482. return recs.Select(r =>
  1483. {
  1484. r.ServiceName = i.Name;
  1485. return new Tuple<SeriesTimerInfo, ILiveTvService>(r, i);
  1486. });
  1487. }
  1488. catch (Exception ex)
  1489. {
  1490. _logger.LogError(ex, "Error getting recordings");
  1491. return new List<Tuple<SeriesTimerInfo, ILiveTvService>>();
  1492. }
  1493. });
  1494. var results = await Task.WhenAll(tasks).ConfigureAwait(false);
  1495. var timers = results.SelectMany(i => i.ToList());
  1496. if (string.Equals(query.SortBy, "Priority", StringComparison.OrdinalIgnoreCase))
  1497. {
  1498. timers = query.SortOrder == SortOrder.Descending ?
  1499. timers.OrderBy(i => i.Item1.Priority).ThenByStringDescending(i => i.Item1.Name) :
  1500. timers.OrderByDescending(i => i.Item1.Priority).ThenByString(i => i.Item1.Name);
  1501. }
  1502. else
  1503. {
  1504. timers = query.SortOrder == SortOrder.Descending ?
  1505. timers.OrderByStringDescending(i => i.Item1.Name) :
  1506. timers.OrderByString(i => i.Item1.Name);
  1507. }
  1508. var returnArray = timers
  1509. .Select(i =>
  1510. {
  1511. return i.Item1;
  1512. })
  1513. .ToArray();
  1514. return new QueryResult<SeriesTimerInfo>
  1515. {
  1516. Items = returnArray,
  1517. TotalRecordCount = returnArray.Length
  1518. };
  1519. }
  1520. public async Task<QueryResult<SeriesTimerInfoDto>> GetSeriesTimers(SeriesTimerQuery query, CancellationToken cancellationToken)
  1521. {
  1522. var tasks = _services.Select(async i =>
  1523. {
  1524. try
  1525. {
  1526. var recs = await i.GetSeriesTimersAsync(cancellationToken).ConfigureAwait(false);
  1527. return recs.Select(r => new Tuple<SeriesTimerInfo, ILiveTvService>(r, i));
  1528. }
  1529. catch (Exception ex)
  1530. {
  1531. _logger.LogError(ex, "Error getting recordings");
  1532. return new List<Tuple<SeriesTimerInfo, ILiveTvService>>();
  1533. }
  1534. });
  1535. var results = await Task.WhenAll(tasks).ConfigureAwait(false);
  1536. var timers = results.SelectMany(i => i.ToList());
  1537. if (string.Equals(query.SortBy, "Priority", StringComparison.OrdinalIgnoreCase))
  1538. {
  1539. timers = query.SortOrder == SortOrder.Descending ?
  1540. timers.OrderBy(i => i.Item1.Priority).ThenByStringDescending(i => i.Item1.Name) :
  1541. timers.OrderByDescending(i => i.Item1.Priority).ThenByString(i => i.Item1.Name);
  1542. }
  1543. else
  1544. {
  1545. timers = query.SortOrder == SortOrder.Descending ?
  1546. timers.OrderByStringDescending(i => i.Item1.Name) :
  1547. timers.OrderByString(i => i.Item1.Name);
  1548. }
  1549. var returnArray = timers
  1550. .Select(i =>
  1551. {
  1552. string channelName = null;
  1553. if (!string.IsNullOrEmpty(i.Item1.ChannelId))
  1554. {
  1555. var internalChannelId = _tvDtoService.GetInternalChannelId(i.Item2.Name, i.Item1.ChannelId);
  1556. var channel = _libraryManager.GetItemById(internalChannelId);
  1557. channelName = channel == null ? null : channel.Name;
  1558. }
  1559. return _tvDtoService.GetSeriesTimerInfoDto(i.Item1, i.Item2, channelName);
  1560. })
  1561. .ToArray();
  1562. return new QueryResult<SeriesTimerInfoDto>
  1563. {
  1564. Items = returnArray,
  1565. TotalRecordCount = returnArray.Length
  1566. };
  1567. }
  1568. public BaseItem GetLiveTvChannel(TimerInfo timer, ILiveTvService service)
  1569. {
  1570. var internalChannelId = _tvDtoService.GetInternalChannelId(service.Name, timer.ChannelId);
  1571. return _libraryManager.GetItemById(internalChannelId);
  1572. }
  1573. public void AddChannelInfo(List<Tuple<BaseItemDto, LiveTvChannel>> tuples, DtoOptions options, User user)
  1574. {
  1575. var now = DateTime.UtcNow;
  1576. var channelIds = tuples.Select(i => i.Item2.Id).Distinct().ToArray();
  1577. var programs = options.AddCurrentProgram ? _libraryManager.GetItemList(new InternalItemsQuery(user)
  1578. {
  1579. IncludeItemTypes = new[] { typeof(LiveTvProgram).Name },
  1580. ChannelIds = channelIds,
  1581. MaxStartDate = now,
  1582. MinEndDate = now,
  1583. Limit = channelIds.Length,
  1584. OrderBy = new[] { new ValueTuple<string, SortOrder>(ItemSortBy.StartDate, SortOrder.Ascending) },
  1585. TopParentIds = new[] { GetInternalLiveTvFolder(CancellationToken.None).Id },
  1586. DtoOptions = options
  1587. }) : new List<BaseItem>();
  1588. RemoveFields(options);
  1589. var currentProgramsList = new List<BaseItem>();
  1590. var currentChannelsDict = new Dictionary<Guid, BaseItemDto>();
  1591. var addCurrentProgram = options.AddCurrentProgram;
  1592. foreach (var tuple in tuples)
  1593. {
  1594. var dto = tuple.Item1;
  1595. var channel = tuple.Item2;
  1596. dto.Number = channel.Number;
  1597. dto.ChannelNumber = channel.Number;
  1598. dto.ChannelType = channel.ChannelType;
  1599. currentChannelsDict[dto.Id] = dto;
  1600. if (addCurrentProgram)
  1601. {
  1602. var currentProgram = programs.FirstOrDefault(i => channel.Id.Equals(i.ChannelId));
  1603. if (currentProgram != null)
  1604. {
  1605. currentProgramsList.Add(currentProgram);
  1606. }
  1607. }
  1608. }
  1609. if (addCurrentProgram)
  1610. {
  1611. var currentProgramDtos = _dtoService.GetBaseItemDtos(currentProgramsList, options, user);
  1612. foreach (var programDto in currentProgramDtos)
  1613. {
  1614. BaseItemDto channelDto;
  1615. if (currentChannelsDict.TryGetValue(programDto.ChannelId, out channelDto))
  1616. {
  1617. channelDto.CurrentProgram = programDto;
  1618. }
  1619. }
  1620. }
  1621. }
  1622. private async Task<Tuple<SeriesTimerInfo, ILiveTvService>> GetNewTimerDefaultsInternal(CancellationToken cancellationToken, LiveTvProgram program = null)
  1623. {
  1624. ILiveTvService service = null;
  1625. ProgramInfo programInfo = null;
  1626. if(program != null)
  1627. {
  1628. service = GetService(program);
  1629. var channel = _libraryManager.GetItemById(program.ChannelId);
  1630. programInfo = new ProgramInfo
  1631. {
  1632. Audio = program.Audio,
  1633. ChannelId = channel.ExternalId,
  1634. CommunityRating = program.CommunityRating,
  1635. EndDate = program.EndDate ?? DateTime.MinValue,
  1636. EpisodeTitle = program.EpisodeTitle,
  1637. Genres = program.Genres.ToList(),
  1638. Id = program.ExternalId,
  1639. IsHD = program.IsHD,
  1640. IsKids = program.IsKids,
  1641. IsLive = program.IsLive,
  1642. IsMovie = program.IsMovie,
  1643. IsNews = program.IsNews,
  1644. IsPremiere = program.IsPremiere,
  1645. IsRepeat = program.IsRepeat,
  1646. IsSeries = program.IsSeries,
  1647. IsSports = program.IsSports,
  1648. OriginalAirDate = program.PremiereDate,
  1649. Overview = program.Overview,
  1650. StartDate = program.StartDate,
  1651. //ImagePath = program.ExternalImagePath,
  1652. Name = program.Name,
  1653. OfficialRating = program.OfficialRating
  1654. };
  1655. }
  1656. if (service == null)
  1657. {
  1658. service = _services.First();
  1659. }
  1660. var info = await service.GetNewTimerDefaultsAsync(cancellationToken, programInfo).ConfigureAwait(false);
  1661. info.RecordAnyTime = true;
  1662. info.Days = new List<DayOfWeek>
  1663. {
  1664. DayOfWeek.Sunday,
  1665. DayOfWeek.Monday,
  1666. DayOfWeek.Tuesday,
  1667. DayOfWeek.Wednesday,
  1668. DayOfWeek.Thursday,
  1669. DayOfWeek.Friday,
  1670. DayOfWeek.Saturday
  1671. };
  1672. info.Id = null;
  1673. return new Tuple<SeriesTimerInfo, ILiveTvService>(info, service);
  1674. }
  1675. public async Task<SeriesTimerInfoDto> GetNewTimerDefaults(CancellationToken cancellationToken)
  1676. {
  1677. var info = await GetNewTimerDefaultsInternal(cancellationToken).ConfigureAwait(false);
  1678. var obj = _tvDtoService.GetSeriesTimerInfoDto(info.Item1, info.Item2, null);
  1679. return obj;
  1680. }
  1681. public async Task<SeriesTimerInfoDto> GetNewTimerDefaults(string programId, CancellationToken cancellationToken)
  1682. {
  1683. var program = (LiveTvProgram)_libraryManager.GetItemById(programId);
  1684. var programDto = await GetProgram(programId, cancellationToken).ConfigureAwait(false);
  1685. var defaults = await GetNewTimerDefaultsInternal(cancellationToken, program).ConfigureAwait(false);
  1686. var info = _tvDtoService.GetSeriesTimerInfoDto(defaults.Item1, defaults.Item2, null);
  1687. info.Days = defaults.Item1.Days.ToArray();
  1688. info.DayPattern = _tvDtoService.GetDayPattern(info.Days);
  1689. info.Name = program.Name;
  1690. info.ChannelId = programDto.ChannelId;
  1691. info.ChannelName = programDto.ChannelName;
  1692. info.StartDate = program.StartDate;
  1693. info.Name = program.Name;
  1694. info.Overview = program.Overview;
  1695. info.ProgramId = programDto.Id.ToString("N");
  1696. info.ExternalProgramId = program.ExternalId;
  1697. if (program.EndDate.HasValue)
  1698. {
  1699. info.EndDate = program.EndDate.Value;
  1700. }
  1701. return info;
  1702. }
  1703. public async Task CreateTimer(TimerInfoDto timer, CancellationToken cancellationToken)
  1704. {
  1705. var service = GetService(timer.ServiceName);
  1706. var info = await _tvDtoService.GetTimerInfo(timer, true, this, cancellationToken).ConfigureAwait(false);
  1707. // Set priority from default values
  1708. var defaultValues = await service.GetNewTimerDefaultsAsync(cancellationToken).ConfigureAwait(false);
  1709. info.Priority = defaultValues.Priority;
  1710. string newTimerId = null;
  1711. if (service is ISupportsNewTimerIds supportsNewTimerIds)
  1712. {
  1713. newTimerId = await supportsNewTimerIds.CreateTimer(info, cancellationToken).ConfigureAwait(false);
  1714. newTimerId = _tvDtoService.GetInternalTimerId(newTimerId);
  1715. }
  1716. else
  1717. {
  1718. await service.CreateTimerAsync(info, cancellationToken).ConfigureAwait(false);
  1719. }
  1720. _logger.LogInformation("New recording scheduled");
  1721. if (!(service is EmbyTV.EmbyTV))
  1722. {
  1723. TimerCreated?.Invoke(this, new GenericEventArgs<TimerEventInfo>
  1724. {
  1725. Argument = new TimerEventInfo
  1726. {
  1727. ProgramId = _tvDtoService.GetInternalProgramId(info.ProgramId),
  1728. Id = newTimerId
  1729. }
  1730. });
  1731. }
  1732. }
  1733. public async Task CreateSeriesTimer(SeriesTimerInfoDto timer, CancellationToken cancellationToken)
  1734. {
  1735. var service = GetService(timer.ServiceName);
  1736. var info = await _tvDtoService.GetSeriesTimerInfo(timer, true, this, cancellationToken).ConfigureAwait(false);
  1737. // Set priority from default values
  1738. var defaultValues = await service.GetNewTimerDefaultsAsync(cancellationToken).ConfigureAwait(false);
  1739. info.Priority = defaultValues.Priority;
  1740. string newTimerId = null;
  1741. if (service is ISupportsNewTimerIds supportsNewTimerIds)
  1742. {
  1743. newTimerId = await supportsNewTimerIds.CreateSeriesTimer(info, cancellationToken).ConfigureAwait(false);
  1744. newTimerId = _tvDtoService.GetInternalSeriesTimerId(newTimerId).ToString("N");
  1745. }
  1746. else
  1747. {
  1748. await service.CreateSeriesTimerAsync(info, cancellationToken).ConfigureAwait(false);
  1749. }
  1750. SeriesTimerCreated?.Invoke(this, new GenericEventArgs<TimerEventInfo>
  1751. {
  1752. Argument = new TimerEventInfo
  1753. {
  1754. ProgramId = _tvDtoService.GetInternalProgramId(info.ProgramId),
  1755. Id = newTimerId
  1756. }
  1757. });
  1758. }
  1759. public async Task UpdateTimer(TimerInfoDto timer, CancellationToken cancellationToken)
  1760. {
  1761. var info = await _tvDtoService.GetTimerInfo(timer, false, this, cancellationToken).ConfigureAwait(false);
  1762. var service = GetService(timer.ServiceName);
  1763. await service.UpdateTimerAsync(info, cancellationToken).ConfigureAwait(false);
  1764. }
  1765. public async Task UpdateSeriesTimer(SeriesTimerInfoDto timer, CancellationToken cancellationToken)
  1766. {
  1767. var info = await _tvDtoService.GetSeriesTimerInfo(timer, false, this, cancellationToken).ConfigureAwait(false);
  1768. var service = GetService(timer.ServiceName);
  1769. await service.UpdateSeriesTimerAsync(info, cancellationToken).ConfigureAwait(false);
  1770. }
  1771. public GuideInfo GetGuideInfo()
  1772. {
  1773. var startDate = DateTime.UtcNow;
  1774. var endDate = startDate.AddDays(GetGuideDays());
  1775. return new GuideInfo
  1776. {
  1777. StartDate = startDate,
  1778. EndDate = endDate
  1779. };
  1780. }
  1781. /// <summary>
  1782. /// Performs application-defined tasks associated with freeing, releasing, or resetting unmanaged resources.
  1783. /// </summary>
  1784. public void Dispose()
  1785. {
  1786. Dispose(true);
  1787. }
  1788. private bool _isDisposed = false;
  1789. /// <summary>
  1790. /// Releases unmanaged and - optionally - managed resources.
  1791. /// </summary>
  1792. /// <param name="dispose"><c>true</c> to release both managed and unmanaged resources; <c>false</c> to release only unmanaged resources.</param>
  1793. protected virtual void Dispose(bool dispose)
  1794. {
  1795. if (dispose)
  1796. {
  1797. _isDisposed = true;
  1798. }
  1799. }
  1800. private LiveTvServiceInfo[] GetServiceInfos()
  1801. {
  1802. return Services.Select(GetServiceInfo).ToArray();
  1803. }
  1804. private static LiveTvServiceInfo GetServiceInfo(ILiveTvService service)
  1805. {
  1806. return new LiveTvServiceInfo
  1807. {
  1808. Name = service.Name
  1809. };
  1810. }
  1811. public LiveTvInfo GetLiveTvInfo(CancellationToken cancellationToken)
  1812. {
  1813. var services = GetServiceInfos();
  1814. var info = new LiveTvInfo
  1815. {
  1816. Services = services,
  1817. IsEnabled = services.Length > 0
  1818. };
  1819. info.EnabledUsers = _userManager.Users
  1820. .Where(IsLiveTvEnabled)
  1821. .Select(i => i.Id.ToString("N"))
  1822. .ToArray();
  1823. return info;
  1824. }
  1825. private bool IsLiveTvEnabled(User user)
  1826. {
  1827. return user.Policy.EnableLiveTvAccess && (Services.Count > 1 || GetConfiguration().TunerHosts.Length > 0);
  1828. }
  1829. public IEnumerable<User> GetEnabledUsers()
  1830. {
  1831. return _userManager.Users
  1832. .Where(IsLiveTvEnabled);
  1833. }
  1834. /// <summary>
  1835. /// Resets the tuner.
  1836. /// </summary>
  1837. /// <param name="id">The identifier.</param>
  1838. /// <param name="cancellationToken">The cancellation token.</param>
  1839. /// <returns>Task.</returns>
  1840. public Task ResetTuner(string id, CancellationToken cancellationToken)
  1841. {
  1842. var parts = id.Split(new[] { '_' }, 2);
  1843. var service = _services.FirstOrDefault(i => string.Equals(i.GetType().FullName.GetMD5().ToString("N"), parts[0], StringComparison.OrdinalIgnoreCase));
  1844. if (service == null)
  1845. {
  1846. throw new ArgumentException("Service not found.");
  1847. }
  1848. return service.ResetTuner(parts[1], cancellationToken);
  1849. }
  1850. private static void RemoveFields(DtoOptions options)
  1851. {
  1852. var fields = options.Fields.ToList();
  1853. fields.Remove(ItemFields.CanDelete);
  1854. fields.Remove(ItemFields.CanDownload);
  1855. fields.Remove(ItemFields.DisplayPreferencesId);
  1856. fields.Remove(ItemFields.Etag);
  1857. options.Fields = fields.ToArray();
  1858. }
  1859. public Folder GetInternalLiveTvFolder(CancellationToken cancellationToken)
  1860. {
  1861. var name = _localization.GetLocalizedString("HeaderLiveTV");
  1862. return _libraryManager.GetNamedView(name, CollectionType.LiveTv, name);
  1863. }
  1864. public async Task<TunerHostInfo> SaveTunerHost(TunerHostInfo info, bool dataSourceChanged = true)
  1865. {
  1866. info = _jsonSerializer.DeserializeFromString<TunerHostInfo>(_jsonSerializer.SerializeToString(info));
  1867. var provider = _tunerHosts.FirstOrDefault(i => string.Equals(info.Type, i.Type, StringComparison.OrdinalIgnoreCase));
  1868. if (provider == null)
  1869. {
  1870. throw new ResourceNotFoundException();
  1871. }
  1872. if (provider is IConfigurableTunerHost configurable)
  1873. {
  1874. await configurable.Validate(info).ConfigureAwait(false);
  1875. }
  1876. var config = GetConfiguration();
  1877. var list = config.TunerHosts.ToList();
  1878. var index = list.FindIndex(i => string.Equals(i.Id, info.Id, StringComparison.OrdinalIgnoreCase));
  1879. if (index == -1 || string.IsNullOrWhiteSpace(info.Id))
  1880. {
  1881. info.Id = Guid.NewGuid().ToString("N");
  1882. list.Add(info);
  1883. config.TunerHosts = list.ToArray();
  1884. }
  1885. else
  1886. {
  1887. config.TunerHosts[index] = info;
  1888. }
  1889. _config.SaveConfiguration("livetv", config);
  1890. if (dataSourceChanged)
  1891. {
  1892. _taskManager.CancelIfRunningAndQueue<RefreshChannelsScheduledTask>();
  1893. }
  1894. return info;
  1895. }
  1896. public async Task<ListingsProviderInfo> SaveListingProvider(ListingsProviderInfo info, bool validateLogin, bool validateListings)
  1897. {
  1898. // Hack to make the object a pure ListingsProviderInfo instead of an AddListingProvider
  1899. // ServerConfiguration.SaveConfiguration crashes during xml serialization for AddListingProvider
  1900. info = _jsonSerializer.DeserializeFromString<ListingsProviderInfo>(_jsonSerializer.SerializeToString(info));
  1901. IListingsProvider provider = _listingProviders.FirstOrDefault(i => string.Equals(info.Type, i.Type, StringComparison.OrdinalIgnoreCase));
  1902. if (provider == null)
  1903. {
  1904. throw new ResourceNotFoundException(
  1905. string.Format("Couldn't find provider of type: '{0}'", info.Type)
  1906. );
  1907. }
  1908. await provider.Validate(info, validateLogin, validateListings).ConfigureAwait(false);
  1909. LiveTvOptions config = GetConfiguration();
  1910. List<ListingsProviderInfo> list = config.ListingProviders.ToList();
  1911. int index = list.FindIndex(i => string.Equals(i.Id, info.Id, StringComparison.OrdinalIgnoreCase));
  1912. if (index == -1 || string.IsNullOrWhiteSpace(info.Id))
  1913. {
  1914. info.Id = Guid.NewGuid().ToString("N");
  1915. list.Add(info);
  1916. config.ListingProviders = list.ToArray();
  1917. }
  1918. else
  1919. {
  1920. config.ListingProviders[index] = info;
  1921. }
  1922. _config.SaveConfiguration("livetv", config);
  1923. _taskManager.CancelIfRunningAndQueue<RefreshChannelsScheduledTask>();
  1924. return info;
  1925. }
  1926. public void DeleteListingsProvider(string id)
  1927. {
  1928. var config = GetConfiguration();
  1929. config.ListingProviders = config.ListingProviders.Where(i => !string.Equals(id, i.Id, StringComparison.OrdinalIgnoreCase)).ToArray();
  1930. _config.SaveConfiguration("livetv", config);
  1931. _taskManager.CancelIfRunningAndQueue<RefreshChannelsScheduledTask>();
  1932. }
  1933. public async Task<TunerChannelMapping> SetChannelMapping(string providerId, string tunerChannelId, string providerChannelId)
  1934. {
  1935. var config = GetConfiguration();
  1936. var listingsProviderInfo = config.ListingProviders.First(i => string.Equals(providerId, i.Id, StringComparison.OrdinalIgnoreCase));
  1937. listingsProviderInfo.ChannelMappings = listingsProviderInfo.ChannelMappings.Where(i => !string.Equals(i.Name, tunerChannelId, StringComparison.OrdinalIgnoreCase)).ToArray();
  1938. if (!string.Equals(tunerChannelId, providerChannelId, StringComparison.OrdinalIgnoreCase))
  1939. {
  1940. var list = listingsProviderInfo.ChannelMappings.ToList();
  1941. list.Add(new NameValuePair
  1942. {
  1943. Name = tunerChannelId,
  1944. Value = providerChannelId
  1945. });
  1946. listingsProviderInfo.ChannelMappings = list.ToArray();
  1947. }
  1948. _config.SaveConfiguration("livetv", config);
  1949. var tunerChannels = await GetChannelsForListingsProvider(providerId, CancellationToken.None)
  1950. .ConfigureAwait(false);
  1951. var providerChannels = await GetChannelsFromListingsProviderData(providerId, CancellationToken.None)
  1952. .ConfigureAwait(false);
  1953. var mappings = listingsProviderInfo.ChannelMappings;
  1954. var tunerChannelMappings =
  1955. tunerChannels.Select(i => GetTunerChannelMapping(i, mappings, providerChannels)).ToList();
  1956. _taskManager.CancelIfRunningAndQueue<RefreshChannelsScheduledTask>();
  1957. return tunerChannelMappings.First(i => string.Equals(i.Id, tunerChannelId, StringComparison.OrdinalIgnoreCase));
  1958. }
  1959. public TunerChannelMapping GetTunerChannelMapping(ChannelInfo tunerChannel, NameValuePair[] mappings, List<ChannelInfo> epgChannels)
  1960. {
  1961. var result = new TunerChannelMapping
  1962. {
  1963. Name = tunerChannel.Name,
  1964. Id = tunerChannel.Id
  1965. };
  1966. if (!string.IsNullOrWhiteSpace(tunerChannel.Number))
  1967. {
  1968. result.Name = tunerChannel.Number + " " + result.Name;
  1969. }
  1970. var providerChannel = EmbyTV.EmbyTV.Current.GetEpgChannelFromTunerChannel(mappings, tunerChannel, epgChannels);
  1971. if (providerChannel != null)
  1972. {
  1973. result.ProviderChannelName = providerChannel.Name;
  1974. result.ProviderChannelId = providerChannel.Id;
  1975. }
  1976. return result;
  1977. }
  1978. public Task<List<NameIdPair>> GetLineups(string providerType, string providerId, string country, string location)
  1979. {
  1980. var config = GetConfiguration();
  1981. if (string.IsNullOrWhiteSpace(providerId))
  1982. {
  1983. var provider = _listingProviders.FirstOrDefault(i => string.Equals(providerType, i.Type, StringComparison.OrdinalIgnoreCase));
  1984. if (provider == null)
  1985. {
  1986. throw new ResourceNotFoundException();
  1987. }
  1988. return provider.GetLineups(null, country, location);
  1989. }
  1990. else
  1991. {
  1992. var info = config.ListingProviders.FirstOrDefault(i => string.Equals(i.Id, providerId, StringComparison.OrdinalIgnoreCase));
  1993. var provider = _listingProviders.FirstOrDefault(i => string.Equals(info.Type, i.Type, StringComparison.OrdinalIgnoreCase));
  1994. if (provider == null)
  1995. {
  1996. throw new ResourceNotFoundException();
  1997. }
  1998. return provider.GetLineups(info, country, location);
  1999. }
  2000. }
  2001. public Task<List<ChannelInfo>> GetChannelsForListingsProvider(string id, CancellationToken cancellationToken)
  2002. {
  2003. var info = GetConfiguration().ListingProviders.First(i => string.Equals(i.Id, id, StringComparison.OrdinalIgnoreCase));
  2004. return EmbyTV.EmbyTV.Current.GetChannelsForListingsProvider(info, cancellationToken);
  2005. }
  2006. public Task<List<ChannelInfo>> GetChannelsFromListingsProviderData(string id, CancellationToken cancellationToken)
  2007. {
  2008. var info = GetConfiguration().ListingProviders.First(i => string.Equals(i.Id, id, StringComparison.OrdinalIgnoreCase));
  2009. var provider = _listingProviders.First(i => string.Equals(i.Type, info.Type, StringComparison.OrdinalIgnoreCase));
  2010. return provider.GetChannels(info, cancellationToken);
  2011. }
  2012. public Guid GetInternalChannelId(string serviceName, string externalId)
  2013. {
  2014. return _tvDtoService.GetInternalChannelId(serviceName, externalId);
  2015. }
  2016. public Guid GetInternalProgramId(string externalId)
  2017. {
  2018. return _tvDtoService.GetInternalProgramId(externalId);
  2019. }
  2020. public List<BaseItem> GetRecordingFolders(User user)
  2021. {
  2022. return GetRecordingFolders(user, false);
  2023. }
  2024. private List<BaseItem> GetRecordingFolders(User user, bool refreshChannels)
  2025. {
  2026. var folders = EmbyTV.EmbyTV.Current.GetRecordingFolders()
  2027. .SelectMany(i => i.Locations)
  2028. .Distinct(StringComparer.OrdinalIgnoreCase)
  2029. .Select(i => _libraryManager.FindByPath(i, true))
  2030. .Where(i => i != null)
  2031. .Where(i => i.IsVisibleStandalone(user))
  2032. .SelectMany(i => _libraryManager.GetCollectionFolders(i))
  2033. .DistinctBy(i => i.Id)
  2034. .OrderBy(i => i.SortName)
  2035. .ToList();
  2036. folders.AddRange(_channelManager().GetChannelsInternal(new MediaBrowser.Model.Channels.ChannelQuery
  2037. {
  2038. UserId = user.Id,
  2039. IsRecordingsFolder = true,
  2040. RefreshLatestChannelItems = refreshChannels
  2041. }).Items);
  2042. return folders.Cast<BaseItem>().ToList();
  2043. }
  2044. }
  2045. }