Make Range parameter take Range type of int[] type arg

Author: Kapil Borle

Engine/Commands/InvokeFormatterCommand.cs

@@ -26,6 +26,7 @@ public class InvokeFormatterCommand : PSCmdlet, IOutputWriter
     {
         private const string defaultSettingsPreset = "CodeFormatting";
         private Settings inputSettings;
+        private Range range;
 
         /// <summary>
         /// The script text to be formated.
@@ -43,19 +44,10 @@ public class InvokeFormatterCommand : PSCmdlet, IOutputWriter
         [ValidateNotNull]
         public object Settings { get; set; } = defaultSettingsPreset;
 
-#if DEBUG
         [Parameter(Mandatory = false)]
-        public Range Range { get; set; }
-
-        [Parameter(Mandatory = false, ParameterSetName = "NoRange")]
-        public int StartLineNumber { get; set; } = -1;
-        [Parameter(Mandatory = false, ParameterSetName = "NoRange")]
-        public int StartColumnNumber { get; set; } = -1;
-        [Parameter(Mandatory = false, ParameterSetName = "NoRange")]
-        public int EndLineNumber { get; set; } = -1;
-        [Parameter(Mandatory = false, ParameterSetName = "NoRange")]
-        public int EndColumnNumber { get; set; } = -1;
-
+        [ValidateNotNull]
+        public object Range { get; set; }
+#if DEBUG
         /// <summary>
         /// Attaches to an instance of a .Net debugger
         /// </summary>
@@ -85,6 +77,22 @@ protected override void BeginProcessing()
             }
 #endif
 
+            if (Range != null)
+            {
+                try
+                {
+                    range = GetRange(Range);
+                }
+                catch (Exception e)
+                {
+                    this.ThrowTerminatingError(new ErrorRecord(
+                            e,
+                            "RANGE_ERROR",
+                            ErrorCategory.InvalidArgument,
+                            Range));
+                }
+            }
+
             try
             {
                 inputSettings = PSSASettings.Create(Settings, this.MyInvocation.PSScriptRoot, this);
@@ -93,7 +101,7 @@ protected override void BeginProcessing()
             {
                 this.ThrowTerminatingError(new ErrorRecord(
                         e,
-                        "SETTNGS_ERROR",
+                        "SETTINGS_ERROR",
                         ErrorCategory.InvalidData,
                         Settings));
             }
@@ -114,18 +122,30 @@ protected override void ProcessRecord()
         {
             // todo add tests to check range formatting
             string formattedScriptDefinition;
-#if DEBUG
-            var range = Range;
-            if (this.ParameterSetName.Equals("NoRange"))
+            formattedScriptDefinition = Formatter.Format(ScriptDefinition, inputSettings, range, this);
+            this.WriteObject(formattedScriptDefinition);
+        }
+
+        private static Range GetRange(object rangeObj)
+        {
+            var range = rangeObj as Range;
+            if (range == null)
             {
-                range = new Range(StartLineNumber, StartColumnNumber, EndLineNumber, EndColumnNumber);
-            }
+                var intArr = rangeObj as int[];
+                if (intArr == null)
+                {
+                    throw new ArgumentException("Argument should be of type Range or int[].");
+                }
 
-            formattedScriptDefinition = Formatter.Format(ScriptDefinition, inputSettings, range, this);
-#endif // DEBUG
+                if (intArr.Length != 4)
+                {
+                    throw new ArgumentException("Integer array should be of length 4.");
+                }
 
-            formattedScriptDefinition = Formatter.Format(ScriptDefinition, inputSettings, null, this);
-            this.WriteObject(formattedScriptDefinition);
+                range = new Range(intArr[0], intArr[1], intArr[2], intArr[3]);
+            }
+
+            return range;
         }
 
         private void ValidateInputSettings()