com.mountainminds.eclemma.internal.core.instr.SourceLocation.java Source code

Introduction

Here is the source code for com.mountainminds.eclemma.internal.core.instr.SourceLocation.java

Source

/*******************************************************************************
 * Copyright (c) 2006 Mountainminds GmbH & Co. KG
 * This software is provided under the terms of the Eclipse Public License v1.0
 * See http://www.eclipse.org/legal/epl-v10.html.
 *
 * $Id: SourceLocation.java 84 2006-09-14 20:00:01Z mho $
 ******************************************************************************/
package com.mountainminds.eclemma.internal.core.instr;

import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.zip.ZipEntry;
import java.util.zip.ZipInputStream;

import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IPath;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.Path;
import org.eclipse.jdt.core.IPackageFragmentRoot;
import org.eclipse.jdt.core.JavaModelException;
import org.eclipse.osgi.util.NLS;

import com.mountainminds.eclemma.core.EclEmmaStatus;
import com.mountainminds.eclemma.core.ISourceLocation;
import com.mountainminds.eclemma.internal.core.CoreMessages;
import com.mountainminds.eclemma.internal.core.EclEmmaCorePlugin;

/**
 * Implementation of {@link ISourceLocation}.
 * 
 * @author  Marc R. Hoffmann
 * @version $Revision: 84 $
 */
public class SourceLocation implements ISourceLocation {

    private static final String JAVA_EXT = ".java"; //$NON-NLS-1$

    public static ISourceLocation findLocation(IPackageFragmentRoot root) throws JavaModelException {
        if (root.getKind() == IPackageFragmentRoot.K_SOURCE) {
            IPath path = EclEmmaCorePlugin.getAbsolutePath(root.getPath());
            return new SourceLocation(path, new Path(IPackageFragmentRoot.DEFAULT_PACKAGEROOT_PATH));
        } else {
            IPath path = root.getSourceAttachmentPath();
            if (path != null) {
                path = EclEmmaCorePlugin.getAbsolutePath(path);
                return new SourceLocation(path, root.getSourceAttachmentRootPath());
            } else {
                return null;
            }
        }
    }

    public static ISourceLocation[] findLocations(IPackageFragmentRoot[] roots) throws JavaModelException {
        List l = new ArrayList();
        for (int i = 0; i < roots.length; i++) {
            ISourceLocation loc = findLocation(roots[i]);
            if (loc != null) {
                l.add(loc);
            }
        }
        return (ISourceLocation[]) l.toArray(new ISourceLocation[l.size()]);
    }

    private IPath path;
    private IPath rootpath;

    /**
     * Creates a source location for the given path and root path.
     * 
     * @param path
     * @param rootpath
     */
    public SourceLocation(IPath path, IPath rootpath) {
        this.path = path;
        this.rootpath = rootpath;
    }

    public IPath getPath() {
        return path;
    }

    public IPath getRootPath() {
        return rootpath;
    }

    public boolean isArchive() {
        return path.toFile().isFile();
    }

    public void extract(IProgressMonitor monitor) throws CoreException {
        if (isArchive()) {
            monitor.beginTask(NLS.bind(CoreMessages.ExtractingSourceArchive_task, path), 1);
            String prefix = rootpath == null ? "" : rootpath.toString(); //$NON-NLS-1$
            byte[] buffer = new byte[0x1000];
            IPath srcfolder = EclEmmaCorePlugin.getInstance().getStateFiles().getSourceFolder(path);
            if (!srcfolder.toFile().exists()) {
                try {
                    ZipInputStream zip = new ZipInputStream(new FileInputStream(path.toFile()));
                    while (true) {
                        ZipEntry entry = zip.getNextEntry();
                        if (entry == null)
                            break;
                        if (!entry.isDirectory() && entry.getName().startsWith(prefix)
                                && entry.getName().endsWith(JAVA_EXT)) {
                            IPath path = srcfolder.append(entry.getName().substring(prefix.length()));
                            path.removeLastSegments(1).toFile().mkdirs();
                            OutputStream out = new FileOutputStream(path.toFile());
                            int len;
                            while ((len = zip.read(buffer)) != -1) {
                                out.write(buffer, 0, len);
                            }
                            out.close();
                        }
                        zip.closeEntry();
                    }
                    zip.close();
                } catch (IOException e) {
                    throw new CoreException(EclEmmaStatus.SOURCE_EXTRACTION_ERROR.getStatus(path, e));
                }
            }
            path = srcfolder;
            rootpath = null;
        }
        monitor.done();
    }

}