diff --git a/src/org/ntlab/pushPullRefactoring/PushPullProcessor.java b/src/org/ntlab/pushPullRefactoring/PushPullProcessor.java index b60bbbc..5c7e46b 100644 --- a/src/org/ntlab/pushPullRefactoring/PushPullProcessor.java +++ b/src/org/ntlab/pushPullRefactoring/PushPullProcessor.java @@ -157,10 +157,10 @@ return false; } }); - final TypeDeclaration typeDeclaration = (TypeDeclaration)typedec[0]; + final FieldDeclaration typeDeclaration = searchFieldDeclaration(unit, "State"); ASTRewrite rewrite = ASTRewrite.create(typeDeclaration.getAST()); - ListRewrite rewrites = rewrite.getListRewrite(typeDeclaration, TypeDeclaration.MODIFIERS2_PROPERTY); - Statement mycomment = (Statement) rewrite.createStringPlaceholder("//mycomment"+System.getProperty("line.separator"), ASTNode.EMPTY_STATEMENT); + ListRewrite rewrites = rewrite.getListRewrite(typeDeclaration, FieldDeclaration.MODIFIERS2_PROPERTY); + Statement pushable = (Statement) rewrite.createStringPlaceholder("@Pushable"+System.getProperty("line.separator"), ASTNode.EMPTY_STATEMENT); Statement statement = (Statement) rewrite.createStringPlaceholder("Object obj = new Object();", ASTNode.VARIABLE_DECLARATION_STATEMENT); PackageDeclaration packageDeclaration = unit.getPackage(); FieldDeclaration fd = searchFieldDeclaration(unit, "State"); @@ -170,7 +170,7 @@ TextEditGroup editGroup = new TextEditGroup("new class edit group"); //rewrites.insertFirst(mycomment, editGroup); - rewrites.insertFirst(mycomment, editGroup); + rewrites.insertFirst(pushable, editGroup); IPath path = unit.getJavaElement().getPath(); //�V�K�t�@�C���쐬