ModelioMetaModelResourceFactory.java

/*******************************************************************************
 * Copyright (c) 2011-2014 The University of York.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 * 
 * Contributors:
 *     Konstantinos Barmpis - initial API and implementation
 ******************************************************************************/
package org.hawk.modelio.mm;

import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.util.HashSet;

import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.EPackage;
import org.eclipse.emf.ecore.EPackage.Registry;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.emf.ecore.resource.ResourceSet;
import org.eclipse.emf.ecore.resource.impl.ResourceSetImpl;
import org.eclipse.emf.ecore.xmi.impl.EcoreResourceFactoryImpl;
import org.eclipse.emf.ecore.xmi.impl.XMIResourceFactoryImpl;
import org.hawk.modelio.ModelioPackage;
import org.hawk.modelio.mm.ModelioMetaModelResource;
import org.hawk.core.IMetaModelResourceFactory;
import org.hawk.core.model.IHawkMetaModelResource;
import org.hawk.core.model.IHawkPackage;

public class ModelioMetaModelResourceFactory implements IMetaModelResourceFactory {

	String type = "com.googlecode.hawk.emf.metamodel.ModelioMetaModelResourceFactory";
	// GraphDatabase graph;

	ResourceSet resourceSet = null;

	public ModelioMetaModelResourceFactory() {

		if (resourceSet == null) {
			resourceSet = new ResourceSetImpl();
			resourceSet.getResourceFactoryRegistry().getExtensionToFactoryMap()
					.put("ecore", new EcoreResourceFactoryImpl());
			resourceSet.getResourceFactoryRegistry().getExtensionToFactoryMap()
					.put("*", new XMIResourceFactoryImpl());
		}
	}

	@Override
	public String getType() {
		return type;
	}

	@Override
	public void shutdown() {
		type = null;
		resourceSet = null;
	}

	@Override
	public IHawkMetaModelResource parse(File f) throws Exception {

		ModelioMetaModelResource ret;

		Resource r = resourceSet.createResource(URI.createFileURI(f
				.getAbsolutePath()));
		r.load(null);

		//
		RegisterMeta.registerPackages(r);

		ret = new ModelioMetaModelResource(r, this);

		return ret;

	}

	@Override
	public HashSet<String> getMetaModelExtensions() {

		return new HashSet<String>();
	}

	@Override
	public IHawkMetaModelResource createMetamodelWithSinglePackage(String s,
			IHawkPackage p) {

		Resource r = resourceSet.createResource(URI.createURI(s));

		r.getContents().add(((ModelioPackage) p).getEObject());

		return new ModelioMetaModelResource(r, this);

	}

	@Override
	public IHawkMetaModelResource parseFromString(String name, String contents)
			throws Exception {

		if (name != null && contents != null) {

			Resource r = resourceSet.createResource(URI.createURI(name));

			InputStream input = new ByteArrayInputStream(
					contents.getBytes("UTF-8"));

			r.load(input, null);

			//
			RegisterMeta.registerPackages(r);

			return new ModelioMetaModelResource(r, this);
		} else
			return null;
	}

	@Override
	public void removeMetamodel(String property) {

		boolean found = false;
		Resource rem = null;

		for (Resource r : resourceSet.getResources())
			if (r.getURI().toString().contains(property)) {
				rem = r;
				found = true;
				break;
			}

		if (found)
			try {
				rem.delete(null);
				EPackage.Registry.INSTANCE.remove(property);
			} catch (Exception e) {
				e.printStackTrace();
			}

		System.err.println(found ? "removed: " + property : property
				+ " not present in this EMF parser");

	}

	@Override
	public boolean canParse(File f) {

		return false;
	}

	@Override
	public HashSet<IHawkMetaModelResource> getStaticMetamodels() {


		HashSet<IHawkMetaModelResource> set = new HashSet<>();

		Registry globalRegistry = EPackage.Registry.INSTANCE;

		HashSet<String> keys = new HashSet<>();

		keys.addAll(globalRegistry.keySet());
		
		File f = new File("../uk.ac.york.cs.mde.hawk.modelio/models/UML.ecore");

		File fr = new File("../uk.ac.york.cs.mde.hawk.modelio/models/Types.ecore");


		Resource r = resourceSet.createResource(URI.createFileURI(f
				.getAbsolutePath()));
		Resource rr = resourceSet.createResource(URI.createFileURI(fr
				.getAbsolutePath()));
		try {
			r.load(null);
			rr.load(null);
		} catch (IOException e) {
			System.err.print("could not load UML metamodel");
		}

		//
		RegisterMeta.registerPackages(r);
		RegisterMeta.registerPackages(rr);
		
		

		ModelioMetaModelResource ret = new ModelioMetaModelResource(r, this);
		ModelioMetaModelResource ret2 = new ModelioMetaModelResource(rr, this);

		set.add(ret);
		set.add(ret2);
		
		return set;
	}

	private boolean notDefaultPackage(String e) {
		// System.err.println(">" + e);

		// new eclipse populates the registry with MANY random metamodels so no
		// way to pre-populate this in emf without ignoring www.eclipse

		// http://www.eclipse.org/emf/2003/XMLType,
		// http://www.eclipse.org/emf/2002/Ecore,
		// http://www.w3.org/XML/1998/namespace

		// if (e.contains("www.eclipse.org/emf/") && e.contains("XMLType")
		// || e.contains("www.eclipse.org/emf/") && e.contains("Ecore")
		// || e.contains("www.w3.org/XML") && e.contains("namespace"))
		if (e.contains("http://www.eclipse.org/")||e.contains("http:///org/eclipse/")
				|| e.contains("www.w3.org/XML") && e.contains("namespace"))
			return false;
		else
			return true;
	}

	@Override
	public String getHumanReadableName() {
		// TODO Auto-generated method stub
		return "Modelio Meta model parser for Hawk";
	}
}