TextSerializationWriter.cs

using System;
using System.IO;
using System.Collections.Generic;
using System.Xml;
using Microsoft.Kiota.Abstractions;
using Microsoft.Kiota.Abstractions.Serialization;
using Microsoft.Kiota.Abstractions.Extensions;
using System.Linq;

namespace Microsoft.Kiota.Serialization.Text;

/// <summary>
/// The <see cref="ISerializationWriter"/> implementation for text content types.
/// </summary>
public class TextSerializationWriter : ISerializationWriter, IDisposable {
    private readonly MemoryStream _stream = new MemoryStream();
    private readonly StreamWriter writer;
    /// <summary>
    /// Initializes a new instance of the <see cref="TextSerializationWriter"/> class.
    /// </summary>
    public TextSerializationWriter()
    {
        writer = new(_stream);
    }
    private bool written;
    /// <inheritdoc />
    public Action<IParsable>? OnBeforeObjectSerialization { get; set; }
    /// <inheritdoc />
    public Action<IParsable>? OnAfterObjectSerialization { get; set; }
    /// <inheritdoc />
    public Action<IParsable, ISerializationWriter>? OnStartObjectSerialization { get; set; }
    /// <inheritdoc />
    public void Dispose()
    {
        _stream?.Dispose();
        writer?.Dispose();
        GC.SuppressFinalize(this);
    }
    /// <inheritdoc />
    public Stream GetSerializedContent() {
        writer.Flush();
        return _stream;
    }
    /// <inheritdoc />
    public void WriteAdditionalData(IDictionary<string, object> value) => throw new InvalidOperationException(TextParseNode.NoStructuredDataMessage);
    /// <inheritdoc />
    public void WriteBoolValue(string? key, bool? value) => WriteStringValue(key, value?.ToString());
    /// <inheritdoc />
    public void WriteByteArrayValue(string? key, byte[]? value) => WriteStringValue(key, value?.Any() ?? false ? Convert.ToBase64String(value) : string.Empty);
    /// <inheritdoc />
    public void WriteByteValue(string? key, byte? value) => WriteStringValue(key, value?.ToString());
    /// <inheritdoc />
    public void WriteCollectionOfObjectValues<T>(string? key, IEnumerable<T>? values) where T : IParsable => throw new InvalidOperationException(TextParseNode.NoStructuredDataMessage);
    /// <inheritdoc />
    public void WriteCollectionOfPrimitiveValues<T>(string? key, IEnumerable<T>? values) => throw new InvalidOperationException(TextParseNode.NoStructuredDataMessage);
    /// <inheritdoc />
    public void WriteDateTimeOffsetValue(string? key, DateTimeOffset? value) => WriteStringValue(key, value.HasValue ? value.Value.ToString() : null);
    /// <inheritdoc />
    public void WriteDateValue(string? key, Date? value) => WriteStringValue(key, value?.ToString());
    /// <inheritdoc />
    public void WriteDecimalValue(string? key, decimal? value) => WriteStringValue(key, value?.ToString());
    /// <inheritdoc />
    public void WriteDoubleValue(string? key, double? value) => WriteStringValue(key, value?.ToString());
    /// <inheritdoc />
    public void WriteFloatValue(string? key, float? value) => WriteStringValue(key, value?.ToString());
    /// <inheritdoc />
    public void WriteGuidValue(string? key, Guid? value) => WriteStringValue(key, value?.ToString());
    /// <inheritdoc />
    public void WriteIntValue(string? key, int? value) => WriteStringValue(key, value?.ToString());
    /// <inheritdoc />
    public void WriteLongValue(string? key, long? value) => WriteStringValue(key, value?.ToString());
    /// <inheritdoc />
    public void WriteNullValue(string? key) => WriteStringValue(key, "null");
    /// <inheritdoc />
    public void WriteObjectValue<T>(string? key, T? value, params IParsable?[] additionalValuesToMerge) where T : IParsable => throw new InvalidOperationException(TextParseNode.NoStructuredDataMessage);
    /// <inheritdoc />
    public void WriteSbyteValue(string? key, sbyte? value) => WriteStringValue(key, value?.ToString());
    /// <inheritdoc />
    public void WriteStringValue(string? key, string? value)
    {
        if(!string.IsNullOrEmpty(key))
            throw new InvalidOperationException(TextParseNode.NoStructuredDataMessage);
        if(!string.IsNullOrEmpty(value))
            if(written)
                throw new InvalidOperationException("a value was already written for this serialization writer, text content only supports a single value");
            else {
                writer.Write(value);
                written = true;
            }
    }
    /// <inheritdoc />
    public void WriteTimeSpanValue(string? key, TimeSpan? value) => WriteStringValue(key, value.HasValue ? XmlConvert.ToString(value.Value) : null);
    /// <inheritdoc />
    public void WriteTimeValue(string? key, Time? value) => WriteStringValue(key, value?.ToString());
    /// <inheritdoc />
    void ISerializationWriter.WriteCollectionOfEnumValues<T>(string? key, IEnumerable<T?>? values) => throw new InvalidOperationException(TextParseNode.NoStructuredDataMessage);
    /// <inheritdoc />
    void ISerializationWriter.WriteEnumValue<T>(string? key, T? value) => WriteStringValue(key, value.HasValue ? value.Value.ToString().ToFirstCharacterLowerCase() : null);
}