ref #8477 Set allow duplicates to false when creating a term tree
[taxeditor.git] / eu.etaxonomy.taxeditor.store / src / main / java / eu / etaxonomy / taxeditor / featuretree / AvailableFeaturesWizard.java
index 64736e58869ede921eaec35d43a20773837bb605..7508c9c7ee2b6f6369e91b4161079b5747908d5b 100644 (file)
@@ -9,12 +9,17 @@
 
 package eu.etaxonomy.taxeditor.featuretree;
 
+import java.util.List;
 import java.util.Set;
 
 import org.eclipse.jface.wizard.Wizard;
 
+import eu.etaxonomy.cdm.api.service.IVocabularyService;
 import eu.etaxonomy.cdm.model.term.DefinedTermBase;
 import eu.etaxonomy.cdm.model.term.TermType;
+import eu.etaxonomy.cdm.persistence.dto.TermVocabularyDto;
+import eu.etaxonomy.taxeditor.model.MessagingUtils;
+import eu.etaxonomy.taxeditor.store.CdmStore;
 
 /**
  * <p>AvailableFeaturesWizard class.</p>
@@ -29,27 +34,46 @@ public class AvailableFeaturesWizard extends Wizard {
 
        private Set<DefinedTermBase> additionalFeatures;
 
-    private TermType termType;
+    private List<TermVocabularyDto> vocabularyDtos;
+
+    private boolean singleSelection = false;
 
        public AvailableFeaturesWizard(TermType termType){
-           this.termType = termType;
+        vocabularyDtos = CdmStore.getService(IVocabularyService.class).findVocabularyDtoByTermType(termType);
+       }
+
+       public AvailableFeaturesWizard(List<TermVocabularyDto> vocabularyDtos){
+           this.vocabularyDtos = vocabularyDtos;
        }
 
        /** {@inheritDoc} */
        @Override
        public void addPages() {
-               page = new AvailableFeaturesWizardPage(termType);
+               page = new AvailableFeaturesWizardPage(vocabularyDtos,singleSelection);
                addPage(page);
        }
 
+       @Override
+       public boolean canFinish() {
+        int size = page.getSelectedFeatures().size();
+        return singleSelection?size==1:size>0;
+       }
+
        /** {@inheritDoc} */
        @Override
        public boolean performFinish() {
                additionalFeatures = page.getSelectedFeatures();
+               if(singleSelection && additionalFeatures.size()>1){
+                   MessagingUtils.warningDialog("Too many terms selected", this, "You can only select 1 term");
+               }
                return additionalFeatures.size() > 0;
        }
 
        public Set<DefinedTermBase> getAdditionalFeatures() {
                return additionalFeatures;
        }
+
+    public void setSingleSelection(boolean singleSelection) {
+        this.singleSelection = singleSelection;
+    }
 }