Add support to comment out CtStatement

Fixes INRIA#2013
LakshyAAAgrawal · Jul 13, 2020 · fbeaa96 · fbeaa96
1 parent d62c3eb
commit fbeaa96
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/src/main/java/spoon/reflect/code/CtStatement.java b/src/main/java/spoon/reflect/code/CtStatement.java
@@ -56,6 +56,11 @@ public interface CtStatement extends CtCodeElement {
 	@PropertySetter(role = LABEL)
 	<T extends CtStatement> T setLabel(String label);
 
+	/**
+	 * Replace the statement with a CtComment having the statement as text
+	 */
+	void comment();
+
 	@Override
 	CtStatement clone();
 }
diff --git a/src/main/java/spoon/support/reflect/code/CtStatementImpl.java b/src/main/java/spoon/support/reflect/code/CtStatementImpl.java
@@ -25,6 +25,7 @@
 import spoon.reflect.visitor.CtInheritanceScanner;
 
 import java.util.ArrayList;
+import java.util.Collections;
 import java.util.List;
 
 import static spoon.reflect.path.CtRole.LABEL;
@@ -282,6 +283,16 @@ public <T extends CtStatement> T setLabel(String label) {
 		this.label = label;
 		return (T) this;
 	}
+
+	@Override
+	public void comment() {
+		if (!isParentInitialized()) {
+			// already not in a tree, commenting wouldn't make a difference
+			return;
+		}
+		// comment is implemented as replace by a comment
+		replace(getFactory().Code().createInlineComment(toString()));
+	}
 
 	@Override
 	public CtStatement clone() {