src/main/java/cuchaz/enigma/analysis/DropImportAstTransform.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

package cuchaz.enigma.analysis;

import com.strobel.decompiler.languages.java.ast.AstNode;
import com.strobel.decompiler.languages.java.ast.DepthFirstAstVisitor;
import com.strobel.decompiler.languages.java.ast.ImportDeclaration;
import com.strobel.decompiler.languages.java.ast.PackageDeclaration;
import com.strobel.decompiler.languages.java.ast.transforms.IAstTransform;

public final class DropImportAstTransform implements IAstTransform {
	public static final DropImportAstTransform INSTANCE = new DropImportAstTransform();

	private DropImportAstTransform() {
	}

	@Override
	public void run(AstNode compilationUnit) {
		compilationUnit.acceptVisitor(new Visitor(), null);
	}

	static class Visitor extends DepthFirstAstVisitor<Void, Void> {
		@Override
		public Void visitPackageDeclaration(PackageDeclaration node, Void data) {
			node.remove();
			return null;
		}

		@Override
		public Void visitImportDeclaration(ImportDeclaration node, Void data) {
			node.remove();
			return null;
		}
	}
}