SerializationPartialBuilder.cs

Go to the documentation of this file.

//-----------------------------------------------------------------------
// <copyright file="SerializationPartialBuilder.cs" company="Marimer LLC">
//     Copyright (c) Marimer LLC. All rights reserved.
//     Website: https://cslanet.com
// </copyright>
// <summary>Builds the text of a partial type to implement the IMobileObject interface</summary>
//-----------------------------------------------------------------------
using Microsoft.CodeAnalysis;
using Microsoft.CodeAnalysis.CSharp.Syntax;
using System;
using System.CodeDom.Compiler;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
 
namespace Csla.Generators.CSharp.AutoSerialization
{
 
  internal class SerializationPartialBuilder
  {
 
    internal GenerationResults BuildPartialTypeDefinition(ExtractedTypeDefinition typeDefinition)
    {
      GenerationResults generationResults;
      IndentedTextWriter textWriter;
 
      using (StringWriter stringWriter = new StringWriter())
      {
        textWriter = new IndentedTextWriter(stringWriter, "\t");
 
        AppendUsingStatements(textWriter, typeDefinition);
        AppendContainerDefinitions(textWriter, typeDefinition);
 
        AppendTypeDefinition(textWriter, typeDefinition);
        AppendBlockStart(textWriter);
 
        AppendGetChildrenMethod(textWriter, typeDefinition);
        AppendGetStateMethod(textWriter, typeDefinition);
        AppendSetChildrenMethod(textWriter, typeDefinition);
        AppendSetStateMethod(textWriter, typeDefinition);
 
        AppendBlockEnd(textWriter);
 
        AppendContainerDefinitionClosures(textWriter, typeDefinition);
 
        generationResults = new GenerationResults()
        {
          FullyQualifiedName = typeDefinition.FullyQualifiedName,
          GeneratedSource = stringWriter.ToString()
        };
      }
 
      return generationResults;
    }
 
    #region Private Helper Methods
 
    private void AppendBlockStart(IndentedTextWriter textWriter)
    {
      textWriter.WriteLine("{");
      textWriter.Indent++;
    }
 
    private void AppendBlockEnd(IndentedTextWriter textWriter)
    {
      textWriter.Indent--;
      textWriter.WriteLine("}");
    }
 
    private void AppendUsingStatements(IndentedTextWriter textWriter, ExtractedTypeDefinition typeDefinition)
    {
      HashSet<string> requiredNamespaces;
 
      requiredNamespaces = GetRequiredNamespaces(typeDefinition);
 
      foreach (string requiredNamespace in requiredNamespaces)
      {
        textWriter.Write("using ");
        textWriter.Write(requiredNamespace);
        textWriter.WriteLine(";");
      }
 
      textWriter.WriteLine();
    }
 
    private HashSet<string> GetRequiredNamespaces(ExtractedTypeDefinition typeDefinition)
    {
      HashSet<string> requiredNamespaces = new HashSet<string>() { "System", "Csla.Serialization.Mobile" };
 
      foreach (ExtractedFieldDefinition fieldDefinition in typeDefinition.Fields)
      {
        requiredNamespaces.Add(fieldDefinition.TypeDefinition.TypeNamespace);
      }
 
      foreach (ExtractedPropertyDefinition propertyDefinition in typeDefinition.Properties)
      {
        requiredNamespaces.Add(propertyDefinition.TypeDefinition.TypeNamespace);
      }
 
      requiredNamespaces.Remove(typeDefinition.Namespace);
 
      return requiredNamespaces;
    }
 
    private void AppendContainerDefinitions(IndentedTextWriter textWriter, ExtractedTypeDefinition typeDefinition)
    {
      foreach (ExtractedContainerDefinition containerDefinition in typeDefinition.ContainerDefinitions)
      {
        textWriter.WriteLine(containerDefinition.FullDefinition);
        AppendBlockStart(textWriter);
      }
    }
 
    private void AppendContainerDefinitionClosures(IndentedTextWriter textWriter, ExtractedTypeDefinition typeDefinition)
    {
      foreach (ExtractedContainerDefinition containerDefinition in typeDefinition.ContainerDefinitions)
      {
        AppendBlockEnd(textWriter);
      }
    }
 
    private void AppendTypeDefinition(IndentedTextWriter textWriter, ExtractedTypeDefinition typeDefinition)
    {
      textWriter.WriteLine("[Serializable]");
      textWriter.Write(typeDefinition.Scope);
      textWriter.Write(" partial ");
      textWriter.Write(typeDefinition.TypeKind);
      textWriter.Write(" ");
      textWriter.Write(typeDefinition.TypeName);
      textWriter.Write(" : IMobileObject");
      textWriter.WriteLine();
    }
 
    private void AppendGetChildrenMethod(IndentedTextWriter textWriter, ExtractedTypeDefinition typeDefinition)
    {
      textWriter.WriteLine("void IMobileObject.GetChildren(SerializationInfo info, MobileFormatter formatter)");
      AppendBlockStart(textWriter);
      if (HasChildrenToExpose(typeDefinition))
      {
        textWriter.WriteLine("IMobileObject mobileObject;");
        textWriter.WriteLine("SerializationInfo childInfo;");
        textWriter.WriteLine();
      }
 
      foreach (ExtractedFieldDefinition fieldDefinition in typeDefinition.Fields)
      {
        if (!IsChildToExpose(fieldDefinition)) continue;
 
        AppendSerializeChildFragment(textWriter, fieldDefinition);
      }
 
      foreach (ExtractedPropertyDefinition propertyDefinition in typeDefinition.Properties)
      {
        if (!IsChildToExpose(propertyDefinition)) continue;
 
        AppendSerializeChildFragment(textWriter, propertyDefinition);
      }
 
      AppendBlockEnd(textWriter);
      textWriter.WriteLine();
    }
 
    private void AppendSerializeChildFragment(IndentedTextWriter textWriter, IMemberDefinition memberDefinition)
    {
      textWriter.Write("mobileObject = ");
      textWriter.Write(memberDefinition.MemberName);
      textWriter.WriteLine(" as IMobileObject;");
      textWriter.WriteLine("childInfo = formatter.SerializeObject(mobileObject);");
      textWriter.Write("info.AddChild(nameof(");
      textWriter.Write(memberDefinition.MemberName);
      textWriter.WriteLine("), childInfo.ReferenceId, true);");
    }
 
    private void AppendSetChildrenMethod(IndentedTextWriter textWriter, ExtractedTypeDefinition typeDefinition)
    {
      textWriter.WriteLine("void IMobileObject.SetChildren(SerializationInfo info, MobileFormatter formatter)");
      AppendBlockStart(textWriter);
 
      if (HasChildrenToExpose(typeDefinition))
      {
        textWriter.WriteLine("SerializationInfo.ChildData childData;");
        textWriter.WriteLine();
      }
 
      foreach (ExtractedFieldDefinition fieldDefinition in typeDefinition.Fields)
      {
        if (!IsChildToExpose(fieldDefinition)) continue;
 
        AppendDeserializeChildFragment(textWriter, fieldDefinition);
      }
 
      foreach (ExtractedPropertyDefinition propertyDefinition in typeDefinition.Properties)
      {
        if (!IsChildToExpose(propertyDefinition)) continue;
 
        AppendDeserializeChildFragment(textWriter, propertyDefinition);
      }
 
      AppendBlockEnd(textWriter);
      textWriter.WriteLine();
    }
 
    private void AppendDeserializeChildFragment(IndentedTextWriter textWriter, IMemberDefinition memberDefinition)
    {
      textWriter.Write("if (info.Children.ContainsKey(nameof(");
      textWriter.Write(memberDefinition.MemberName);
      textWriter.WriteLine(")))");
      AppendBlockStart(textWriter);
 
      textWriter.Write("childData = info.Children[nameof(");
      textWriter.Write(memberDefinition.MemberName);
      textWriter.WriteLine(")];");
      textWriter.WriteLine();
      textWriter.Write(memberDefinition.MemberName);
      textWriter.Write(" = formatter.GetObject(childData.ReferenceId) as ");
      textWriter.Write(memberDefinition.TypeDefinition.TypeName);
      textWriter.WriteLine(";");
 
      AppendBlockEnd(textWriter);
    }
 
    private void AppendGetStateMethod(IndentedTextWriter textWriter, ExtractedTypeDefinition typeDefinition)
    {
      textWriter.WriteLine("void IMobileObject.GetState(SerializationInfo info)");
      AppendBlockStart(textWriter);
 
      foreach (ExtractedFieldDefinition fieldDefinition in typeDefinition.Fields)
      {
        if (IsChildToExpose(fieldDefinition)) continue;
 
        AppendGetMemberStateFragment(textWriter, fieldDefinition);
      }
 
      foreach (ExtractedPropertyDefinition propertyDefinition in typeDefinition.Properties)
      {
        if (IsChildToExpose(propertyDefinition)) continue;
 
        AppendGetMemberStateFragment(textWriter, propertyDefinition);
      }
 
      AppendBlockEnd(textWriter);
      textWriter.WriteLine();
    }
 
    private void AppendGetMemberStateFragment(IndentedTextWriter textWriter, IMemberDefinition memberDefinition)
    {
      textWriter.Write("info.AddValue(nameof(");
      textWriter.Write(memberDefinition.MemberName);
      textWriter.Write("), ");
      textWriter.Write(memberDefinition.MemberName);
      textWriter.WriteLine(");");
    }
 
    private void AppendSetStateMethod(IndentedTextWriter textWriter, ExtractedTypeDefinition typeDefinition)
    {
      textWriter.WriteLine("void IMobileObject.SetState(SerializationInfo info)");
      AppendBlockStart(textWriter);
 
      foreach (ExtractedFieldDefinition fieldDefinition in typeDefinition.Fields)
      {
        if (IsChildToExpose(fieldDefinition)) continue;
 
        AppendSetMemberStateMethod(textWriter, fieldDefinition);
      }
 
      foreach (ExtractedPropertyDefinition propertyDefinition in typeDefinition.Properties)
      {
        if (IsChildToExpose(propertyDefinition)) continue;
 
        AppendSetMemberStateMethod(textWriter, propertyDefinition);
      }
 
      AppendBlockEnd(textWriter);
      textWriter.WriteLine();
    }
 
    private void AppendSetMemberStateMethod(IndentedTextWriter textWriter, IMemberDefinition memberDefinition)
    {
      textWriter.Write(memberDefinition.MemberName);
      textWriter.Write(" = info.GetValue<");
      textWriter.Write(memberDefinition.TypeDefinition.TypeName);
      textWriter.Write(">(nameof(");
      textWriter.Write(memberDefinition.MemberName);
      textWriter.WriteLine("));");
    }
 
    private bool HasChildrenToExpose(ExtractedTypeDefinition typeDefinition)
    {
      return typeDefinition.Properties.Any(p => IsChildToExpose(p)) ||
        typeDefinition.Fields.Any(f => IsChildToExpose(f));
    }
 
    private bool IsChildToExpose(IMemberDefinition memberDefinition)
    {
      return memberDefinition.TypeDefinition.IsAutoSerializable || memberDefinition.TypeDefinition.ImplementsIMobileObject;
    }
 
    #endregion
 
  }
 
}