KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > netbeans > modules > refactoring > java > plugins > RenameTransformer


1 /*
2  * The contents of this file are subject to the terms of the Common Development
3  * and Distribution License (the License). You may not use this file except in
4  * compliance with the License.
5  *
6  * You can obtain a copy of the License at http://www.netbeans.org/cddl.html
7  * or http://www.netbeans.org/cddl.txt.
8  *
9  * When distributing Covered Code, include this CDDL Header Notice in each file
10  * and include the License file at http://www.netbeans.org/cddl.txt.
11  * If applicable, add the following below the CDDL Header, with the fields
12  * enclosed by brackets [] replaced by your own identifying information:
13  * "Portions Copyrighted [year] [name of copyright owner]"
14  *
15  * The Original Software is NetBeans. The Initial Developer of the Original
16  * Software is Sun Microsystems, Inc. Portions Copyright 1997-2006 Sun
17  * Microsystems, Inc. All Rights Reserved.
18  */
19
20 package org.netbeans.modules.refactoring.java.plugins;
21
22 import com.sun.source.tree.*;
23 import com.sun.source.util.TreePath;
24 import javax.lang.model.element.*;
25 import org.netbeans.api.java.source.WorkingCopy;
26
27 /**
28  *
29  * @author Jan Becicka
30  */
31 public class RenameTransformer extends SearchVisitor {
32
33     private String JavaDoc newName;
34
35     public RenameTransformer(String JavaDoc newName, WorkingCopy workingCopy) {
36         super(workingCopy);
37         this.newName = newName;
38     }
39
40     @Override JavaDoc
41     public Tree visitIdentifier(IdentifierTree node, Element p) {
42         renameUsageIfMatch(getCurrentPath(), node,p);
43         return super.visitIdentifier(node, p);
44     }
45
46     @Override JavaDoc
47     public Tree visitMemberSelect(MemberSelectTree node, Element p) {
48         renameUsageIfMatch(getCurrentPath(), node,p);
49         return super.visitMemberSelect(node, p);
50     }
51     
52     private void renameUsageIfMatch(TreePath path, Tree tree, Element elementToFind) {
53         if (workingCopy.getTreeUtilities().isSynthetic(path))
54             return;
55         Element el = workingCopy.getTrees().getElement(path);
56         if (el==null)
57             return;
58         
59         if (el != null && elementToFind!=null && elementToFind.getKind() == ElementKind.METHOD && el.getKind() == ElementKind.METHOD) {
60             if (el.equals(elementToFind) || workingCopy.getElements().overrides(((ExecutableElement) el), (ExecutableElement) elementToFind, (TypeElement) elementToFind.getEnclosingElement())) {
61                 Tree nju = make.setLabel(tree, newName);
62                 workingCopy.rewrite(tree, nju);
63             }
64         } else if (el.equals(elementToFind)) {
65                 Tree nju = make.setLabel(tree, newName);
66                 workingCopy.rewrite(tree, nju);
67         }
68     }
69
70     @Override JavaDoc
71     public Tree visitMethod(MethodTree tree, Element p) {
72         renameDeclIfMatch(getCurrentPath(), tree, p);
73         return super.visitMethod(tree, p);
74     }
75
76     @Override JavaDoc
77     public Tree visitClass(ClassTree tree, Element p) {
78         renameDeclIfMatch(getCurrentPath(), tree, p);
79         return super.visitClass(tree, p);
80     }
81
82     @Override JavaDoc
83     public Tree visitVariable(VariableTree tree, Element p) {
84         renameDeclIfMatch(getCurrentPath(), tree, p);
85         return super.visitVariable(tree, p);
86     }
87     
88     private void renameDeclIfMatch(TreePath path, Tree tree, Element elementToFind) {
89         if (workingCopy.getTreeUtilities().isSynthetic(path))
90             return;
91         Element el = workingCopy.getTrees().getElement(path);
92         if (elementToFind.equals(el)) {
93             Tree nju = make.setLabel(tree, newName);
94             workingCopy.rewrite(tree, nju);
95         }
96     }
97     
98 }
99
Popular Tags