EMFURIConverter.java

package org.andromda.repositories.emf;

import java.io.InputStream;
import java.net.URL;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.andromda.core.common.AndroMDALogger;
import org.andromda.core.common.ResourceUtils;
import org.apache.log4j.Logger;
import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.resource.impl.ExtensibleURIConverterImpl;

/**
 * Extends the default URIConverterImpl to be able to discover the physical path of URIs when
 * given the moduleSearchPaths.
 *
 * @author Chad Brandon
 * @author Bob Fields
 */
public class EMFURIConverter
    extends ExtensibleURIConverterImpl
    // Use URIConverterImpl for UML2 2.x
{
    /**
     * Creates a new instance of EMFURIConverter taking the <code>moduleSearchPaths</code>
     * and the existing URI Map as arguments. These are the paths used to attempt to normalize a given URI during
     * the call to {@link #normalize(URI)} provided that it couldn't be found in the normal manner.
     *
     * @param moduleSearchPaths the paths to search for modules.
     * @param uriMap
     */
    public EMFURIConverter(final List<String> moduleSearchPaths, Map<URI, URI> uriMap)
    {
        this.normalizedUris = uriMap;
        this.moduleSearchPaths = moduleSearchPaths;
        if (logger.isDebugEnabled())
        {
            for (final String path : moduleSearchPaths)
            {
                logger.debug("Model search path:" + path);
            }
        }
    }

    /**
     * Creates a new instance of EMFURIConverter taking the <code>moduleSearchPaths</code>
     * as an argument. These are the paths used to attempt to normalize a given URI during
     * the call to {@link #normalize(URI)} provided that it couldn't be found in the normal manner.
     *
     * @param moduleSearchPaths the paths to search for modules.
     */
    public EMFURIConverter(final List<String> moduleSearchPaths)
    {
        this.moduleSearchPaths = moduleSearchPaths;
        if (logger.isDebugEnabled())
        {
            for (final String path : moduleSearchPaths)
            {
                logger.debug("Model search path:" + path);
            }
        }
    }

    /**
     * Stores the module search paths.
     */
    private List<String> moduleSearchPaths;

    /**
     * Stores the URIs that have been normalized.
     */
    private Map<URI, URI> normalizedUris = new HashMap<URI, URI>();

    /**
     * The logger instance.
     */
    private static final Logger logger = Logger.getLogger(EMFURIConverter.class);

    /**
     * Overridden to provide the normalization of uris given the module search paths.
     *
     * @see org.eclipse.emf.ecore.resource.URIConverter#normalize(org.eclipse.emf.common.util.URI)
     */
    public URI normalize(final URI uri)
    {
        URI normalizedUri = super.normalize(uri);
        if (normalizedUri.equals(uri))
        {
            if (this.moduleSearchPaths != null)
            {
                if (!this.normalizedUris.containsKey(uri))
                {
                    final String resourceName = uri.toString().replaceAll(
                            ".*(\\\\+|/)",
                            "");
                    for (final String searchPath : moduleSearchPaths)
                    {
                        Date now1 = new Date();
                        final URI fileURI = EMFRepositoryFacadeUtils.createUri(ResourceUtils.normalizePath(searchPath));
                        long ms1 = new Date().getTime() - now1.getTime();
                        if (fileURI != null && fileURI.lastSegment() != null && fileURI.lastSegment().equals(resourceName))
                        {
                            String ms = (ms1 > 100 ? ms1 + " ms" : "");
                            AndroMDALogger.info("referenced model --> '" + fileURI + "' " + ms);
                            normalizedUri = fileURI;
                            this.normalizedUris.put(
                                uri,
                                normalizedUri);
                            break;
                        }

                        final String completePath = ResourceUtils.normalizePath(searchPath + '/' + resourceName);

                        try
                        {
                            Date now = new Date();
                            InputStream stream;
                            URL url = ResourceUtils.toURL(completePath);
                            if (url != null)
                            {
                                try
                                {
                                    stream = url.openStream();
                                    stream.close();
                                    long ms2 = new Date().getTime() - now.getTime();
                                    String ms = (ms2 > 100 ? ms2 + " ms" : "");
                                    AndroMDALogger.info("referenced model --> '" + url + "' " + ms);
                                }
                                catch (final Exception exception)
                                {
                                    url = null;
                                }
                                finally
                                {
                                    stream = null;
                                }
                                if (url != null)
                                {
                                    long ms2 = new Date().getTime() - now.getTime();
                                    normalizedUri = EMFRepositoryFacadeUtils.createUri(url.toString());
                                    this.normalizedUris.put(
                                        uri,
                                        normalizedUri);
                                    if (AndroMDALogger.isDebugEnabled())
                                    {
                                        AndroMDALogger.debug("loaded model --> '" + url + "' " + ms2 + " ms");
                                    }
                                    break;
                                }
                            }
                        }
                        catch (final Exception exception)
                        {
                            logger.debug(
                                "Caught exception in EMFURIConverter",
                                exception);
                        }
                        long ms2 = new Date().getTime() - now1.getTime();
                        String ms = (ms2 > 100 ? ms2 + " ms" : "");
                        if (AndroMDALogger.isDebugEnabled())
                        {
                            AndroMDALogger.debug("loaded model --> '" + fileURI + "' " + ms);
                        }
                    }

                    // - if the normalized URI isn't part of the module search path,
                    //   still store it so we don't continue to look it up each time (which is really slow)
                    if (!this.normalizedUris.containsKey(uri))
                    {
                        this.normalizedUris.put(
                            uri,
                            normalizedUri);
                    }
                }
                else
                {
                    normalizedUri = this.normalizedUris.get(uri);
                }
            }
        }

        return normalizedUri;
    }
}