Add EncapsulateFieldCodeBuilder and factory

Author: BZngr

Rubberduck.Refactorings/EncapsulateField/EncapsulateFieldInsertNewCode/EncapsulateFieldCodeBuilder.cs

@@ -0,0 +1,91 @@
+using Rubberduck.Parsing.Grammar;
+using Rubberduck.Parsing.Symbols;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+
+namespace Rubberduck.Refactorings.EncapsulateField
+{
+    public interface IEncapsulateFieldCodeBuilder
+    {
+        (string Get, string Let, string Set) BuildPropertyBlocks(PropertyAttributeSet propertyAttributeSet);
+        string BuildUserDefinedTypeDeclaration(IObjectStateUDT objectStateUDT, IEnumerable<IEncapsulateFieldCandidate> candidates);
+        string BuildObjectStateFieldDeclaration(IObjectStateUDT objectStateUDT);
+    }
+
+    public class EncapsulateFieldCodeBuilder : IEncapsulateFieldCodeBuilder
+    {
+        private const string FourSpaces = "    ";
+        private static string _doubleSpace = $"{Environment.NewLine}{Environment.NewLine}";
+
+        private readonly ICodeBuilder _codeBuilder;
+
+        public EncapsulateFieldCodeBuilder(ICodeBuilder codeBuilder)
+        {
+            _codeBuilder = codeBuilder;
+        }
+
+        public (string Get, string Let, string Set) BuildPropertyBlocks(PropertyAttributeSet propertyAttributes)
+        {
+            string propertyLet = null;
+            string propertySet = null;
+
+            if (propertyAttributes.GenerateLetter)
+            {
+                var letterContent = $"{FourSpaces}{propertyAttributes.BackingField} = {propertyAttributes.ParameterName}";
+                if (!_codeBuilder.TryBuildPropertyLetCodeBlock(propertyAttributes.Declaration, propertyAttributes.PropertyName, out propertyLet, content: letterContent))
+                {
+                    throw new ArgumentException();
+                }
+            }
+
+            if (propertyAttributes.GenerateSetter)
+            {
+                var setterContent = $"{FourSpaces}{Tokens.Set} {propertyAttributes.BackingField} = {propertyAttributes.ParameterName}";
+                if (!_codeBuilder.TryBuildPropertySetCodeBlock(propertyAttributes.Declaration, propertyAttributes.PropertyName, out propertySet, content: setterContent))
+                {
+                    throw new ArgumentException();
+                }
+            }
+
+            var getterContent = $"{propertyAttributes.PropertyName} = {propertyAttributes.BackingField}";
+            if (propertyAttributes.UsesSetAssignment)
+            {
+                getterContent = $"{Tokens.Set} {getterContent}";
+            }
+
+            if (propertyAttributes.AsTypeName.Equals(Tokens.Variant) && !propertyAttributes.Declaration.IsArray)
+            {
+                getterContent = string.Join(Environment.NewLine,
+                    $"{Tokens.If} IsObject({propertyAttributes.BackingField}) {Tokens.Then}",
+                    $"{FourSpaces}{Tokens.Set} {propertyAttributes.PropertyName} = {propertyAttributes.BackingField}",
+                    Tokens.Else,
+                    $"{FourSpaces}{propertyAttributes.PropertyName} = {propertyAttributes.BackingField}",
+                    $"{Tokens.End} {Tokens.If}",
+                    Environment.NewLine);
+            }
+
+            if (!_codeBuilder.TryBuildPropertyGetCodeBlock(propertyAttributes.Declaration, propertyAttributes.PropertyName, out var propertyGet, content: $"{FourSpaces}{getterContent}"))
+            {
+                throw new ArgumentException();
+            }
+
+            return (propertyGet, propertyLet, propertySet);
+        }
+
+        public string BuildUserDefinedTypeDeclaration(IObjectStateUDT objectStateUDT, IEnumerable<IEncapsulateFieldCandidate> candidates)
+        {
+            var selected = candidates.Where(c => c.EncapsulateFlag);
+
+            var newUDTMembers = selected
+                .Select(m => (m.Declaration as VariableDeclaration, m.BackingIdentifier));
+
+            return _codeBuilder.BuildUserDefinedTypeDeclaration(objectStateUDT.AsTypeName, newUDTMembers);
+        }
+
+        public string BuildObjectStateFieldDeclaration(IObjectStateUDT objectStateUDT)
+        {
+            return $"{Accessibility.Private} {objectStateUDT.IdentifierName} {Tokens.As} {objectStateUDT.AsTypeName}";
+        }
+    }
+}

Rubberduck.Refactorings/EncapsulateField/EncapsulateFieldInsertNewCode/EncapsulateFieldCodeBuilderFactory.cs

@@ -0,0 +1,22 @@
+
+namespace Rubberduck.Refactorings.EncapsulateField
+{
+    public interface IEncapsulateFieldCodeBuilderFactory
+    {
+        IEncapsulateFieldCodeBuilder Create();
+    }
+
+    public class EncapsulateFieldCodeBuilderFactory : IEncapsulateFieldCodeBuilderFactory
+    {
+        private readonly ICodeBuilder _codeBuilder;
+        public EncapsulateFieldCodeBuilderFactory(ICodeBuilder codeBuilder)
+        {
+            _codeBuilder = codeBuilder;
+        }
+
+        public IEncapsulateFieldCodeBuilder Create()
+        {
+            return new EncapsulateFieldCodeBuilder(_codeBuilder);
+        }
+    }
+}