@@ -4,9 +4,9 @@ namespace Discord | |||
{ | |||
public enum PermissionTarget | |||
{ | |||
[ModelEnum("role")] | |||
[ModelEnumValue("role")] | |||
Role, | |||
[ModelEnum("user")] | |||
[ModelEnumValue("user")] | |||
User | |||
} | |||
} |
@@ -1,10 +1,13 @@ | |||
namespace Discord | |||
using Discord.Serialization; | |||
namespace Discord | |||
{ | |||
public struct Overwrite | |||
{ | |||
/// <summary> Gets the unique identifier for the object this overwrite is targeting. </summary> | |||
public ulong TargetId { get; } | |||
/// <summary> Gets the type of object this overwrite is targeting. </summary> | |||
[ModelStringEnum] | |||
public PermissionTarget TargetType { get; } | |||
/// <summary> Gets the permissions associated with this overwrite entry. </summary> | |||
public OverwritePermissions Permissions { get; } | |||
@@ -5,13 +5,13 @@ namespace Discord | |||
public enum UserStatus | |||
{ | |||
Offline, | |||
[ModelEnum("online")] | |||
[ModelEnumValue("online")] | |||
Online, | |||
[ModelEnum("idle")] | |||
[ModelEnumValue("idle")] | |||
Idle, | |||
[ModelEnum("idle", EnumValueType.WriteOnly)] | |||
[ModelEnumValue("idle", EnumValueType.WriteOnly)] | |||
AFK, | |||
[ModelEnum("dnd")] | |||
[ModelEnumValue("dnd")] | |||
DoNotDisturb, | |||
Invisible, | |||
} | |||
@@ -10,6 +10,7 @@ namespace Discord.API | |||
[ModelProperty("guild_id")] | |||
public Optional<ulong> GuildId { get; set; } | |||
[ModelProperty("status")] | |||
[ModelStringEnum] | |||
public UserStatus Status { get; set; } | |||
[ModelProperty("game")] | |||
public Game Game { get; set; } | |||
@@ -191,7 +191,7 @@ namespace Discord.API | |||
options.BucketId = AuthTokenType == TokenType.User ? ClientBucket.Get(clientBucket).Id : bucketId; | |||
options.IsClientBucket = AuthTokenType == TokenType.User; | |||
if (_formatters.TryDequeue(out var data)) | |||
if (!_formatters.TryDequeue(out var data)) | |||
data = new ArrayFormatter(128, SymbolTable.InvariantUtf8); | |||
try | |||
{ | |||
@@ -240,7 +240,7 @@ namespace Discord.API | |||
options.BucketId = AuthTokenType == TokenType.User ? ClientBucket.Get(clientBucket).Id : bucketId; | |||
options.IsClientBucket = AuthTokenType == TokenType.User; | |||
if (_formatters.TryDequeue(out var data)) | |||
if (!_formatters.TryDequeue(out var data)) | |||
data = new ArrayFormatter(128, SymbolTable.InvariantUtf8); | |||
try | |||
{ | |||
@@ -1168,13 +1168,12 @@ namespace Discord.API | |||
throw new InvalidOperationException("Client is not logged in."); | |||
} | |||
protected static double ToMilliseconds(Stopwatch stopwatch) => Math.Round((double)stopwatch.ElapsedTicks / (double)Stopwatch.Frequency * 1000.0, 2); | |||
protected ReadOnlyBuffer<byte> SerializeJson(ArrayFormatter data, object value) | |||
protected ReadOnlyBuffer<byte> SerializeJson<T>(ArrayFormatter data, T value) | |||
{ | |||
_serializer.Write(data, value); | |||
return new ReadOnlyBuffer<byte>(data.Formatted.Array, 0, data.Formatted.Count); | |||
} | |||
protected T DeserializeJson<T>(ReadOnlyBuffer<byte> data) | |||
where T : class, new() | |||
{ | |||
return _serializer.Read<T>(data); | |||
} | |||
@@ -12,19 +12,19 @@ namespace Discord.Serialization.Json.Converters | |||
_innerConverter = innerConverter; | |||
} | |||
public EntityOrId<T> Read(PropertyMap map, ref JsonReader reader, bool isTopLevel) | |||
public EntityOrId<T> Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
reader.Read(); | |||
if (reader.ValueType == JsonValueType.Number) | |||
return new EntityOrId<T>(reader.ParseUInt64()); | |||
return new EntityOrId<T>(_innerConverter.Read(map, ref reader, false)); | |||
return new EntityOrId<T>(_innerConverter.Read(map, model, ref reader, false)); | |||
} | |||
public void Write(PropertyMap map, ref JsonWriter writer, EntityOrId<T> value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, EntityOrId<T> value, bool isTopLevel) | |||
{ | |||
if (value.Object != null) | |||
_innerConverter.Write(map, ref writer, value.Object, isTopLevel); | |||
_innerConverter.Write(map, model, ref writer, value.Object, isTopLevel); | |||
else | |||
{ | |||
if (isTopLevel) | |||
@@ -5,7 +5,7 @@ namespace Discord.Serialization.Json.Converters | |||
{ | |||
internal class ImagePropertyConverter : IJsonPropertyConverter<API.Image> | |||
{ | |||
public API.Image Read(PropertyMap map, ref JsonReader reader, bool isTopLevel) | |||
public API.Image Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
reader.Read(); | |||
@@ -13,7 +13,7 @@ namespace Discord.Serialization.Json.Converters | |||
throw new SerializationException("Bad input, expected String"); | |||
return new API.Image(reader.ParseString()); | |||
} | |||
public void Write(PropertyMap map, ref JsonWriter writer, API.Image value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, API.Image value, bool isTopLevel) | |||
{ | |||
string str; | |||
if (value.Stream != null) | |||
@@ -4,7 +4,7 @@ namespace Discord.Serialization.Json.Converters | |||
{ | |||
internal class Int53PropertyConverter : IJsonPropertyConverter<long> | |||
{ | |||
public long Read(PropertyMap map, ref JsonReader reader, bool isTopLevel) | |||
public long Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
reader.Read(); | |||
@@ -12,7 +12,7 @@ namespace Discord.Serialization.Json.Converters | |||
throw new SerializationException("Bad input, expected Number"); | |||
return reader.ParseInt64(); | |||
} | |||
public void Write(PropertyMap map, ref JsonWriter writer, long value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, long value, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
writer.WriteAttribute(map.Key, value); | |||
@@ -11,13 +11,13 @@ namespace Discord.Serialization.Json.Converters | |||
_innerConverter = innerConverter; | |||
} | |||
public Optional<T> Read(PropertyMap map, ref JsonReader reader, bool isTopLevel) | |||
=> new Optional<T>(_innerConverter.Read(map, ref reader, isTopLevel)); | |||
public Optional<T> Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
=> new Optional<T>(_innerConverter.Read(map, model, ref reader, isTopLevel)); | |||
public void Write(PropertyMap map, ref JsonWriter writer, Optional<T> value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, Optional<T> value, bool isTopLevel) | |||
{ | |||
if (value.IsSpecified) | |||
_innerConverter.Write(map, ref writer, value.Value, isTopLevel); | |||
_innerConverter.Write(map, model, ref writer, value.Value, isTopLevel); | |||
} | |||
} | |||
} |
@@ -4,7 +4,7 @@ namespace Discord.Serialization.Json.Converters | |||
{ | |||
internal class UInt53PropertyConverter : IJsonPropertyConverter<ulong> | |||
{ | |||
public ulong Read(PropertyMap map, ref JsonReader reader, bool isTopLevel) | |||
public ulong Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
reader.Read(); | |||
@@ -12,7 +12,7 @@ namespace Discord.Serialization.Json.Converters | |||
throw new SerializationException("Bad input, expected Number"); | |||
return reader.ParseUInt64(); | |||
} | |||
public void Write(PropertyMap map, ref JsonWriter writer, ulong value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, ulong value, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
writer.WriteAttribute(map.Key, value); | |||
@@ -0,0 +1,11 @@ | |||
namespace Discord.Serialization | |||
{ | |||
public static class ModelSelectorGroups | |||
{ | |||
public const string GatewayFrame = nameof(GatewayFrame); | |||
public const string GatewayDispatchFrame = nameof(GatewayDispatchFrame); | |||
public const string VoiceFrame = nameof(VoiceFrame); | |||
public const string VoiceDispatchFrame = nameof(VoiceDispatchFrame); | |||
public const string RpcFrame = nameof(RpcFrame); | |||
} | |||
} |
@@ -14,7 +14,9 @@ namespace Discord.API.Rpc | |||
public Optional<string> Event { get; set; } | |||
[ModelProperty("data")] | |||
public Optional<ReadOnlyBuffer<byte>> Data { get; set; } | |||
[ModelProperty("args")] | |||
[ModelSelector(ModelSelectorGroups.RpcFrame, nameof(Event))] | |||
public object Args { get; set; } | |||
} | |||
} |
@@ -5,6 +5,7 @@ using Discord.Net.Rest; | |||
using Discord.Net.WebSockets; | |||
using Discord.Rpc; | |||
using Discord.Serialization; | |||
using Discord.Serialization.Json; | |||
using System; | |||
using System.Collections.Concurrent; | |||
using System.Collections.Generic; | |||
@@ -41,11 +42,11 @@ namespace Discord.API | |||
} | |||
public Task SetResultAsync(ReadOnlyBuffer<byte> data) | |||
{ | |||
return Promise.TrySetResultAsync(Serializer.Json.Read<T>(data)); | |||
return Promise.TrySetResultAsync(DiscordJsonSerializer.Global.Read<T>(data)); | |||
} | |||
public Task SetExceptionAsync(ReadOnlyBuffer<byte> data) | |||
{ | |||
var error = Serializer.Json.Read<ErrorEvent>(data); | |||
var error = DiscordJsonSerializer.Global.Read<ErrorEvent>(data); | |||
return Promise.TrySetExceptionAsync(new RpcException(error.Code, error.Message)); | |||
} | |||
} | |||
@@ -231,12 +232,12 @@ namespace Discord.API | |||
try | |||
{ | |||
var guid = Guid.NewGuid(); | |||
payload = new API.Rpc.RpcFrame { Cmd = cmd, Event = evt, Args = SerializeJson(data1, payload), Nonce = guid }; | |||
var frame = new API.Rpc.RpcFrame { Cmd = cmd, Event = evt, Args = SerializeJson(data1, payload), Nonce = guid }; | |||
var requestTracker = new RpcRequest<TResponse>(options); | |||
_requests[guid] = requestTracker; | |||
await RequestQueue.SendAsync(new WebSocketRequest(_webSocketClient, null, SerializeJson(data2, payload), true, options)).ConfigureAwait(false); | |||
await RequestQueue.SendAsync(new WebSocketRequest(_webSocketClient, null, SerializeJson(data2, frame), true, options)).ConfigureAwait(false); | |||
await _sentRpcMessageEvent.InvokeAsync(cmd).ConfigureAwait(false); | |||
return await requestTracker.Promise.Task.ConfigureAwait(false); | |||
} | |||
@@ -8,6 +8,7 @@ using System.Linq; | |||
using System.Threading.Tasks; | |||
using System.Threading; | |||
using Discord.Serialization; | |||
using Discord.Serialization.Json; | |||
namespace Discord.Rpc | |||
{ | |||
@@ -37,7 +38,7 @@ namespace Discord.Rpc | |||
_authorizeLock = new SemaphoreSlim(1, 1); | |||
_rpcLogger = LogManager.CreateLogger("RPC"); | |||
_serializer = new Serializer(SerializationFormat.Json); | |||
_serializer = DiscordJsonSerializer.Global.CreateScope(); | |||
_serializer.Error += ex => | |||
{ | |||
_rpcLogger.WarningAsync("Serializer Error", ex).GetAwaiter().GetResult(); | |||
@@ -10,12 +10,12 @@ namespace Discord.Serialization | |||
} | |||
[AttributeUsage(AttributeTargets.Field, AllowMultiple = false)] | |||
public class ModelEnumAttribute : Attribute | |||
public class ModelEnumValueAttribute : Attribute | |||
{ | |||
public string Key { get; } | |||
public EnumValueType Type { get; } | |||
public ModelEnumAttribute(string key, EnumValueType type = EnumValueType.ReadWrite) | |||
public ModelEnumValueAttribute(string key, EnumValueType type = EnumValueType.ReadWrite) | |||
{ | |||
Key = key; | |||
Type = type; |
@@ -8,7 +8,9 @@ namespace Discord.Serialization | |||
public string Key { get; } | |||
public bool ExcludeNull { get; set; } | |||
public ModelPropertyAttribute(string key = null) | |||
public ModelPropertyAttribute() | |||
: this(null) { } | |||
public ModelPropertyAttribute(string key) | |||
{ | |||
Key = key; | |||
} |
@@ -0,0 +1,17 @@ | |||
using System; | |||
namespace Discord.Serialization | |||
{ | |||
[AttributeUsage(AttributeTargets.Property, AllowMultiple = true)] | |||
public class ModelSelectorAttribute : Attribute | |||
{ | |||
public string SelectorProperty { get; } | |||
public string SelectorGroup { get; } | |||
public ModelSelectorAttribute(string selectorProperty, string selectorGroup) | |||
{ | |||
SelectorProperty = selectorProperty; | |||
SelectorGroup = selectorGroup; | |||
} | |||
} | |||
} |
@@ -0,0 +1,9 @@ | |||
using System; | |||
namespace Discord.Serialization | |||
{ | |||
[AttributeUsage(AttributeTargets.Property, AllowMultiple = false)] | |||
public class ModelStringEnumAttribute : Attribute | |||
{ | |||
} | |||
} |
@@ -17,14 +17,20 @@ namespace Discord.Serialization | |||
private static readonly TypeInfo _serializerType = typeof(Serializer).GetTypeInfo(); | |||
private readonly Serializer _serializer; | |||
private readonly ConcurrentDictionary<Type, object> _cache = new ConcurrentDictionary<Type, object>(); | |||
private readonly Dictionary<Type, ConverterTypeCollection> _types = new Dictionary<Type, ConverterTypeCollection>(); | |||
private readonly Dictionary<Type, ConverterTypeCollection> _mappedGenericTypes = new Dictionary<Type, ConverterTypeCollection>(); | |||
private readonly ConverterTypeCollection _genericTypes = new ConverterTypeCollection(); | |||
private readonly ConcurrentDictionary<Type, object> _cache; | |||
private readonly Dictionary<Type, ConverterTypeCollection> _types; | |||
private readonly Dictionary<Type, ConverterTypeCollection> _mappedGenericTypes; | |||
private readonly ConverterTypeCollection _genericTypes; | |||
private readonly ConcurrentDictionary<Type, ConcurrentDictionary<string, ISelectorGroup>> _selectorGroups; | |||
internal ConverterCollection(Serializer serializer) | |||
{ | |||
_serializer = serializer; | |||
_cache = new ConcurrentDictionary<Type, object>(); | |||
_types = new Dictionary<Type, ConverterTypeCollection>(); | |||
_mappedGenericTypes = new Dictionary<Type, ConverterTypeCollection>(); | |||
_genericTypes = new ConverterTypeCollection(); | |||
_selectorGroups = new ConcurrentDictionary<Type, ConcurrentDictionary<string, ISelectorGroup>>(); | |||
} | |||
public void Add(Type type, Type converterType) | |||
@@ -66,15 +72,22 @@ namespace Discord.Serialization | |||
_mappedGenericTypes.Add(openType, converters = new ConverterTypeCollection()); | |||
converters.Conditionals.Add((condition, openConverterType)); | |||
} | |||
public void AddSelector(string groupKey, Type keyType, object keyValue, Type converterType) | |||
{ | |||
var group = GetSelectorGroup(keyType, groupKey); | |||
group.AddDynamicConverter(keyValue, converterType); | |||
} | |||
public object Get(Type type, PropertyInfo propInfo = null) | |||
{ | |||
if (!_cache.TryGetValue(type, out var result)) | |||
if (propInfo == null) //Can only cache top-level due to attribute influences | |||
{ | |||
object converter = Create(type, propInfo); | |||
result = _cache.GetOrAdd(type, converter); | |||
if (_cache.TryGetValue(type, out var result)) | |||
return result; | |||
return _cache.GetOrAdd(type, Create(type, propInfo)); | |||
} | |||
return result; | |||
return Create(type, propInfo); | |||
} | |||
private object Create(Type type, PropertyInfo propInfo) | |||
{ | |||
@@ -108,7 +121,7 @@ namespace Discord.Serialization | |||
converterType = converterType.MakeGenericType(type); | |||
var converterTypeInfo = converterType.GetTypeInfo(); | |||
var constructors = converterTypeInfo.DeclaredConstructors.ToArray(); | |||
var constructors = converterTypeInfo.DeclaredConstructors.Where(x => !x.IsStatic).ToArray(); | |||
if (constructors.Length == 0) | |||
throw new SerializationException($"{converterType.Name} is missing a constructor"); | |||
if (constructors.Length != 1) | |||
@@ -148,5 +161,17 @@ namespace Discord.Serialization | |||
return converters.DefaultConverterType; | |||
return null; | |||
} | |||
public ISelectorGroup GetSelectorGroup(Type keyType, string groupKey) | |||
{ | |||
var keyGroup = GetSelectorKeyGroup(keyType); | |||
if (keyGroup.TryGetValue(groupKey, out var selectorGroup)) | |||
return selectorGroup; | |||
return CreateSelectorGroup(keyType, keyGroup, groupKey); | |||
} | |||
private ISelectorGroup CreateSelectorGroup(Type keyType, ConcurrentDictionary<string, ISelectorGroup> keyGroup, string groupKey) | |||
=> keyGroup.GetOrAdd(groupKey, Activator.CreateInstance(typeof(SelectorGroup<>).MakeGenericType(keyType)) as ISelectorGroup); | |||
private ConcurrentDictionary<string, ISelectorGroup> GetSelectorKeyGroup(Type keyType) | |||
=> _selectorGroups.GetOrAdd(keyType, _ => new ConcurrentDictionary<string, ISelectorGroup>()); | |||
} | |||
} |
@@ -16,9 +16,12 @@ namespace Discord.Serialization | |||
{ | |||
public static readonly EnumMap<T> Instance = new EnumMap<T>(); | |||
private readonly BufferDictionary<T> _keyToValue; | |||
private readonly Dictionary<string, T> _keyToValue; | |||
private readonly BufferDictionary<T> _utf8KeyToValue; | |||
private readonly Dictionary<long, T> _intToValue; | |||
private readonly Dictionary<T, string> _valueToKey; | |||
private readonly Dictionary<T, ReadOnlyBuffer<byte>> _valueToUtf8Key; | |||
private readonly Dictionary<T, long> _valueToInt; | |||
public EnumMap() | |||
{ | |||
@@ -26,50 +29,104 @@ namespace Discord.Serialization | |||
if (!typeInfo.IsEnum) | |||
throw new InvalidOperationException($"{typeInfo.Name} is not an Enum"); | |||
_keyToValue = new BufferDictionary<T>(); | |||
_keyToValue = new Dictionary<string, T>(); | |||
_utf8KeyToValue = new BufferDictionary<T>(); | |||
_intToValue = new Dictionary<long, T>(); | |||
_valueToKey = new Dictionary<T, string>(); | |||
_valueToUtf8Key = new Dictionary<T, ReadOnlyBuffer<byte>>(); | |||
_valueToInt = new Dictionary<T, long>(); | |||
foreach (T val in Enum.GetValues(typeof(T)).OfType<T>()) | |||
{ | |||
var fieldInfo = typeInfo.GetDeclaredField(Enum.GetName(typeof(T), val)); | |||
var attr = fieldInfo.GetCustomAttribute<ModelEnumAttribute>(); | |||
var attr = fieldInfo.GetCustomAttribute<ModelEnumValueAttribute>(); | |||
if (attr != null) | |||
{ | |||
var key = new ReadOnlyBuffer<byte>(new Utf8String(attr.Key).Bytes.ToArray()); | |||
string key = attr.Key; | |||
var keyBuffer = new ReadOnlyBuffer<byte>(new Utf8String(attr.Key).Bytes.ToArray()); | |||
if (attr.Type != EnumValueType.WriteOnly) | |||
_keyToValue.Add(key, val); | |||
if (attr.Type != EnumValueType.ReadOnly) | |||
{ | |||
_valueToUtf8Key.Add(val, key); | |||
_valueToKey.Add(val, attr.Key); | |||
_keyToValue.Add(key, val); | |||
_utf8KeyToValue.Add(keyBuffer, val); | |||
} | |||
if (attr.Type != EnumValueType.ReadOnly) | |||
_valueToKey.Add(val, key); | |||
} | |||
var underlyingType = Enum.GetUnderlyingType(typeof(T)); | |||
long baseVal; | |||
if (underlyingType == typeof(sbyte)) | |||
baseVal = (sbyte)(ValueType)val; | |||
else if (underlyingType == typeof(short)) | |||
baseVal = (short)(ValueType)val; | |||
else if (underlyingType == typeof(int)) | |||
baseVal = (int)(ValueType)val; | |||
else if (underlyingType == typeof(long)) | |||
baseVal = (long)(ValueType)val; | |||
else if (underlyingType == typeof(byte)) | |||
baseVal = (byte)(ValueType)val; | |||
else if (underlyingType == typeof(ushort)) | |||
baseVal = (ushort)(ValueType)val; | |||
else if (underlyingType == typeof(uint)) | |||
baseVal = (uint)(ValueType)val; | |||
else if (underlyingType == typeof(ulong)) | |||
baseVal = (long)(ulong)(ValueType)val; | |||
else | |||
throw new SerializationException($"Unsupported underlying enum type: {underlyingType.Name}"); | |||
_intToValue.Add(baseVal, val); | |||
_valueToInt.Add(val, baseVal); | |||
} | |||
} | |||
public T GetValue(ReadOnlyBuffer<byte> key) | |||
public T FromKey(ReadOnlyBuffer<byte> key) | |||
{ | |||
if (_keyToValue.TryGetValue(key, out var value)) | |||
if (_utf8KeyToValue.TryGetValue(key, out var value)) | |||
return value; | |||
throw new SerializationException($"Unknown enum key: {new Utf8String(key.Span).ToString()}"); | |||
} | |||
public T GetValue(ReadOnlySpan<byte> key) | |||
public T FromKey(ReadOnlySpan<byte> key) | |||
{ | |||
if (_keyToValue.TryGetValue(key, out var value)) | |||
if (_utf8KeyToValue.TryGetValue(key, out var value)) | |||
return value; | |||
throw new SerializationException($"Unknown enum key: {new Utf8String(key).ToString()}"); | |||
} | |||
public string GetKey(T value) | |||
public string ToUtf16Key(T value) | |||
{ | |||
if (_valueToKey.TryGetValue(value, out var key)) | |||
return key; | |||
throw new SerializationException($"Unknown enum value: {value}"); | |||
} | |||
public ReadOnlyBuffer<byte> GetUtf8Key(T value) | |||
public T FromInt64(ReadOnlyBuffer<byte> intBuffer) | |||
=> FromInt64(intBuffer.Span); | |||
public T FromInt64(ReadOnlySpan<byte> intBuffer) | |||
{ | |||
if (_valueToUtf8Key.TryGetValue(value, out var key)) | |||
return key; | |||
long intValue = intBuffer.ParseInt64(); | |||
if (_intToValue.TryGetValue(intValue, out var value)) | |||
return value; | |||
throw new SerializationException($"Unknown enum value: {intValue}"); | |||
} | |||
public long ToInt64(T value) | |||
{ | |||
if (_valueToInt.TryGetValue(value, out var intValue)) | |||
return intValue; | |||
throw new SerializationException($"Unknown enum value: {value}"); | |||
} | |||
public T FromUInt64(ReadOnlyBuffer<byte> intBuffer) | |||
=> FromUInt64(intBuffer.Span); | |||
public T FromUInt64(ReadOnlySpan<byte> intBuffer) | |||
{ | |||
ulong intValue = intBuffer.ParseUInt64(); | |||
if (_intToValue.TryGetValue((long)intValue, out var value)) | |||
return value; | |||
throw new SerializationException($"Unknown enum value: {intValue}"); | |||
} | |||
public ulong ToUInt64(T value) | |||
{ | |||
if (_valueToInt.TryGetValue(value, out var intValue)) | |||
return (ulong)intValue; | |||
throw new SerializationException($"Unknown enum value: {value}"); | |||
} | |||
} | |||
@@ -29,7 +29,7 @@ namespace Discord.Serialization | |||
public static Guid ParseGuid(this JsonReader reader) => reader.Value.ParseGuid(); | |||
} | |||
public static class JsonUtils | |||
public static class JsonReaderUtils | |||
{ | |||
public static void Skip(ref JsonReader reader) | |||
{ | |||
@@ -3,7 +3,7 @@ using System.Text.Json; | |||
namespace Discord.Serialization.Json.Converters | |||
{ | |||
internal class ListPropertyConverter<T> : IJsonPropertyConverter<List<T>> | |||
public class ListPropertyConverter<T> : IJsonPropertyConverter<List<T>> | |||
{ | |||
private readonly IJsonPropertyConverter<T> _innerConverter; | |||
@@ -12,25 +12,25 @@ namespace Discord.Serialization.Json.Converters | |||
_innerConverter = innerConverter; | |||
} | |||
public List<T> Read(PropertyMap map, ref JsonReader reader, bool isTopLevel) | |||
public List<T> Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if ((isTopLevel && !reader.Read()) || reader.TokenType != JsonTokenType.StartArray) | |||
throw new SerializationException("Bad input, expected StartArray"); | |||
var list = new List<T>(); | |||
while (reader.Read() && reader.TokenType != JsonTokenType.EndArray) | |||
list.Add(_innerConverter.Read(map, ref reader, false)); | |||
list.Add(_innerConverter.Read(map, model, ref reader, false)); | |||
return list; | |||
} | |||
public void Write(PropertyMap map, ref JsonWriter writer, List<T> value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, List<T> value, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
writer.WriteArrayStart(map.Key); | |||
else | |||
writer.WriteArrayStart(); | |||
for (int i = 0; i < value.Count; i++) | |||
_innerConverter.Write(map, ref writer, value[i], false); | |||
_innerConverter.Write(map, model, ref writer, value[i], false); | |||
writer.WriteArrayEnd(); | |||
} | |||
} | |||
@@ -0,0 +1,35 @@ | |||
using System.Text.Json; | |||
namespace Discord.Serialization.Json.Converters | |||
{ | |||
//TODO: Only supports cases where the key arrives first | |||
public class DynamicPropertyConverter : IJsonPropertyConverter<object> | |||
{ | |||
public object Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (map.GetDynamicConverter(model, false) is IJsonPropertyReader<object> converter) | |||
return converter.Read(map, model, ref reader, isTopLevel); | |||
else | |||
{ | |||
JsonReaderUtils.Skip(ref reader); | |||
return null; | |||
} | |||
} | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, object value, bool isTopLevel) | |||
{ | |||
if (value == null) | |||
{ | |||
if (isTopLevel) | |||
writer.WriteAttributeNull(map.Key); | |||
else | |||
writer.WriteNull(); | |||
} | |||
else | |||
{ | |||
var converter = map.GetDynamicConverter(model, true) as IJsonPropertyWriter<object>; | |||
converter.Write(map, model, ref writer, value, isTopLevel); | |||
} | |||
} | |||
} | |||
} |
@@ -2,22 +2,68 @@ | |||
namespace Discord.Serialization.Json.Converters | |||
{ | |||
internal class EnumPropertyConverter<T> : IJsonPropertyConverter<T> | |||
public class Int64EnumPropertyConverter<T> : IJsonPropertyConverter<T> | |||
where T : struct | |||
{ | |||
private static readonly EnumMap<T> _map = EnumMap.For<T>(); | |||
public T Read(PropertyMap map, ref JsonReader reader, bool isTopLevel) | |||
public T Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
reader.Read(); | |||
if (reader.ValueType != JsonValueType.Number && reader.ValueType != JsonValueType.String) | |||
throw new SerializationException("Bad input, expected Number or String"); | |||
return _map.FromInt64(reader.Value); | |||
} | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, T value, bool isTopLevel) | |||
{ | |||
long key = _map.ToInt64(value); | |||
if (isTopLevel) | |||
writer.WriteAttribute(map.Key, key); | |||
else | |||
writer.WriteValue(key); | |||
} | |||
} | |||
public class UInt64EnumPropertyConverter<T> : IJsonPropertyConverter<T> | |||
where T : struct | |||
{ | |||
private static readonly EnumMap<T> _map = EnumMap.For<T>(); | |||
public T Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
reader.Read(); | |||
if (reader.ValueType != JsonValueType.Number && reader.ValueType != JsonValueType.String) | |||
throw new SerializationException("Bad input, expected Number or String"); | |||
return _map.FromUInt64(reader.Value); | |||
} | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, T value, bool isTopLevel) | |||
{ | |||
ulong key = _map.ToUInt64(value); | |||
if (isTopLevel) | |||
writer.WriteAttribute(map.Key, key); | |||
else | |||
writer.WriteValue(key); | |||
} | |||
} | |||
public class StringEnumPropertyConverter<T> : IJsonPropertyConverter<T> | |||
where T : struct | |||
{ | |||
private static readonly EnumMap<T> _map = EnumMap.For<T>(); | |||
public T Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
reader.Read(); | |||
if (reader.ValueType != JsonValueType.String) | |||
throw new SerializationException("Bad input, expected String"); | |||
return _map.GetValue(reader.Value); | |||
return _map.FromKey(reader.Value); | |||
} | |||
public void Write(PropertyMap map, ref JsonWriter writer, T value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, T value, bool isTopLevel) | |||
{ | |||
string key = _map.GetKey(value); | |||
string key = _map.ToUtf16Key(value); | |||
if (isTopLevel) | |||
writer.WriteAttribute(map.Key, key); | |||
else | |||
@@ -2,7 +2,7 @@ | |||
namespace Discord.Serialization.Json.Converters | |||
{ | |||
internal class NullablePropertyConverter<T> : IJsonPropertyConverter<T?> | |||
public class NullablePropertyConverter<T> : IJsonPropertyConverter<T?> | |||
where T : struct | |||
{ | |||
private readonly IJsonPropertyConverter<T> _innerConverter; | |||
@@ -12,19 +12,19 @@ namespace Discord.Serialization.Json.Converters | |||
_innerConverter = innerConverter; | |||
} | |||
public T? Read(PropertyMap map, ref JsonReader reader, bool isTopLevel) | |||
public T? Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
reader.Read(); | |||
if (reader.ValueType == JsonValueType.Null) | |||
return null; | |||
return _innerConverter.Read(map, ref reader, false); | |||
return _innerConverter.Read(map, model, ref reader, false); | |||
} | |||
public void Write(PropertyMap map, ref JsonWriter writer, T? value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, T? value, bool isTopLevel) | |||
{ | |||
if (value.HasValue) | |||
_innerConverter.Write(map, ref writer, value.Value, isTopLevel); | |||
_innerConverter.Write(map, model, ref writer, value.Value, isTopLevel); | |||
else | |||
{ | |||
if (isTopLevel) | |||
@@ -2,7 +2,7 @@ | |||
namespace Discord.Serialization.Json.Converters | |||
{ | |||
internal class ObjectPropertyConverter<T> : IJsonPropertyConverter<T> | |||
public class ObjectPropertyConverter<T> : IJsonPropertyConverter<T> | |||
where T : class, new() | |||
{ | |||
private readonly ModelMap<T> _map; | |||
@@ -12,27 +12,27 @@ namespace Discord.Serialization.Json.Converters | |||
_map = serializer.MapModel<T>(); | |||
} | |||
public T Read(PropertyMap map, ref JsonReader reader, bool isTopLevel) | |||
public T Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
var model = new T(); | |||
var subModel = new T(); | |||
if ((isTopLevel && !reader.Read()) || reader.TokenType != JsonTokenType.StartObject) | |||
throw new SerializationException("Bad input, expected StartObject"); | |||
while (reader.Read()) | |||
{ | |||
if (reader.TokenType == JsonTokenType.EndObject) | |||
return model; | |||
return subModel; | |||
if (reader.TokenType != JsonTokenType.PropertyName) | |||
throw new SerializationException("Bad input, expected PropertyName"); | |||
if (_map.TryGetProperty(reader.Value, out var property)) | |||
(property as IJsonPropertyMap<T>).Read(model, ref reader); | |||
(property as IJsonPropertyMap<T>).Read(subModel, ref reader); | |||
else | |||
JsonUtils.Skip(ref reader); //Unknown property, skip | |||
JsonReaderUtils.Skip(ref reader); //Unknown property, skip | |||
} | |||
throw new SerializationException("Bad input, expected EndObject"); | |||
} | |||
public void Write(PropertyMap map, ref JsonWriter writer, T value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, T value, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
writer.WriteObjectStart(map.Key); | |||
@@ -3,9 +3,9 @@ using System.Text.Json; | |||
namespace Discord.Serialization.Json.Converters | |||
{ | |||
internal class DateTimePropertyConverter : IJsonPropertyConverter<DateTime> | |||
public class DateTimePropertyConverter : IJsonPropertyConverter<DateTime> | |||
{ | |||
public DateTime Read(PropertyMap map, ref JsonReader reader, bool isTopLevel) | |||
public DateTime Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
reader.Read(); | |||
@@ -13,7 +13,7 @@ namespace Discord.Serialization.Json.Converters | |||
throw new SerializationException("Bad input, expected String"); | |||
return reader.ParseDateTime(); | |||
} | |||
public void Write(PropertyMap map, ref JsonWriter writer, DateTime value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, DateTime value, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
writer.WriteAttribute(map.Key, value); | |||
@@ -22,9 +22,9 @@ namespace Discord.Serialization.Json.Converters | |||
} | |||
} | |||
internal class DateTimeOffsetPropertyConverter : IJsonPropertyConverter<DateTimeOffset> | |||
public class DateTimeOffsetPropertyConverter : IJsonPropertyConverter<DateTimeOffset> | |||
{ | |||
public DateTimeOffset Read(PropertyMap map, ref JsonReader reader, bool isTopLevel) | |||
public DateTimeOffset Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
reader.Read(); | |||
@@ -32,7 +32,7 @@ namespace Discord.Serialization.Json.Converters | |||
throw new SerializationException("Bad input, expected String"); | |||
return reader.ParseDateTimeOffset(); | |||
} | |||
public void Write(PropertyMap map, ref JsonWriter writer, DateTimeOffset value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, DateTimeOffset value, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
writer.WriteAttribute(map.Key, value); | |||
@@ -2,9 +2,9 @@ | |||
namespace Discord.Serialization.Json.Converters | |||
{ | |||
internal class SinglePropertyConverter : IJsonPropertyConverter<float> | |||
public class SinglePropertyConverter : IJsonPropertyConverter<float> | |||
{ | |||
public float Read(PropertyMap map, ref JsonReader reader, bool isTopLevel) | |||
public float Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
reader.Read(); | |||
@@ -12,7 +12,7 @@ namespace Discord.Serialization.Json.Converters | |||
throw new SerializationException("Bad input, expected Number or String"); | |||
return reader.ParseSingle(); | |||
} | |||
public void Write(PropertyMap map, ref JsonWriter writer, float value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, float value, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
writer.WriteAttribute(map.Key, value.ToString()); | |||
@@ -21,9 +21,9 @@ namespace Discord.Serialization.Json.Converters | |||
} | |||
} | |||
internal class DoublePropertyConverter : IJsonPropertyConverter<double> | |||
public class DoublePropertyConverter : IJsonPropertyConverter<double> | |||
{ | |||
public double Read(PropertyMap map, ref JsonReader reader, bool isTopLevel) | |||
public double Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
reader.Read(); | |||
@@ -31,7 +31,7 @@ namespace Discord.Serialization.Json.Converters | |||
throw new SerializationException("Bad input, expected Number or String"); | |||
return reader.ParseDouble(); | |||
} | |||
public void Write(PropertyMap map, ref JsonWriter writer, double value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, double value, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
writer.WriteAttribute(map.Key, value.ToString()); | |||
@@ -42,7 +42,7 @@ namespace Discord.Serialization.Json.Converters | |||
internal class DecimalPropertyConverter : IJsonPropertyConverter<decimal> | |||
{ | |||
public decimal Read(PropertyMap map, ref JsonReader reader, bool isTopLevel) | |||
public decimal Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
reader.Read(); | |||
@@ -50,7 +50,7 @@ namespace Discord.Serialization.Json.Converters | |||
throw new SerializationException("Bad input, expected Number or String"); | |||
return reader.ParseDecimal(); | |||
} | |||
public void Write(PropertyMap map, ref JsonWriter writer, decimal value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, decimal value, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
writer.WriteAttribute(map.Key, value.ToString()); | |||
@@ -3,9 +3,9 @@ using System.Text.Json; | |||
namespace Discord.Serialization.Json.Converters | |||
{ | |||
internal class BooleanPropertyConverter : IJsonPropertyConverter<bool> | |||
public class BooleanPropertyConverter : IJsonPropertyConverter<bool> | |||
{ | |||
public bool Read(PropertyMap map, ref JsonReader reader, bool isTopLevel) | |||
public bool Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
reader.Read(); | |||
@@ -16,7 +16,7 @@ namespace Discord.Serialization.Json.Converters | |||
default: throw new SerializationException("Bad input, expected False or True"); | |||
} | |||
} | |||
public void Write(PropertyMap map, ref JsonWriter writer, bool value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, bool value, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
writer.WriteAttribute(map.Key, value); | |||
@@ -25,9 +25,9 @@ namespace Discord.Serialization.Json.Converters | |||
} | |||
} | |||
internal class GuidPropertyConverter : IJsonPropertyConverter<Guid> | |||
public class GuidPropertyConverter : IJsonPropertyConverter<Guid> | |||
{ | |||
public Guid Read(PropertyMap map, ref JsonReader reader, bool isTopLevel) | |||
public Guid Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
reader.Read(); | |||
@@ -35,7 +35,7 @@ namespace Discord.Serialization.Json.Converters | |||
throw new SerializationException("Bad input, expected String"); | |||
return reader.ParseGuid(); | |||
} | |||
public void Write(PropertyMap map, ref JsonWriter writer, Guid value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, Guid value, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
writer.WriteAttribute(map.Key, value.ToString()); | |||
@@ -2,9 +2,9 @@ | |||
namespace Discord.Serialization.Json.Converters | |||
{ | |||
internal class Int8PropertyConverter : IJsonPropertyConverter<sbyte> | |||
public class Int8PropertyConverter : IJsonPropertyConverter<sbyte> | |||
{ | |||
public sbyte Read(PropertyMap map, ref JsonReader reader, bool isTopLevel) | |||
public sbyte Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
reader.Read(); | |||
@@ -12,7 +12,7 @@ namespace Discord.Serialization.Json.Converters | |||
throw new SerializationException("Bad input, expected Number or String"); | |||
return reader.ParseInt8(); | |||
} | |||
public void Write(PropertyMap map, ref JsonWriter writer, sbyte value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, sbyte value, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
writer.WriteAttribute(map.Key, value); | |||
@@ -21,9 +21,9 @@ namespace Discord.Serialization.Json.Converters | |||
} | |||
} | |||
internal class Int16PropertyConverter : IJsonPropertyConverter<short> | |||
public class Int16PropertyConverter : IJsonPropertyConverter<short> | |||
{ | |||
public short Read(PropertyMap map, ref JsonReader reader, bool isTopLevel) | |||
public short Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
reader.Read(); | |||
@@ -31,7 +31,7 @@ namespace Discord.Serialization.Json.Converters | |||
throw new SerializationException("Bad input, expected Number or String"); | |||
return reader.ParseInt16(); | |||
} | |||
public void Write(PropertyMap map, ref JsonWriter writer, short value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, short value, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
writer.WriteAttribute(map.Key, value); | |||
@@ -40,9 +40,9 @@ namespace Discord.Serialization.Json.Converters | |||
} | |||
} | |||
internal class Int32PropertyConverter : IJsonPropertyConverter<int> | |||
public class Int32PropertyConverter : IJsonPropertyConverter<int> | |||
{ | |||
public int Read(PropertyMap map, ref JsonReader reader, bool isTopLevel) | |||
public int Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
reader.Read(); | |||
@@ -50,7 +50,7 @@ namespace Discord.Serialization.Json.Converters | |||
throw new SerializationException("Bad input, expected Number or String"); | |||
return reader.ParseInt32(); | |||
} | |||
public void Write(PropertyMap map, ref JsonWriter writer, int value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, int value, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
writer.WriteAttribute(map.Key, value); | |||
@@ -59,9 +59,9 @@ namespace Discord.Serialization.Json.Converters | |||
} | |||
} | |||
internal class Int64PropertyConverter : IJsonPropertyConverter<long> | |||
public class Int64PropertyConverter : IJsonPropertyConverter<long> | |||
{ | |||
public long Read(PropertyMap map, ref JsonReader reader, bool isTopLevel) | |||
public long Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
reader.Read(); | |||
@@ -69,7 +69,7 @@ namespace Discord.Serialization.Json.Converters | |||
throw new SerializationException("Bad input, expected Number or String"); | |||
return reader.ParseInt64(); | |||
} | |||
public void Write(PropertyMap map, ref JsonWriter writer, long value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, long value, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
writer.WriteAttribute(map.Key, value.ToString()); | |||
@@ -1,11 +1,10 @@ | |||
using System.Text.Json; | |||
using System.Text.Utf8; | |||
namespace Discord.Serialization.Json.Converters | |||
{ | |||
/*internal class CharPropertyConverter : IJsonPropertyConverter<char> | |||
/*public class CharPropertyConverter : IJsonPropertyConverter<char> | |||
{ | |||
public char Read(PropertyMap map, JsonReader reader, bool isTopLevel) | |||
public char Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
reader.Read(); | |||
@@ -13,7 +12,7 @@ namespace Discord.Serialization.Json.Converters | |||
throw new SerializationException("Bad input, expected String"); | |||
return reader.ParseChar(); | |||
} | |||
public void Write(PropertyMap map, JsonWriter writer, char value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, char value, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
writer.WriteAttribute(map.Key, value); | |||
@@ -22,17 +21,19 @@ namespace Discord.Serialization.Json.Converters | |||
} | |||
}*/ | |||
internal class StringPropertyConverter : IJsonPropertyConverter<string> | |||
public class StringPropertyConverter : IJsonPropertyConverter<string> | |||
{ | |||
public string Read(PropertyMap map, ref JsonReader reader, bool isTopLevel) | |||
public string Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
reader.Read(); | |||
if (reader.ValueType != JsonValueType.String) | |||
if (reader.ValueType == JsonValueType.Null) | |||
return null; | |||
else if (reader.ValueType != JsonValueType.String) | |||
throw new SerializationException("Bad input, expected String"); | |||
return reader.ParseString(); | |||
} | |||
public void Write(PropertyMap map, ref JsonWriter writer, string value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, string value, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
writer.WriteAttribute(map.Key, value); | |||
@@ -2,9 +2,9 @@ | |||
namespace Discord.Serialization.Json.Converters | |||
{ | |||
internal class UInt8PropertyConverter : IJsonPropertyConverter<byte> | |||
public class UInt8PropertyConverter : IJsonPropertyConverter<byte> | |||
{ | |||
public byte Read(PropertyMap map, ref JsonReader reader, bool isTopLevel) | |||
public byte Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
reader.Read(); | |||
@@ -12,7 +12,7 @@ namespace Discord.Serialization.Json.Converters | |||
throw new SerializationException("Bad input, expected Number or String"); | |||
return reader.ParseUInt8(); | |||
} | |||
public void Write(PropertyMap map, ref JsonWriter writer, byte value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, byte value, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
writer.WriteAttribute(map.Key, value); | |||
@@ -21,9 +21,9 @@ namespace Discord.Serialization.Json.Converters | |||
} | |||
} | |||
internal class UInt16PropertyConverter : IJsonPropertyConverter<ushort> | |||
public class UInt16PropertyConverter : IJsonPropertyConverter<ushort> | |||
{ | |||
public ushort Read(PropertyMap map, ref JsonReader reader, bool isTopLevel) | |||
public ushort Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
reader.Read(); | |||
@@ -31,7 +31,7 @@ namespace Discord.Serialization.Json.Converters | |||
throw new SerializationException("Bad input, expected Number or String"); | |||
return reader.ParseUInt16(); | |||
} | |||
public void Write(PropertyMap map, ref JsonWriter writer, ushort value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, ushort value, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
writer.WriteAttribute(map.Key, value); | |||
@@ -40,9 +40,9 @@ namespace Discord.Serialization.Json.Converters | |||
} | |||
} | |||
internal class UInt32PropertyConverter : IJsonPropertyConverter<uint> | |||
public class UInt32PropertyConverter : IJsonPropertyConverter<uint> | |||
{ | |||
public uint Read(PropertyMap map, ref JsonReader reader, bool isTopLevel) | |||
public uint Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
reader.Read(); | |||
@@ -50,7 +50,7 @@ namespace Discord.Serialization.Json.Converters | |||
throw new SerializationException("Bad input, expected Number or String"); | |||
return reader.ParseUInt32(); | |||
} | |||
public void Write(PropertyMap map, ref JsonWriter writer, uint value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, uint value, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
writer.WriteAttribute(map.Key, value); | |||
@@ -59,9 +59,9 @@ namespace Discord.Serialization.Json.Converters | |||
} | |||
} | |||
internal class UInt64PropertyConverter : IJsonPropertyConverter<ulong> | |||
public class UInt64PropertyConverter : IJsonPropertyConverter<ulong> | |||
{ | |||
public ulong Read(PropertyMap map, ref JsonReader reader, bool isTopLevel) | |||
public ulong Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
reader.Read(); | |||
@@ -69,7 +69,7 @@ namespace Discord.Serialization.Json.Converters | |||
throw new SerializationException("Bad input, expected Number or String"); | |||
return reader.ParseUInt64(); | |||
} | |||
public void Write(PropertyMap map, ref JsonWriter writer, ulong value, bool isTopLevel) | |||
public void Write(PropertyMap map, object model, ref JsonWriter writer, ulong value, bool isTopLevel) | |||
{ | |||
if (isTopLevel) | |||
writer.WriteAttribute(map.Key, value.ToString()); | |||
@@ -2,9 +2,14 @@ | |||
namespace Discord.Serialization.Json | |||
{ | |||
public interface IJsonPropertyConverter<T> | |||
public interface IJsonPropertyConverter<T> : IJsonPropertyReader<T>, IJsonPropertyWriter<T> { } | |||
public interface IJsonPropertyReader<out T> | |||
{ | |||
T Read(PropertyMap map, object model, ref JsonReader reader, bool isTopLevel); | |||
} | |||
public interface IJsonPropertyWriter<in T> | |||
{ | |||
T Read(PropertyMap map, ref JsonReader reader, bool isTopLevel); | |||
void Write(PropertyMap map, ref JsonWriter writer, T value, bool isTopLevel); | |||
void Write(PropertyMap map, object model, ref JsonWriter writer, T value, bool isTopLevel); | |||
} | |||
} |
@@ -1,14 +0,0 @@ | |||
using System; | |||
using System.Text.Json; | |||
namespace Discord.Serialization | |||
{ | |||
internal interface IJsonPropertyMap<TModel> | |||
{ | |||
string Key { get; } | |||
ReadOnlyBuffer<byte> Utf8Key { get; } | |||
void Write(TModel model, ref JsonWriter writer); | |||
void Read(TModel model, ref JsonReader reader); | |||
} | |||
} |
@@ -4,19 +4,28 @@ using System.Text.Json; | |||
namespace Discord.Serialization.Json | |||
{ | |||
internal class JsonPropertyMap<TModel, TType> : PropertyMap, IJsonPropertyMap<TModel> | |||
internal interface IJsonPropertyMap<TModel> | |||
{ | |||
private readonly IJsonPropertyConverter<TType> _converter; | |||
private readonly Func<TModel, TType> _getFunc; | |||
private readonly Action<TModel, TType> _setFunc; | |||
string Key { get; } | |||
ReadOnlyBuffer<byte> Utf8Key { get; } | |||
public JsonPropertyMap(PropertyInfo propInfo, IJsonPropertyConverter<TType> converter) | |||
: base(propInfo) | |||
void Write(TModel model, ref JsonWriter writer); | |||
void Read(TModel model, ref JsonReader reader); | |||
} | |||
internal class JsonPropertyMap<TModel, TValue> : PropertyMap<TModel, TValue>, IJsonPropertyMap<TModel> | |||
{ | |||
private readonly IJsonPropertyConverter<TValue> _converter; | |||
private readonly Func<TModel, TValue> _getFunc; | |||
private readonly Action<TModel, TValue> _setFunc; | |||
public JsonPropertyMap(Serializer serializer, PropertyInfo propInfo, IJsonPropertyConverter<TValue> converter) | |||
: base(serializer, propInfo) | |||
{ | |||
_converter = converter; | |||
_getFunc = propInfo.GetMethod.CreateDelegate(typeof(Func<TModel, TType>)) as Func<TModel, TType>; | |||
_setFunc = propInfo.SetMethod.CreateDelegate(typeof(Action<TModel, TType>)) as Action<TModel, TType>; | |||
_getFunc = propInfo.GetMethod.CreateDelegate(typeof(Func<TModel, TValue>)) as Func<TModel, TValue>; | |||
_setFunc = propInfo.SetMethod.CreateDelegate(typeof(Action<TModel, TValue>)) as Action<TModel, TValue>; | |||
} | |||
public void Write(TModel model, ref JsonWriter writer) | |||
@@ -24,11 +33,11 @@ namespace Discord.Serialization.Json | |||
var value = _getFunc(model); | |||
if (value == null && ExcludeNull) | |||
return; | |||
_converter.Write(this, ref writer, value, true); | |||
_converter.Write(this, model, ref writer, value, true); | |||
} | |||
public void Read(TModel model, ref JsonReader reader) | |||
{ | |||
var value = _converter.Read(this, ref reader, true); | |||
var value = _converter.Read(this, model, ref reader, true); | |||
_setFunc(model, value); | |||
} | |||
} | |||
@@ -1,5 +1,6 @@ | |||
using System; | |||
using System.Collections.Generic; | |||
using System.Linq; | |||
using System.Reflection; | |||
using System.Text; | |||
using System.Text.Formatting; | |||
@@ -28,7 +29,7 @@ namespace Discord.Serialization.Json | |||
AddConverter<double, Converters.DoublePropertyConverter>(); | |||
AddConverter<decimal, Converters.DecimalPropertyConverter>(); | |||
//AddConverter<char, Converters.CharPropertyConverter>(); //char.Parse does not support Json.Net's serialization | |||
//AddConverter<char, Converters.CharPropertyConverter>(); //TODO: char.Parse does not support Json.Net's serialization | |||
AddConverter<string, Converters.StringPropertyConverter>(); | |||
AddConverter<DateTime, Converters.DateTimePropertyConverter>(); | |||
@@ -37,11 +38,20 @@ namespace Discord.Serialization.Json | |||
AddConverter<bool, Converters.BooleanPropertyConverter>(); | |||
AddConverter<Guid, Converters.GuidPropertyConverter>(); | |||
AddConverter<object, Converters.DynamicPropertyConverter>( | |||
(type, prop) => prop.GetCustomAttributes<ModelSelectorAttribute>().Any()); | |||
AddGenericConverter(typeof(List<>), typeof(Converters.ListPropertyConverter<>)); | |||
AddGenericConverter(typeof(Nullable<>), typeof(Converters.NullablePropertyConverter<>)); | |||
AddGenericConverter(typeof(Converters.EnumPropertyConverter<>), (type, prop) => type.IsEnum); | |||
AddGenericConverter(typeof(Converters.ObjectPropertyConverter<>), (type, prop) => type.IsClass); | |||
AddGenericConverter(typeof(Converters.StringEnumPropertyConverter<>), | |||
(type, prop) => type.IsEnum && prop.GetCustomAttribute<ModelStringEnumAttribute>() != null); | |||
AddGenericConverter(typeof(Converters.Int64EnumPropertyConverter<>), | |||
(type, prop) => type.IsEnum && IsSignedEnum(Enum.GetUnderlyingType(type.AsType()))); | |||
AddGenericConverter(typeof(Converters.UInt64EnumPropertyConverter<>), | |||
(type, prop) => type.IsEnum && IsUnsignedEnum(Enum.GetUnderlyingType(type.AsType()))); | |||
AddGenericConverter(typeof(Converters.ObjectPropertyConverter<>), | |||
(type, prop) => type.IsClass); | |||
} | |||
protected JsonSerializer(JsonSerializer parent) : base(parent) { } | |||
public JsonSerializer CreateScope() => new JsonSerializer(this); | |||
@@ -56,22 +66,33 @@ namespace Discord.Serialization.Json | |||
protected override PropertyMap CreatePropertyMap<TModel, TValue>(PropertyInfo propInfo) | |||
{ | |||
var converter = (IJsonPropertyConverter<TValue>)GetConverter(typeof(TValue), propInfo); | |||
return new JsonPropertyMap<TModel, TValue>(propInfo, converter); | |||
return new JsonPropertyMap<TModel, TValue>(this, propInfo, converter); | |||
} | |||
public override TModel Read<TModel>(ReadOnlyBuffer<byte> data) | |||
{ | |||
var reader = new JsonReader(data.Span, SymbolTable.InvariantUtf8); | |||
if (!reader.Read()) | |||
return null; | |||
return default; | |||
var converter = GetConverter(typeof(TModel)) as IJsonPropertyConverter<TModel>; | |||
return converter.Read(null, ref reader, false); | |||
return converter.Read(null, null, ref reader, false); | |||
} | |||
public override void Write<TModel>(ArrayFormatter stream, TModel model) | |||
{ | |||
var writer = new JsonWriter(stream); | |||
var converter = GetConverter(typeof(TModel)) as IJsonPropertyConverter<TModel>; | |||
converter.Write(null, ref writer, model, false); | |||
converter.Write(null, null, ref writer, model, false); | |||
} | |||
private static bool IsSignedEnum(Type underlyingType) | |||
=> underlyingType == typeof(sbyte) || | |||
underlyingType == typeof(short) || | |||
underlyingType == typeof(int) || | |||
underlyingType == typeof(long); | |||
private static bool IsUnsignedEnum(Type underlyingType) | |||
=> underlyingType == typeof(byte) || | |||
underlyingType == typeof(ushort) || | |||
underlyingType == typeof(uint) || | |||
underlyingType == typeof(ulong); | |||
} | |||
} |
@@ -1,24 +1,27 @@ | |||
using System; | |||
using System.Collections.Generic; | |||
using System.Linq; | |||
using System.Reflection; | |||
namespace Discord.Serialization | |||
{ | |||
public class ModelMap<TModel> | |||
where TModel : class, new() | |||
{ | |||
private BufferDictionary<PropertyMap> _dictionary; | |||
private BufferDictionary<PropertyMap> _propDict; | |||
public bool HasDynamics { get; } | |||
public PropertyMap[] Properties { get; } | |||
public ModelMap(List<PropertyMap> properties) | |||
public ModelMap(Serializer serializer, TypeInfo type, List<PropertyMap> properties) | |||
{ | |||
Properties = properties.ToArray(); | |||
_dictionary = new BufferDictionary<PropertyMap>(properties.ToDictionary(x => x.Utf8Key)); | |||
_propDict = new BufferDictionary<PropertyMap>(properties.ToDictionary(x => x.Utf8Key)); | |||
} | |||
public bool TryGetProperty(ReadOnlyBuffer<byte> key, out PropertyMap value) | |||
=> _dictionary.TryGetValue(key, out value); | |||
=> _propDict.TryGetValue(key, out value); | |||
public bool TryGetProperty(ReadOnlySpan<byte> key, out PropertyMap value) | |||
=> _dictionary.TryGetValue(key, out value); | |||
=> _propDict.TryGetValue(key, out value); | |||
} | |||
} |
@@ -1,4 +1,6 @@ | |||
using System; | |||
using System.Collections.Generic; | |||
using System.Linq; | |||
using System.Reflection; | |||
using System.Text.Utf8; | |||
@@ -8,15 +10,83 @@ namespace Discord.Serialization | |||
{ | |||
public string Key { get; } | |||
public ReadOnlyBuffer<byte> Utf8Key { get; } | |||
public string Name { get; } | |||
public bool ExcludeNull { get; } | |||
public PropertyMap(PropertyInfo propInfo) | |||
public PropertyMap(Serializer serializer, PropertyInfo propInfo) | |||
{ | |||
var jsonProperty = propInfo.GetCustomAttribute<ModelPropertyAttribute>(); | |||
Name = propInfo.Name; | |||
Key = jsonProperty?.Key ?? propInfo.Name; | |||
var attr = propInfo.GetCustomAttribute<ModelPropertyAttribute>(); | |||
Key = attr.Key ?? propInfo.Name; | |||
Utf8Key = new ReadOnlyBuffer<byte>(new Utf8String(Key).Bytes.ToArray()); | |||
ExcludeNull = jsonProperty?.ExcludeNull ?? false; | |||
ExcludeNull = attr.ExcludeNull; | |||
} | |||
public abstract object GetDynamicConverter(object model, bool throwOnMissing); | |||
} | |||
public abstract class PropertyMap<TModel, TValue> : PropertyMap | |||
{ | |||
private class Selector | |||
{ | |||
private static readonly MethodInfo _getSelectorKeyFunc | |||
= typeof(Selector).GetTypeInfo().GetDeclaredMethod(nameof(GetSelectorKey)); | |||
private readonly ISelectorGroup _group; | |||
private readonly Delegate _getSelectorFunc; | |||
private readonly Delegate _getWrappedSelectorFunc; | |||
public Selector(PropertyInfo prop, ISelectorGroup group, Delegate getSelectorFunc) | |||
{ | |||
_group = group; | |||
_getSelectorFunc = getSelectorFunc; | |||
var funcType = typeof(Func<,>).MakeGenericType(typeof(object), prop.PropertyType); | |||
_getWrappedSelectorFunc = _getSelectorKeyFunc.MakeGenericMethod(prop.PropertyType).CreateDelegate(funcType, this); | |||
} | |||
private TKey GetSelectorKey<TKey>(object model) | |||
=> (_getSelectorFunc as Func<TModel, TKey>)((TModel)model); | |||
public object GetDynamicConverter(object model) | |||
=> _group.GetDynamicConverter(_getWrappedSelectorFunc, model); | |||
} | |||
private readonly Delegate _getSelectorFunc, _getWrappedSelectorFunc; | |||
private readonly IReadOnlyList<Selector> _selectors; | |||
public PropertyMap(Serializer serializer, PropertyInfo propInfo) | |||
: base(serializer, propInfo) | |||
{ | |||
_selectors = propInfo.GetCustomAttributes<ModelSelectorAttribute>() | |||
.Select(x => | |||
{ | |||
var prop = typeof(TModel).GetTypeInfo().DeclaredProperties.FirstOrDefault(y => y.Name == x.SelectorProperty); | |||
if (prop == null) | |||
throw new SerializationException($"Selector key \"{x.SelectorProperty}\" not found"); | |||
var selectorGroup = serializer.GetSelectorGroup(prop.PropertyType, x.SelectorGroup); | |||
var funcType = typeof(Func<,>).MakeGenericType(typeof(TModel), prop.PropertyType); | |||
var selectorFunc = prop.GetMethod.CreateDelegate(funcType); | |||
return new Selector(prop, selectorGroup, selectorFunc); | |||
}) | |||
.ToList(); | |||
} | |||
public override object GetDynamicConverter(object model, bool throwOnMissing) | |||
{ | |||
for (int i = 0; i < _selectors.Count; i++) | |||
{ | |||
object converter = _selectors[i].GetDynamicConverter(model); | |||
if (converter != null) | |||
return converter; | |||
} | |||
if (throwOnMissing) | |||
throw new SerializationException($"Unable to find a converter for {Name}."); | |||
return null; | |||
} | |||
} | |||
} |
@@ -0,0 +1,40 @@ | |||
using System; | |||
using System.Collections.Concurrent; | |||
namespace Discord.Serialization | |||
{ | |||
public interface ISelectorGroup | |||
{ | |||
Type Type { get; } | |||
void AddDynamicConverter(object key, object converter); | |||
object GetDynamicConverter(Delegate getKeyFunc, object model); | |||
} | |||
internal class SelectorGroup<TKey> : ISelectorGroup | |||
{ | |||
private ConcurrentDictionary<TKey, object> _mapping; | |||
public Type Type => typeof(TKey); | |||
public SelectorGroup() | |||
{ | |||
_mapping = new ConcurrentDictionary<TKey, object>(); | |||
} | |||
public void AddDynamicConverter(object key, object converter) | |||
=> _mapping[(TKey)key] = converter; | |||
public object GetDynamicConverter(Delegate getKeyFunc, object model) | |||
{ | |||
var keyFunc = getKeyFunc as Func<object, TKey>; | |||
var key = keyFunc(model); | |||
if (key == null) | |||
return null; | |||
if (_mapping.TryGetValue(key, out var converter)) | |||
return converter; | |||
return null; | |||
} | |||
} | |||
} |
@@ -32,41 +32,49 @@ namespace Discord.Serialization | |||
IsScoped = true; | |||
} | |||
protected object GetConverter(Type type, PropertyInfo propInfo = null) | |||
=> _converters.Get(type, propInfo); | |||
protected object GetConverter(Type valueType, PropertyInfo propInfo = null) | |||
=> _converters.Get(valueType, propInfo); | |||
public void AddConverter(Type type, Type converter) | |||
public void AddConverter(Type valueType, Type converterType) | |||
{ | |||
CheckScoped(); | |||
_converters.Add(type, converter); | |||
_converters.Add(valueType, converterType); | |||
} | |||
public void AddConverter(Type type, Type converter, Func<TypeInfo, PropertyInfo, bool> condition) | |||
public void AddConverter(Type valueType, Type converterType, Func<TypeInfo, PropertyInfo, bool> condition) | |||
{ | |||
CheckScoped(); | |||
_converters.Add(type, converter, condition); | |||
_converters.Add(valueType, converterType, condition); | |||
} | |||
public void AddGenericConverter(Type converter) | |||
public void AddGenericConverter(Type converterType) | |||
{ | |||
CheckScoped(); | |||
_converters.AddGeneric(converter); | |||
_converters.AddGeneric(converterType); | |||
} | |||
public void AddGenericConverter(Type converter, Func<TypeInfo, PropertyInfo, bool> condition) | |||
public void AddGenericConverter(Type converterType, Func<TypeInfo, PropertyInfo, bool> condition) | |||
{ | |||
CheckScoped(); | |||
_converters.AddGeneric(converter, condition); | |||
_converters.AddGeneric(converterType, condition); | |||
} | |||
public void AddGenericConverter(Type value, Type converter) | |||
public void AddGenericConverter(Type valueType, Type converterType) | |||
{ | |||
CheckScoped(); | |||
_converters.AddGeneric(value, converter); | |||
_converters.AddGeneric(valueType, converterType); | |||
} | |||
public void AddGenericConverter(Type value, Type converter, Func<TypeInfo, PropertyInfo, bool> condition) | |||
public void AddGenericConverter(Type valueType, Type converterType, Func<TypeInfo, PropertyInfo, bool> condition) | |||
{ | |||
CheckScoped(); | |||
_converters.AddGeneric(value, converter, condition); | |||
_converters.AddGeneric(valueType, converterType, condition); | |||
} | |||
public void AddSelectorConverter(string groupKey, Type keyType, object keyValue, Type converterType) | |||
{ | |||
CheckScoped(); | |||
_converters.AddSelector(groupKey, keyType, keyValue, converterType); | |||
} | |||
public ISelectorGroup GetSelectorGroup(Type keyType, string groupKey) | |||
=> _converters.GetSelectorGroup(keyType, groupKey); | |||
protected internal ModelMap<TModel> MapModel<TModel>() | |||
where TModel : class, new() | |||
{ | |||
@@ -80,24 +88,23 @@ namespace Discord.Serialization | |||
var properties = new List<PropertyMap>(); | |||
for (int i = 0; i < propInfos.Length; i++) | |||
{ | |||
var propMap = MapProperty<TModel>(propInfos[i]); | |||
properties.Add(propMap); | |||
if (propInfos[i].GetCustomAttribute<ModelPropertyAttribute>() != null) | |||
{ | |||
var propMap = MapProperty<TModel>(propInfos[i]); | |||
properties.Add(propMap); | |||
} | |||
} | |||
return new ModelMap<TModel>(properties); | |||
return new ModelMap<TModel>(this, type, properties); | |||
}) as ModelMap<TModel>; | |||
} | |||
private PropertyMap MapProperty<TModel>(PropertyInfo propInfo) | |||
where TModel : class, new() | |||
=> _createPropertyMapMethod.MakeGenericMethod(typeof(TModel), propInfo.PropertyType).Invoke(this, new object[] { propInfo }) as PropertyMap; | |||
protected abstract PropertyMap CreatePropertyMap<TModel, TValue>(PropertyInfo propInfo) | |||
where TModel : class, new(); | |||
public abstract TModel Read<TModel>(ReadOnlyBuffer<byte> data) | |||
where TModel : class, new(); | |||
public abstract void Write<TModel>(ArrayFormatter stream, TModel model) | |||
where TModel : class, new(); | |||
protected abstract PropertyMap CreatePropertyMap<TModel, TValue>(PropertyInfo propInfo); | |||
public abstract TModel Read<TModel>(ReadOnlyBuffer<byte> data); | |||
public abstract void Write<TModel>(ArrayFormatter stream, TModel model); | |||
private void CheckScoped() | |||
{ | |||
if (IsScoped) | |||
@@ -0,0 +1,20 @@ | |||
#pragma warning disable CS1591 | |||
using Discord.Serialization; | |||
namespace Discord.API.Gateway | |||
{ | |||
internal class GatewaySocketFrame | |||
{ | |||
[ModelProperty("op")] | |||
public GatewayOpCode Operation { get; set; } | |||
[ModelProperty("t", ExcludeNull = true)] | |||
public string Type { get; set; } | |||
[ModelProperty("s", ExcludeNull = true)] | |||
public int? Sequence { get; set; } | |||
[ModelProperty("d")] | |||
[ModelSelector(nameof(Operation), ModelSelectorGroups.GatewayFrame)] | |||
[ModelSelector(nameof(Type), ModelSelectorGroups.GatewayDispatchFrame)] | |||
public object Payload { get; set; } | |||
} | |||
} |
@@ -1,18 +1,20 @@ | |||
#pragma warning disable CS1591 | |||
using Discord.Serialization; | |||
using System; | |||
namespace Discord.API | |||
namespace Discord.API.Voice | |||
{ | |||
internal class SocketFrame | |||
internal class VoiceSocketFrame | |||
{ | |||
[ModelProperty("op")] | |||
public int Operation { get; set; } | |||
public VoiceOpCode Operation { get; set; } | |||
[ModelProperty("t", ExcludeNull = true)] | |||
public string Type { get; set; } | |||
[ModelProperty("s", ExcludeNull = true)] | |||
public int? Sequence { get; set; } | |||
[ModelProperty("d")] | |||
public ReadOnlyBuffer<byte> Payload { get; set; } | |||
[ModelSelector(nameof(Operation), ModelSelectorGroups.VoiceFrame)] | |||
[ModelSelector(nameof(Type), ModelSelectorGroups.VoiceDispatchFrame)] | |||
public object Payload { get; set; } | |||
} | |||
} |
@@ -11,6 +11,7 @@ using System.Threading.Tasks; | |||
using System.Collections.Generic; | |||
using System.Runtime.InteropServices; | |||
using Discord.Serialization; | |||
using Discord.Serialization.Json; | |||
namespace Discord.Audio | |||
{ | |||
@@ -66,7 +67,7 @@ namespace Discord.Audio | |||
ChannelId = channelId; | |||
_audioLogger = Discord.LogManager.CreateLogger($"Audio #{clientId}"); | |||
_serializer = new Serializer(SerializationFormat.Json); | |||
_serializer = DiscordJsonSerializer.Global.CreateScope(); | |||
_serializer.Error += ex => | |||
{ | |||
_audioLogger.WarningAsync("Serializer Error", ex).GetAwaiter().GetResult(); | |||
@@ -225,7 +226,7 @@ namespace Discord.Audio | |||
_streams.Clear(); | |||
} | |||
private async Task ProcessMessageAsync(VoiceOpCode opCode, ReadOnlyBuffer<byte> payload) | |||
private async Task ProcessMessageAsync(VoiceOpCode opCode, object payload) | |||
{ | |||
_lastMessageTime = Environment.TickCount; | |||
@@ -236,7 +237,7 @@ namespace Discord.Audio | |||
case VoiceOpCode.Ready: | |||
{ | |||
await _audioLogger.DebugAsync("Received Ready").ConfigureAwait(false); | |||
var data = _serializer.Read<ReadyEvent>(payload); | |||
var data = payload as ReadyEvent; | |||
_ssrc = data.SSRC; | |||
@@ -252,7 +253,7 @@ namespace Discord.Audio | |||
case VoiceOpCode.SessionDescription: | |||
{ | |||
await _audioLogger.DebugAsync("Received SessionDescription").ConfigureAwait(false); | |||
var data = _serializer.Read<SessionDescriptionEvent>(payload); | |||
var data = payload as SessionDescriptionEvent; | |||
if (data.Mode != DiscordVoiceApiClient.Mode) | |||
throw new InvalidOperationException($"Discord selected an unexpected mode: {data.Mode}"); | |||
@@ -288,7 +289,7 @@ namespace Discord.Audio | |||
{ | |||
await _audioLogger.DebugAsync("Received Speaking").ConfigureAwait(false); | |||
var data = _serializer.Read<SpeakingEvent>(payload); | |||
var data = payload as SpeakingEvent; | |||
_ssrcMap[data.Ssrc] = data.UserId; //TODO: Memory Leak: SSRCs are never cleaned up | |||
await _speakingUpdatedEvent.InvokeAsync(data.UserId, data.Speaking); | |||
@@ -7,6 +7,7 @@ using System.Linq; | |||
using System.Threading.Tasks; | |||
using System.Threading; | |||
using Discord.Serialization; | |||
using Discord.Serialization.Json; | |||
namespace Discord.WebSocket | |||
{ | |||
@@ -54,7 +55,7 @@ namespace Discord.WebSocket | |||
_baseConfig = config; | |||
_connectionGroupLock = new SemaphoreSlim(1, 1); | |||
_serializer = new Serializer(SerializationFormat.Json); | |||
_serializer = DiscordJsonSerializer.Global.CreateScope(); | |||
_serializer.Error += ex => | |||
{ | |||
_restLogger.WarningAsync("Serializer Error", ex).GetAwaiter().GetResult(); | |||
@@ -21,8 +21,8 @@ namespace Discord.API | |||
{ | |||
public event Func<GatewayOpCode, Task> SentGatewayMessage { add { _sentGatewayMessageEvent.Add(value); } remove { _sentGatewayMessageEvent.Remove(value); } } | |||
private readonly AsyncEvent<Func<GatewayOpCode, Task>> _sentGatewayMessageEvent = new AsyncEvent<Func<GatewayOpCode, Task>>(); | |||
public event Func<GatewayOpCode, int?, string, ReadOnlyBuffer<byte>, Task> ReceivedGatewayEvent { add { _receivedGatewayEvent.Add(value); } remove { _receivedGatewayEvent.Remove(value); } } | |||
private readonly AsyncEvent<Func<GatewayOpCode, int?, string, ReadOnlyBuffer<byte>, Task>> _receivedGatewayEvent = new AsyncEvent<Func<GatewayOpCode, int?, string, ReadOnlyBuffer<byte>, Task>>(); | |||
public event Func<GatewayOpCode, int?, string, object, Task> ReceivedGatewayEvent { add { _receivedGatewayEvent.Add(value); } remove { _receivedGatewayEvent.Remove(value); } } | |||
private readonly AsyncEvent<Func<GatewayOpCode, int?, string, object, Task>> _receivedGatewayEvent = new AsyncEvent<Func<GatewayOpCode, int?, string, object, Task>>(); | |||
public event Func<Exception, Task> Disconnected { add { _disconnectedEvent.Add(value); } remove { _disconnectedEvent.Remove(value); } } | |||
private readonly AsyncEvent<Func<Exception, Task>> _disconnectedEvent = new AsyncEvent<Func<Exception, Task>>(); | |||
@@ -60,16 +60,16 @@ namespace Discord.API | |||
_decompressionStream.SetLength(_decompressionStream.Position); | |||
_decompressionStream.Position = 0; | |||
var msg = _serializer.Read<SocketFrame>(_decompressionStream.ToReadOnlyBuffer()); | |||
var msg = _serializer.Read<GatewaySocketFrame>(_decompressionStream.ToReadOnlyBuffer()); | |||
if (msg != null) | |||
await _receivedGatewayEvent.InvokeAsync((GatewayOpCode)msg.Operation, msg.Sequence, msg.Type, msg.Payload).ConfigureAwait(false); | |||
await _receivedGatewayEvent.InvokeAsync(msg.Operation, msg.Sequence, msg.Type, msg.Payload).ConfigureAwait(false); | |||
} | |||
} | |||
else | |||
{ | |||
var msg = _serializer.Read<SocketFrame>(data); | |||
var msg = _serializer.Read<GatewaySocketFrame>(data); | |||
if (msg != null) | |||
await _receivedGatewayEvent.InvokeAsync((GatewayOpCode)msg.Operation, msg.Sequence, msg.Type, msg.Payload).ConfigureAwait(false); | |||
await _receivedGatewayEvent.InvokeAsync(msg.Operation, msg.Sequence, msg.Type, msg.Payload).ConfigureAwait(false); | |||
} | |||
}; | |||
WebSocketClient.Closed += async ex => | |||
@@ -173,20 +173,17 @@ namespace Discord.API | |||
{ | |||
CheckState(); | |||
if (_formatters.TryDequeue(out var data1)) | |||
data1 = new ArrayFormatter(128, SymbolTable.InvariantUtf8); | |||
if (_formatters.TryDequeue(out var data2)) | |||
data2 = new ArrayFormatter(128, SymbolTable.InvariantUtf8); | |||
if (!_formatters.TryDequeue(out var data)) | |||
data = new ArrayFormatter(128, SymbolTable.InvariantUtf8); | |||
try | |||
{ | |||
payload = new SocketFrame { Operation = (int)opCode, Payload = SerializeJson(data1, payload) }; | |||
await RequestQueue.SendAsync(new WebSocketRequest(WebSocketClient, null, SerializeJson(data2, payload), true, options)).ConfigureAwait(false); | |||
var frame = new GatewaySocketFrame { Operation = opCode, Payload = payload }; | |||
await RequestQueue.SendAsync(new WebSocketRequest(WebSocketClient, null, SerializeJson(data, frame), true, options)).ConfigureAwait(false); | |||
await _sentGatewayMessageEvent.InvokeAsync(opCode).ConfigureAwait(false); | |||
} | |||
finally | |||
{ | |||
_formatters.Enqueue(data1); | |||
_formatters.Enqueue(data2); | |||
_formatters.Enqueue(data); | |||
} | |||
} | |||
@@ -5,6 +5,7 @@ using Discord.Net.Udp; | |||
using Discord.Net.WebSockets; | |||
using Discord.Rest; | |||
using Discord.Serialization; | |||
using Discord.Serialization.Json; | |||
using System; | |||
using System.Collections.Concurrent; | |||
using System.Collections.Generic; | |||
@@ -87,7 +88,7 @@ namespace Discord.WebSocket | |||
_stateLock = new SemaphoreSlim(1, 1); | |||
_gatewayLogger = LogManager.CreateLogger(ShardId == 0 && TotalShards == 1 ? "Gateway" : $"Shard #{ShardId}"); | |||
_serializer = new Serializer(SerializationFormat.Json); | |||
_serializer = DiscordJsonSerializer.Global.CreateScope(); | |||
_serializer.Error += ex => | |||
{ | |||
_restLogger.WarningAsync("Serializer Error", ex).GetAwaiter().GetResult(); | |||
@@ -382,7 +383,7 @@ namespace Discord.WebSocket | |||
gameModel).ConfigureAwait(false); | |||
} | |||
private async Task ProcessMessageAsync(GatewayOpCode opCode, int? seq, string type, ReadOnlyBuffer<byte> payload) | |||
private async Task ProcessMessageAsync(GatewayOpCode opCode, int? seq, string type, object payload) | |||
{ | |||
if (seq != null) | |||
_lastSeq = seq.Value; | |||
@@ -395,7 +396,7 @@ namespace Discord.WebSocket | |||
case GatewayOpCode.Hello: | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Hello").ConfigureAwait(false); | |||
var data = _serializer.Read<HelloEvent>(payload); | |||
var data = payload as HelloEvent; | |||
_heartbeatTask = RunHeartbeatAsync(data.HeartbeatInterval, _connection.CancelToken); | |||
} | |||
@@ -428,7 +429,7 @@ namespace Discord.WebSocket | |||
_sessionId = null; | |||
_lastSeq = 0; | |||
bool retry = IsTrue(); | |||
bool retry = (bool)payload; | |||
if (retry) | |||
_connection.Reconnect(); //TODO: Untested | |||
else | |||
@@ -451,7 +452,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (READY)").ConfigureAwait(false); | |||
var data = _serializer.Read<ReadyEvent>(payload); | |||
var data = payload as ReadyEvent; | |||
var state = new ClientState(data.Guilds.Length, data.PrivateChannels.Length); | |||
var currentUser = SocketSelfUser.Create(this, state, data.User); | |||
@@ -521,7 +522,7 @@ namespace Discord.WebSocket | |||
//Guilds | |||
case "GUILD_CREATE": | |||
{ | |||
var data = _serializer.Read<ExtendedGuild>(payload); | |||
var data = payload as ExtendedGuild; | |||
if (data.Unavailable == false) | |||
{ | |||
@@ -573,7 +574,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_UPDATE)").ConfigureAwait(false); | |||
var data = _serializer.Read<API.Guild>(payload); | |||
var data = payload as API.Guild; | |||
var guild = State.GetGuild(data.Id); | |||
if (guild != null) | |||
{ | |||
@@ -592,7 +593,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_EMOJIS_UPDATE)").ConfigureAwait(false); | |||
var data = _serializer.Read<API.Gateway.GuildEmojiUpdateEvent>(payload); | |||
var data = payload as API.Gateway.GuildEmojiUpdateEvent; | |||
var guild = State.GetGuild(data.GuildId); | |||
if (guild != null) | |||
{ | |||
@@ -610,7 +611,7 @@ namespace Discord.WebSocket | |||
case "GUILD_SYNC": | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_SYNC)").ConfigureAwait(false); | |||
var data = _serializer.Read<GuildSyncEvent>(payload); | |||
var data = payload as GuildSyncEvent; | |||
var guild = State.GetGuild(data.Id); | |||
if (guild != null) | |||
{ | |||
@@ -631,7 +632,7 @@ namespace Discord.WebSocket | |||
break; | |||
case "GUILD_DELETE": | |||
{ | |||
var data = _serializer.Read<ExtendedGuild>(payload); | |||
var data = payload as ExtendedGuild; | |||
if (data.Unavailable == true) | |||
{ | |||
type = "GUILD_UNAVAILABLE"; | |||
@@ -673,7 +674,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (CHANNEL_CREATE)").ConfigureAwait(false); | |||
var data = _serializer.Read<API.Channel>(payload); | |||
var data = payload as API.Channel; | |||
SocketChannel channel = null; | |||
if (data.GuildId.IsSpecified) | |||
{ | |||
@@ -705,7 +706,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (CHANNEL_UPDATE)").ConfigureAwait(false); | |||
var data = _serializer.Read<API.Channel>(payload); | |||
var data = payload as API.Channel; | |||
var channel = State.GetChannel(data.Id); | |||
if (channel != null) | |||
{ | |||
@@ -733,7 +734,7 @@ namespace Discord.WebSocket | |||
await _gatewayLogger.DebugAsync("Received Dispatch (CHANNEL_DELETE)").ConfigureAwait(false); | |||
SocketChannel channel = null; | |||
var data = _serializer.Read<API.Channel>(payload); | |||
var data = payload as API.Channel; | |||
if (data.GuildId.IsSpecified) | |||
{ | |||
var guild = State.GetGuild(data.GuildId.Value); | |||
@@ -771,7 +772,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_MEMBER_ADD)").ConfigureAwait(false); | |||
var data = _serializer.Read<GuildMemberAddEvent>(payload); | |||
var data = payload as GuildMemberAddEvent; | |||
var guild = State.GetGuild(data.GuildId); | |||
if (guild != null) | |||
{ | |||
@@ -797,7 +798,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_MEMBER_UPDATE)").ConfigureAwait(false); | |||
var data = _serializer.Read<GuildMemberUpdateEvent>(payload); | |||
var data = payload as GuildMemberUpdateEvent; | |||
var guild = State.GetGuild(data.GuildId); | |||
if (guild != null) | |||
{ | |||
@@ -835,7 +836,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_MEMBER_REMOVE)").ConfigureAwait(false); | |||
var data = _serializer.Read<GuildMemberRemoveEvent>(payload); | |||
var data = payload as GuildMemberRemoveEvent; | |||
var guild = State.GetGuild(data.GuildId); | |||
if (guild != null) | |||
{ | |||
@@ -870,7 +871,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_MEMBERS_CHUNK)").ConfigureAwait(false); | |||
var data = _serializer.Read<GuildMembersChunkEvent>(payload); | |||
var data = payload as GuildMembersChunkEvent; | |||
var guild = State.GetGuild(data.GuildId); | |||
if (guild != null) | |||
{ | |||
@@ -894,7 +895,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (CHANNEL_RECIPIENT_ADD)").ConfigureAwait(false); | |||
var data = _serializer.Read<RecipientEvent>(payload); | |||
var data = payload as RecipientEvent; | |||
if (State.GetChannel(data.ChannelId) is SocketGroupChannel channel) | |||
{ | |||
var user = channel.GetOrAddUser(data.User); | |||
@@ -911,7 +912,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (CHANNEL_RECIPIENT_REMOVE)").ConfigureAwait(false); | |||
var data = _serializer.Read<RecipientEvent>(payload); | |||
var data = payload as RecipientEvent; | |||
if (State.GetChannel(data.ChannelId) is SocketGroupChannel channel) | |||
{ | |||
var user = channel.RemoveUser(data.User.Id); | |||
@@ -936,7 +937,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_ROLE_CREATE)").ConfigureAwait(false); | |||
var data = _serializer.Read<GuildRoleCreateEvent>(payload); | |||
var data = payload as GuildRoleCreateEvent; | |||
var guild = State.GetGuild(data.GuildId); | |||
if (guild != null) | |||
{ | |||
@@ -960,7 +961,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_ROLE_UPDATE)").ConfigureAwait(false); | |||
var data = _serializer.Read<GuildRoleUpdateEvent>(payload); | |||
var data = payload as GuildRoleUpdateEvent; | |||
var guild = State.GetGuild(data.GuildId); | |||
if (guild != null) | |||
{ | |||
@@ -995,7 +996,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_ROLE_DELETE)").ConfigureAwait(false); | |||
var data = _serializer.Read<GuildRoleDeleteEvent>(payload); | |||
var data = payload as GuildRoleDeleteEvent; | |||
var guild = State.GetGuild(data.GuildId); | |||
if (guild != null) | |||
{ | |||
@@ -1029,7 +1030,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_BAN_ADD)").ConfigureAwait(false); | |||
var data = _serializer.Read<GuildBanEvent>(payload); | |||
var data = payload as GuildBanEvent; | |||
var guild = State.GetGuild(data.GuildId); | |||
if (guild != null) | |||
{ | |||
@@ -1055,7 +1056,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_BAN_REMOVE)").ConfigureAwait(false); | |||
var data = _serializer.Read<GuildBanEvent>(payload); | |||
var data = payload as GuildBanEvent; | |||
var guild = State.GetGuild(data.GuildId); | |||
if (guild != null) | |||
{ | |||
@@ -1083,7 +1084,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (MESSAGE_CREATE)").ConfigureAwait(false); | |||
var data = _serializer.Read<API.Message>(payload); | |||
var data = payload as API.Message; | |||
if (State.GetChannel(data.ChannelId) is ISocketMessageChannel channel) | |||
{ | |||
var guild = (channel as SocketGuildChannel)?.Guild; | |||
@@ -1130,7 +1131,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (MESSAGE_UPDATE)").ConfigureAwait(false); | |||
var data = _serializer.Read<API.Message>(payload); | |||
var data = payload as API.Message; | |||
if (State.GetChannel(data.ChannelId) is ISocketMessageChannel channel) | |||
{ | |||
var guild = (channel as SocketGuildChannel)?.Guild; | |||
@@ -1177,7 +1178,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (MESSAGE_DELETE)").ConfigureAwait(false); | |||
var data = _serializer.Read<API.Message>(payload); | |||
var data = payload as API.Message; | |||
if (State.GetChannel(data.ChannelId) is ISocketMessageChannel channel) | |||
{ | |||
var guild = (channel as SocketGuildChannel)?.Guild; | |||
@@ -1204,7 +1205,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (MESSAGE_REACTION_ADD)").ConfigureAwait(false); | |||
var data = _serializer.Read<API.Gateway.Reaction>(payload); | |||
var data = payload as API.Gateway.Reaction; | |||
if (State.GetChannel(data.ChannelId) is ISocketMessageChannel channel) | |||
{ | |||
var cachedMsg = channel.GetCachedMessage(data.MessageId) as SocketUserMessage; | |||
@@ -1228,7 +1229,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (MESSAGE_REACTION_REMOVE)").ConfigureAwait(false); | |||
var data = _serializer.Read<API.Gateway.Reaction>(payload); | |||
var data = payload as API.Gateway.Reaction; | |||
if (State.GetChannel(data.ChannelId) is ISocketMessageChannel channel) | |||
{ | |||
var cachedMsg = channel.GetCachedMessage(data.MessageId) as SocketUserMessage; | |||
@@ -1252,7 +1253,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (MESSAGE_REACTION_REMOVE_ALL)").ConfigureAwait(false); | |||
var data = _serializer.Read<RemoveAllReactionsEvent>(payload); | |||
var data = payload as RemoveAllReactionsEvent; | |||
if (State.GetChannel(data.ChannelId) is ISocketMessageChannel channel) | |||
{ | |||
var cachedMsg = channel.GetCachedMessage(data.MessageId) as SocketUserMessage; | |||
@@ -1274,7 +1275,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (MESSAGE_DELETE_BULK)").ConfigureAwait(false); | |||
var data = _serializer.Read<MessageDeleteBulkEvent>(payload); | |||
var data = payload as MessageDeleteBulkEvent; | |||
if (State.GetChannel(data.ChannelId) is ISocketMessageChannel channel) | |||
{ | |||
var guild = (channel as SocketGuildChannel)?.Guild; | |||
@@ -1305,7 +1306,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (PRESENCE_UPDATE)").ConfigureAwait(false); | |||
var data = _serializer.Read<API.Presence>(payload); | |||
var data = payload as API.Presence; | |||
if (data.GuildId.IsSpecified) | |||
{ | |||
@@ -1364,7 +1365,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (TYPING_START)").ConfigureAwait(false); | |||
var data = _serializer.Read<TypingStartEvent>(payload); | |||
var data = payload as TypingStartEvent; | |||
if (State.GetChannel(data.ChannelId) is ISocketMessageChannel channel) | |||
{ | |||
var guild = (channel as SocketGuildChannel)?.Guild; | |||
@@ -1386,7 +1387,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (USER_UPDATE)").ConfigureAwait(false); | |||
var data = _serializer.Read<API.User>(payload); | |||
var data = payload as API.User; | |||
if (data.Id == CurrentUser.Id) | |||
{ | |||
var before = CurrentUser.Clone(); | |||
@@ -1406,7 +1407,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (VOICE_STATE_UPDATE)").ConfigureAwait(false); | |||
var data = _serializer.Read<API.VoiceState>(payload); | |||
var data = payload as API.VoiceState; | |||
SocketUser user; | |||
SocketVoiceState before, after; | |||
if (data.GuildId != null) | |||
@@ -1478,7 +1479,7 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.DebugAsync("Received Dispatch (VOICE_SERVER_UPDATE)").ConfigureAwait(false); | |||
var data = _serializer.Read<VoiceServerUpdateEvent>(payload); | |||
var data = payload as VoiceServerUpdateEvent; | |||
var guild = State.GetGuild(data.GuildId); | |||
if (guild != null) | |||
{ | |||
@@ -1528,15 +1529,6 @@ namespace Discord.WebSocket | |||
{ | |||
await _gatewayLogger.ErrorAsync($"Error handling {opCode}{(type != null ? $" ({type})" : "")}", ex).ConfigureAwait(false); | |||
} | |||
bool IsTrue() | |||
{ | |||
ref var ptr = ref payload.Span.DangerousGetPinnableReference(); | |||
return Unsafe.Add(ref ptr, 0) == (byte)'t' && | |||
Unsafe.Add(ref ptr, 1) == (byte)'r' && | |||
Unsafe.Add(ref ptr, 2) == (byte)'u' && | |||
Unsafe.Add(ref ptr, 3) == (byte)'e'; | |||
} | |||
} | |||
private async Task RunHeartbeatAsync(int intervalMillis, CancellationToken cancelToken) | |||
@@ -31,8 +31,8 @@ namespace Discord.Audio | |||
public event Func<int, Task> SentData { add { _sentDataEvent.Add(value); } remove { _sentDataEvent.Remove(value); } } | |||
private readonly AsyncEvent<Func<int, Task>> _sentDataEvent = new AsyncEvent<Func<int, Task>>(); | |||
public event Func<VoiceOpCode, ReadOnlyBuffer<byte>, Task> ReceivedEvent { add { _receivedEvent.Add(value); } remove { _receivedEvent.Remove(value); } } | |||
private readonly AsyncEvent<Func<VoiceOpCode, ReadOnlyBuffer<byte>, Task>> _receivedEvent = new AsyncEvent<Func<VoiceOpCode, ReadOnlyBuffer<byte>, Task>>(); | |||
public event Func<VoiceOpCode, object, Task> ReceivedEvent { add { _receivedEvent.Add(value); } remove { _receivedEvent.Remove(value); } } | |||
private readonly AsyncEvent<Func<VoiceOpCode, object, Task>> _receivedEvent = new AsyncEvent<Func<VoiceOpCode, object, Task>>(); | |||
public event Func<byte[], int, int, Task> ReceivedPacket { add { _receivedPacketEvent.Add(value); } remove { _receivedPacketEvent.Remove(value); } } | |||
private readonly AsyncEvent<Func<byte[], int, int, Task>> _receivedPacketEvent = new AsyncEvent<Func<byte[], int, int, Task>>(); | |||
public event Func<Exception, Task> Disconnected { add { _disconnectedEvent.Add(value); } remove { _disconnectedEvent.Remove(value); } } | |||
@@ -79,16 +79,16 @@ namespace Discord.Audio | |||
_decompressionStream.SetLength(_decompressionStream.Position); | |||
_decompressionStream.Position = 0; | |||
var msg = _serializer.Read<SocketFrame>(_decompressionStream.ToReadOnlyBuffer()); | |||
var msg = _serializer.Read<VoiceSocketFrame>(_decompressionStream.ToReadOnlyBuffer()); | |||
if (msg != null) | |||
await _receivedEvent.InvokeAsync((VoiceOpCode)msg.Operation, msg.Payload).ConfigureAwait(false); | |||
await _receivedEvent.InvokeAsync(msg.Operation, msg.Payload).ConfigureAwait(false); | |||
} | |||
} | |||
else | |||
{ | |||
var msg = _serializer.Read<SocketFrame>(data); | |||
var msg = _serializer.Read<VoiceSocketFrame>(data); | |||
if (msg != null) | |||
await _receivedEvent.InvokeAsync((VoiceOpCode)msg.Operation, msg.Payload).ConfigureAwait(false); | |||
await _receivedEvent.InvokeAsync(msg.Operation, msg.Payload).ConfigureAwait(false); | |||
} | |||
}; | |||
WebSocketClient.Closed += async ex => | |||
@@ -114,20 +114,17 @@ namespace Discord.Audio | |||
public async Task SendAsync(VoiceOpCode opCode, object payload, RequestOptions options = null) | |||
{ | |||
if (_formatters.TryDequeue(out var data1)) | |||
data1 = new ArrayFormatter(128, SymbolTable.InvariantUtf8); | |||
if (_formatters.TryDequeue(out var data2)) | |||
data2 = new ArrayFormatter(128, SymbolTable.InvariantUtf8); | |||
if (!_formatters.TryDequeue(out var data)) | |||
data = new ArrayFormatter(128, SymbolTable.InvariantUtf8); | |||
try | |||
{ | |||
payload = new SocketFrame { Operation = (int)opCode, Payload = SerializeJson(data1, payload) }; | |||
await WebSocketClient.SendAsync(SerializeJson(data2, payload), true).ConfigureAwait(false); | |||
var frame = new VoiceSocketFrame { Operation = opCode, Payload = payload }; | |||
await WebSocketClient.SendAsync(SerializeJson(data, frame), true).ConfigureAwait(false); | |||
await _sentGatewayMessageEvent.InvokeAsync(opCode).ConfigureAwait(false); | |||
} | |||
finally | |||
{ | |||
_formatters.Enqueue(data1); | |||
_formatters.Enqueue(data2); | |||
_formatters.Enqueue(data); | |||
} | |||
} | |||
public async Task SendAsync(byte[] data, int offset, int bytes) | |||
@@ -6,6 +6,7 @@ using System.Threading.Tasks; | |||
using System.Linq; | |||
using Discord.Logging; | |||
using Discord.Serialization; | |||
using Discord.Serialization.Json; | |||
namespace Discord.Webhook | |||
{ | |||
@@ -29,7 +30,7 @@ namespace Discord.Webhook | |||
{ | |||
_webhookId = webhookId; | |||
_serializer = new Serializer(SerializationFormat.Json); | |||
_serializer = DiscordJsonSerializer.Global.CreateScope(); | |||
_serializer.Error += ex => | |||
{ | |||
_restLogger.WarningAsync("Serializer Error", ex).GetAwaiter().GetResult(); | |||