using MediaBrowser.Common.IO ;
using MediaBrowser.Common.Net ;
using MediaBrowser.Controller.Channels ;
using MediaBrowser.Controller.Configuration ;
using MediaBrowser.Controller.Dlna ;
using MediaBrowser.Controller.Dto ;
using MediaBrowser.Controller.Library ;
using MediaBrowser.Controller.LiveTv ;
using MediaBrowser.Controller.MediaEncoding ;
using MediaBrowser.Controller.Persistence ;
using MediaBrowser.Model.IO ;
using ServiceStack ;
using System ;
using System.Collections.Generic ;
using System.Globalization ;
using System.IO ;
using System.Text ;
using System.Threading ;
using System.Threading.Tasks ;
using MimeTypes = ServiceStack . MimeTypes ;
namespace MediaBrowser.Api.Playback.Hls
{
[Route("/Videos/{Id}/master.m3u8", "GET")]
[Api(Description = "Gets a video stream using HTTP live streaming.")]
public class GetMasterHlsVideoStream : VideoStreamRequest
{
[ApiMember(Name = "BaselineStreamAudioBitRate", Description = "Optional. Specify the audio bitrate for the baseline stream.", IsRequired = false, DataType = "int", ParameterType = "query", Verb = "GET")]
public int? BaselineStreamAudioBitRate { get ; set ; }
[ApiMember(Name = "AppendBaselineStream", Description = "Optional. Whether or not to include a baseline audio-only stream in the master playlist.", IsRequired = false, DataType = "bool", ParameterType = "query", Verb = "GET")]
public bool AppendBaselineStream { get ; set ; }
}
[Route("/Videos/{Id}/main.m3u8", "GET")]
[Api(Description = "Gets a video stream using HTTP live streaming.")]
public class GetMainHlsVideoStream : VideoStreamRequest
{
}
[Route("/Videos/{Id}/baseline.m3u8", "GET")]
[Api(Description = "Gets a video stream using HTTP live streaming.")]
public class GetBaselineHlsVideoStream : VideoStreamRequest
{
}
/// <summary>
/// Class GetHlsVideoSegment
/// </summary>
[Route("/Videos/{Id}/hlsdynamic/{PlaylistId}/{SegmentId}.ts", "GET")]
[Api(Description = "Gets an Http live streaming segment file. Internal use only.")]
public class GetDynamicHlsVideoSegment : VideoStreamRequest
{
public string PlaylistId { get ; set ; }
/// <summary>
/// Gets or sets the segment id.
/// </summary>
/// <value>The segment id.</value>
public string SegmentId { get ; set ; }
}
public class DynamicHlsService : BaseHlsService
{
public DynamicHlsService ( IServerConfigurationManager serverConfig , IUserManager userManager , ILibraryManager libraryManager , IIsoManager isoManager , IMediaEncoder mediaEncoder , IDtoService dtoService , IFileSystem fileSystem , IItemRepository itemRepository , ILiveTvManager liveTvManager , IEncodingManager encodingManager , IDlnaManager dlnaManager , IChannelManager channelManager , IHttpClient httpClient ) : base ( serverConfig , userManager , libraryManager , isoManager , mediaEncoder , dtoService , fileSystem , itemRepository , liveTvManager , encodingManager , dlnaManager , channelManager , httpClient )
{
}
public object Get ( GetMasterHlsVideoStream request )
{
var result = GetAsync ( request ) . Result ;
return result ;
}
public object Get ( GetDynamicHlsVideoSegment request )
{
if ( string . Equals ( "baseline" , request . PlaylistId , StringComparison . OrdinalIgnoreCase ) )
{
return GetDynamicSegment ( request , false ) . Result ;
}
return GetDynamicSegment ( request , true ) . Result ;
}
private async Task < object > GetDynamicSegment ( GetDynamicHlsVideoSegment request , bool isMain )
{
var index = int . Parse ( request . SegmentId , NumberStyles . Integer , UsCulture ) ;
var state = await GetState ( request , CancellationToken . None ) . ConfigureAwait ( false ) ;
var playlistPath = Path . ChangeExtension ( GetOutputFilePath ( state ) , ".m3u8" ) ;
var path = GetSegmentPath ( playlistPath , index ) ;
if ( File . Exists ( path ) )
{
return GetSegementResult ( path ) ;
}
if ( ! File . Exists ( playlistPath ) )
{
await StartFfMpeg ( state , playlistPath , new CancellationTokenSource ( ) ) . ConfigureAwait ( false ) ;
await WaitForMinimumSegmentCount ( playlistPath , GetSegmentWait ( ) , CancellationToken . None ) . ConfigureAwait ( false ) ;
}
return GetSegementResult ( path ) ;
}
private string GetSegmentPath ( string playlist , int index )
{
var folder = Path . GetDirectoryName ( playlist ) ;
var filename = Path . GetFileNameWithoutExtension ( playlist ) ;
return Path . Combine ( folder , filename + index . ToString ( UsCulture ) + ".ts" ) ;
}
private object GetSegementResult ( string path )
{
// TODO: Handle if it's currently being written to
return ResultFactory . GetStaticFileResult ( Request , path ) ;
}
private async Task < object > GetAsync ( GetMasterHlsVideoStream request )
{
var state = await GetState ( request , CancellationToken . None ) . ConfigureAwait ( false ) ;
int audioBitrate ;
int videoBitrate ;
GetPlaylistBitrates ( state , out audioBitrate , out videoBitrate ) ;
var appendBaselineStream = false ;
var baselineStreamBitrate = 64000 ;
var hlsVideoRequest = state . VideoRequest as GetMasterHlsVideoStream ;
if ( hlsVideoRequest ! = null )
{
appendBaselineStream = hlsVideoRequest . AppendBaselineStream ;
baselineStreamBitrate = hlsVideoRequest . BaselineStreamAudioBitRate ? ? baselineStreamBitrate ;
}
var playlistText = GetMasterPlaylistFileText ( videoBitrate + audioBitrate , appendBaselineStream , baselineStreamBitrate ) ;
return ResultFactory . GetResult ( playlistText , MimeTypes . GetMimeType ( "playlist.m3u8" ) , new Dictionary < string , string > ( ) ) ;
}
private string GetMasterPlaylistFileText ( int bitrate , bool includeBaselineStream , int baselineStreamBitrate )
{
var builder = new StringBuilder ( ) ;
builder . AppendLine ( "#EXTM3U" ) ;
// Pad a little to satisfy the apple hls validator
var paddedBitrate = Convert . ToInt32 ( bitrate * 1.05 ) ;
var queryStringIndex = Request . RawUrl . IndexOf ( '?' ) ;
var queryString = queryStringIndex = = - 1 ? string . Empty : Request . RawUrl . Substring ( queryStringIndex ) ;
// Main stream
builder . AppendLine ( "#EXT-X-STREAM-INF:PROGRAM-ID=1,BANDWIDTH=" + paddedBitrate . ToString ( UsCulture ) ) ;
var playlistUrl = "main.m3u8" + queryString ;
builder . AppendLine ( playlistUrl ) ;
// Low bitrate stream
if ( includeBaselineStream )
{
builder . AppendLine ( "#EXT-X-STREAM-INF:PROGRAM-ID=1,BANDWIDTH=" + baselineStreamBitrate . ToString ( UsCulture ) ) ;
playlistUrl = "baseline.m3u8" + queryString ;
builder . AppendLine ( playlistUrl ) ;
}
return builder . ToString ( ) ;
}
public object Get ( GetMainHlsVideoStream request )
{
var result = GetPlaylistAsync ( request , "main" ) . Result ;
return result ;
}
public object Get ( GetBaselineHlsVideoStream request )
{
var result = GetPlaylistAsync ( request , "baseline" ) . Result ;
return result ;
}
private async Task < object > GetPlaylistAsync ( VideoStreamRequest request , string name )
{
var state = await GetState ( request , CancellationToken . None ) . ConfigureAwait ( false ) ;
var builder = new StringBuilder ( ) ;
builder . AppendLine ( "#EXTM3U" ) ;
builder . AppendLine ( "#EXT-X-VERSION:3" ) ;
builder . AppendLine ( "#EXT-X-TARGETDURATION:" + state . SegmentLength . ToString ( UsCulture ) ) ;
builder . AppendLine ( "#EXT-X-MEDIA-SEQUENCE:0" ) ;
var queryStringIndex = Request . RawUrl . IndexOf ( '?' ) ;
var queryString = queryStringIndex = = - 1 ? string . Empty : Request . RawUrl . Substring ( queryStringIndex ) ;
var seconds = TimeSpan . FromTicks ( state . RunTimeTicks ? ? 0 ) . TotalSeconds ;
var index = 0 ;
while ( seconds > 0 )
{
var length = seconds > = state . SegmentLength ? state . SegmentLength : seconds ;
builder . AppendLine ( "#EXTINF:" + length . ToString ( UsCulture ) ) ;
builder . AppendLine ( string . Format ( "hlsdynamic/{0}/{1}.ts{2}" ,
name ,
index . ToString ( UsCulture ) ,
queryString ) ) ;
seconds - = state . SegmentLength ;
index + + ;
}
builder . AppendLine ( "#EXT-X-ENDLIST" ) ;
var playlistText = builder . ToString ( ) ;
return ResultFactory . GetResult ( playlistText , MimeTypes . GetMimeType ( "playlist.m3u8" ) , new Dictionary < string , string > ( ) ) ;
}
protected override string GetAudioArguments ( StreamState state )
{
var codec = GetAudioCodec ( state . Request ) ;
if ( codec . Equals ( "copy" , StringComparison . OrdinalIgnoreCase ) )
{
return "-codec:a:0 copy" ;
}
var args = "-codec:a:0 " + codec ;
if ( state . AudioStream ! = null )
{
var channels = state . OutputAudioChannels ;
if ( channels . HasValue )
{
args + = " -ac " + channels . Value ;
}
var bitrate = state . OutputAudioBitrate ;
if ( bitrate . HasValue )
{
args + = " -ab " + bitrate . Value . ToString ( UsCulture ) ;
}
args + = " " + GetAudioFilterParam ( state , true ) ;
return args ;
}
return args ;
}
protected override string GetVideoArguments ( StreamState state , bool performSubtitleConversion )
{
var codec = GetVideoCodec ( state . VideoRequest ) ;
// See if we can save come cpu cycles by avoiding encoding
if ( codec . Equals ( "copy" , StringComparison . OrdinalIgnoreCase ) )
{
return IsH264 ( state . VideoStream ) ? "-codec:v:0 copy -bsf h264_mp4toannexb" : "-codec:v:0 copy" ;
}
const string keyFrameArg = " -force_key_frames expr:if(isnan(prev_forced_t),gte(t,.1),gte(t,prev_forced_t+5))" ;
var hasGraphicalSubs = state . SubtitleStream ! = null & & state . SubtitleStream . IsGraphicalSubtitleStream ;
var args = "-codec:v:0 " + codec + " " + GetVideoQualityParam ( state , "libx264" , true ) + keyFrameArg ;
// Add resolution params, if specified
if ( ! hasGraphicalSubs )
{
if ( state . VideoRequest . Width . HasValue | | state . VideoRequest . Height . HasValue | | state . VideoRequest . MaxHeight . HasValue | | state . VideoRequest . MaxWidth . HasValue )
{
args + = GetOutputSizeParam ( state , codec , performSubtitleConversion , CancellationToken . None ) ;
}
}
// This is for internal graphical subs
if ( hasGraphicalSubs )
{
args + = GetInternalGraphicalSubtitleParam ( state , codec ) ;
}
return args ;
}
/// <summary>
/// Gets the segment file extension.
/// </summary>
/// <param name="state">The state.</param>
/// <returns>System.String.</returns>
protected override string GetSegmentFileExtension ( StreamState state )
{
return ".ts" ;
}
}
}