/*
* Copyright 2010 JBoss Inc
*
* 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 org.drools.eclipse.wizard.project;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.lang.reflect.InvocationTargetException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import org.drools.eclipse.DroolsEclipsePlugin;
import org.drools.eclipse.builder.DroolsBuilder;
import org.drools.eclipse.util.DroolsClasspathContainer;
import org.drools.eclipse.util.DroolsRuntime;
import org.eclipse.core.resources.ICommand;
import org.eclipse.core.resources.IContainer;
import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IFolder;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.IProjectDescription;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.resources.IResourceStatus;
import org.eclipse.core.resources.IWorkspace;
import org.eclipse.core.resources.IncrementalProjectBuilder;
import org.eclipse.core.resources.ResourcesPlugin;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IPath;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.OperationCanceledException;
import org.eclipse.core.runtime.Path;
import org.eclipse.core.runtime.SubProgressMonitor;
import org.eclipse.jdt.core.IClasspathContainer;
import org.eclipse.jdt.core.IClasspathEntry;
import org.eclipse.jdt.core.IJavaProject;
import org.eclipse.jdt.core.IPackageFragment;
import org.eclipse.jdt.core.IPackageFragmentRoot;
import org.eclipse.jdt.core.JavaCore;
import org.eclipse.jdt.core.JavaModelException;
import org.eclipse.jdt.ui.PreferenceConstants;
import org.eclipse.jface.dialogs.ErrorDialog;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jface.resource.ImageDescriptor;
import org.eclipse.ui.actions.WorkspaceModifyOperation;
import org.eclipse.ui.dialogs.WizardNewProjectCreationPage;
import org.eclipse.ui.wizards.newresource.BasicNewResourceWizard;
/**
* A wizard to create a new Drools project.
*/
public class NewDroolsProjectWizard extends BasicNewResourceWizard {
public static final String DROOLS_CLASSPATH_CONTAINER_PATH = "DROOLS/Drools";
private IProject newProject;
private WizardNewProjectCreationPage mainPage;
private NewDroolsProjectWizardPage extraPage;
private NewDroolsProjectRuntimeWizardPage runtimePage;
public void addPages() {
super.addPages();
mainPage = new WizardNewProjectCreationPage("basicNewProjectPage");
mainPage.setTitle("New Drools Project");
mainPage.setDescription("Create a new Drools Project");
this.addPage(mainPage);
extraPage = new NewDroolsProjectWizardPage();
addPage(extraPage);
runtimePage = new NewDroolsProjectRuntimeWizardPage();
addPage(runtimePage);
setNeedsProgressMonitor(true);
}
public boolean performFinish() {
createDroolsProject();
if (newProject == null) {
return false;
}
selectAndReveal(newProject);
return true;
}
private void createDroolsProject() {
newProject = createNewProject();
WorkspaceModifyOperation op = new WorkspaceModifyOperation() {
protected void execute(IProgressMonitor monitor)
throws CoreException {
try {
IJavaProject project = JavaCore.create(newProject);
createDroolsRuntime(project, monitor);
createOutputLocation(project, monitor);
addJavaBuilder(project, monitor);
setClasspath(project, monitor);
createInitialContent(project, monitor);
newProject.build(IncrementalProjectBuilder.FULL_BUILD, monitor);
} catch (IOException _ex) {
ErrorDialog.openError(getShell(), "Problem creating Drools project",
null, null);
}
}
};
try {
getContainer().run(true, true, op);
} catch (Throwable t) {
DroolsEclipsePlugin.log(t);
}
}
private IProject createNewProject() {
if (newProject != null) {
return newProject;
}
final IProject newProjectHandle = mainPage.getProjectHandle();
// get a project descriptor
IPath newPath = null;
if (!mainPage.useDefaults())
newPath = mainPage.getLocationPath();
IWorkspace workspace = ResourcesPlugin.getWorkspace();
final IProjectDescription description = workspace
.newProjectDescription(newProjectHandle.getName());
description.setLocation(newPath);
addNatures(description);
// create the new project operation
WorkspaceModifyOperation op = new WorkspaceModifyOperation() {
protected void execute(IProgressMonitor monitor)
throws CoreException {
createProject(description, newProjectHandle, monitor);
}
};
// run the new project creation operation
try {
getContainer().run(true, true, op);
} catch (InterruptedException e) {
return null;
} catch (InvocationTargetException e) {
Throwable t = e.getTargetException();
if (t instanceof CoreException) {
if (((CoreException) t).getStatus().getCode() == IResourceStatus.CASE_VARIANT_EXISTS) {
MessageDialog.openError(getShell(),
"NewProject.errorMessage",
"NewProject.caseVariantExistsError"
+ newProjectHandle.getName());
} else {
ErrorDialog.openError(getShell(),
"NewProject.errorMessage", null, // no special message
((CoreException) t).getStatus());
}
} else {
DroolsEclipsePlugin.log(e);
}
return null;
}
return newProjectHandle;
}
List<String> list = new ArrayList<String>();
private void addNatures(IProjectDescription projectDescription) {
list.addAll(Arrays.asList(projectDescription.getNatureIds()));
list.add("org.eclipse.jdt.core.javanature");
projectDescription.setNatureIds((String[]) list
.toArray(new String[list.size()]));
}
private void createProject(IProjectDescription description,
IProject projectHandle, IProgressMonitor monitor)
throws CoreException, OperationCanceledException {
try {
monitor.beginTask("", 2000);
projectHandle.create(description, new SubProgressMonitor(monitor,
1000));
if (monitor.isCanceled()) {
throw new OperationCanceledException();
}
projectHandle.open(IResource.BACKGROUND_REFRESH,
new SubProgressMonitor(monitor, 1000));
} finally {
monitor.done();
}
}
private void createDroolsRuntime(IJavaProject project, IProgressMonitor monitor) throws CoreException {
DroolsRuntime runtime = runtimePage.getDroolsRuntime();
if (runtime != null) {
IFile file = project.getProject().getFile(".settings/.drools.runtime");
String runtimeString = "<runtime>" + runtime.getName() + "</runtime>";
if (!file.exists()) {
IFolder folder = project.getProject().getFolder(".settings");
if (!folder.exists()) {
folder.create(true, true, null);
}
file.create(new ByteArrayInputStream(runtimeString.getBytes()), true, null);
} else {
file.setContents(new ByteArrayInputStream(runtimeString.getBytes()), true, false, null);
}
}
}
private void createOutputLocation(IJavaProject project, IProgressMonitor monitor)
throws JavaModelException, CoreException {
IFolder folder = createFolder(project, "target", monitor);
IPath path = folder.getFullPath();
project.setOutputLocation(path, null);
}
private void addJavaBuilder(IJavaProject project, IProgressMonitor monitor) throws CoreException {
IProjectDescription description = project.getProject().getDescription();
ICommand[] commands = description.getBuildSpec();
ICommand[] newCommands = new ICommand[commands.length + 2];
System.arraycopy(commands, 0, newCommands, 0, commands.length);
ICommand javaCommand = description.newCommand();
javaCommand.setBuilderName("org.eclipse.jdt.core.javabuilder");
newCommands[commands.length] = javaCommand;
ICommand droolsCommand = description.newCommand();
droolsCommand.setBuilderName(DroolsBuilder.BUILDER_ID);
newCommands[commands.length + 1] = droolsCommand;
description.setBuildSpec(newCommands);
project.getProject().setDescription(description, monitor);
}
private void setClasspath(IJavaProject project, IProgressMonitor monitor)
throws JavaModelException, CoreException {
project.setRawClasspath(new IClasspathEntry[0], monitor);
addSourceFolders(project, monitor);
addJRELibraries(project, monitor);
addDroolsLibraries(project, monitor);
}
private void addSourceFolders(IJavaProject project, IProgressMonitor monitor) throws JavaModelException, CoreException {
List<IClasspathEntry> list = new ArrayList<IClasspathEntry>();
list.addAll(Arrays.asList(project.getRawClasspath()));
addSourceFolder(project, list, "src/main/java", monitor);
if (runtimePage.getGenerationType() == NewDroolsProjectRuntimeWizardPage.DROOLS6) {
addSourceFolder(project, list, "src/main/resources", monitor);
createFolder(project, "src/main/resources/META-INF", monitor);
createFolder(project, "src/main/resources/META-INF/maven", monitor);
} else {
addSourceFolder(project, list, "src/main/rules", monitor);
}
project.setRawClasspath((IClasspathEntry[]) list.toArray(new IClasspathEntry[list.size()]), null);
}
private void addJRELibraries(IJavaProject project, IProgressMonitor monitor) throws JavaModelException {
List<IClasspathEntry> list = new ArrayList<IClasspathEntry>();
list.addAll(Arrays.asList(project.getRawClasspath()));
list.addAll(Arrays.asList(PreferenceConstants.getDefaultJRELibrary()));
project.setRawClasspath((IClasspathEntry[]) list
.toArray(new IClasspathEntry[list.size()]), monitor);
}
private static IPath getClassPathContainerPath() {
return new Path(DROOLS_CLASSPATH_CONTAINER_PATH);
}
private static void createDroolsLibraryContainer(IJavaProject project, IProgressMonitor monitor)
throws JavaModelException {
JavaCore.setClasspathContainer(getClassPathContainerPath(),
new IJavaProject[] { project },
new IClasspathContainer[] { new DroolsClasspathContainer(
project, getClassPathContainerPath()) }, monitor);
}
public static void addDroolsLibraries(IJavaProject project, IProgressMonitor monitor)
throws JavaModelException {
createDroolsLibraryContainer(project, monitor);
List<IClasspathEntry> list = new ArrayList<IClasspathEntry>();
list.addAll(Arrays.asList(project.getRawClasspath()));
list.add(JavaCore.newContainerEntry(getClassPathContainerPath()));
project.setRawClasspath((IClasspathEntry[]) list
.toArray(new IClasspathEntry[list.size()]), monitor);
}
private void createInitialContent(IJavaProject project, IProgressMonitor monitor)
throws CoreException, JavaModelException, IOException {
try {
boolean createKModule = false;
if (extraPage.createJavaRuleFile()) {
createRuleSampleLauncher(project);
createKModule = true;
}
if (extraPage.createRuleFile()) {
createRule(project, monitor);
createKModule = true;
}
if (extraPage.createDecisionTableFile()) {
createDecisionTable(project, monitor);
createKModule = true;
}
if (extraPage.createJavaDecisionTableFile()) {
createDecisionTableSampleLauncher(project);
createKModule = true;
}
if (extraPage.createRuleFlowFile()) {
createRuleFlow(project, monitor);
createKModule = true;
}
if (extraPage.createJavaRuleFlowFile()) {
createRuleFlowSampleLauncher(project);
createKModule = true;
}
if (createKModule) {
createKModule(project, monitor);
createPom(project, monitor);
}
} catch (Throwable t) {
t.printStackTrace();
}
}
/**
* Create the sample rule launcher file.
*/
private void createRuleSampleLauncher(IJavaProject project)
throws JavaModelException, IOException {
if (runtimePage.getGenerationType() == NewDroolsProjectRuntimeWizardPage.DROOLS4) {
createProjectJavaFile(project, "org/drools/eclipse/wizard/project/RuleLauncherSample_4.java.template", "DroolsTest.java");
} else if (runtimePage.getGenerationType() == NewDroolsProjectRuntimeWizardPage.DROOLS5 ||
runtimePage.getGenerationType() == NewDroolsProjectRuntimeWizardPage.DROOLS5_1) {
createProjectJavaFile(project, "org/drools/eclipse/wizard/project/RuleLauncherSample_5.java.template", "DroolsTest.java");
} else {
createProjectJavaFile(project, "org/drools/eclipse/wizard/project/RuleLauncherSample_6.java.template", "DroolsTest.java");
}
}
/**
* Create the sample decision table launcher file.
*/
private void createDecisionTableSampleLauncher(IJavaProject project)
throws JavaModelException, IOException {
if (runtimePage.getGenerationType() == NewDroolsProjectRuntimeWizardPage.DROOLS4) {
createProjectJavaFile(project, "org/drools/eclipse/wizard/project/DecisionTableLauncherSample_4.java.template", "DecisionTableTest.java");
} else if (runtimePage.getGenerationType() == NewDroolsProjectRuntimeWizardPage.DROOLS5 ||
runtimePage.getGenerationType() == NewDroolsProjectRuntimeWizardPage.DROOLS5_1) {
createProjectJavaFile(project, "org/drools/eclipse/wizard/project/DecisionTableLauncherSample_5.java.template", "DecisionTableTest.java");
} else {
createProjectJavaFile(project, "org/drools/eclipse/wizard/project/DecisionTableLauncherSample_6.java.template", "DecisionTableTest.java");
}
}
private void createProjectJavaFile(IJavaProject project, String templateFile, String javaFile) throws JavaModelException, IOException {
IFolder folder = project.getProject().getFolder("src/main/java");
IPackageFragmentRoot packageFragmentRoot = project.getPackageFragmentRoot(folder);
IPackageFragment packageFragment = packageFragmentRoot.createPackageFragment("com.sample", true, null);
InputStream inputstream = getClass().getClassLoader().getResourceAsStream(templateFile);
packageFragment.createCompilationUnit(javaFile, new String(readStream(inputstream)), true, null);
}
private void createProjectFile(IJavaProject project, IProgressMonitor monitor, String templateFile, String folderName, String fileName) throws CoreException {
InputStream inputstream = getClass().getClassLoader().getResourceAsStream(templateFile);
createProjectFile(project, monitor, inputstream, folderName, fileName);
}
private void createProjectFile(IJavaProject project, IProgressMonitor monitor, InputStream inputstream, String folderName, String fileName) throws CoreException {
IFolder folder = project.getProject().getFolder(folderName);
IFile file = folder.getFile(fileName);
if (!file.exists()) {
file.create(inputstream, true, monitor);
} else {
file.setContents(inputstream, true, false, monitor);
}
}
/**
* Create the sample rule file.
*/
private void createRule(IJavaProject project, IProgressMonitor monitor) throws CoreException {
if (runtimePage.getGenerationType() == NewDroolsProjectRuntimeWizardPage.DROOLS6) {
createFolder(project, "src/main/resources/rules", monitor);
createProjectFile(project, monitor, "org/drools/eclipse/wizard/project/Sample.drl.template", "src/main/resources/rules", "Sample.drl");
} else {
createProjectFile(project, monitor, "org/drools/eclipse/wizard/project/Sample.drl.template", "src/main/rules", "Sample.drl");
}
}
private void createKModule(IJavaProject project, IProgressMonitor monitor) throws CoreException {
if (runtimePage.getGenerationType() == NewDroolsProjectRuntimeWizardPage.DROOLS6) {
createProjectFile(project, monitor, generateKModule(), "src/main/resources/META-INF", "kmodule.xml");
}
}
private void createPom(IJavaProject project, IProgressMonitor monitor) throws CoreException {
if (runtimePage.getGenerationType() == NewDroolsProjectRuntimeWizardPage.DROOLS6) {
createProjectFile(project, monitor, generatePomProperties(runtimePage.getGroupId(), runtimePage.getArtifactId(), runtimePage.getVersion()), "src/main/resources/META-INF/maven", "pom.properties");
}
}
/**
* Create the sample decision table file.
*/
private void createDecisionTable(IJavaProject project, IProgressMonitor monitor) throws CoreException {
if (runtimePage.getGenerationType() == NewDroolsProjectRuntimeWizardPage.DROOLS6) {
createFolder(project, "src/main/resources/dtables", monitor);
createProjectFile(project, monitor, "org/drools/eclipse/wizard/project/Sample.xls.template", "src/main/resources/dtables", "Sample.xls");
} else {
createProjectFile(project, monitor, "org/drools/eclipse/wizard/project/Sample.xls.template", "src/main/rules", "Sample.xls");
}
}
/**
* Create the sample RuleFlow file.
*/
private void createRuleFlow(IJavaProject project, IProgressMonitor monitor) throws CoreException {
String generationType = runtimePage.getGenerationType();
if (NewDroolsProjectRuntimeWizardPage.DROOLS4.equals(generationType)) {
createProjectFile(project, monitor, "org/drools/eclipse/wizard/project/ruleflow_4.rf.template", "src/main/rules", "ruleflow.rf");
createProjectFile(project, monitor, "org/drools/eclipse/wizard/project/ruleflow_4.rfm.template", "src/main/rules", "ruleflow.rfm");
createProjectFile(project, monitor, "org/drools/eclipse/wizard/project/ruleflow_4.drl.template", "src/main/rules", "ruleflow.drl");
} else if (NewDroolsProjectRuntimeWizardPage.DROOLS5.equals(generationType)) {
createProjectFile(project, monitor, "org/drools/eclipse/wizard/project/ruleflow.rf.template", "src/main/rules", "ruleflow.rf");
} else if (NewDroolsProjectRuntimeWizardPage.DROOLS5_1.equals(generationType)) {
createProjectFile(project, monitor, "org/drools/eclipse/wizard/project/sample.bpmn.template", "src/main/rules", "sample.bpmn");
} else {
createFolder(project, "src/main/resources/process", monitor);
createProjectFile(project, monitor, "org/drools/eclipse/wizard/project/sample.bpmn.template", "src/main/resources/process", "sample.bpmn");
}
}
/**
* Create the sample RuleFlow launcher file.
*/
private void createRuleFlowSampleLauncher(IJavaProject project)
throws JavaModelException, IOException {
String s;
String generationType = runtimePage.getGenerationType();
if (NewDroolsProjectRuntimeWizardPage.DROOLS4.equals(generationType)) {
s = "org/drools/eclipse/wizard/project/RuleFlowLauncherSample_4.java.template";
} else if (NewDroolsProjectRuntimeWizardPage.DROOLS5.equals(generationType)) {
s = "org/drools/eclipse/wizard/project/RuleFlowLauncherSample.java.template";
} else if (NewDroolsProjectRuntimeWizardPage.DROOLS5_1.equals(generationType)) {
s = "org/drools/eclipse/wizard/project/ProcessLauncherSample_bpmn_5.java.template";
} else {
s = "org/drools/eclipse/wizard/project/ProcessLauncherSample_bpmn_6.java.template";
}
createProjectJavaFile(project, s, "ProcessTest.java");
}
protected void initializeDefaultPageImageDescriptor() {
ImageDescriptor desc = DroolsEclipsePlugin.getImageDescriptor("icons/drools-large.PNG");
setDefaultPageImageDescriptor(desc);
}
private byte[] readStream(InputStream inputstream) throws IOException {
byte bytes[] = (byte[]) null;
int i = 0;
byte tempBytes[] = new byte[1024];
for (int j = inputstream.read(tempBytes); j != -1; j = inputstream.read(tempBytes)) {
byte tempBytes2[] = new byte[i + j];
if (i > 0) {
System.arraycopy(bytes, 0, tempBytes2, 0, i);
}
System.arraycopy(tempBytes, 0, tempBytes2, i, j);
bytes = tempBytes2;
i += j;
}
return bytes;
}
private void addSourceFolder(IJavaProject project, List<IClasspathEntry> list, String s, IProgressMonitor monitor) throws CoreException {
IFolder folder = project.getProject().getFolder(s);
createFolder(folder, monitor);
IPackageFragmentRoot ipackagefragmentroot = project.getPackageFragmentRoot(folder);
list.add(JavaCore.newSourceEntry(ipackagefragmentroot.getPath()));
}
private IFolder createFolder(IJavaProject project, String s, IProgressMonitor monitor) throws CoreException {
IFolder folder = project.getProject().getFolder(s);
createFolder(folder, monitor);
return folder;
}
private void createFolder(IFolder folder, IProgressMonitor monitor) throws CoreException {
IContainer container = folder.getParent();
if (container != null && !container.exists()
&& (container instanceof IFolder))
createFolder((IFolder) container, monitor);
if (!folder.exists()) {
folder.create(true, true, monitor);
}
}
private InputStream generateKModule() {
StringBuilder sb = new StringBuilder();
sb.append("<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n");
sb.append("<kmodule xmlns=\"http://jboss.org/kie/6.0.0/kmodule\">\n");
if (extraPage.createJavaRuleFile() || extraPage.createRuleFile()) {
sb.append(" <kbase name=\"rules\" packages=\"rules\">\n");
sb.append(" <ksession name=\"ksession-rules\"/>\n");
sb.append(" </kbase>\n");
}
if (extraPage.createDecisionTableFile() || extraPage.createJavaDecisionTableFile()) {
sb.append(" <kbase name=\"dtables\" packages=\"dtables\">\n");
sb.append(" <ksession name=\"ksession-dtables\"/>\n");
sb.append(" </kbase>\n");
}
if (extraPage.createRuleFlowFile() || extraPage.createJavaRuleFlowFile()) {
sb.append(" <kbase name=\"process\" packages=\"process\">\n");
sb.append(" <ksession name=\"ksession-process\"/>\n");
sb.append(" </kbase>\n");
}
sb.append("</kmodule>\n");
return new ByteArrayInputStream(sb.toString().getBytes());
}
private InputStream generatePom(String groupId, String artifactId, String version) {
String pom =
"<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n" +
"<project xmlns=\"http://maven.apache.org/POM/4.0.0\" xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\"\n" +
" xsi:schemaLocation=\"http://maven.apache.org/POM/4.0.0 http://maven.apache.org/maven-v4_0_0.xsd\">\n" +
" <modelVersion>4.0.0</modelVersion>\n" +
"\n" +
" <groupId>" + groupId + "</groupId>\n" +
" <artifactId>" + artifactId + "</artifactId>\n" +
" <version>" + version + "</version>\n" +
"</project>\n";
return new ByteArrayInputStream(pom.getBytes());
}
private InputStream generatePomProperties(String groupId, String artifactId, String version) {
String pom =
"groupId=" + groupId + "\n" +
"artifactId=" + artifactId + "\n" +
"version=" + version + "\n";
return new ByteArrayInputStream(pom.getBytes());
}
}