|
@@ -0,0 +1,188 @@
|
|
|
+using System;
|
|
|
+using System.Collections.Generic;
|
|
|
+using System.Globalization;
|
|
|
+using System.Net;
|
|
|
+using System.Threading;
|
|
|
+using System.Threading.Tasks;
|
|
|
+using MediaBrowser.Model.IO;
|
|
|
+using MediaBrowser.Model.Logging;
|
|
|
+using MediaBrowser.Model.Services;
|
|
|
+
|
|
|
+namespace Emby.Server.Implementations.HttpServer
|
|
|
+{
|
|
|
+ public class FileWriter : IHttpResult
|
|
|
+ {
|
|
|
+ private ILogger Logger { get; set; }
|
|
|
+
|
|
|
+ private string RangeHeader { get; set; }
|
|
|
+ private bool IsHeadRequest { get; set; }
|
|
|
+
|
|
|
+ private long RangeStart { get; set; }
|
|
|
+ private long RangeEnd { get; set; }
|
|
|
+ private long RangeLength { get; set; }
|
|
|
+ private long TotalContentLength { get; set; }
|
|
|
+
|
|
|
+ public Action OnComplete { get; set; }
|
|
|
+ public Action OnError { get; set; }
|
|
|
+ private static readonly CultureInfo UsCulture = new CultureInfo("en-US");
|
|
|
+ public List<Cookie> Cookies { get; private set; }
|
|
|
+
|
|
|
+ /// <summary>
|
|
|
+ /// The _options
|
|
|
+ /// </summary>
|
|
|
+ private readonly IDictionary<string, string> _options = new Dictionary<string, string>();
|
|
|
+ /// <summary>
|
|
|
+ /// Gets the options.
|
|
|
+ /// </summary>
|
|
|
+ /// <value>The options.</value>
|
|
|
+ public IDictionary<string, string> Headers
|
|
|
+ {
|
|
|
+ get { return _options; }
|
|
|
+ }
|
|
|
+
|
|
|
+ public string Path { get; set; }
|
|
|
+
|
|
|
+ public FileWriter(string path, string contentType, string rangeHeader, ILogger logger, IFileSystem fileSystem)
|
|
|
+ {
|
|
|
+ if (string.IsNullOrEmpty(contentType))
|
|
|
+ {
|
|
|
+ throw new ArgumentNullException("contentType");
|
|
|
+ }
|
|
|
+
|
|
|
+ Path = path;
|
|
|
+ Logger = logger;
|
|
|
+ RangeHeader = rangeHeader;
|
|
|
+
|
|
|
+ Headers["Content-Type"] = contentType;
|
|
|
+
|
|
|
+ TotalContentLength = fileSystem.GetFileInfo(path).Length;
|
|
|
+
|
|
|
+ if (string.IsNullOrWhiteSpace(rangeHeader))
|
|
|
+ {
|
|
|
+ Headers["Content-Length"] = TotalContentLength.ToString(UsCulture);
|
|
|
+ StatusCode = HttpStatusCode.OK;
|
|
|
+ }
|
|
|
+ else
|
|
|
+ {
|
|
|
+ Headers["Accept-Ranges"] = "bytes";
|
|
|
+ StatusCode = HttpStatusCode.PartialContent;
|
|
|
+ SetRangeValues();
|
|
|
+ }
|
|
|
+
|
|
|
+ Cookies = new List<Cookie>();
|
|
|
+ }
|
|
|
+
|
|
|
+ /// <summary>
|
|
|
+ /// Sets the range values.
|
|
|
+ /// </summary>
|
|
|
+ private void SetRangeValues()
|
|
|
+ {
|
|
|
+ var requestedRange = RequestedRanges[0];
|
|
|
+
|
|
|
+ // If the requested range is "0-", we can optimize by just doing a stream copy
|
|
|
+ if (!requestedRange.Value.HasValue)
|
|
|
+ {
|
|
|
+ RangeEnd = TotalContentLength - 1;
|
|
|
+ }
|
|
|
+ else
|
|
|
+ {
|
|
|
+ RangeEnd = requestedRange.Value.Value;
|
|
|
+ }
|
|
|
+
|
|
|
+ RangeStart = requestedRange.Key;
|
|
|
+ RangeLength = 1 + RangeEnd - RangeStart;
|
|
|
+
|
|
|
+ // Content-Length is the length of what we're serving, not the original content
|
|
|
+ Headers["Content-Length"] = RangeLength.ToString(UsCulture);
|
|
|
+ Headers["Content-Range"] = string.Format("bytes {0}-{1}/{2}", RangeStart, RangeEnd, TotalContentLength);
|
|
|
+ }
|
|
|
+
|
|
|
+ /// <summary>
|
|
|
+ /// The _requested ranges
|
|
|
+ /// </summary>
|
|
|
+ private List<KeyValuePair<long, long?>> _requestedRanges;
|
|
|
+ /// <summary>
|
|
|
+ /// Gets the requested ranges.
|
|
|
+ /// </summary>
|
|
|
+ /// <value>The requested ranges.</value>
|
|
|
+ protected List<KeyValuePair<long, long?>> RequestedRanges
|
|
|
+ {
|
|
|
+ get
|
|
|
+ {
|
|
|
+ if (_requestedRanges == null)
|
|
|
+ {
|
|
|
+ _requestedRanges = new List<KeyValuePair<long, long?>>();
|
|
|
+
|
|
|
+ // Example: bytes=0-,32-63
|
|
|
+ var ranges = RangeHeader.Split('=')[1].Split(',');
|
|
|
+
|
|
|
+ foreach (var range in ranges)
|
|
|
+ {
|
|
|
+ var vals = range.Split('-');
|
|
|
+
|
|
|
+ long start = 0;
|
|
|
+ long? end = null;
|
|
|
+
|
|
|
+ if (!string.IsNullOrEmpty(vals[0]))
|
|
|
+ {
|
|
|
+ start = long.Parse(vals[0], UsCulture);
|
|
|
+ }
|
|
|
+ if (!string.IsNullOrEmpty(vals[1]))
|
|
|
+ {
|
|
|
+ end = long.Parse(vals[1], UsCulture);
|
|
|
+ }
|
|
|
+
|
|
|
+ _requestedRanges.Add(new KeyValuePair<long, long?>(start, end));
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return _requestedRanges;
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ public async Task WriteToAsync(IResponse response, CancellationToken cancellationToken)
|
|
|
+ {
|
|
|
+ try
|
|
|
+ {
|
|
|
+ // Headers only
|
|
|
+ if (IsHeadRequest)
|
|
|
+ {
|
|
|
+ return;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (string.IsNullOrWhiteSpace(RangeHeader) || (RangeStart <= 0 && RangeEnd >= TotalContentLength - 1))
|
|
|
+ {
|
|
|
+ Logger.Info("Transmit file {0}", Path);
|
|
|
+ await response.TransmitFile(Path, 0, 0, cancellationToken).ConfigureAwait(false);
|
|
|
+ return;
|
|
|
+ }
|
|
|
+
|
|
|
+ await response.TransmitFile(Path, RangeStart, RangeEnd, cancellationToken).ConfigureAwait(false);
|
|
|
+ }
|
|
|
+ finally
|
|
|
+ {
|
|
|
+ if (OnComplete != null)
|
|
|
+ {
|
|
|
+ OnComplete();
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ public string ContentType { get; set; }
|
|
|
+
|
|
|
+ public IRequest RequestContext { get; set; }
|
|
|
+
|
|
|
+ public object Response { get; set; }
|
|
|
+
|
|
|
+ public int Status { get; set; }
|
|
|
+
|
|
|
+ public HttpStatusCode StatusCode
|
|
|
+ {
|
|
|
+ get { return (HttpStatusCode)Status; }
|
|
|
+ set { Status = (int)value; }
|
|
|
+ }
|
|
|
+
|
|
|
+ public string StatusDescription { get; set; }
|
|
|
+
|
|
|
+ }
|
|
|
+}
|