MDateField.java

package org.vaadin.viritin.v7.fields;

import com.vaadin.event.FieldEvents.BlurListener;
import com.vaadin.v7.data.Property;
import com.vaadin.v7.event.FieldEvents;
import com.vaadin.server.Resource;
import com.vaadin.v7.shared.ui.datefield.Resolution;
import com.vaadin.v7.ui.DateField;

import java.util.Date;
import java.util.Map;

/**
 *
 * @author Matti Tahvonen
 */
public class MDateField extends DateField {

    public MDateField() {
    }

    public MDateField(String caption) {
        super(caption);
    }

    public MDateField(String caption, Property dataSource) {
        super(caption, dataSource);
    }

    public MDateField(Property dataSource) throws IllegalArgumentException {
        super(dataSource);
    }

    public MDateField(String caption, Date value) {
        super(caption, value);
    }

    public enum InitialTimeMode {
        START_OF_DAY, END_OF_DAY, NOW
    }

    private InitialTimeMode initialTimeMode;

    /**
     * Flag to tell if setting the initial (from null to something) value by the
     * end user
     */
    private boolean settingInitialValue;

    /**
     * @return the method how value of the time part is to be initialized when
     * resolution setting don't show time.
     */
    public InitialTimeMode getInitialTimeMode() {
        return initialTimeMode;
    }

    /**
     *
     * @param initialTimeMode the method how value of the time part is to be initialized when
     * resolution setting don't show time.
     */
    public void setInitialTimeMode(InitialTimeMode initialTimeMode) {
        this.initialTimeMode = initialTimeMode;
    }

    @Override
    public void changeVariables(Object source, Map<String, Object> variables) {
        if (getValue() == null) {
            settingInitialValue = true;
        }
        super.changeVariables(source, variables);
        settingInitialValue = false;
    }

    @Override
    protected void setValue(Date newValue, boolean repaintIsNotNeeded) throws ReadOnlyException {
        if (settingInitialValue && getResolution().ordinal() > Resolution.HOUR.ordinal()) {
            if (getInitialTimeMode() == InitialTimeMode.START_OF_DAY) {
                newValue.setHours(0);
                newValue.setMinutes(0);
                newValue.setSeconds(0);
                newValue.setTime(
                        newValue.getTime() - newValue.getTime() % 1000l);
            } else if (getInitialTimeMode() == InitialTimeMode.END_OF_DAY) {
                newValue.setHours(23);
                newValue.setMinutes(59);
                newValue.setSeconds(59);
                newValue.setTime(
                        newValue.getTime() - newValue.getTime() % 1000l + 999l);
            }
        }
        super.setValue(newValue, repaintIsNotNeeded);
    }

    public MDateField withIcon(Resource icon) {
        setIcon(icon);
        return this;
    }

    public MDateField withResolution(Resolution resolution) {
        setResolution(resolution);
        return this;
    }

    public MDateField withStyleName(String... styleNames) {
        for (String styleName : styleNames) {
            addStyleName(styleName);
        }
        return this;
    }

    public MDateField withRequired(boolean required) {
        setRequired(required);
        return this;
    }

    public MDateField withRequiredError(String requiredError) {
        setRequiredError(requiredError);
        return this;
    }

    public MDateField withFullWidth() {
        setWidth("100%");
        return this;
    }

    public MDateField withValue(Date newDate) {
        setValue(newDate);
        return this;
    }

    public MDateField withValueChangeListener(Property.ValueChangeListener listener) {
        addValueChangeListener(listener);
        return this;
    }

    public MDateField withBlurListener(BlurListener listener) {
        addBlurListener(listener);
        return this;
    }

    public MDateField withWidth(float width, Unit unit) {
        setWidth(width, unit);
        return this;
    }

    public MDateField withWidth(String width) {
        setWidth(width);
        return this;
    }

    public MDateField withId(String id) {
        setId(id);
        return this;
    }

    public MDateField withVisible(boolean visible) {
        setVisible(visible);
        return this;
    }

    public MDateField withDateFormat(String dateFormat){
        setDateFormat(dateFormat);
        return this;
    }
}