plugins/org.eclipse.ease.lang.unittest.ui/src/org/eclipse/ease/lang/unittest/ui/handlers/AddFolderForVariablesHandler.java

/*******************************************************************************
 * Copyright (c) 2016 Madalina Hodorog and others.
 * 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:
 *     Madalina Hodorog - initial API and implementation
 *******************************************************************************/
package org.eclipse.ease.lang.unittest.ui.handlers;

import org.eclipse.core.commands.AbstractHandler;
import org.eclipse.core.commands.ExecutionEvent;
import org.eclipse.core.commands.ExecutionException;
import org.eclipse.core.commands.IHandler;
import org.eclipse.core.runtime.IPath;
import org.eclipse.core.runtime.Path;
import org.eclipse.ease.lang.unittest.definition.IVariable;
import org.eclipse.ease.lang.unittest.ui.editor.TestSuiteEditor;
import org.eclipse.ease.lang.unittest.ui.editor.VariablesPage;
import org.eclipse.jface.viewers.ISelection;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.ui.IEditorPart;
import org.eclipse.ui.forms.editor.IFormPage;
import org.eclipse.ui.handlers.HandlerUtil;

public class AddFolderForVariablesHandler extends AbstractHandler implements IHandler {

	private static final String DEFAULT_FOLDER_NAME = "folder";

	private static boolean isUniquePath(IPath path, VariablesPage variablesPage) {
		if (variablesPage.containsPath(path))
			return false;

		for (final IVariable variable : variablesPage.getEditor().getTestSuite().getVariables()) {
			if (path.isPrefixOf(variable.getFullName()))
				return false;
		}

		return true;
	}

	@Override
	public Object execute(ExecutionEvent event) throws ExecutionException {

		final IEditorPart editorWindow = HandlerUtil.getActiveEditor(event);
		if (editorWindow instanceof TestSuiteEditor) {
			final IFormPage variablesPage = ((TestSuiteEditor) editorWindow).getActivePageInstance();
			if (variablesPage instanceof VariablesPage) {
				final ISelection selection = HandlerUtil.getActiveMenuSelection(event);

				IPath rootPath = null;
				if (selection.isEmpty()) {
					// add new root group
					rootPath = Path.ROOT;

				} else {
					final Object firstElement = ((IStructuredSelection) selection).getFirstElement();
					if (firstElement instanceof IVariable)
						rootPath = ((IVariable) firstElement).getPath();
					else if (firstElement instanceof IPath)
						rootPath = (IPath) firstElement;
				}

				if (rootPath != null) {
					IPath candidate = rootPath.append(DEFAULT_FOLDER_NAME);
					int index = 1;
					while (!isUniquePath(candidate, (VariablesPage) variablesPage))
						candidate = rootPath.append(DEFAULT_FOLDER_NAME + " " + index++);

					((VariablesPage) variablesPage).addPath(candidate);
				}
			}
		}

		return null;
	}
}