46 lines
1.4 KiB
C#
46 lines
1.4 KiB
C#
|
using System.Text.Json;
|
||
|
|
using System.Text.Json.Serialization;
|
||
|
|
|
||
|
|
using AnthropicClient.Models;
|
||
|
|
|
||
|
|
namespace AnthropicClient.Json;
|
||
|
|
|
||
|
|
class SourceConverter : JsonConverter<Source>
|
||
|
|
{
|
||
|
|
public override Source Read(ref Utf8JsonReader reader, Type typeToConvert, JsonSerializerOptions options)
|
||
|
|
{
|
||
|
|
using var jsonDocument = JsonDocument.ParseValue(ref reader);
|
||
|
|
var root = jsonDocument.RootElement;
|
||
|
|
var type = root.GetProperty("type").GetString();
|
||
|
|
return type switch
|
||
|
|
{
|
||
|
|
SourceType.Text => JsonSerializer.Deserialize<TextSource>(root.GetRawText(), options)!,
|
||
|
|
SourceType.Content => JsonSerializer.Deserialize<CustomSource>(root.GetRawText(), options)!,
|
||
|
|
SourceType.Base64 => JsonSerializer.Deserialize<Base64Source>(root.GetRawText(), options)!,
|
||
|
|
_ => throw new JsonException($"Unknown content type: {type}")
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
public override void Write(Utf8JsonWriter writer, Source value, JsonSerializerOptions options)
|
||
|
|
{
|
||
|
|
if (value is TextSource textSource)
|
||
|
|
{
|
||
|
|
JsonSerializer.Serialize(writer, textSource, options);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (value is CustomSource customSource)
|
||
|
|
{
|
||
|
|
JsonSerializer.Serialize(writer, customSource, options);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (value is Base64Source base64Source)
|
||
|
|
{
|
||
|
|
JsonSerializer.Serialize(writer, base64Source, options);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
JsonSerializer.Serialize(writer, value, value.GetType(), options);
|
||
|
|
}
|
||
|
|
}
|