/*
    * Copyright 2000-2018 Vaadin Ltd.
    *
    * Licensed under the Apache License, Version 2.0 (the "License"); you may not
    * use this file except in compliance with the License. You may obtain a copy of
    * the License at
    *
    * http://www.apache.org/licenses/LICENSE-2.0
    *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
   * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
   * License for the specific language governing permissions and limitations under
   * the License.
   */
  
  package com.vaadin.contextmenu;
  
  import com.vaadin.contextmenu.ContextMenu.ContextMenuOpenListener.ContextMenuOpenEvent;
  import com.vaadin.contextmenu.GridContextMenu.GridContextMenuOpenListener.GridContextMenuOpenEvent;
  import com.vaadin.v7.shared.ui.grid.GridConstants.Section;
  import com.vaadin.v7.ui.Grid;
  import com.vaadin.v7.ui.Grid.Column;
  import com.vaadin.v7.ui.Grid.GridContextClickEvent;
  
  @SuppressWarnings("serial")
  public class GridContextMenu<T> extends ContextMenu {
  
      public GridContextMenu(Grid<T> parentComponent) {
          super(parentComponent, true);
      }
  
      private void addGridSectionContextMenuListener(final Section section,
              final GridContextMenuOpenListener<T> listener) {
          addContextMenuOpenListener((final ContextMenuOpenEvent event) -> {
              if (event
                      .getContextClickEvent() instanceof Grid.GridContextClickEvent) {
                  @SuppressWarnings("unchecked")
                  GridContextClickEvent<T> gridEvent = (GridContextClickEvent<T>) event
                          .getContextClickEvent();
                  if (gridEvent.getSection() == section) {
                      listener.onContextMenuOpen(new GridContextMenuOpenEvent<>(
                              GridContextMenu.this, gridEvent));
                  }
              }
          });
      }
  
      public void addGridHeaderContextMenuListener(
              GridContextMenuOpenListener<T> listener) {
          addGridSectionContextMenuListener(Section.HEADER, listener);
      }
  
      public void addGridFooterContextMenuListener(
              GridContextMenuOpenListener<T> listener) {
          addGridSectionContextMenuListener(Section.FOOTER, listener);
      }
  
      public void addGridBodyContextMenuListener(
              GridContextMenuOpenListener<T> listener) {
          addGridSectionContextMenuListener(Section.BODY, listener);
      }
  
      public interface GridContextMenuOpenListener<T>
              extends java.util.EventListener, java.io.Serializable {
  
          public void onContextMenuOpen(GridContextMenuOpenEvent<T> event);
  
          public static class GridContextMenuOpenEvent<T>
                  extends ContextMenuOpenEvent {
  
              private final T item;
              private final Grid<T> component;
              private final int rowIndex;
              private final Column<T, ?> column;
              private final Section section;
  
              public GridContextMenuOpenEvent(ContextMenu contextMenu,
                      GridContextClickEvent<T> contextClickEvent) {
                  super(contextMenu, contextClickEvent);
                  item = contextClickEvent.getItem();
                  component = contextClickEvent.getComponent();
                  rowIndex = contextClickEvent.getRowIndex();
                  column = contextClickEvent.getColumn();
                  section = contextClickEvent.getSection();
              }
  
              public T getItem() {
                  return item;
              }
  
              public Grid<T> getComponent() {
                  return component;
              }
  
              public int getRowIndex() {
                  return rowIndex;
              }
  
             public Column<T, ?> getColumn() {
                 return column;
             }
 
             public Section getSection() {
                 return section;
             }
         }
     }
 }