mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-06 03:26:07 +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.
158 lines
4.8 KiB
C#
158 lines
4.8 KiB
C#
using System;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using System.Net.WebSockets;
|
|
using Emby.Server.Implementations.Net;
|
|
using Microsoft.Extensions.Logging;
|
|
|
|
namespace Jellyfin.SocketSharp
|
|
{
|
|
public class SharpWebSocket : IWebSocket
|
|
{
|
|
/// <summary>
|
|
/// The logger
|
|
/// </summary>
|
|
private readonly ILogger _logger;
|
|
|
|
public event EventHandler<EventArgs> Closed;
|
|
|
|
/// <summary>
|
|
/// Gets or sets the web socket.
|
|
/// </summary>
|
|
/// <value>The web socket.</value>
|
|
private SocketHttpListener.WebSocket WebSocket { get; set; }
|
|
|
|
private TaskCompletionSource<bool> _taskCompletionSource = new TaskCompletionSource<bool>();
|
|
private readonly CancellationTokenSource _cancellationTokenSource = new CancellationTokenSource();
|
|
|
|
public SharpWebSocket(SocketHttpListener.WebSocket socket, ILogger logger)
|
|
{
|
|
if (socket == null)
|
|
{
|
|
throw new ArgumentNullException(nameof(socket));
|
|
}
|
|
|
|
if (logger == null)
|
|
{
|
|
throw new ArgumentNullException(nameof(logger));
|
|
}
|
|
|
|
_logger = logger;
|
|
WebSocket = socket;
|
|
|
|
socket.OnMessage += socket_OnMessage;
|
|
socket.OnClose += socket_OnClose;
|
|
socket.OnError += socket_OnError;
|
|
|
|
WebSocket.ConnectAsServer();
|
|
}
|
|
|
|
public Task StartReceive()
|
|
{
|
|
return _taskCompletionSource.Task;
|
|
}
|
|
|
|
void socket_OnError(object sender, SocketHttpListener.ErrorEventArgs e)
|
|
{
|
|
_logger.LogError("Error in SharpWebSocket: {Message}", e.Message ?? string.Empty);
|
|
//Closed?.Invoke(this, EventArgs.Empty);
|
|
}
|
|
|
|
void socket_OnClose(object sender, SocketHttpListener.CloseEventArgs e)
|
|
{
|
|
_taskCompletionSource.TrySetResult(true);
|
|
|
|
Closed?.Invoke(this, EventArgs.Empty);
|
|
}
|
|
|
|
void socket_OnMessage(object sender, SocketHttpListener.MessageEventArgs e)
|
|
{
|
|
//if (!string.IsNullOrEmpty(e.Data))
|
|
//{
|
|
// if (OnReceive != null)
|
|
// {
|
|
// OnReceive(e.Data);
|
|
// }
|
|
// return;
|
|
//}
|
|
if (OnReceiveBytes != null)
|
|
{
|
|
OnReceiveBytes(e.RawData);
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets or sets the state.
|
|
/// </summary>
|
|
/// <value>The state.</value>
|
|
public WebSocketState State
|
|
{
|
|
get
|
|
{
|
|
return WebSocket.ReadyState;
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Sends the async.
|
|
/// </summary>
|
|
/// <param name="bytes">The bytes.</param>
|
|
/// <param name="endOfMessage">if set to <c>true</c> [end of message].</param>
|
|
/// <param name="cancellationToken">The cancellation token.</param>
|
|
/// <returns>Task.</returns>
|
|
public Task SendAsync(byte[] bytes, bool endOfMessage, CancellationToken cancellationToken)
|
|
{
|
|
return WebSocket.SendAsync(bytes);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Sends the asynchronous.
|
|
/// </summary>
|
|
/// <param name="text">The text.</param>
|
|
/// <param name="endOfMessage">if set to <c>true</c> [end of message].</param>
|
|
/// <param name="cancellationToken">The cancellation token.</param>
|
|
/// <returns>Task.</returns>
|
|
public Task SendAsync(string text, bool endOfMessage, CancellationToken cancellationToken)
|
|
{
|
|
return WebSocket.SendAsync(text);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Performs application-defined tasks associated with freeing, releasing, or resetting unmanaged resources.
|
|
/// </summary>
|
|
public void Dispose()
|
|
{
|
|
Dispose(true);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Releases unmanaged and - optionally - managed resources.
|
|
/// </summary>
|
|
/// <param name="dispose"><c>true</c> to release both managed and unmanaged resources; <c>false</c> to release only unmanaged resources.</param>
|
|
protected virtual void Dispose(bool dispose)
|
|
{
|
|
if (dispose)
|
|
{
|
|
WebSocket.OnMessage -= socket_OnMessage;
|
|
WebSocket.OnClose -= socket_OnClose;
|
|
WebSocket.OnError -= socket_OnError;
|
|
|
|
_cancellationTokenSource.Cancel();
|
|
|
|
WebSocket.Close();
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets or sets the receive action.
|
|
/// </summary>
|
|
/// <value>The receive action.</value>
|
|
public Action<byte[]> OnReceiveBytes { get; set; }
|
|
|
|
/// <summary>
|
|
/// Gets or sets the on receive.
|
|
/// </summary>
|
|
/// <value>The on receive.</value>
|
|
public Action<string> OnReceive { get; set; }
|
|
}
|
|
}
|