Application: gvSIG desktop

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2013 gvSIG Association.

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 3

 * of the License, or (at your option) any later version.

 *

 * This program 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 General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.app.extension.clipboard;

import java.io.StringReader;

import java.util.List;

import java.util.function.Predicate;

import javax.swing.JOptionPane;

import org.apache.commons.lang.mutable.MutableInt;

import org.apache.commons.lang3.StringUtils;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.andami.IconThemeHelper;

import org.gvsig.andami.plugins.Extension;

import org.gvsig.app.ApplicationLocator;

import org.gvsig.app.ApplicationManager;

import org.gvsig.app.project.documents.view.ViewDocument;

import org.gvsig.app.project.documents.view.ViewManager;

import org.gvsig.fmap.dal.feature.EditableFeature;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.fmap.mapcontext.MapContext;

import org.gvsig.fmap.mapcontext.layers.FLayer;

import org.gvsig.fmap.mapcontext.layers.vectorial.FLyrVect;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.tools.logger.FilteredLogger;

import javax.json.JsonObject;

import javax.json.Json;

import javax.json.JsonArray;

import javax.json.JsonReader;

import org.gvsig.app.project.documents.Document;

import org.gvsig.fmap.dal.exception.DataException;

import org.gvsig.fmap.dal.feature.Feature;

import org.gvsig.fmap.dal.feature.FeatureAttributeDescriptor;

import org.gvsig.fmap.dal.feature.FeatureSelection;

import org.gvsig.fmap.dal.feature.FeatureType;

@SuppressWarnings({"UseSpecificCatch", "Convert2Lambda", "ConvertToTryWithResources"})

public class PasteAlphanumericValuesFromClipboardExtension extends Extension {

    private static final Logger LOGGER = LoggerFactory.getLogger(PasteAlphanumericValuesFromClipboardExtension.class);

    @Override

    public void initialize() {

        IconThemeHelper.registerIcon("action", "layer-modify-clipboard-paste-alphanumeric", this);

    }

    @Override

    public void execute(String actionCommand) {

        if (actionCommand.compareToIgnoreCase("layer-modify-clipboard-paste-alphanumeric") != 0) {

            return;

        }

        ApplicationManager application = ApplicationLocator.getManager();

        ViewDocument viewdoc = (ViewDocument) application.getActiveDocument(ViewManager.TYPENAME);

        if (viewdoc == null) {

            return;

        }

        I18nManager i18n = ToolsLocator.getI18nManager();

        String clipboardValue = application.getFromClipboard();

        if (StringUtils.isBlank(clipboardValue)) {

            application.messageDialog(

                    i18n.getTranslation("_Clipboard_has_no_valid_info_or_is_empty"),

                    null,

                    i18n.getTranslation("_Pasting_features_from_clipboard"),

                    JOptionPane.INFORMATION_MESSAGE,

                    "ClipboardEmptyWhenPasteFeatures"

            );

            return;

        }

        JsonArray jsonArray;

        try {

            JsonReader jsonReader = Json.createReader(new StringReader(clipboardValue));

            jsonArray = jsonReader.readArray();

            jsonReader.close();

        } catch (Exception ex) {

            LOGGER.warn("Can't get JSON from the clipboard.", ex);

            application.messageDialog(

                    i18n.getTranslation("_Clipboard_has_no_valid_info_or_is_empty"),

                    null,

                    i18n.getTranslation("_Pasting_features_from_clipboard"),

                    JOptionPane.WARNING_MESSAGE,

                    "ClipboardNotValidWhenPasteFeatures"

            );

            return;

        }

        if (jsonArray.size() < 1) {

            application.messageDialog(

                    i18n.getTranslation("_No_features_found_in_clipboard"),

                    null,

                    i18n.getTranslation("_Pasting_features_from_clipboard"),

                    JOptionPane.WARNING_MESSAGE,

                    "ClipboardNotHasElementsWhenPasteFeatures"

            );

            return;

        }

        MapContext mapContext = viewdoc.getMapContext();

        List<FLayer> layers = mapContext.getLayers().getLayers((FLayer layer) -> layer instanceof FLyrVect

                && layer.isActive()

                && layer.isAvailable()

                && layer.isEditing());

        if (layers.isEmpty()) {

            return;

        }

        FLyrVect layer = (FLyrVect) layers.get(0);

        FeatureStore store = layer.getFeatureStore();

        FeatureSelection selectionFeatures;

        try {

            selectionFeatures = store.getFeatureSelection();

        } catch (DataException ex) {

            return;

        }

        try {

            if (selectionFeatures.isEmpty()) {

                return;

            }

        } catch (Exception ex) {

            return;

        }

        FilteredLogger theLogger = new FilteredLogger(LOGGER, "PasteFeatures", 10);

        Object obj = jsonArray.get(0);

        if (!(obj instanceof JsonObject)) {

            return;

        }

        JsonObject fjson = (JsonObject) obj;

        try {

            boolean hasPk = false;

            FeatureType ft = store.getDefaultFeatureTypeQuietly();

            for (String name : fjson.keySet()) {

                FeatureAttributeDescriptor attr = ft.getAttributeDescriptor(name);

                if( attr!=null ) {

                    if( attr.isPrimaryKey() ) {

                        hasPk = true;

                        break;

                    }

                }

            }

            final boolean copyPk = hasPk && (application.confirmDialog(

                        i18n.getTranslation("_Copy_primary_key_values_XQuestionX"),

                        i18n.getTranslation("_Pasting_features_from_clipboard"),

                        JOptionPane.YES_NO_OPTION,

                        JOptionPane.QUESTION_MESSAGE, 

                        "_CopyPrimaryKeyWhenPasteAlphanumericValuesFromClipboard"

                ) == JOptionPane.YES_OPTION);

            EditableFeature f = store.createNewFeature(fjson);

            for (Feature selectionFeature : selectionFeatures) {

                EditableFeature eds = selectionFeature.getEditable();

                eds.copyFrom(f, new Predicate<FeatureAttributeDescriptor>() {

                    @Override

                    public boolean test(FeatureAttributeDescriptor attr) {

                        String nameGeom;

                        try {

                            nameGeom = store.getDefaultFeatureType().getDefaultGeometryAttributeName();

                        } catch (DataException ex) {

                            nameGeom = null;

                        }

                        if (StringUtils.equals(attr.getName(), nameGeom)) {

                            return false;

                        }

                        if( !copyPk && attr.isPrimaryKey() ) {

                            return false;

                        }

                        return true;

                    }

                });

                selectionFeatures.update(eds);

            }

        } catch (Exception ex) {

            theLogger.warn("Can't create feature from JSON.", ex);

        }

        mapContext.invalidate();

    }

    @Override

    public boolean isEnabled() {

        ApplicationManager application = ApplicationLocator.getManager();

        ViewDocument viewdoc = (ViewDocument) application.getActiveDocument(ViewManager.TYPENAME);

        if (viewdoc == null) {

            return false;

        }

        String clipboardValue = application.getFromClipboard();

        if (StringUtils.isBlank(clipboardValue)) {

            return false;

        }

        JsonArray jsonArray;

        try {

            JsonReader jsonReader = Json.createReader(new StringReader(clipboardValue));

            jsonArray = jsonReader.readArray();

            jsonReader.close();

        } catch (Exception ex) {

            return false;

        }

        if (jsonArray.size() != 1) {

            return false;

        }

        MapContext mapContext = viewdoc.getMapContext();

        MutableInt numActiveVectorial = new MutableInt(0);

        MutableInt numActiveVectorialEditable = new MutableInt(0);

        MutableInt hasSelection = new MutableInt(0);

        List<FLayer> layers = mapContext.getLayers().getLayers(new Predicate<FLayer>() {

            @Override

            public boolean test(FLayer layer) {

                if (layer instanceof FLyrVect

                        && layer.isActive()

                        && layer.isAvailable()) {

                    numActiveVectorial.increment();

                    if (layer.isEditing()) {

                        numActiveVectorialEditable.increment();

                    }

                    if (!((FLyrVect) layer).getFeatureStore().isFeatureSelectionEmpty()) {

                        hasSelection.increment();

                    }

                }

                // No recuperamos ninguna capa, solo contamos las activa en edicion

                return false;

            }

        });

        if (numActiveVectorialEditable.intValue() == 1

                && numActiveVectorial.intValue() == 1

                && hasSelection.intValue() > 0) {

            return true;

        }

        return false;

    }

    @Override

    public boolean isVisible() {

        ApplicationManager application = ApplicationLocator.getManager();

        Document activeDocument = application.getActiveDocument(ViewManager.TYPENAME);

        return activeDocument != null;

    }

}