[ASTParser 2]メソッドの中身を削除したり，フィールドを削除する

今回はフィールドの削除や、メソッドの中身を削除する方法について。

例によってソースコード全体についてはここ。

Sample2.java

1. package info.haxahaxa.astparser.sample;
3. import info.haxahaxa.astparser.Envs;
4. import info.haxahaxa.astparser.SourceFile;
5. import info.haxahaxa.astparser.sample.visitor.*;
7. import java.io.File;
9. import org.eclipse.core.runtime.NullProgressMonitor;
10. import org.eclipse.jdt.core.dom.AST;
11. import org.eclipse.jdt.core.dom.ASTParser;
12. import org.eclipse.jdt.core.dom.ASTVisitor;
13. import org.eclipse.jdt.core.dom.CompilationUnit;
14. import org.eclipse.jface.text.Document;
15. import org.eclipse.text.edits.TextEdit;
17. /**
18. * importを削除したり，メソッドを削除したり，フィールドを削除したり，javadocコメントを削除するサンプル
19. *
20. * @author satanabe1
21. *
22. */
23. public class Sample2 {
25.     private static ASTVisitor visitor = new SampleVisitor2();
27.     public static void main(String[] arg) throws Exception {
28.         SourceFile sourceFile = new SourceFile("src" + File.separator
29.                 + "samples" + File.separator + "AntFileGen.java");
30.         CompilationUnit unit;
31.         ASTParser astParser = ASTParser.newParser(AST.JLS4);
32.         // 以下の setBindingsRecovery setStatementsRecovery はおまじない．
33.         // 完成しているソースコードを解析する時は呼ぶ必要ない．
34.         // 詳しく知りたいならば，IMBのASTParser関連のドキュメントとかを参照すべき．
35.         astParser.setBindingsRecovery(true);
36.         astParser.setStatementsRecovery(true);
37.         // 次の setResolveBindings と setEnvironment が重要！！
38.         // setResolveBindings(true) をしておかないとまともに解析はできない．
39.         // setResolveBindings をまともに機能させるために setEnvironment が必要．
40.         astParser.setResolveBindings(true);
41.         // setEnvironment の第一引数にはクラスパスの配列．第二引数にはソースコードを検索するパスの配列
42.         // 第三第四については何も考えず null, true ．納得いかない時はIBMのASTPa...
43.         astParser.setEnvironment(Envs.getClassPath(), Envs.getSourcePath(),
44.                 null, true);
46.         // 解析対象のソースコードの入力とか
47.         astParser.setUnitName(sourceFile.getFilePath());// なんでもいいから名前を設定しておく
48.         astParser.setSource(sourceFile.getSourceCode().toCharArray());// 解析対象コードを設定する
49.         unit = (CompilationUnit) astParser.createAST(new NullProgressMonitor());
50.         unit.recordModifications();// ASTへの操作履歴のようなものを有効に
51.         // 解析 & 変換実行
52.         unit.accept(visitor);
54.         // 変換結果を文字列で取得
55.         String code = getCode(sourceFile.getSourceCode(), unit);
56.         System.out.println(code);
57.     }
59.     /**
60.      * ASTを文字列のコードに戻すメソッド
61.      *
62.      * @param code
63.      *            元のコード
64.      * @param unit
65.      *            ASTVisitorで操作を行ったヤツ
66.      * @return ソースコード
67.      */
68.     private static String getCode(String code, CompilationUnit unit) {
69.         org.eclipse.jface.text.IDocument eDoc = new Document(code);
70.         TextEdit edit = unit.rewrite(eDoc, null);
71.         try {
72.             edit.apply(eDoc);
73.             return eDoc.get();
74.         } catch (Exception ex) {
75.             ex.printStackTrace();
76.             return null;
77.         }
78.     }
79. }

SampleVisitor2.java

1. package info.haxahaxa.astparser.sample.visitor;
3. import org.eclipse.jdt.core.dom.ASTVisitor;
4. import org.eclipse.jdt.core.dom.FieldDeclaration;
5. import org.eclipse.jdt.core.dom.ImportDeclaration;
6. import org.eclipse.jdt.core.dom.Javadoc;
7. import org.eclipse.jdt.core.dom.MethodDeclaration;
9. /**
10. * メソッドの中身を削除したり，フィールドを削除するサンプル
11. *
12. * @author satanabe1
13. *
14. */
15. public class SampleVisitor2 extends ASTVisitor {
16.     /**
17.      * フィールド宣言をdelete!
18.      */
19.     public boolean visit(FieldDeclaration node) {
20.         node.delete();
21.         return super.visit(node);
22.     }
24.     /**
25.      * javadocもついでにdelete!
26.      */
27.     public boolean visit(Javadoc node) {
28.         node.delete();
29.         return super.visit(node);
30.     }
32.     /**
33.      * ダメ押しにimportもdelete!
34.      */
35.     public boolean visit(ImportDeclaration node) {
36.         node.delete();
37.         return super.visit(node);
38.     }
40.     /**
41.      * メソッド内の処理をdelete!
42.      */
43.     public boolean visit(MethodDeclaration node) {
44.         node.getBody().statements().clear();
45.         return super.visit(node);
46.     }
47. }

Sample2.javaについては前回の記事にあるSample1.javaとほぼ変わらない。
変更点としては、

1. 25行目のSampleVisitor2クラスのインスタンスを生成している
2. 68-78行に新しいメソッドの追加
3. 55-56行でのソースコードの標準出力への出力

の3点となっている。
新たに追加されたgetCodeメソッドのしている事をざっくり言ってしまえば、CompilationUnitを使っていじくり回された変更を、元のソースコードに適用して文字列に直した内容を返している。


で、
今回色々削除したり削除したり削除したりするコードがSampleVisitor2.javaとなっている。
見ての通り各visitメソッドは2行しかない。
基本はdeleteメソッドを呼ぶだけであっさり何もかも無くなっていく。

少しだけ違う処理をしているのは44行目の
node.getBody().statements().clear();
なんのことはない、これはListクラスのclearメソッドを呼んでいるだけ。
このあたりで難しい事は何もないはず。

ちなみにpackage宣言なんかはCompilationUnita側からじゃないといじれない。

まとめ
  delete!
  delete!!
  delete!!!
  clear!!!!!