KustoMappingBuilder.cs

// Copyright (c) .NET Foundation. All rights reserved.
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.

using Microsoft.CodeAnalysis;

namespace NuGet.Insights
{
    public class KustoMappingBuilder : IPropertyVisitor
    {
        private readonly int _indent;
        private readonly bool _escapeQuotes;
        private readonly StringBuilder _builder;
        private int _nextOrdinal;

        public KustoMappingBuilder(int indent, bool escapeQuotes)
        {
            _indent = indent;
            _escapeQuotes = escapeQuotes;
            _builder = new StringBuilder();
            _nextOrdinal = 0;
        }

        public void OnProperty(SourceProductionContext context, CsvRecordModel model, CsvPropertyModel property)
        {
            var field = new DataMapping
            {
                Column = property.Name,
                DataType = PropertyHelper.GetKustoDataType(property),
                Properties = new CsvProperties
                {
                    Ordinal = _nextOrdinal,
                }
            };
            _nextOrdinal++;

            if (property.IsKustoIgnore)
            {
                return;
            }

            if (_builder.Length > 1)
            {
                _builder.Append(",'");
                _builder.AppendLine();
            }

            _builder.Append(' ', _indent);
            _builder.Append("'");

            var json = JsonSerializer.Serialize(field).Replace("'", "\\'");

            if (_escapeQuotes)
            {
                json = json.Replace("\"", "\"\"");
            }

            _builder.Append(json);
        }

        public void Finish(SourceProductionContext context, CsvRecordModel model)
        {
            _builder.Append("'");
        }

        public string GetResult()
        {
            return _builder.ToString();
        }

        /// <summary>
        /// Source: https://docs.microsoft.com/en-us/azure/data-explorer/kusto/management/mappings
        /// </summary>
        private class DataMapping
        {
            public string Column { get; set; }

            public string DataType { get; set; }

            public CsvProperties Properties { get; set; }
        }

        private class CsvProperties
        {
            public int Ordinal { get; set; }
        }
    }
}