aboutsummaryrefslogtreecommitdiffstats
path: root/uitest/src/com/vaadin/tests/components/grid/GridEditorCustomField.java
blob: 4703c9899480988a77c80479b9b3320cae657060 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
/*
 * Copyright 2000-2014 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.HashSet;
import java.util.Set;

import com.vaadin.annotations.Theme;
import com.vaadin.server.VaadinRequest;
import com.vaadin.tests.components.AbstractTestUIWithLog;
import com.vaadin.tests.fieldgroup.ComplexPerson;
import com.vaadin.ui.Button;
import com.vaadin.ui.ComboBox;
import com.vaadin.ui.Component;
import com.vaadin.ui.CustomField;
import com.vaadin.ui.Grid;
import com.vaadin.ui.HorizontalLayout;

@Theme("valo")
public class GridEditorCustomField extends AbstractTestUIWithLog {

    @Override
    protected void setup(VaadinRequest request) {
        Grid grid = new PersonTestGrid(100);
        grid.setWidth("800px");
        grid.setColumns("firstName", "lastName", "address.city");
        grid.setEditorEnabled(true);
        Set<String> cities = new HashSet<String>();
        for (Object o : grid.getContainerDataSource().getItemIds()) {
            ComplexPerson p = (ComplexPerson) o;
            cities.add(p.getAddress().getCity());
        }
        CustomCitySelect cityEditor = new CustomCitySelect(
                cities.toArray(new String[cities.size()]));
        grid.getColumn("address.city").setEditorField(cityEditor);
        addComponent(grid);
    }

    public static class CustomCitySelect extends CustomField<String> {
        private HorizontalLayout fieldLayout;
        private String[] values;
        private ComboBox cityComboBox;

        public CustomCitySelect(String... values) {
            this.values = values;
        }

        @Override
        protected Component initContent() {
            fieldLayout = new HorizontalLayout();
            fieldLayout.setWidth("100%");

            cityComboBox = new ComboBox();
            for (String value : values) {
                cityComboBox.addItem(value);
            }
            fieldLayout.addComponent(cityComboBox);
            fieldLayout.setExpandRatio(cityComboBox, 1.0f);

            Button addCountryButton = new Button("New");
            fieldLayout.addComponent(addCountryButton);

            return fieldLayout;
        }

        @Override
        public Class<String> getType() {
            return String.class;
        }

        @Override
        protected void setInternalValue(String newValue) {
            super.setInternalValue(newValue);
            if (cityComboBox == null) {
                return;
            }
            cityComboBox.setValue(newValue);
        }

        @Override
        public String getInternalValue() {
            if (cityComboBox == null) {
                return null;
            }
            return (String) cityComboBox.getValue();
        }
    }

}