Обнновлено до .net10
This commit is contained in:
FrigaT
@@ -1,179 +1,142 @@
|
||||
using System.Net.WebSockets;
|
||||
using System.Text;
|
||||
using System.Text.Json;
|
||||
|
||||
namespace YandexMusic.API.Common.Ynison
|
||||
namespace YandexMusic.API.Common.Ynison;
|
||||
|
||||
/// <summary>WebSocket-клиент для взаимодействия с протоколом Ynison.</summary>
|
||||
public class YnisonWebSocket : IDisposable
|
||||
{
|
||||
public class YnisonWebSocket : IDisposable
|
||||
private readonly ClientWebSocket _socketClient = new();
|
||||
private CancellationTokenSource? _cancellationTokenSource;
|
||||
private CancellationToken _cancellationToken;
|
||||
private readonly StringBuilder _data = new();
|
||||
private const int BufferSize = 4096;
|
||||
|
||||
/// <summary>Флаг, указывает, открыто ли соединение.</summary>
|
||||
public bool IsConnected => _socketClient.State == WebSocketState.Open;
|
||||
|
||||
/// <summary>Событие получения сообщения.</summary>
|
||||
public event EventHandler<ReceiveEventArgs>? OnReceive;
|
||||
|
||||
/// <summary>Событие закрытия соединения.</summary>
|
||||
public event EventHandler<CloseEventArgs>? OnClose;
|
||||
|
||||
/// <summary>Аргументы события получения данных.</summary>
|
||||
public class ReceiveEventArgs : EventArgs
|
||||
{
|
||||
#region Поля
|
||||
/// <summary>Полученные данные (JSON-строка).</summary>
|
||||
public string Data { get; init; } = null!;
|
||||
}
|
||||
|
||||
private readonly JsonSerializerSettings jsonSettings = new()
|
||||
/// <summary>Аргументы события закрытия соединения.</summary>
|
||||
public class CloseEventArgs : EventArgs
|
||||
{
|
||||
/// <summary>Статус закрытия.</summary>
|
||||
public WebSocketCloseStatus? Status { get; init; }
|
||||
/// <summary>Описание причины закрытия.</summary>
|
||||
public string? Description { get; init; }
|
||||
}
|
||||
|
||||
private static string GetProtocolData(string deviceId, string? redirectTicket)
|
||||
{
|
||||
var deviceInfo = new Dictionary<string, object>
|
||||
{
|
||||
Converters = new List<JsonConverter> {
|
||||
new StringEnumConverter {
|
||||
NamingStrategy = new CamelCaseNamingStrategy()
|
||||
}
|
||||
},
|
||||
NullValueHandling = NullValueHandling.Ignore
|
||||
{ "app_name", "Chrome" },
|
||||
{ "type", 1 }
|
||||
};
|
||||
|
||||
private readonly ClientWebSocket socketClient = new();
|
||||
|
||||
private CancellationTokenSource cancellationTokenSource = new();
|
||||
private CancellationToken cancellation;
|
||||
|
||||
private readonly StringBuilder data = new();
|
||||
private readonly int size = 4096;
|
||||
|
||||
#endregion Поля
|
||||
|
||||
#region Свойства
|
||||
|
||||
public bool IsConnected => socketClient.State == WebSocketState.Open;
|
||||
|
||||
#endregion Свойства
|
||||
|
||||
#region События
|
||||
|
||||
public class ReceiveEventArgs
|
||||
var protocol = new Dictionary<string, string>
|
||||
{
|
||||
public string Data { get; internal set; }
|
||||
}
|
||||
{ "Ynison-Device-Id", deviceId },
|
||||
{ "Ynison-Device-Info", JsonSerializer.Serialize(deviceInfo) }
|
||||
};
|
||||
if (!string.IsNullOrEmpty(redirectTicket))
|
||||
protocol.Add("Ynison-Redirect-Ticket", redirectTicket);
|
||||
return JsonSerializer.Serialize(protocol);
|
||||
}
|
||||
|
||||
public delegate void OnReceiveEventHandler(YnisonWebSocket socket, ReceiveEventArgs args);
|
||||
/// <summary>
|
||||
/// Получение данных
|
||||
/// </summary>
|
||||
public event OnReceiveEventHandler OnReceive;
|
||||
|
||||
public class CloseEventArgs
|
||||
private async Task<string> ReadSocketContentAsync()
|
||||
{
|
||||
var buffer = new byte[BufferSize];
|
||||
WebSocketReceiveResult result;
|
||||
do
|
||||
{
|
||||
public WebSocketCloseStatus? Status { get; set; }
|
||||
public string Description { get; set; }
|
||||
}
|
||||
result = await _socketClient.ReceiveAsync(new ArraySegment<byte>(buffer), _cancellationToken);
|
||||
_data.Append(Encoding.UTF8.GetString(buffer, 0, result.Count));
|
||||
} while (!result.EndOfMessage);
|
||||
return _data.ToString();
|
||||
}
|
||||
|
||||
public delegate void OnCloseEventHandler(YnisonWebSocket socket, CloseEventArgs args);
|
||||
/// <summary>
|
||||
/// Закрытие соединения
|
||||
/// </summary>
|
||||
public event OnCloseEventHandler OnClose;
|
||||
/// <summary>Подключается к WebSocket.</summary>
|
||||
/// <param name="storage">Хранилище авторизации.</param>
|
||||
/// <param name="url">URL WebSocket.</param>
|
||||
/// <param name="redirectTicket">Тикет перенаправления (опционально).</param>
|
||||
public async Task ConnectAsync(AuthStorage storage, string url, string? redirectTicket = null)
|
||||
{
|
||||
_socketClient.Options.AddSubProtocol("Bearer");
|
||||
var protocolData = GetProtocolData(storage.DeviceId, redirectTicket);
|
||||
_socketClient.Options.SetRequestHeader("Sec-WebSocket-Protocol", $"Bearer, v2, {protocolData}");
|
||||
_socketClient.Options.SetRequestHeader("Origin", "https://music.yandex.ru");
|
||||
_socketClient.Options.SetRequestHeader("Authorization", $"OAuth {storage.Token}");
|
||||
_socketClient.Options.Proxy = storage.Context.WebProxy;
|
||||
|
||||
#endregion События
|
||||
_cancellationTokenSource = new CancellationTokenSource();
|
||||
_cancellationToken = _cancellationTokenSource.Token;
|
||||
|
||||
#region Вспомогательные функции
|
||||
await _socketClient.ConnectAsync(new Uri(url), CancellationToken.None);
|
||||
}
|
||||
|
||||
private string SerializeJson(object obj)
|
||||
/// <summary>Начинает асинхронный приём сообщений.</summary>
|
||||
public async Task BeginReceiveAsync()
|
||||
{
|
||||
if (_socketClient.State != WebSocketState.Open)
|
||||
return;
|
||||
|
||||
try
|
||||
{
|
||||
return JsonConvert.SerializeObject(obj, jsonSettings);
|
||||
}
|
||||
|
||||
private string GetProtocolData(string deviceId, string redirectTicket)
|
||||
{
|
||||
Dictionary<string, object> deviceInfo = new() {
|
||||
{ "app_name", "Chrome" },
|
||||
{ "type", 1 }
|
||||
};
|
||||
|
||||
Dictionary<string, string> protocol = new() {
|
||||
{ "Ynison-Device-Id", deviceId },
|
||||
{ "Ynison-Device-Info", SerializeJson(deviceInfo) }
|
||||
};
|
||||
|
||||
if (!string.IsNullOrEmpty(redirectTicket))
|
||||
protocol.Add("Ynison-Redirect-Ticket", redirectTicket);
|
||||
|
||||
return SerializeJson(protocol);
|
||||
}
|
||||
|
||||
private async Task<string> ReadSocketContent()
|
||||
{
|
||||
byte[] buffer = new byte[size];
|
||||
WebSocketReceiveResult result;
|
||||
|
||||
do
|
||||
while (!_cancellationToken.IsCancellationRequested && _socketClient.State == WebSocketState.Open)
|
||||
{
|
||||
result = await socketClient.ReceiveAsync(new ArraySegment<byte>(buffer), CancellationToken.None);
|
||||
data.Append(Encoding.UTF8.GetString(buffer, 0, result.Count));
|
||||
} while (!result.EndOfMessage);
|
||||
|
||||
return data.ToString();
|
||||
var content = await ReadSocketContentAsync();
|
||||
OnReceive?.Invoke(this, new ReceiveEventArgs { Data = content });
|
||||
_data.Clear();
|
||||
}
|
||||
}
|
||||
|
||||
#endregion Вспомогательные функции
|
||||
|
||||
|
||||
|
||||
public bool Connect(AuthStorage storage, string url, string redirectTicket = null)
|
||||
catch (OperationCanceledException)
|
||||
{
|
||||
socketClient.Options.AddSubProtocol("Bearer");
|
||||
|
||||
socketClient.Options.SetRequestHeader("Sec-WebSocket-Protocol", $"Bearer, v2, {GetProtocolData(storage.DeviceId, redirectTicket)}");
|
||||
socketClient.Options.SetRequestHeader("Origin", "https://music.yandex.ru");
|
||||
socketClient.Options.SetRequestHeader("Authorization", $"OAuth {storage.Token}");
|
||||
|
||||
socketClient.Options.Proxy = storage.Context.WebProxy;
|
||||
|
||||
socketClient.ConnectAsync(new Uri(url), CancellationToken.None)
|
||||
.GetAwaiter()
|
||||
.GetResult();
|
||||
|
||||
cancellation = cancellationTokenSource.Token;
|
||||
|
||||
return socketClient.State == WebSocketState.Open;
|
||||
// Ожидаемая отмена
|
||||
}
|
||||
|
||||
public async Task BeginReceive()
|
||||
finally
|
||||
{
|
||||
if (socketClient.State != WebSocketState.Open)
|
||||
return;
|
||||
|
||||
do
|
||||
{
|
||||
string content = await ReadSocketContent();
|
||||
OnReceive?.Invoke(this, new ReceiveEventArgs
|
||||
{
|
||||
Data = content
|
||||
});
|
||||
|
||||
data.Clear();
|
||||
} while (!cancellation.IsCancellationRequested && socketClient.State == WebSocketState.Open);
|
||||
|
||||
OnClose?.Invoke(this, new CloseEventArgs
|
||||
{
|
||||
Status = socketClient.CloseStatus,
|
||||
Description = socketClient.CloseStatusDescription
|
||||
});
|
||||
|
||||
await socketClient.CloseAsync(WebSocketCloseStatus.NormalClosure, string.Empty, CancellationToken.None);
|
||||
var closeStatus = _socketClient.CloseStatus;
|
||||
var closeDesc = _socketClient.CloseStatusDescription;
|
||||
OnClose?.Invoke(this, new CloseEventArgs { Status = closeStatus, Description = closeDesc });
|
||||
if (_socketClient.State == WebSocketState.Open)
|
||||
await _socketClient.CloseAsync(WebSocketCloseStatus.NormalClosure, string.Empty, CancellationToken.None);
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>Отправляет JSON-сообщение.</summary>
|
||||
/// <param name="json">JSON-строка.</param>
|
||||
public async ValueTask SendAsync(string json)
|
||||
{
|
||||
var bytes = Encoding.UTF8.GetBytes(json);
|
||||
await _socketClient.SendAsync(new ArraySegment<byte>(bytes), WebSocketMessageType.Text, true, _cancellationToken);
|
||||
}
|
||||
|
||||
public ValueTask Send(string json)
|
||||
/// <summary>Останавливает приём сообщений.</summary>
|
||||
public async Task StopReceiveAsync()
|
||||
{
|
||||
if (_cancellationTokenSource != null && !_cancellationTokenSource.IsCancellationRequested)
|
||||
{
|
||||
ReadOnlyMemory<byte> message = new(Encoding.UTF8.GetBytes(json));
|
||||
return socketClient.SendAsync(message, WebSocketMessageType.Text, false, CancellationToken.None);
|
||||
await _cancellationTokenSource.CancelAsync();
|
||||
}
|
||||
}
|
||||
|
||||
public Task StopReceive()
|
||||
{
|
||||
if (socketClient.State != WebSocketState.Open)
|
||||
return Task.CompletedTask;
|
||||
|
||||
cancellationTokenSource.Cancel(false);
|
||||
|
||||
return Task.CompletedTask;
|
||||
}
|
||||
|
||||
|
||||
|
||||
#region IDisposable
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
socketClient?.Dispose();
|
||||
cancellationTokenSource?.Dispose();
|
||||
}
|
||||
|
||||
#endregion IDisposable
|
||||
/// <summary>Освобождает ресурсы.</summary>
|
||||
public void Dispose()
|
||||
{
|
||||
_cancellationTokenSource?.Dispose();
|
||||
_socketClient.Dispose();
|
||||
GC.SuppressFinalize(this);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user