/**
    * 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.poc;
  
  import info.magnolia.icons.MagnoliaIcons;
  import info.magnolia.poc.customer.Customer;
  import info.magnolia.poc.customer.CustomerStatusRenderer;
  import info.magnolia.poc.customer.CustomerService;
  
  import java.util.List;
  
  import javax.servlet.annotation.WebServlet;
  
  import org.jsoup.Jsoup;
  import org.jsoup.safety.Whitelist;
  
  import com.vaadin.annotations.Theme;
  import com.vaadin.annotations.Title;
  import com.vaadin.annotations.VaadinServletConfiguration;
  import com.vaadin.annotations.Widgetset;
  import com.vaadin.contextmenu.ContextMenu;
  import com.vaadin.icons.VaadinIcons;
  import com.vaadin.server.VaadinRequest;
  import com.vaadin.server.VaadinServlet;
  import com.vaadin.ui.Grid.SelectionMode;
  import com.vaadin.ui.HorizontalLayout;
  import com.vaadin.ui.TreeGrid;
  import com.vaadin.ui.UI;
  import com.vaadin.ui.VerticalLayout;
  import com.vaadin.ui.components.grid.ItemClickListener;
  import com.vaadin.ui.renderers.HtmlRenderer;
  
  @Theme("poctheme")
  @Title("Magnolia 6 Resurface")
  @Widgetset("info.magnolia.poc.Widgetset")
  public class TreeGridUI extends UI {
  
      @WebServlet(value = "/grid/*", asyncSupported = true)
      @VaadinServletConfiguration(productionMode = false, ui = TreeGridUI.class)
      public static class Servlet extends VaadinServlet {
      }
  
      private TreeGrid<Customer> treeGrid = new TreeGrid<>();
      private CustomerService service = CustomerService.getInstance();
  
      @Override
      protected void init(VaadinRequest request) {
          final VerticalLayout layout = new VerticalLayout();
          layout.setSpacing(false);
          HorizontalLayout main = new HorizontalLayout(treeGrid);
          main.setSizeFull();
          treeGrid.setSizeFull();
  
          treeGrid.addColumn(customer -> {
              String iconHtml;
              if (customer.isPersisted()) {
                  iconHtml = VaadinIcons.FOLDER_O.getHtml();
              } else {
                  iconHtml = VaadinIcons.FILE_O.getHtml();
              }
              return iconHtml + " " + Jsoup.clean(customer.getLastName(), Whitelist.simpleText());
          }, new HtmlRenderer()).setCaption("Last Name").setExpandRatio(5);
          treeGrid.addColumn(Customer::getFirstName).setCaption("First Name").setExpandRatio(2);
          treeGrid.addColumn(customer -> customer.getStatus().name(), new CustomerStatusRenderer()).setCaption("Status").setExpandRatio(0);
          treeGrid.addColumn(Customer::getBirthDate).setCaption("Date of Birth").setExpandRatio(2);
          treeGrid.addColumn(Customer::getEmail).setCaption("Email").setExpandRatio(2);
  
          treeGrid.setSelectionMode(SelectionMode.MULTI);
 
         treeGrid.addItemClickListener((ItemClickListener<Customer>) event -> {
             if (event.getSource().getSelectionModel().isSelected(event.getItem())) {
                 event.getSource().getSelectionModel().deselect(event.getItem());
             } else {
                 event.getSource().getSelectionModel().select(event.getItem());
             }
         });
 
         main.setExpandRatio(treeGrid, 1);
         layout.addComponents(main);
 
         // refresh the customer list
         updateList();
 
         ContextMenu contextMenu = new ContextMenu(this, false);
         contextMenu.setAsContextMenuOf(layout);
         contextMenu.addItem("Upload asset", MagnoliaIcons.UPLOAD, null);
         contextMenu.addItem("Upload Zip archive", MagnoliaIcons.UPLOAD, null);
         contextMenu.addItem("Add folder", MagnoliaIcons.ADD_FOLDER, null);
         contextMenu.addItem("Delete folder", MagnoliaIcons.DELETE, null);
         contextMenu.addItem("Publish incl. subnodes", MagnoliaIcons.PUBLISH_INCL_SUB, null);
         contextMenu.addItem("Move folder", MagnoliaIcons.MOVE, null);
 
         setContent(layout);
     }
 
     public void updateList() {
         // fetch list of items from service and assign it to Grid
         // in this case it's a list of customers
         List<Customer> customers = service.findAll();
         treeGrid.setItems(customers, Customer::getChildren);
     }
 
     @Override
     public boolean equals(Object obj) {
         return super.equals(obj);
     }
 
     @Override
     public int hashCode() {
         return super.hashCode();
     }
 }