ref #6746 Fix potential ClassCastException
[taxeditor.git] / eu.etaxonomy.taxeditor.store / src / main / java / eu / etaxonomy / taxeditor / featuretree / e4 / handler / AddFeatureHandler.java
1 /**
2 * Copyright (C) 2017 EDIT
3 * European Distributed Institute of Taxonomy
4 * http://www.e-taxonomy.eu
5 *
6 * The contents of this file are subject to the Mozilla Public License Version 1.1
7 * See LICENSE.TXT at the top of this package for the full license terms.
8 */
9 package eu.etaxonomy.taxeditor.featuretree.e4.handler;
10
11 import java.util.Collection;
12
13 import javax.inject.Named;
14
15 import org.eclipse.core.runtime.IStatus;
16 import org.eclipse.e4.core.di.annotations.CanExecute;
17 import org.eclipse.e4.core.di.annotations.Execute;
18 import org.eclipse.e4.core.di.annotations.Optional;
19 import org.eclipse.e4.ui.model.application.ui.basic.MPart;
20 import org.eclipse.e4.ui.model.application.ui.menu.MHandledMenuItem;
21 import org.eclipse.e4.ui.services.IServiceConstants;
22 import org.eclipse.jface.viewers.IStructuredSelection;
23 import org.eclipse.jface.wizard.WizardDialog;
24 import org.eclipse.swt.widgets.Shell;
25
26 import eu.etaxonomy.cdm.api.service.IFeatureNodeService;
27 import eu.etaxonomy.cdm.model.description.Feature;
28 import eu.etaxonomy.cdm.model.description.FeatureTree;
29 import eu.etaxonomy.taxeditor.featuretree.AvailableFeaturesWizard;
30 import eu.etaxonomy.taxeditor.featuretree.e4.FeatureTreeEditor;
31 import eu.etaxonomy.taxeditor.store.CdmStore;
32
33 /**
34 * @author pplitzner
35 * @since Jul 12, 2017
36 *
37 */
38 public class AddFeatureHandler {
39
40 @Execute
41 public void execute(@Named(IServiceConstants.ACTIVE_SHELL) Shell shell,
42 @Named(IServiceConstants.ACTIVE_SELECTION)IStructuredSelection selection,
43 @Named(IServiceConstants.ACTIVE_PART)MPart thisPart) {
44 FeatureTreeEditor editor = ((FeatureTreeEditor) thisPart.getObject());
45 AvailableFeaturesWizard wizard = new AvailableFeaturesWizard();
46 WizardDialog dialog = new WizardDialog(shell, wizard);
47
48 if (dialog.open() == IStatus.OK) {
49 FeatureTree tree = (FeatureTree) selection.iterator().next();
50 Collection<Feature> additionalFeatures = wizard.getAdditionalFeatures();
51 for (Feature feature : additionalFeatures) {
52 CdmStore.getService(IFeatureNodeService.class).addChildFeatureNode(tree.getRoot().getUuid(), feature.getUuid());
53 }
54 editor.getViewer().refresh();
55 editor.getViewer().expandToLevel(selection, 1);
56 }
57 }
58
59 @CanExecute
60 public boolean canExecute(
61 @Optional@Named(IServiceConstants.ACTIVE_SELECTION)IStructuredSelection selection,
62 @Named(IServiceConstants.ACTIVE_PART)MPart thisPart,
63 MHandledMenuItem menuItem) {
64 boolean canExecute = false;
65 canExecute = selection!=null
66 && selection.size()==1
67 && selection.iterator().next() instanceof FeatureTree
68 && thisPart.getObject() instanceof FeatureTreeEditor;
69 menuItem.setVisible(canExecute);
70 return canExecute;
71 }
72 }