001: /*
002: * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
003: *
004: * Copyright 1997-2007 Sun Microsystems, Inc. All rights reserved.
005: *
006: * The contents of this file are subject to the terms of either the GNU
007: * General Public License Version 2 only ("GPL") or the Common
008: * Development and Distribution License("CDDL") (collectively, the
009: * "License"). You may not use this file except in compliance with the
010: * License. You can obtain a copy of the License at
011: * http://www.netbeans.org/cddl-gplv2.html
012: * or nbbuild/licenses/CDDL-GPL-2-CP. See the License for the
013: * specific language governing permissions and limitations under the
014: * License. When distributing the software, include this License Header
015: * Notice in each file and include the License file at
016: * nbbuild/licenses/CDDL-GPL-2-CP. Sun designates this
017: * particular file as subject to the "Classpath" exception as provided
018: * by Sun in the GPL Version 2 section of the License file that
019: * accompanied this code. If applicable, add the following below the
020: * License Header, with the fields enclosed by brackets [] replaced by
021: * your own identifying information:
022: * "Portions Copyrighted [year] [name of copyright owner]"
023: *
024: * Contributor(s):
025: *
026: * The Original Software is NetBeans. The Initial Developer of the Original
027: * Software is Sun Microsystems, Inc. Portions Copyright 1997-2006 Sun
028: * Microsystems, Inc. All Rights Reserved.
029: *
030: * If you wish your version of this file to be governed by only the CDDL
031: * or only the GPL Version 2, indicate your decision by adding
032: * "[Contributor] elects to include this software in this distribution
033: * under the [CDDL or GPL Version 2] license." If you do not indicate a
034: * single choice of license, a recipient has the option to distribute
035: * your version of this file under either the CDDL, the GPL Version 2 or
036: * to extend the choice of license to its licensees as provided above.
037: * However, if you add GPL Version 2 code and therefore, elected the GPL
038: * Version 2 license, then the option applies only if the new code is
039: * made subject to such option by the copyright holder.
040: */package org.netbeans.modules.vmd.midp.propertyeditors.api.usercode;
041:
042: import com.sun.source.tree.CompilationUnitTree;
043: import com.sun.source.tree.MethodTree;
044: import com.sun.source.tree.Tree;
045: import com.sun.source.tree.Tree.Kind;
046: import java.io.IOException;
047: import javax.lang.model.element.ExecutableElement;
048: import javax.lang.model.element.Modifier;
049: import javax.lang.model.element.TypeElement;
050: import javax.lang.model.util.ElementFilter;
051: import org.netbeans.api.java.source.CompilationController;
052: import org.netbeans.api.java.source.JavaSource;
053: import org.netbeans.api.java.source.Task;
054: import org.netbeans.modules.vmd.api.io.DataObjectContext;
055: import org.openide.ErrorManager;
056:
057: /**
058: *
059: * @author Anton Chechel
060: */
061: public final class CodeUtils {
062:
063: private CodeUtils() {
064: }
065:
066: public static int getMethodOffset(DataObjectContext context) {
067: final int[] offset = new int[1];
068: JavaSource javaSource = JavaSource.forFileObject(context
069: .getDataObject().getPrimaryFile());
070:
071: try {
072: javaSource.runUserActionTask(
073: new Task<CompilationController>() {
074:
075: public void run(
076: CompilationController compilationController)
077: throws Exception {
078: compilationController
079: .toPhase(JavaSource.Phase.ELEMENTS_RESOLVED);
080: CompilationUnitTree unitTree = compilationController
081: .getCompilationUnit();
082: java.util.List<? extends TypeElement> types = compilationController
083: .getTopLevelElements();
084: if (types.size() > 0) {
085: TypeElement type = types.get(0);
086: java.util.List<ExecutableElement> methods = ElementFilter
087: .methodsIn(type
088: .getEnclosedElements());
089: for (ExecutableElement method : methods) {
090: if (!method.getModifiers()
091: .contains(Modifier.STATIC)) {
092: Tree methodTree = compilationController
093: .getTrees().getTree(
094: method);
095: if (methodTree.getKind() == Kind.METHOD) {
096: Tree tree = ((MethodTree) methodTree)
097: .getBody();
098: if (tree != null) {
099: offset[0] = (int) compilationController
100: .getTrees()
101: .getSourcePositions()
102: .getStartPosition(
103: unitTree,
104: tree) + 1;
105: break;
106: }
107: }
108: }
109: }
110: }
111: }
112: }, true);
113: } catch (IOException e) {
114: ErrorManager.getDefault().notify(e);
115: }
116: return offset[0];
117: }
118:
119: }
|