mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-07 03:48:06 +02:00
Add Argument*Exceptions now use proper nameof operators. Added exception messages to quite a few Argument*Exceptions. Fixed rethorwing to be proper syntax. Added a ton of null checkes. (This is only a start, there are about 500 places that need proper null handling) Added some TODOs to log certain exceptions. Fix sln again. Fixed all AssemblyInfo's and added proper copyright (where I could find them) We live in *current year*. Fixed the use of braces. Fixed a ton of properties, and made a fair amount of functions static that should be and can be static. Made more Methods that should be static static. You can now use static to find bad functions! Removed unused variable. And added one more proper XML comment.
124 lines
3.6 KiB
C#
124 lines
3.6 KiB
C#
using Microsoft.Extensions.Logging;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Globalization;
|
|
using System.IO;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
|
|
using MediaBrowser.Model.Services;
|
|
|
|
namespace Emby.Server.Implementations.HttpServer
|
|
{
|
|
/// <summary>
|
|
/// Class StreamWriter
|
|
/// </summary>
|
|
public class StreamWriter : IAsyncStreamWriter, IHasHeaders
|
|
{
|
|
private ILogger Logger { get; set; }
|
|
|
|
private static readonly CultureInfo UsCulture = new CultureInfo("en-US");
|
|
|
|
/// <summary>
|
|
/// Gets or sets the source stream.
|
|
/// </summary>
|
|
/// <value>The source stream.</value>
|
|
private Stream SourceStream { get; set; }
|
|
|
|
private byte[] SourceBytes { get; 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 => _options;
|
|
|
|
public Action OnComplete { get; set; }
|
|
public Action OnError { get; set; }
|
|
|
|
/// <summary>
|
|
/// Initializes a new instance of the <see cref="StreamWriter" /> class.
|
|
/// </summary>
|
|
/// <param name="source">The source.</param>
|
|
/// <param name="contentType">Type of the content.</param>
|
|
/// <param name="logger">The logger.</param>
|
|
public StreamWriter(Stream source, string contentType, ILogger logger)
|
|
{
|
|
if (string.IsNullOrEmpty(contentType))
|
|
{
|
|
throw new ArgumentNullException(nameof(contentType));
|
|
}
|
|
|
|
SourceStream = source;
|
|
Logger = logger;
|
|
|
|
Headers["Content-Type"] = contentType;
|
|
|
|
if (source.CanSeek)
|
|
{
|
|
Headers["Content-Length"] = source.Length.ToString(UsCulture);
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Initializes a new instance of the <see cref="StreamWriter"/> class.
|
|
/// </summary>
|
|
/// <param name="source">The source.</param>
|
|
/// <param name="contentType">Type of the content.</param>
|
|
/// <param name="logger">The logger.</param>
|
|
public StreamWriter(byte[] source, string contentType, int contentLength, ILogger logger)
|
|
{
|
|
if (string.IsNullOrEmpty(contentType))
|
|
{
|
|
throw new ArgumentNullException(nameof(contentType));
|
|
}
|
|
|
|
SourceBytes = source;
|
|
Logger = logger;
|
|
|
|
Headers["Content-Type"] = contentType;
|
|
|
|
Headers["Content-Length"] = contentLength.ToString(UsCulture);
|
|
}
|
|
|
|
public async Task WriteToAsync(Stream responseStream, CancellationToken cancellationToken)
|
|
{
|
|
try
|
|
{
|
|
var bytes = SourceBytes;
|
|
|
|
if (bytes != null)
|
|
{
|
|
await responseStream.WriteAsync(bytes, 0, bytes.Length).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
using (var src = SourceStream)
|
|
{
|
|
await src.CopyToAsync(responseStream).ConfigureAwait(false);
|
|
}
|
|
}
|
|
}
|
|
catch
|
|
{
|
|
if (OnError != null)
|
|
{
|
|
OnError();
|
|
}
|
|
|
|
throw;
|
|
}
|
|
finally
|
|
{
|
|
if (OnComplete != null)
|
|
{
|
|
OnComplete();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|