SubtitleService.cs 7.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178
  1. using MediaBrowser.Controller.Entities;
  2. using MediaBrowser.Controller.Library;
  3. using MediaBrowser.Controller.MediaEncoding;
  4. using MediaBrowser.Controller.Providers;
  5. using MediaBrowser.Controller.Subtitles;
  6. using MediaBrowser.Model.Entities;
  7. using MediaBrowser.Model.Providers;
  8. using ServiceStack;
  9. using System;
  10. using System.Collections.Generic;
  11. using System.IO;
  12. using System.Linq;
  13. using System.Threading;
  14. using System.Threading.Tasks;
  15. namespace MediaBrowser.Api.Library
  16. {
  17. [Route("/Videos/{Id}/Subtitles/{Index}", "GET", Summary = "Gets an external subtitle file")]
  18. [Route("/Videos/{Id}/Subtitles/{Index}/Stream.{Format}", "GET", Summary = "Gets subtitles in a specified format (vtt).")]
  19. public class GetSubtitle
  20. {
  21. /// <summary>
  22. /// Gets or sets the id.
  23. /// </summary>
  24. /// <value>The id.</value>
  25. [ApiMember(Name = "Id", Description = "Item Id", IsRequired = true, DataType = "string", ParameterType = "path", Verb = "GET")]
  26. public string Id { get; set; }
  27. [ApiMember(Name = "MediaSourceId", Description = "MediaSourceId", IsRequired = true, DataType = "string", ParameterType = "query", Verb = "GET")]
  28. public string MediaSourceId { get; set; }
  29. [ApiMember(Name = "Index", Description = "The subtitle stream index", IsRequired = true, DataType = "int", ParameterType = "path", Verb = "GET")]
  30. public int Index { get; set; }
  31. [ApiMember(Name = "Format", Description = "Format", IsRequired = true, DataType = "string", ParameterType = "path", Verb = "GET")]
  32. public string Format { get; set; }
  33. }
  34. [Route("/Videos/{Id}/Subtitles/{Index}", "DELETE", Summary = "Deletes an external subtitle file")]
  35. public class DeleteSubtitle
  36. {
  37. /// <summary>
  38. /// Gets or sets the id.
  39. /// </summary>
  40. /// <value>The id.</value>
  41. [ApiMember(Name = "Id", Description = "Item Id", IsRequired = true, DataType = "string", ParameterType = "path", Verb = "DELETE")]
  42. public string Id { get; set; }
  43. [ApiMember(Name = "Index", Description = "The subtitle stream index", IsRequired = true, DataType = "int", ParameterType = "path", Verb = "DELETE")]
  44. public int Index { get; set; }
  45. }
  46. [Route("/Items/{Id}/RemoteSearch/Subtitles/{Language}", "GET")]
  47. public class SearchRemoteSubtitles : IReturn<List<RemoteSubtitleInfo>>
  48. {
  49. [ApiMember(Name = "Id", Description = "Item Id", IsRequired = true, DataType = "string", ParameterType = "path", Verb = "GET")]
  50. public string Id { get; set; }
  51. [ApiMember(Name = "Language", Description = "Language", IsRequired = true, DataType = "string", ParameterType = "path", Verb = "GET")]
  52. public string Language { get; set; }
  53. }
  54. [Route("/Items/{Id}/RemoteSearch/Subtitles/Providers", "GET")]
  55. public class GetSubtitleProviders : IReturn<List<SubtitleProviderInfo>>
  56. {
  57. [ApiMember(Name = "Id", Description = "Item Id", IsRequired = true, DataType = "string", ParameterType = "path", Verb = "GET")]
  58. public string Id { get; set; }
  59. }
  60. [Route("/Items/{Id}/RemoteSearch/Subtitles/{SubtitleId}", "POST")]
  61. public class DownloadRemoteSubtitles : IReturnVoid
  62. {
  63. [ApiMember(Name = "Id", Description = "Item Id", IsRequired = true, DataType = "string", ParameterType = "path", Verb = "POST")]
  64. public string Id { get; set; }
  65. [ApiMember(Name = "SubtitleId", Description = "SubtitleId", IsRequired = true, DataType = "string", ParameterType = "path", Verb = "POST")]
  66. public string SubtitleId { get; set; }
  67. }
  68. [Route("/Providers/Subtitles/Subtitles/{Id}", "GET")]
  69. public class GetRemoteSubtitles : IReturnVoid
  70. {
  71. [ApiMember(Name = "Id", Description = "Item Id", IsRequired = true, DataType = "string", ParameterType = "path", Verb = "GET")]
  72. public string Id { get; set; }
  73. }
  74. public class SubtitleService : BaseApiService
  75. {
  76. private readonly ILibraryManager _libraryManager;
  77. private readonly ISubtitleManager _subtitleManager;
  78. private readonly ISubtitleEncoder _subtitleEncoder;
  79. public SubtitleService(ILibraryManager libraryManager, ISubtitleManager subtitleManager, ISubtitleEncoder subtitleEncoder)
  80. {
  81. _libraryManager = libraryManager;
  82. _subtitleManager = subtitleManager;
  83. _subtitleEncoder = subtitleEncoder;
  84. }
  85. public object Get(SearchRemoteSubtitles request)
  86. {
  87. var video = (Video)_libraryManager.GetItemById(request.Id);
  88. var response = _subtitleManager.SearchSubtitles(video, request.Language, CancellationToken.None).Result;
  89. return ToOptimizedResult(response);
  90. }
  91. public object Get(GetSubtitle request)
  92. {
  93. if (string.IsNullOrEmpty(request.Format))
  94. {
  95. var item = (Video)_libraryManager.GetItemById(new Guid(request.Id));
  96. var mediaSource = item.GetMediaSources(false)
  97. .First(i => string.Equals(i.Id, request.MediaSourceId ?? request.Id));
  98. var subtitleStream = mediaSource.MediaStreams
  99. .First(i => i.Type == MediaStreamType.Subtitle && i.Index == request.Index);
  100. return ToStaticFileResult(subtitleStream.Path);
  101. }
  102. var stream = GetSubtitles(request).Result;
  103. return ResultFactory.GetResult(stream, Common.Net.MimeTypes.GetMimeType("file." + request.Format));
  104. }
  105. private async Task<Stream> GetSubtitles(GetSubtitle request)
  106. {
  107. var stream = await _subtitleEncoder.GetSubtitles(request.Id, request.MediaSourceId, request.Index, request.Format,
  108. CancellationToken.None);
  109. return stream;
  110. }
  111. public void Delete(DeleteSubtitle request)
  112. {
  113. var task = _subtitleManager.DeleteSubtitles(request.Id, request.Index);
  114. Task.WaitAll(task);
  115. }
  116. public object Get(GetSubtitleProviders request)
  117. {
  118. var result = _subtitleManager.GetProviders(request.Id);
  119. return ToOptimizedResult(result);
  120. }
  121. public object Get(GetRemoteSubtitles request)
  122. {
  123. var result = _subtitleManager.GetRemoteSubtitles(request.Id, CancellationToken.None).Result;
  124. return ResultFactory.GetResult(result.Stream, Common.Net.MimeTypes.GetMimeType("file." + result.Format));
  125. }
  126. public void Post(DownloadRemoteSubtitles request)
  127. {
  128. var video = (Video)_libraryManager.GetItemById(request.Id);
  129. Task.Run(async () =>
  130. {
  131. try
  132. {
  133. await _subtitleManager.DownloadSubtitles(video, request.SubtitleId, CancellationToken.None)
  134. .ConfigureAwait(false);
  135. await video.RefreshMetadata(new MetadataRefreshOptions(), CancellationToken.None).ConfigureAwait(false);
  136. }
  137. catch (Exception ex)
  138. {
  139. Logger.ErrorException("Error downloading subtitles", ex);
  140. }
  141. });
  142. }
  143. }
  144. }