| | | 1 | | using System.Text.Json; |
| | | 2 | | using System.Text.Json.Serialization; |
| | | 3 | | |
| | | 4 | | namespace Syki.Back.Converters; |
| | | 5 | | |
| | | 6 | | public sealed class SykiStringEnumConverter : JsonConverterFactory |
| | | 7 | | { |
| | 12 | 8 | | public SykiStringEnumConverter() { } |
| | | 9 | | |
| | | 10 | | public override bool CanConvert(Type typeToConvert) |
| | | 11 | | { |
| | 230 | 12 | | var underlying = Nullable.GetUnderlyingType(typeToConvert); |
| | 230 | 13 | | return (underlying is not null && underlying.IsEnum) || typeToConvert.IsEnum; |
| | | 14 | | } |
| | | 15 | | |
| | | 16 | | public override JsonConverter CreateConverter(Type typeToConvert, JsonSerializerOptions options) |
| | | 17 | | { |
| | 14 | 18 | | var underlying = Nullable.GetUnderlyingType(typeToConvert); |
| | 14 | 19 | | var enumType = underlying ?? typeToConvert; |
| | | 20 | | |
| | 14 | 21 | | var convType = underlying == null |
| | 14 | 22 | | ? typeof(EnumConverter<>).MakeGenericType(enumType) |
| | 14 | 23 | | : typeof(NullableEnumConverter<>).MakeGenericType(enumType); |
| | | 24 | | |
| | 14 | 25 | | return (JsonConverter)Activator.CreateInstance(convType)!; |
| | | 26 | | } |
| | | 27 | | |
| | | 28 | | private sealed class EnumConverter<TEnum> : JsonConverter<TEnum> where TEnum : struct, Enum |
| | | 29 | | { |
| | 16 | 30 | | public EnumConverter() { } |
| | | 31 | | |
| | | 32 | | public override TEnum Read(ref Utf8JsonReader reader, Type typeToConvert, JsonSerializerOptions options) |
| | | 33 | | { |
| | 0 | 34 | | if (reader.TokenType == JsonTokenType.String) |
| | | 35 | | { |
| | 0 | 36 | | var s = reader.GetString(); |
| | 0 | 37 | | if (!string.IsNullOrWhiteSpace(s) && |
| | 0 | 38 | | Enum.TryParse<TEnum>(s, ignoreCase: true, out var parsed) && |
| | 0 | 39 | | Enum.IsDefined(parsed)) |
| | 0 | 40 | | return parsed; |
| | | 41 | | |
| | 0 | 42 | | if (int.TryParse(s, out var n) && |
| | 0 | 43 | | Enum.IsDefined(typeof(TEnum), n)) |
| | 0 | 44 | | return (TEnum)Enum.ToObject(typeof(TEnum), n); |
| | | 45 | | } |
| | 0 | 46 | | else if (reader.TokenType == JsonTokenType.Number && |
| | 0 | 47 | | reader.TryGetInt32(out var value) && |
| | 0 | 48 | | Enum.IsDefined(typeof(TEnum), value)) |
| | | 49 | | { |
| | 0 | 50 | | return (TEnum)Enum.ToObject(typeof(TEnum), value); |
| | | 51 | | } |
| | | 52 | | |
| | 0 | 53 | | return (TEnum)Enum.ToObject(typeof(TEnum), int.MaxValue); |
| | | 54 | | } |
| | | 55 | | |
| | | 56 | | public override void Write(Utf8JsonWriter writer, TEnum value, JsonSerializerOptions options) |
| | 18 | 57 | | => writer.WriteStringValue(value.ToString()); |
| | | 58 | | } |
| | | 59 | | |
| | | 60 | | private sealed class NullableEnumConverter<TEnum> : JsonConverter<TEnum?> where TEnum : struct, Enum |
| | | 61 | | { |
| | 12 | 62 | | public NullableEnumConverter() { } |
| | | 63 | | |
| | | 64 | | public override TEnum? Read(ref Utf8JsonReader reader, Type typeToConvert, JsonSerializerOptions options) |
| | | 65 | | { |
| | 96 | 66 | | if (reader.TokenType == JsonTokenType.Null) |
| | 0 | 67 | | return null; |
| | | 68 | | |
| | 96 | 69 | | if (reader.TokenType == JsonTokenType.String) |
| | | 70 | | { |
| | 0 | 71 | | var str = reader.GetString(); |
| | 0 | 72 | | if (str.IsEmpty()) return null; |
| | | 73 | | |
| | 0 | 74 | | if (Enum.TryParse<TEnum>(str, ignoreCase: true, out var parsed) && |
| | 0 | 75 | | Enum.IsDefined(parsed)) |
| | 0 | 76 | | return parsed; |
| | | 77 | | |
| | 0 | 78 | | if (int.TryParse(str, out var n) && |
| | 0 | 79 | | Enum.IsDefined(typeof(TEnum), n)) |
| | 0 | 80 | | return (TEnum)Enum.ToObject(typeof(TEnum), n); |
| | | 81 | | |
| | 0 | 82 | | return null; |
| | | 83 | | } |
| | | 84 | | |
| | 96 | 85 | | if (reader.TokenType == JsonTokenType.Number && |
| | 96 | 86 | | reader.TryGetInt32(out var value) && |
| | 96 | 87 | | Enum.IsDefined(typeof(TEnum), value)) |
| | | 88 | | { |
| | 88 | 89 | | return (TEnum)Enum.ToObject(typeof(TEnum), value); |
| | | 90 | | } |
| | | 91 | | |
| | 8 | 92 | | return null; |
| | | 93 | | } |
| | | 94 | | |
| | | 95 | | public override void Write(Utf8JsonWriter writer, TEnum? value, JsonSerializerOptions options) |
| | | 96 | | { |
| | 0 | 97 | | if (value == null) |
| | | 98 | | { |
| | 0 | 99 | | writer.WriteNullValue(); |
| | 0 | 100 | | return; |
| | | 101 | | } |
| | | 102 | | |
| | 0 | 103 | | writer.WriteStringValue(value.Value.ToString()); |
| | 0 | 104 | | } |
| | | 105 | | } |
| | | 106 | | } |