/*
* 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.guvnor.tools.wizards;
import java.util.ArrayList;
import java.util.List;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.jface.action.Action;
import org.eclipse.jface.resource.ImageDescriptor;
import org.eclipse.jface.viewers.DoubleClickEvent;
import org.eclipse.jface.viewers.IDoubleClickListener;
import org.eclipse.jface.viewers.ISelection;
import org.eclipse.jface.viewers.ISelectionChangedListener;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.viewers.SelectionChangedEvent;
import org.eclipse.jface.viewers.TreeViewer;
import org.eclipse.jface.wizard.WizardPage;
import org.eclipse.swt.SWT;
import org.eclipse.swt.layout.GridData;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Label;
import org.guvnor.tools.Activator;
import org.guvnor.tools.GuvnorRepository;
import org.guvnor.tools.Messages;
import org.guvnor.tools.utils.PlatformUtils;
import org.guvnor.tools.views.RepositoryContentProvider;
import org.guvnor.tools.views.RepositoryLabelProvider;
import org.guvnor.tools.views.model.TreeObject;
import org.guvnor.tools.views.model.TreeParent;
/**
* Wizard page for selecting resources in Guvnor.
*/
public class SelectGuvnorResourcesPage extends WizardPage {
private TreeViewer viewer;
private String previousSelection;
private Action doubleClickAction;
public SelectGuvnorResourcesPage(String pageName) {
super(pageName);
}
public SelectGuvnorResourcesPage(String pageName, String title, ImageDescriptor titleImage) {
super(pageName, title, titleImage);
}
public void createControl(Composite parent) {
Composite composite = PlatformUtils.createComposite(parent, 1);
new Label(composite, SWT.NONE).setText(Messages.getString("select.resources")); //$NON-NLS-1$
viewer = new TreeViewer(composite, SWT.BORDER | SWT.MULTI | SWT.H_SCROLL | SWT.V_SCROLL);
viewer.getTree().setLayoutData(new GridData(GridData.FILL_BOTH));
viewer.setLabelProvider(new RepositoryLabelProvider());
viewer.addPostSelectionChangedListener(new ISelectionChangedListener() {
public void selectionChanged(SelectionChangedEvent event) {
updateModel();
}
});
hookDoubleClickAction();
super.setControl(composite);
}
private void hookDoubleClickAction() {
doubleClickAction = new Action() {
public void run() {
ISelection selection = viewer.getSelection();
Object obj = ((IStructuredSelection)selection).getFirstElement();
if (obj instanceof TreeObject) {
doubleClick((TreeObject)obj);
}
}
};
viewer.addDoubleClickListener(new IDoubleClickListener() {
public void doubleClick(DoubleClickEvent event) {
doubleClickAction.run();
}
});
}
private void doubleClick(TreeObject node) {
if (node instanceof TreeParent) {
if (viewer.getExpandedState(node)) {
viewer.collapseToLevel(node, 1);
} else {
viewer.expandToLevel(node, 1);
}
}
}
private void handleRepositoryCreation() {
// First we'll see if the repository already exists
GuvWizardModel model = ((IGuvnorWizard)super.getWizard()).getModel();
if (model.shouldCreateNewRep()
&& model.getRepLocation() != null) {
GuvnorRepository theRep = Activator.getLocationManager().
findRepository(model.getRepLocation());
if (theRep != null) {
// The repository already exists, nothing to do
return;
}
try {
WizardUtils.createGuvnorRepository(model);
} catch (Exception e) {
super.setErrorMessage(e.getMessage());
Activator.getDefault().displayError(IStatus.ERROR, e.getMessage(), e, true);
}
}
}
@Override
public void setVisible(boolean visible) {
if (visible) {
// Need to filter repository list based on currently selected repository
// Will also keep track of which filter is applied, so we don't create
// additional content providers unnecessarily.
GuvWizardModel model = ((IGuvnorWizard)super.getWizard()).getModel();
String currentSelection = model.getRepLocation();
// Not supposed to use this page without setting the target repository in the model
assert(currentSelection != null);
// If we had a repository selection before that is different from the current selection
if (previousSelection != null
&& !currentSelection.equals(previousSelection)) {
handleRepositoryCreation();
RepositoryContentProvider cp = new RepositoryContentProvider();
cp.setRepositorySelection(currentSelection);
viewer.setContentProvider(cp);
viewer.setInput(viewer);
previousSelection = currentSelection;
} else {
// If we didn't have a repository selection before (first time this page is loaded)
if (previousSelection == null) {
handleRepositoryCreation();
RepositoryContentProvider cp = new RepositoryContentProvider();
cp.setRepositorySelection(currentSelection);
viewer.setContentProvider(cp);
viewer.setInput(viewer);
previousSelection = currentSelection;
}
}
}
super.setVisible(visible);
}
@SuppressWarnings("unchecked")
private void updateModel() {
IStructuredSelection selection = (IStructuredSelection)viewer.getSelection();
if (!selection.isEmpty()) {
List<String> resources = new ArrayList<String>();
List<TreeObject> nodes = selection.toList();
for (TreeObject o:nodes) {
if (o.getNodeType() == TreeObject.Type.RESOURCE) {
resources.add(o.getFullPath());
}
}
GuvWizardModel model = ((IGuvnorWizard)super.getWizard()).getModel();
if (resources.size() > 0) {
model.setResources(resources);
} else {
model.setResources(null);
}
}
super.getWizard().getContainer().updateButtons();
}
@Override
public boolean canFlipToNextPage() {
GuvWizardModel model = ((IGuvnorWizard)super.getWizard()).getModel();
return model.getResources() != null && model.getResources().size() > 0;
}
}