Add param block and cmdletbinding in correction

Kapil Borle · Kapil Borle · commit 87a78141991a · 2017-05-01T15:17:33.000-07:00
diff --git a/rules/UseSupportsShouldProcess.cs b/rules/UseSupportsShouldProcess.cs
@@ -179,7 +179,8 @@ private List<CorrectionExtent> GetCorrections(
                 else
                 {
                     // has no cmdletbinding attribute
-                    // add the attribute and supportsshouldprocess argument
+                    // hence, add the attribute and supportsshouldprocess argument
+                    correctionExtents.Add(GetCorrectionExtent(paramBlockAst));
                 }
             }
             else
@@ -238,7 +239,21 @@ private static bool TryGetCmdletBindingAttribute(
             return attributeAst != null;
         }
 
+        private static CorrectionExtent GetCorrectionExtent(ParamBlockAst paramBlockAst)
+        {
+
+            string correctionText = new String(' ', paramBlockAst.Extent.StartColumnNumber - 1)
+                + "CmdletBinding(SupportsShouldProcess)"
+                + Environment.NewLine;
 
+            return new CorrectionExtent(
+                paramBlockAst.Extent.StartLineNumber,
+                paramBlockAst.Extent.EndLineNumber,
+                1,
+                1,
+                correctionText,
+                null);
+        }
         private static CorrectionExtent GetCorrectionExtent(AttributeAst cmdletBindingAttributeAst)
         {
             // 1 for 1 based offset and 1 for the next position.

Original file line number	Diff line number	Diff line change
`@@ -179,7 +179,8 @@ private List<CorrectionExtent> GetCorrections(`
`179`	`179`	`else`
`180`	`180`	`{`
`181`	`181`	`// has no cmdletbinding attribute`
`182`		`- // add the attribute and supportsshouldprocess argument`
	`182`	`+ // hence, add the attribute and supportsshouldprocess argument`
	`183`	`+ correctionExtents.Add(GetCorrectionExtent(paramBlockAst));`
`183`	`184`	`}`
`184`	`185`	`}`
`185`	`186`	`else`
`@@ -238,7 +239,21 @@ private static bool TryGetCmdletBindingAttribute(`
`238`	`239`	`return attributeAst != null;`
`239`	`240`	`}`
`240`	`241`
	`242`	`+ private static CorrectionExtent GetCorrectionExtent(ParamBlockAst paramBlockAst)`
	`243`	`+ {`
	`244`	`+`
	`245`	`+ string correctionText = new String(' ', paramBlockAst.Extent.StartColumnNumber - 1)`
	`246`	`+ + "CmdletBinding(SupportsShouldProcess)"`
	`247`	`+ + Environment.NewLine;`
`241`	`248`
	`249`	`+ return new CorrectionExtent(`
	`250`	`+ paramBlockAst.Extent.StartLineNumber,`
	`251`	`+ paramBlockAst.Extent.EndLineNumber,`
	`252`	`+ 1,`
	`253`	`+ 1,`
	`254`	`+ correctionText,`
	`255`	`+ null);`
	`256`	`+ }`
`242`	`257`	`private static CorrectionExtent GetCorrectionExtent(AttributeAst cmdletBindingAttributeAst)`
`243`	`258`	`{`
`244`	`259`	`// 1 for 1 based offset and 1 for the next position.`