LiveTvService.cs 2.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071
  1. using MediaBrowser.Controller.LiveTv;
  2. using MediaBrowser.Model.LiveTv;
  3. using ServiceStack.ServiceHost;
  4. using System.Collections.Generic;
  5. using System.Linq;
  6. using System.Threading;
  7. using System.Threading.Tasks;
  8. namespace MediaBrowser.Api.LiveTv
  9. {
  10. [Route("/LiveTv/Services", "GET")]
  11. [Api(Description = "Gets available live tv services.")]
  12. public class GetServices : IReturn<List<LiveTvServiceInfo>>
  13. {
  14. }
  15. [Route("/LiveTv/Channels", "GET")]
  16. [Api(Description = "Gets available live tv channels.")]
  17. public class GetChannels : IReturn<List<ChannelInfoDto>>
  18. {
  19. // Add filter by service if needed, and/or other filters
  20. }
  21. public class LiveTvService : BaseApiService
  22. {
  23. private readonly ILiveTvManager _liveTvManager;
  24. public LiveTvService(ILiveTvManager liveTvManager)
  25. {
  26. _liveTvManager = liveTvManager;
  27. }
  28. public object Get(GetServices request)
  29. {
  30. var services = _liveTvManager.Services;
  31. var result = services.Select(GetServiceInfo)
  32. .ToList();
  33. return ToOptimizedResult(result);
  34. }
  35. public object Get(GetChannels request)
  36. {
  37. var result = GetChannelsAsync(request).Result;
  38. return ToOptimizedResult(result);
  39. }
  40. public async Task<IEnumerable<ChannelInfoDto>> GetChannelsAsync(GetChannels request)
  41. {
  42. var services = _liveTvManager.Services;
  43. var channelTasks = services.Select(i => i.GetChannelsAsync(CancellationToken.None));
  44. var channelLists = await Task.WhenAll(channelTasks).ConfigureAwait(false);
  45. // Aggregate all channels from all services
  46. return channelLists.SelectMany(i => i)
  47. .Select(_liveTvManager.GetChannelInfoDto);
  48. }
  49. private LiveTvServiceInfo GetServiceInfo(ILiveTvService service)
  50. {
  51. return new LiveTvServiceInfo
  52. {
  53. Name = service.Name
  54. };
  55. }
  56. }
  57. }