/*
 *    Geotoolkit - An Open Source Java GIS Toolkit
 *    http://www.geotoolkit.org
 *
 *    (C) 2007 - 2008, Open Source Geospatial Foundation (OSGeo)
 *    (C) 2008 - 2009, Johann Sorel
 *    (C) 2011, Geomatys
 *
 *    This library is free software; you can redistribute it and/or
 *    modify it under the terms of the GNU Lesser General Public
 *    License as published by the Free Software Foundation;
 *    version 2.1 of the License.
 *
 *    This library is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 *    Lesser General Public License for more details.
 */
package org.geotoolkit.gui.swing.render2d.control.edition;

import java.awt.event.ActionEvent;
import javax.swing.AbstractAction;
import javax.swing.ImageIcon;

import org.geotoolkit.data.FeatureStoreContentEvent;
import org.geotoolkit.data.FeatureStoreListener;
import org.geotoolkit.data.FeatureStoreManagementEvent;
import org.geotoolkit.font.FontAwesomeIcons;
import org.geotoolkit.font.IconBuilder;
import org.geotoolkit.gui.swing.resource.MessageBundle;
import org.geotoolkit.map.FeatureMapLayer;

/**
 *
 * @author Johann Sorel
 * @module
 */
public class SessionRollbackAction extends AbstractAction implements FeatureStoreListener {

    private static final ImageIcon ICON_ROLLBACK = IconBuilder.createIcon(FontAwesomeIcons.ICON_UNDO, 16, FontAwesomeIcons.DEFAULT_COLOR);
    private static final ImageIcon ICON_WAIT = IconBuilder.createIcon(FontAwesomeIcons.ICON_SPINNER, 16, FontAwesomeIcons.DEFAULT_COLOR);
    
    private final FeatureStoreListener.Weak weakListener = new Weak(this);
    private FeatureMapLayer layer;

    public SessionRollbackAction() {
        this(null);
    }

    public SessionRollbackAction(final FeatureMapLayer layer) {
        putValue(SMALL_ICON, ICON_ROLLBACK);
        putValue(NAME, MessageBundle.format("sessionRollback"));
        putValue(SHORT_DESCRIPTION, MessageBundle.format("sessionRollback"));
        setLayer(layer);
    }

    @Override
    public boolean isEnabled() {
        return super.isEnabled() && (layer != null)
                && (layer.getCollection().getSession().hasPendingChanges());
    }

    public FeatureMapLayer getLayer() {
        return layer;
    }

    public void setLayer(final FeatureMapLayer layer) {
        //remove previous listener
        weakListener.unregisterAll();

        final boolean newst = isEnabled();
        this.layer = layer;
        firePropertyChange("enabled", !newst, newst);

        if(this.layer != null){
            weakListener.registerSource(this.layer.getCollection().getSession());
        }
    }

    @Override
    public void actionPerformed(final ActionEvent event) {

        if (layer != null ) {
            putValue(SMALL_ICON, ICON_WAIT);
            final Thread t = new Thread(){
                @Override
                public void run() {
                    try {
                        layer.getCollection().getSession().rollback();
                    }finally{
                        putValue(SMALL_ICON, ICON_ROLLBACK);
                    }
                }
            };
            t.start();
        }
    }

    @Override
    public void structureChanged(final FeatureStoreManagementEvent event) {
    }

    @Override
    public void contentChanged(final FeatureStoreContentEvent event) {
        if(event.getType() == FeatureStoreContentEvent.Type.SESSION){
            //refresh enable state
            setLayer(layer);
        }
    }

}