CachedXMIFormFactory.java

/**
 * Copyright 2010 The University of North Carolina at Chapel Hill
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package cdr.forms;

import edu.unc.lib.schemas.acl.AclPackage;
import gov.loc.mods.mods.MODSPackage;

import java.io.File;
import java.io.FilenameFilter;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;

import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.EObject;
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.util.EcoreUtil;
import org.eclipse.emf.ecore.xmi.impl.XMIResourceFactoryImpl;
import org.eclipse.emf.ecore.xmi.impl.XMLResourceFactoryImpl;
import org.eclipse.gmf.runtime.notation.NotationPackage;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import crosswalk.CrosswalkPackage;
import crosswalk.Editable;
import crosswalk.EditingContainer;
import crosswalk.Form;

public class CachedXMIFormFactory extends AbstractFormFactory {
	private static final Logger LOG = LoggerFactory.getLogger(CachedXMIFormFactory.class);
	
	ResourceSet rs = null;
	Map<String, Form> cache = new HashMap<String, Form>();
	String formPath = null;
	CrosswalkPackage crosswalkPackage = null;
	MODSPackage modsPackage = null;
	AclPackage aclPackage = null;
	NotationPackage notationPackage = null;

	public CachedXMIFormFactory() {
		rs = new ResourceSetImpl();
		rs.getResourceFactoryRegistry().getExtensionToFactoryMap().put("form", new XMIResourceFactoryImpl());
		rs.getResourceFactoryRegistry().getExtensionToFactoryMap().put("xml", new XMLResourceFactoryImpl());
		crosswalkPackage = CrosswalkPackage.eINSTANCE;
		modsPackage = MODSPackage.eINSTANCE;
		aclPackage = AclPackage.eINSTANCE;
		notationPackage = NotationPackage.eINSTANCE;
	}

	public String getFormPath() {
		return formPath;
	}

	public void setFormPath(String formPath) {
		this.formPath = formPath;
	}

	@Override
	public Form getForm(String id) {
		if (true || !cache.containsKey(id)) {
			loadForm(id);
		}
		Form template = cache.get(id);
		EcoreUtil.Copier copier = new EcoreUtil.Copier(false, true);
		EObject copy = copier.copy(template);
		copier.copyReferences();
		return (Form)copy;
	}
	
	public Map<String, Form> getForms() {
		File[] files = new File(getFormPath()).listFiles();
		HashMap<String, Form> forms = new HashMap<String, Form>(); 
		
		for (File f : files) {
			String name = f.getName();
			
			if (name.endsWith(".form")) {
				String id = name.substring(0, name.length() - 5);
				
				loadForm(id);
				
				if (cache.containsKey(id))
					forms.put(id, cache.get(id));
			}
		}
		
		return forms;
	}

	private void loadForm(String id) {
		File f = new File(getFormPath(), id + ".form");
		if (f.exists()) {
			URI formURI = URI.createFileURI(f.getPath());
			Resource formResource = rs.getResource(formURI, true);
			formResource.unload();
			try {
				formResource.load(Collections.EMPTY_MAP);
			} catch (IOException e) {
				LOG.error("Cannot load form: "+f.getPath(), e);
				cache.remove(id);
				return;
			}
			EObject con = formResource.getContents().get(0);
			if (con instanceof EditingContainer) {
				Editable model = ((EditingContainer) con).getModel();
				if (model instanceof Form) {
					cache.put(id, (Form) model);
				}
			}
		} else {
			LOG.error("Cannot load form: "+f.getPath());
		}
	}

}