Add insertBefore and insertAfter fixes for non-Gradle files

nebula-plugins · Apr 18, 2016 · e1b5db1 · e1b5db1
1 parent 1d2ea1f
commit e1b5db1
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/main/groovy/com/netflix/nebula/lint/GradleViolation.groovy b/src/main/groovy/com/netflix/nebula/lint/GradleViolation.groovy
@@ -66,7 +66,16 @@ class GradleViolation extends Violation {
 
     GradleViolation delete(ASTNode node) {
         fixes += new GradleLintReplaceWith(buildFile, node.lineNumber..node.lastLineNumber, node.columnNumber, node.lastColumnNumber, '')
-//        fixes += new GradleLintDelete(buildFile, node.lineNumber..node.lastLineNumber)
+        this
+    }
+
+    GradleViolation insertAfter(File file, Integer afterLine, String changes) {
+        fixes += new GradleLintInsertAfter(file, afterLine, changes)
+        this
+    }
+
+    GradleViolation insertBefore(File file, Integer beforeLine, String changes) {
+        fixes += new GradleLintInsertBefore(file, beforeLine, changes)
         this
     }
 }