using MediaBrowser.Model.Logging;
using ServiceStack.Web;
using System;
using System.Collections.Generic;
using System.Globalization;
using System.IO;
namespace MediaBrowser.Server.Implementations.HttpServer
{
///
/// Class StreamWriter
///
public class StreamWriter : IStreamWriter, IHasOptions
{
private ILogger Logger { get; set; }
private static readonly CultureInfo UsCulture = new CultureInfo("en-US");
///
/// Gets or sets the source stream.
///
/// The source stream.
private Stream SourceStream { get; set; }
///
/// The _options
///
private readonly IDictionary _options = new Dictionary();
///
/// Gets the options.
///
/// The options.
public IDictionary Options
{
get { return _options; }
}
public bool Throttle { get; set; }
public long ThrottleLimit { get; set; }
public long MinThrottlePosition;
public Func ThrottleCallback { get; set; }
public Action OnComplete { get; set; }
///
/// Initializes a new instance of the class.
///
/// The source.
/// Type of the content.
/// The logger.
public StreamWriter(Stream source, string contentType, ILogger logger)
{
if (string.IsNullOrEmpty(contentType))
{
throw new ArgumentNullException("contentType");
}
SourceStream = source;
Logger = logger;
Options["Content-Type"] = contentType;
if (source.CanSeek)
{
Options["Content-Length"] = source.Length.ToString(UsCulture);
}
}
///
/// Initializes a new instance of the class.
///
/// The source.
/// Type of the content.
/// The logger.
public StreamWriter(byte[] source, string contentType, ILogger logger)
: this(new MemoryStream(source), contentType, logger)
{
}
///
/// Writes to.
///
/// The response stream.
public void WriteTo(Stream responseStream)
{
if (Throttle)
{
responseStream = new ThrottledStream(responseStream, ThrottleLimit)
{
MinThrottlePosition = MinThrottlePosition,
ThrottleCallback = ThrottleCallback
};
}
WriteToInternal(responseStream);
}
///
/// Writes to async.
///
/// The response stream.
/// Task.
private void WriteToInternal(Stream responseStream)
{
try
{
using (var src = SourceStream)
{
src.CopyTo(responseStream);
}
}
catch (Exception ex)
{
Logger.ErrorException("Error streaming data", ex);
throw;
}
finally
{
if (OnComplete != null)
{
OnComplete();
}
}
}
}
}