net.atos.optimus.common.tools.jdt.PackageFragmentHelper.java Source code

Introduction

Here is the source code for net.atos.optimus.common.tools.jdt.PackageFragmentHelper.java

Source

/**
 * Optimus, framework for Model Transformation
 *
 * Copyright (C) 2013 Worldline or third-party contributors as
 * indicated by the @author tags or express copyright attribution
 * statements applied by the authors.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */
/** 
 * Atos WorldLine
 *
 * $Id: PackageFragmentHelper.java,v 1.3 2009/06/18 16:08:09 a125788 Exp $
 * 
 * $Log: PackageFragmentHelper.java,v $
 * Revision 1.3  2009/06/18 16:08:09  a125788
 * MVA : Enhanced file retrieval within workspace, in case projects are interlocked
 *
 * Revision 1.2  2008/03/04 08:08:41  a125788
 * MVA : Merged 1-6-patches branch with HEAD.
 *
 * Revision 1.1.2.1  2008/02/28 15:30:45  tmpffoucart
 * FFo: Add a condition
 *
 * Revision 1.1  2007/07/09 08:14:11  mrifflard
 * MRI : Move
 *
 * Revision 1.1  2007/04/19 14:59:33  cdhalluin
 * Initial version
 *
 *
 **/

package net.atos.optimus.common.tools.jdt;

import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.IWorkspaceRoot;
import org.eclipse.core.resources.ResourcesPlugin;
import org.eclipse.core.runtime.IPath;
import org.eclipse.core.runtime.Path;
import org.eclipse.jdt.core.ICompilationUnit;
import org.eclipse.jdt.core.IJavaElement;
import org.eclipse.jdt.core.IPackageFragment;
import org.eclipse.jdt.core.JavaCore;
import org.eclipse.jdt.core.JavaModelException;

/**
 *  @author Maxence Vanbsien (mvaawl@gmail.com)
 *  @since 1.0
 */
public class PackageFragmentHelper {

    /**
     * Return the IPackageFragment corresponding to the Path
     * 
     * @param Path
     * @return
     */
    public static IPackageFragment getPackageFragment(Path filePath) throws JavaModelException {

        IFile f = getFileForLocation(ResourcesPlugin.getWorkspace().getRoot(), filePath);

        if (f != null) {
            ICompilationUnit compilationUnit = JavaCore.createCompilationUnitFrom(f);
            IPackageFragment packageFragment = null;
            IJavaElement temp = compilationUnit;
            while (packageFragment == null && temp != null) {
                if (temp instanceof IPackageFragment)
                    packageFragment = (IPackageFragment) temp;
                else
                    temp = temp.getParent();
            }
            return packageFragment;
        }
        return null;
    }

    /**
     * Method to replace the WorkspaceRoot.getFileForLocation() methods, since
     * it is not able to retrieve the right file when projects are interlocked
     * 
     * @param workspaceRoot :
     *            WorkspaceRoot Instance
     * @param filePath :
     *            File Path
     * @return IFile if contained by lower level project.
     */
    private static IFile getFileForLocation(IWorkspaceRoot workspaceRoot, IPath filePath) {
        if (filePath.equals(workspaceRoot.getFullPath()))
            return workspaceRoot.getFileForLocation(new Path("/"));
        IProject[] projects = workspaceRoot.getProjects();
        IFile myFile = null;
        int segmentsMatching = 0;
        for (IProject project : projects) {
            IPath projectLocation = project.getLocation();
            if (projectLocation.isPrefixOf(filePath) && projectLocation.segmentCount() > segmentsMatching) {
                IPath fileLocalPath = filePath.removeFirstSegments(projectLocation.segmentCount());
                myFile = project.getFile(fileLocalPath);
                segmentsMatching = projectLocation.segmentCount();
            }
        }
        return myFile;
    }

}