fhir/plugins/org.eclipse.mdht.uml.fhir.ui/src/org/eclipse/mdht/uml/fhir/ui/handlers/ImportFhirHandler.java

/*******************************************************************************
 * Copyright (c) 2015 David A Carlson
 * 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:
 *     David A Carlson (Clinical Cloud Solutions, LLC) - initial API and implementation
 *
 *******************************************************************************/
package org.eclipse.mdht.uml.fhir.ui.handlers;

import java.util.List;

import org.eclipse.core.commands.AbstractHandler;
import org.eclipse.core.commands.ExecutionEvent;
import org.eclipse.core.commands.ExecutionException;
import org.eclipse.core.commands.operations.IUndoableOperation;
import org.eclipse.core.resources.IContainer;
import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IAdaptable;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.NullProgressMonitor;
import org.eclipse.core.runtime.Status;
import org.eclipse.emf.common.ui.dialogs.ResourceDialog;
import org.eclipse.emf.common.ui.dialogs.WorkspaceResourceDialog;
import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.transaction.TransactionalEditingDomain;
import org.eclipse.emf.transaction.util.TransactionUtil;
import org.eclipse.emf.workspace.AbstractEMFOperation;
import org.eclipse.emf.workspace.IWorkspaceCommandStack;
import org.eclipse.mdht.uml.common.ui.util.UMLHandlerUtil;
import org.eclipse.mdht.uml.fhir.common.util.ModelFilter;
import org.eclipse.mdht.uml.fhir.transform.ModelImporter;
import org.eclipse.swt.SWT;
import org.eclipse.ui.IWorkbenchPart;
import org.eclipse.ui.handlers.HandlerUtil;
import org.eclipse.uml2.uml.Package;

public class ImportFhirHandler extends AbstractHandler {
	private static final String IMPORT_FILTERS_PARAMETER = "org.eclipse.mdht.uml.fhir.ui.commandParameter.importFilters";
	
	public ImportFhirHandler() {
		super();
	}
	
	private IContainer getProfileFolder(IWorkbenchPart activePart) {
		IContainer profileFolder = null;
		IContainer[] containers = WorkspaceResourceDialog.openFolderSelection(activePart.getSite().getShell(), 
				"Select Profile Folder", "Folder containing FHIR profiles", false, null, null);
		if (containers.length > 0) {
			profileFolder = containers[0];
		}
		
		return profileFolder;
	}
	
	private List<URI> selectProfileFiles(IWorkbenchPart activePart) {
		ResourceDialog dialog = new ResourceDialog(activePart.getSite().getShell(), "Select FHIR Profiles", SWT.OPEN|SWT.MULTI);
		dialog.open();
		return dialog.getURIs();
	}
	
	protected ModelFilter createFilter(ExecutionEvent event) {
		ModelFilter filter = new ModelFilter();
		String importFiltersParam = event.getParameter(IMPORT_FILTERS_PARAMETER);
		
		if (importFiltersParam != null) {
			// delimiters: ",", "|", " "
			String[] params = importFiltersParam.split("\\s*,\\s*|\\s*\\|\\s*|\\s+");
			for (int i = 0; i < params.length; i++) {
				try {
					ModelFilter.DefinitionType type = ModelFilter.DefinitionType.valueOf(params[i]);
					filter.getFilterTypes().add(type);
				}
				catch (IllegalArgumentException e) {
					e.printStackTrace();
				}
			}
		}
		
		return filter;
	}
	
	public Object execute(final ExecutionEvent event) throws ExecutionException {
		// these APIs throw if the requested thing isn't available
		final Package umlPackage = UMLHandlerUtil.getSelectedElementChecked(event, Package.class);
		IWorkbenchPart activePart = HandlerUtil.getActivePartChecked(event);

		// select the profile folder first, then optionally a list of profiles
		final IContainer profileFolder = getProfileFolder(activePart);
//		List<URI> profiles = selectProfileFiles();
		
		TransactionalEditingDomain editingDomain = TransactionUtil.getEditingDomain(umlPackage);

		IUndoableOperation operation = new AbstractEMFOperation(editingDomain, "Import FHIR definition") {
			@Override
			protected IStatus doExecute(IProgressMonitor monitor, IAdaptable info) {
				if (profileFolder != null) {
					ModelImporter umlImporter = new ModelImporter(umlPackage);
//					for (URI profileURI : profiles) {
//						umlImporter.importProfile(profileURI);
//					}
					
					ModelFilter modelFilter = createFilter(event);
					umlImporter.setModelFilter(modelFilter);
					indexAllFiles(umlImporter, profileFolder);
					umlImporter.importIndexedResources();
					
					return Status.OK_STATUS;
				}

				return Status.CANCEL_STATUS;
			}
		};

		IWorkspaceCommandStack commandStack = (IWorkspaceCommandStack) editingDomain.getCommandStack();
		operation.addContext(commandStack.getDefaultUndoContext());
		commandStack.getOperationHistory().execute(operation, new NullProgressMonitor(), null);

		return null;
	}

	protected void indexAllFiles(ModelImporter umlImporter, IContainer profileFolder) {
		try {
			for (IResource resource : profileFolder.members()) {
				if (resource instanceof IFile) {
					IFile file = (IFile) resource;
					
					// skip file names that contain 'example'
					if (file.getName().contains("example")) {
						continue;
					}

					String fileExt = file.getFileExtension();
					if ("xml".equals(fileExt)) {
						URI fileURI = URI.createFileURI(file.getLocation().toString());
						umlImporter.indexResource(fileURI);
					}
				}
			}
		} catch (CoreException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}

}