vaadin-framework/uitest/src/main/java/com/vaadin/tests/components/grid/GridDragAndDrop.java

/*
 * Copyright 2000-2016 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.tests.components.grid;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.IntStream;

import com.vaadin.server.Page;
import com.vaadin.server.VaadinRequest;
import com.vaadin.shared.ui.grid.DropMode;
import com.vaadin.tests.components.AbstractTestUIWithLog;
import com.vaadin.ui.Grid;
import com.vaadin.ui.GridDragSource;
import com.vaadin.ui.GridDropTarget;
import com.vaadin.ui.HorizontalLayout;
import com.vaadin.ui.Layout;
import com.vaadin.ui.RadioButtonGroup;

import elemental.json.Json;
import elemental.json.JsonObject;

public class GridDragAndDrop extends AbstractTestUIWithLog {
    @Override
    protected void setup(VaadinRequest request) {

        // Drag source Grid
        Grid<Bean> dragSourceComponent = new Grid<>();
        dragSourceComponent.setItems(createItems(50, "left"));
        dragSourceComponent.addColumn(Bean::getId).setCaption("ID");
        dragSourceComponent.addColumn(Bean::getValue).setCaption("Value");

        GridDragSource<Bean> dragSource = new GridDragSource<>(
                dragSourceComponent);
        dragSource.setDragDataGenerator(bean -> {
            JsonObject ret = Json.createObject();
            ret.put("generatedId", bean.getId());
            ret.put("generatedValue", bean.getValue());
            return ret;
        });

        // Drop target Grid
        Grid<Bean> dropTargetComponent = new Grid<>();
        dropTargetComponent.setItems(createItems(5, "right"));
        dropTargetComponent.addColumn(Bean::getId).setCaption("ID");
        dropTargetComponent.addColumn(Bean::getValue).setCaption("Value");

        GridDropTarget<Bean> dropTarget = new GridDropTarget<>(
                dropTargetComponent, DropMode.ON_TOP);
        dropTarget.addGridDropListener(event -> {
            log(event.getDataTransferText() + ", targetId=" + event
                    .getDropTargetRow().getId() + ", location=" + event
                    .getDropLocation());
        });

        // Layout the two grids
        Layout grids = new HorizontalLayout();
        grids.addComponents(dragSourceComponent, dropTargetComponent);

        // Selection modes
        List<Grid.SelectionMode> selectionModes = Arrays.asList(
                Grid.SelectionMode.SINGLE, Grid.SelectionMode.MULTI);
        RadioButtonGroup<Grid.SelectionMode> selectionModeSelect = new RadioButtonGroup<>(
                "Selection mode", selectionModes);
        selectionModeSelect.setSelectedItem(Grid.SelectionMode.SINGLE);
        selectionModeSelect.addValueChangeListener(event -> dragSourceComponent
                .setSelectionMode(event.getValue()));

        // Drop locations
        List<DropMode> dropLocations = Arrays.asList(DropMode.values());
        RadioButtonGroup<DropMode> dropLocationSelect = new RadioButtonGroup<>(
                "Allowed drop location", dropLocations);
        dropLocationSelect.setSelectedItem(DropMode.ON_TOP);
        dropLocationSelect.addValueChangeListener(
                event -> dropTarget.setDropMode(event.getValue()));

        Layout controls = new HorizontalLayout(selectionModeSelect,
                dropLocationSelect);

        addComponents(controls, grids);

        // Set dragover styling
        Page.getCurrent().getStyles().add(".v-drag-over {color: red;}");
    }

    private List<Bean> createItems(int num, String prefix) {
        List<Bean> items = new ArrayList<>(num);

        IntStream.range(0, num)
                .forEach(i -> items
                        .add(new Bean(prefix + "_" + i, "value_" + i)));

        return items;
    }

    public static class Bean {
        private String id;
        private String value;

        public Bean(String id, String value) {
            this.id = id;
            this.value = value;
        }

        public String getId() {

            return id;
        }

        public void setId(String id) {
            this.id = id;
        }

        public String getValue() {
            return value;
        }

        public void setValue(String value) {
            this.value = value;
        }
    }
}