/**
    * This file Copyright (c) 2018 Magnolia International
    * Ltd.  (http://www.magnolia-cms.com). All rights reserved.
    *
    *
    * This file is dual-licensed under both the Magnolia
    * Network Agreement and the GNU General Public License.
    * You may elect to use one or the other of these licenses.
    *
   * This file is distributed in the hope that it will be
   * useful, but AS-IS and WITHOUT ANY WARRANTY; without even the
   * implied warranty of MERCHANTABILITY or FITNESS FOR A
   * PARTICULAR PURPOSE, TITLE, or NONINFRINGEMENT.
   * Redistribution, except as permitted by whichever of the GPL
   * or MNA you select, is prohibited.
   *
   * 1. For the GPL license (GPL), you can redistribute and/or
   * modify this file under the terms of the GNU General
   * Public License, Version 3, as published by the Free Software
   * Foundation.  You should have received a copy of the GNU
   * General Public License, Version 3 along with this program;
   * if not, write to the Free Software Foundation, Inc., 51
   * Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
   *
   * 2. For the Magnolia Network Agreement (MNA), this file
   * and the accompanying materials are made available under the
   * terms of the MNA which accompanies this distribution, and
   * is available at http://www.magnolia-cms.com/mna.html
   *
   * Any modifications to this file must keep this entire header
   * intact.
   *
   */
  package info.magnolia.ui.contentapp.browser;
  
  import static java.util.stream.Collectors.toSet;
  
  import info.magnolia.ui.contentapp.DataFilter;
  import info.magnolia.ui.contentapp.FilterableHierarchicalDataProvider;
  import info.magnolia.ui.contentapp.browser.context.ValueContext;
  import info.magnolia.ui.contentapp.browser.drop.RowDragger;
  import info.magnolia.ui.contentapp.configuration.TreeViewDefinition;
  import info.magnolia.ui.framework.ContextProperty;
  import info.magnolia.ui.framework.FilterContext;
  import info.magnolia.ui.framework.datasource.DatasourceComponent;
  import info.magnolia.ui.framework.datasource.components.SelectedItems;
  import info.magnolia.util.OptionalConsumer;
  
  import java.util.Collection;
  import java.util.HashSet;
  import java.util.Optional;
  import java.util.stream.Stream;
  
  import javax.inject.Inject;
  
  import com.vaadin.shared.ui.grid.DropMode;
  import com.vaadin.ui.CustomComponent;
  import com.vaadin.ui.Grid;
  import com.vaadin.ui.TextField;
  import com.vaadin.ui.TreeGrid;
  import com.vaadin.ui.components.grid.HeaderRow;
  
  /**
   * Tree content view implementation.
   * @param <T> item type.
   */
  public class TreeView<T> extends CustomComponent implements ContentView {
  
      private final TreeGrid<T> treeGrid;
      private final TreePresenter<T> treePresenter;
      private final boolean multiSelect;
  
      @SuppressWarnings("unchecked")
      @Inject
      public TreeView(
              ValueContext<T> valueContext,
              TreeViewDefinition<T> definition,
              @DatasourceComponent ItemInteractionAvailability<T> itemInteractionAvailability) {
  
          this.multiSelect = definition.isMultiSelect();
  
          final ContextProperty<DataFilter> filter = bindContext(FilterContext.class).getFilter();
          filter.set(new DataFilter());
  
          this.treePresenter = create(TreePresenter.class, this, definition);
          bindInstance(TreePresenter.class, treePresenter);
  
          final FilterableHierarchicalDataProvider<T> dataProvider = treePresenter.getDataProvider();
          this.treeGrid = new MagnoliaTreeGrid<>(this.treePresenter.getPropertySet(), itemInteractionAvailability::isItemAvailable, this.treePresenter.getCellEditorAvailability());
          this.treeGrid.setDataProvider(dataProvider);
          this.treeGrid.addCollapseListener(e -> treeGrid.getEditor().cancel());
          this.treeGrid.setSizeFull();
          this.treeGrid.setSelectionMode(this.multiSelect ? Grid.SelectionMode.MULTI : Grid.SelectionMode.SINGLE);
          this.treeGrid.addSelectionListener(event -> valueContext.current().set(SelectedItems.of(event.getAllSelectedItems())));
          this.treeGrid.addSelectionListener(e -> e.getAllSelectedItems().forEach(this::expandParents));
          this.treeGrid.addItemClickListener(event -> {
              this.treeGrid.deselectAll();
              this.treeGrid.select(event.getItem());
          });
         this.treeGrid.getEditor().addSaveListener(e -> {
             this.treePresenter.saveInlineEditingResults(e.getBean());
             this.treeGrid.getDataProvider().refreshAll();
         });
         this.treePresenter.createColumns(treeGrid);
 
 
         Optional.ofNullable(definition.getDropConstraint()).ifPresent(dropConstraint -> getComponentProvider().newInstance(RowDragger.class, treeGrid, DropMode.ON_TOP_OR_BETWEEN, dropConstraint));
         Optional.ofNullable(treeGrid.getColumn("jcrPath")).ifPresent(tColumn -> {
             final HeaderRow headerRow = this.treeGrid.appendHeaderRow();
             headerRow
                     .getCell(tColumn)
                     .setComponent(new TextField(e -> filter.mutate(dataFilter -> dataFilter.setPathFilter(e.getValue()))));
 
         });
 
         valueContext.current()
                 .observe(selectedItemsOptional -> OptionalConsumer.of(selectedItemsOptional)
                         .ifNotPresent(treeGrid::deselectAll)
                         .ifPresent(selectedItems -> {
                             selectedItems.stream().forEach(this::expandParents);
                             select(selectedItems.stream());
                         }));
 
         setSizeFull();
         setCompositionRoot(treeGrid);
     }
 
     private void expandParents(T item) {
         final Collection<T> parents = this.treePresenter.getParents(item).collect(toSet());
         this.treeGrid.expand(parents);
     }
 
     private void select(Stream<T> values) {
         if (this.multiSelect) {
             this.treeGrid.asMultiSelect().updateSelection(values.collect(toSet()), new HashSet<>());
         } else {
             values.findFirst().ifPresent(this.treeGrid.asSingleSelect()::setValue);
         }
     }
 }