CollectionsSelectionPanel.java

package com.constellio.app.ui.framework.components;

import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import com.constellio.app.ui.framework.components.converters.CollectionCodeToLabelConverter;
import com.vaadin.ui.CheckBox;
import com.vaadin.ui.Label;
import com.vaadin.ui.Layout;
import com.vaadin.ui.Panel;
import com.vaadin.ui.VerticalLayout;
import com.vaadin.ui.themes.ValoTheme;

public class CollectionsSelectionPanel extends Panel {
	
    Map<CheckBox, String> collectionCheckBoxes = new LinkedHashMap<>();
    
    private CollectionCodeToLabelConverter collectionCodeToLabelConverter = new CollectionCodeToLabelConverter();
    
    public CollectionsSelectionPanel(String title, List<String> collections) {
        this(title, collections, new ArrayList<String>());
    }

    public CollectionsSelectionPanel(String title, List<String> collections, List<String> selectedCollections) {
        VerticalLayout layout = new VerticalLayout();
        Label titleLabel = new Label(title);
        layout.addComponent(titleLabel);
        for(String collection: collections){
            addCollectionCheckBox(layout, collection, selectedCollections);
        }
        setContent(layout);
        addStyleName(ValoTheme.PANEL_BORDERLESS);
    }

    private void addCollectionCheckBox(Layout layout, String collection, List<String> selectedCollections) {
        CheckBox checkBox = new CheckBox();
        if(selectedCollections.contains(collection)){
            checkBox.setValue(true);
        }
        checkBox.setCaption(collectionCodeToLabelConverter.getCollectionCaption(collection));
        collectionCheckBoxes.put(checkBox, collection);
        layout.addComponent(checkBox);
    }

    public List<String> getSelectedCollections() {
        List<String> selectedCollections = new ArrayList<>();
        for (CheckBox collectionCheckBox: collectionCheckBoxes.keySet()){
            if (collectionCheckBox.getValue()){
                selectedCollections.add(collectionCheckBoxes.get(collectionCheckBox));
            }
        }
        return selectedCollections;
    }
}