Add corrections for operator violations

Author: Kapil Borle

Rules/UseWhitespace.cs

@@ -16,6 +16,7 @@
 using System.Globalization;
 using System.Linq;
 using System.Management.Automation.Language;
+using System.Text;
 using Microsoft.Windows.PowerShell.ScriptAnalyzer.Generic;
 
 namespace Microsoft.Windows.PowerShell.ScriptAnalyzer.BuiltinRules
@@ -156,10 +157,7 @@ private IEnumerable<DiagnosticRecord> FindOpenParenViolations(TokenOperations to
         private IEnumerable<CorrectionExtent> GetOpenBracketCorrections(Token token)
         {
             yield return new CorrectionExtent(
-                token.Extent.StartLineNumber,
-                token.Extent.EndLineNumber,
-                token.Extent.StartColumnNumber,
-                token.Extent.EndColumnNumber,
+                token.Extent,
                 whiteSpace + token.Text,
                 token.Extent.File,
                 GetError(ErrorKind.Brace));
@@ -173,7 +171,8 @@ private bool IsPreviousTokenApartByWhitespace(LinkedListNode<Token> tokenNode)
 
         private IEnumerable<DiagnosticRecord> FindOperatorViolations(TokenOperations tokenOperations)
         {
-            Func<LinkedListNode<Token>, bool> predicate = tokenNode => {
+            Func<LinkedListNode<Token>, bool> predicate = tokenNode =>
+            {
                 return tokenNode.Previous != null
                     && IsPreviousTokenOnSameLine(tokenNode)
                     && IsPreviousTokenApartByWhitespace(tokenNode);
@@ -202,11 +201,35 @@ private IEnumerable<DiagnosticRecord> FindOperatorViolations(TokenOperations tok
                         GetDiagnosticSeverity(),
                         tokenOperations.Ast.Extent.File,
                         null,
-                        null);
+                        GetOperatorCorrections(tokenNode.Value, hasWhitespaceBefore, hasWhitespaceAfter).ToList());
                 }
             }
         }
 
+        private IEnumerable<CorrectionExtent> GetOperatorCorrections(
+            Token token,
+            bool hasWhitespaceBefore,
+            bool hasWhitespaceAfter)
+        {
+            var sb = new StringBuilder();
+            if (!hasWhitespaceBefore)
+            {
+                sb.Append(whiteSpace);
+            }
+
+            sb.Append(token.Text);
+            if (!hasWhitespaceAfter)
+            {
+                sb.Append(whiteSpace);
+            }
+
+            yield return new CorrectionExtent(
+                token.Extent,
+                sb.ToString(),
+                token.Extent.File,
+                GetError(ErrorKind.Operator));
+        }
+
         private bool IsOperator(Token token)
         {
             return TokenTraits.HasTrait(token.Kind, TokenFlags.AssignmentOperator)