package info.magnolia.ui.contentapp.browser;
   
   import info.magnolia.ui.contentapp.Datasource;
   
   import java.util.Set;
   import java.util.function.Predicate;
   
   import com.vaadin.ui.Grid;
   import com.vaadin.ui.components.grid.MultiSelectionModelImpl;
  import com.vaadin.ui.components.grid.SingleSelectionModelImpl;
  
  /**
   * Functionality common to Magnolia-specific {@link com.vaadin.ui.Grid} implementations.
   */
  class MagnoliaGridDecoration {
  
      static <T> void addCommonConfiguration(Grid<T> grid,
                                         Predicate<T> itemInteractionAvailability,
                                         Datasource<T> datasource) {
          grid.getEditor().addOpenListener(e -> {
              if (!itemInteractionAvailability.test(e.getBean())) {
                  grid.getEditor().cancel();
                  grid.getDataProvider().refreshAll();
              }
          });
          grid.addSelectionListener(e -> e.getAllSelectedItems().forEach(item -> {
              if (!itemInteractionAvailability.test(item)) {
                  grid.deselect(item);
              }
          }));
      }
  
      static class MultiSelectionModel<T> extends MultiSelectionModelImpl<T> {
          @Override
          protected void updateSelection(Set<T> addedItems, Set<T> removedItems, boolean userOriginated) {
              super.updateSelection(addedItems, removedItems, userOriginated);
          }
      }
  
      static class SingleSelectionModel<T> extends SingleSelectionModelImpl<T> {
          @Override
          protected void setSelectedFromClient(String key) {
              super.setSelectedFromClient(key);
          }
  
          @Override
          protected T getData(String key) {
              return super.getData(key);
          }
      }
  }