| | 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 | | { |
| 4 | 8 | | public SykiStringEnumConverter() { } |
| | 9 | |
|
| | 10 | | public override bool CanConvert(Type typeToConvert) |
| | 11 | | { |
| 524 | 12 | | var underlying = Nullable.GetUnderlyingType(typeToConvert); |
| 524 | 13 | | return (underlying is not null && underlying.IsEnum) || typeToConvert.IsEnum; |
| | 14 | | } |
| | 15 | |
|
| | 16 | | public override JsonConverter CreateConverter(Type typeToConvert, JsonSerializerOptions options) |
| | 17 | | { |
| 56 | 18 | | var underlying = Nullable.GetUnderlyingType(typeToConvert); |
| 56 | 19 | | var enumType = underlying ?? typeToConvert; |
| | 20 | |
|
| 56 | 21 | | var convType = underlying is null |
| 56 | 22 | | ? typeof(EnumConverter<>).MakeGenericType(enumType) |
| 56 | 23 | | : typeof(NullableEnumConverter<>).MakeGenericType(enumType); |
| | 24 | |
|
| 56 | 25 | | return (JsonConverter)Activator.CreateInstance(convType)!; |
| | 26 | | } |
| | 27 | |
|
| | 28 | | private sealed class EnumConverter<TEnum> : JsonConverter<TEnum> where TEnum : struct, Enum |
| | 29 | | { |
| 100 | 30 | | public EnumConverter() { } |
| | 31 | |
|
| | 32 | | public override TEnum Read(ref Utf8JsonReader reader, Type typeToConvert, JsonSerializerOptions options) |
| | 33 | | { |
| 2434 | 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 | | } |
| 2434 | 46 | | else if (reader.TokenType == JsonTokenType.Number && |
| 2434 | 47 | | reader.TryGetInt32(out var value) && |
| 2434 | 48 | | Enum.IsDefined(typeof(TEnum), value)) |
| | 49 | | { |
| 2434 | 50 | | return (TEnum)Enum.ToObject(typeof(TEnum), value); |
| | 51 | | } |
| | 52 | |
|
| 0 | 53 | | throw new JsonException($"The JSON value could not be converted to {typeof(TEnum).FullName}."); |
| | 54 | | } |
| | 55 | |
|
| | 56 | | public override void Write(Utf8JsonWriter writer, TEnum value, JsonSerializerOptions options) |
| 23098 | 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 | | { |
| 1530 | 66 | | if (reader.TokenType == JsonTokenType.Null) |
| 0 | 67 | | return null; |
| | 68 | |
|
| 1530 | 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 | |
|
| 1530 | 85 | | if (reader.TokenType == JsonTokenType.Number && |
| 1530 | 86 | | reader.TryGetInt32(out var value) && |
| 1530 | 87 | | Enum.IsDefined(typeof(TEnum), value)) |
| | 88 | | { |
| 1522 | 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 is null) |
| | 98 | | { |
| 0 | 99 | | writer.WriteNullValue(); |
| 0 | 100 | | return; |
| | 101 | | } |
| | 102 | |
|
| 0 | 103 | | writer.WriteStringValue(value.Value.ToString()); |
| 0 | 104 | | } |
| | 105 | | } |
| | 106 | | } |