StaticTypeInspectorFile.cs

// This file is part of YamlDotNet - A .NET library for YAML.
// Copyright (c) Antoine Aubry and contributors
//
// Permission is hereby granted, free of charge, to any person obtaining a copy of
// this software and associated documentation files (the "Software"), to deal in
// the Software without restriction, including without limitation the rights to
// use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
// of the Software, and to permit persons to whom the Software is furnished to do
// so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in all
// copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
// SOFTWARE.

using System;
using System.Collections.Immutable;
using System.Text;
using Microsoft.CodeAnalysis;
using Microsoft.CodeAnalysis.CSharp;

namespace YamlDotNet.Analyzers.StaticGenerator
{
    public class StaticTypeInspectorFile : File
    {
        public StaticTypeInspectorFile(Action<string, bool> Write, Action indent, Action unindent, GeneratorExecutionContext context) : base(Write, indent, unindent, context)
        {
        }

        public override void Write(ClassSyntaxReceiver classSyntaxReceiver)
        {
            Write("public class StaticTypeInspector : YamlDotNet.Serialization.ITypeInspector");
            Write("{"); Indent();

            Write("private readonly YamlDotNet.Serialization.ITypeResolver _typeResolver;");
            Write("public StaticTypeInspector(YamlDotNet.Serialization.ITypeResolver typeResolver)");
            Write("{"); Indent();
            Write("_typeResolver = typeResolver;");
            UnIndent(); Write("}");

            #region GetProperties
            Write("public IEnumerable<YamlDotNet.Serialization.IPropertyDescriptor> GetProperties(Type type, object container)");
            Write("{"); Indent();
            foreach (var o in classSyntaxReceiver.Classes)
            {
                var classObject = o.Value;
                Write($"if (type == typeof({classObject.ModuleSymbol.GetFullName().Replace("?", string.Empty)}))");
                Write("{"); Indent();
                Write($"var accessor = new {classObject.SanitizedClassName}_{classObject.GuidSuffix}();");
                Write("return new YamlDotNet.Serialization.IPropertyDescriptor[]");
                Write("{"); Indent();
                foreach (var field in classObject.FieldSymbols)
                {
                    WritePropertyDescriptor(field.Name, field.Type, !field.IsReadOnly, field.GetAttributes(), ',');
                }
                foreach (var property in classObject.PropertySymbols)
                {
                    WritePropertyDescriptor(property.Name, property.Type, property.SetMethod == null, property.GetAttributes(), ',');
                }
                UnIndent(); Write("};");
                UnIndent(); Write("}");
            }
            Write("return new List<YamlDotNet.Serialization.IPropertyDescriptor>();");
            UnIndent(); Write("}");
            #endregion

            #region GetProperty
            Write("public YamlDotNet.Serialization.IPropertyDescriptor GetProperty(Type type, object container, string name, bool ignoreUnmatched)");
            Write("{"); Indent();
            foreach (var o in classSyntaxReceiver.Classes)
            {
                var classObject = o.Value;
                Write($"if (type == typeof({classObject.ModuleSymbol.GetFullName().Replace("?", string.Empty)}))");
                Write("{"); Indent();
                Write($"var accessor = new {classObject.SanitizedClassName}_{classObject.GuidSuffix}();");
                foreach (var field in classObject.FieldSymbols)
                {
                    Write($"if (name == \"{field.Name}\") return ");
                    WritePropertyDescriptor(field.Name, field.Type, field.IsReadOnly, field.GetAttributes(), ';');
                }
                foreach (var property in classObject.PropertySymbols)
                {
                    Write($"if (name == \"{property.Name}\") return ", false);
                    WritePropertyDescriptor(property.Name, property.Type, property.SetMethod == null, property.GetAttributes(), ';');
                }
                UnIndent(); Write("}");

            }
            Write("if (ignoreUnmatched) return null;");
            Write("throw new ArgumentOutOfRangeException(\"Field or property not valid: \" + name);");
            UnIndent(); Write("}");
            #endregion

            UnIndent(); Write("}");
        }

        private void WritePropertyDescriptor(string name, ITypeSymbol type, bool isReadonly, ImmutableArray<AttributeData> attributes, char finalChar)
        {
            Write($"new StaticPropertyDescriptor(_typeResolver, accessor, \"{name}\", {(!isReadonly).ToString().ToLower()}, typeof({type.GetFullName().Replace("?", string.Empty)}), new Attribute[] {{");
            foreach (var attribute in attributes)
            {
                switch (attribute.AttributeClass?.ToDisplayString())
                {
                    case "YamlDotNet.Serialization.YamlIgnore":
                        Write("new YamlDotNet.Serialization.YamlIgnoreAttribute(),");
                        break;
                    case "YamlDotNet.Serialization.YamlMemberAttribute":
                        Write("new YamlDotNet.Serialization.YamlMemberAttribute() {");
                        for (var index = 0; index < attribute.NamedArguments.Length; index++)
                        {
                            if (index > 0)
                            {
                                Write(", ");
                            }
                            var attributeData = attribute.NamedArguments[index];
                            switch (attributeData.Key)
                            {
                                case "Description":
                                    Indent(); Write($"Description = {attributeData.Value.ToCSharpString()}"); UnIndent();
                                    break;
                                case "Order":
                                    Indent(); Write($"Order = {attributeData.Value.Value}"); UnIndent();
                                    break;
                                case "Alias":
                                    Indent(); Write($"Alias = {attributeData.Value.ToCSharpString()}"); UnIndent();
                                    break;
                                case "ApplyNamingConventions":
                                    Indent(); Write($"ApplyNamingConventions = {attributeData.Value.ToCSharpString()}"); UnIndent();
                                    break;
                                case "DefaultValuesHandling":
                                    Indent(); Write($"DefaultValuesHandling = (YamlDotNet.Serialization.DefaultValuesHandling){attributeData.Value.Value}"); UnIndent();
                                    break;
                                case "ScalarStyle":
                                    Indent(); Write($"ScalarStyle = (YamlDotNet.Core.ScalarStyle){attributeData.Value.Value}"); UnIndent();
                                    break;
                            }
                        }
                        Write($"}},");
                        break;
                }
            }
            Write($"}}){finalChar}");

        }
    }
}