NewProcessDefinitionWizard.java

/*
 * JBoss, Home of Professional Open Source
 * Copyright 2005, JBoss Inc., and individual contributors as indicated
 * by the @authors tag. See the copyright.txt in the distribution for a
 * full listing of individual contributors.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */
package org.jbpm.gd.jpdl.wizard;

import java.io.ByteArrayInputStream;
import java.io.File;
import java.net.MalformedURLException;

import org.dom4j.Attribute;
import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.io.SAXReader;
import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.Path;
import org.eclipse.core.runtime.QualifiedName;
import org.eclipse.core.variables.VariablesPlugin;
import org.eclipse.jdt.core.JavaModelException;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.wizard.Wizard;
import org.eclipse.ui.INewWizard;
import org.eclipse.ui.IWorkbench;
import org.eclipse.ui.IWorkbenchPage;
import org.eclipse.ui.IWorkbenchWindow;
import org.eclipse.ui.PartInitException;
import org.eclipse.ui.PlatformUI;
import org.eclipse.ui.ide.IDE;
import org.eclipse.ui.wizards.newresource.BasicNewResourceWizard;
import org.jboss.tools.jbpm.preferences.JbpmInstallation;
import org.jboss.tools.jbpm.preferences.PreferencesManager;
import org.jbpm.gd.jpdl.Logger;
import org.jbpm.gd.jpdl.Plugin;

public class NewProcessDefinitionWizard extends Wizard implements INewWizard {

	private IStructuredSelection selection;
	private NewProcessDefinitionWizardPage page;
	
	public NewProcessDefinitionWizard() {
		setWindowTitle("New Process Definition");
	}

	public void init(IWorkbench workbench, IStructuredSelection currentSelection) {
		this.selection= currentSelection;
	}
	
	public void addPages() {
		page = new NewProcessDefinitionWizardPage();
		addPage(page);
		page.init(selection);
	}
	
	public boolean performFinish() {
		try {
			IFile processDefinitionFile = page.getProcessDefinitionFile();
			processDefinitionFile.create(createInitialProcessDefinition(), true, null);
			IFile gpdFile = page.getGpdFile();
			gpdFile.create(createInitialGpdInfo(), true, null);
			IDE.openEditor(getActivePage(), processDefinitionFile);
			openPropertiesView();
			BasicNewResourceWizard.selectAndReveal(processDefinitionFile, getActiveWorkbenchWindow());
			return true;
		} catch (CoreException e) {
			e.printStackTrace();
			return false;
		}
	}

	private void openPropertiesView() {
		try {
			if (getActivePage().findView("org.eclipse.ui.views.PropertySheet") == null) {
				getActivePage().showView("org.eclipse.ui.views.PropertySheet");
			}
		} catch (PartInitException e) {
			e.printStackTrace();
		}
	}

	private IWorkbenchPage getActivePage() {
		return getActiveWorkbenchWindow().getActivePage();
	}

	private IWorkbenchWindow getActiveWorkbenchWindow() {
		return PlatformUI.getWorkbench().getActiveWorkbenchWindow();
	}
	
	private ByteArrayInputStream createInitialProcessDefinition() throws JavaModelException {
		StringBuffer buffer = new StringBuffer();
		buffer.append("<?xml version=\"1.0\" encoding=\"UTF-8\"?>");
		buffer.append("\n");
		buffer.append("\n");
		buffer.append("<process-definition" + getSchemaText() + "  name=\"" + page.getProcessName() + "\"></process-definition>");	
		return new ByteArrayInputStream(buffer.toString().getBytes());
	}
	
	private String getSchemaText() throws JavaModelException {
		if (!Plugin.getDefault().getPreferenceStore().getBoolean("exclude xsd")) {
			return "  xmlns=\"" + getJbpmSchemaNameSpace() + "\"";
		} else {
			return "";
		}
		
	}

	private ByteArrayInputStream createInitialGpdInfo() {
		StringBuffer buffer = new StringBuffer();
		buffer.append("<?xml version=\"1.0\" encoding=\"UTF-8\"?>");
		buffer.append("\n");
		buffer.append("\n");
		buffer.append("<root-container></root-container>");	
		return new ByteArrayInputStream(buffer.toString().getBytes());
	}
	
	private String getJbpmSchemaNameSpace() throws JavaModelException {
		try {
			IProject project = page.getProcessDefinitionFile().getProject();
			String jbpmName = project.getPersistentProperty(new QualifiedName("", "jbpmName"));
			if (jbpmName == null) return "";
			JbpmInstallation jbpmInstallation = PreferencesManager.getInstance().getJbpmInstallation(jbpmName);
			if (jbpmInstallation == null) return "";
			String location = VariablesPlugin.getDefault().getStringVariableManager().performStringSubstitution(jbpmInstallation.location);
			if (location == null) return "";
			File file = new Path(location).append("src/resources/gpd/version.info.xml").toFile();
			if (!file.exists()) return "";
			return getNameSpace(file);
		} catch (CoreException e) {
			return "";
		}
	}
	
	private String getNameSpace(File file) {
		try {
			Document document = new SAXReader().read(file);
			Attribute attribute = document.getRootElement().attribute("namespace");
			if (attribute != null) {
				return attribute.getValue();
			} else {
				return "";
			}
		} catch (DocumentException e) {
			Logger.logError(e);
			return "";
		} catch (MalformedURLException e) {
			Logger.logError(e);
			return "";
		}
	}

}