Application: gvSIG desktop

package org.gvsig.fmap.dal.swing.impl.actions;

import java.awt.event.ActionEvent;

import java.net.URL;

import java.util.LinkedHashMap;

import java.util.List;

import java.util.Map;

import javax.json.JsonObject;

import javax.swing.AbstractAction;

import javax.swing.Action;

import javax.swing.JComponent;

import javax.swing.JMenuItem;

import javax.swing.JPopupMenu;

import org.apache.commons.lang3.tuple.ImmutablePair;

import org.apache.commons.lang3.tuple.Pair;

import org.gvsig.fmap.dal.DataStore;

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

import org.gvsig.fmap.dal.swing.DALActionFactory;

import org.gvsig.fmap.dal.swing.DALActionFactory.DALActionContext;

import org.gvsig.fmap.dal.swing.DALSwingLocator;

import org.gvsig.fmap.dal.swing.DataSwingManager;

import static org.gvsig.fmap.dal.swing.impl.actions.ReportUtils.getReportActions;

import static org.gvsig.fmap.dal.swing.impl.actions.ReportUtils.getReports;

import org.gvsig.fmap.dal.swing.report.ReportAction;

import org.gvsig.fmap.dal.swing.report.ReportActionFactory;

import org.gvsig.fmap.dal.swing.report.ReportActionFactory.ReportCustomActionFactory;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.tools.swing.api.ToolsSwingLocator;

import org.gvsig.tools.swing.icontheme.IconTheme;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 *

 * @author jjdelcerro

 */

@SuppressWarnings("UseSpecificCatch")

public class ReportStoreAction

        extends AbstractAction {

    public static String ACTION_NAME = "Reports";

    public static class ReportStoreActionFactory implements DALActionFactory {

        private final Map<String,Pair<ReportCustomActionFactory,Boolean>> customActions;

        public ReportStoreActionFactory() {

            this.customActions = new LinkedHashMap<>();

        }

        @Override

        public String getName() {

            return ACTION_NAME;

        }

        @Override

        public Action createAction(DALActionContext context) {

            return new ReportStoreAction(this, context);

        }

        @Override

        public boolean isApplicable(Object... args) {

            return true;

        }

        public void registerCustomAction(ReportCustomActionFactory factory, boolean insertSeparator) {

            this.customActions.put(factory.getName(), new ImmutablePair<>(factory,insertSeparator));

        }

        public Map<String, Pair<ReportCustomActionFactory, Boolean>> getCustomActions() {

            return this.customActions;

        }

        public static void selfRegister() {

            DataSwingManager dalSwingManager = DALSwingLocator.getSwingManager();

            dalSwingManager.registerStoreAction(new ReportStoreActionFactory());

            String[][] iconNames = new String[][]{

                new String[]{"Report", "report-select-reports"}

            };

            IconTheme theme = ToolsSwingLocator.getIconThemeManager().getCurrent();

            for (String[] icon : iconNames) {

                URL url = ReportStoreAction.class.getResource("images/" + icon[1] + ".png");

                theme.registerDefault("Report", icon[0], icon[1], null, url);

            }

        }

    }

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

    private final DALActionContext context;

    private final ReportStoreActionFactory factory;

    @SuppressWarnings("OverridableMethodCallInConstructor")

    public ReportStoreAction(ReportStoreActionFactory factory, DALActionContext context) {

        this.factory = factory;

        this.context = context;

        I18nManager i18n = ToolsLocator.getI18nManager();

        this.putValue(

                Action.ACTION_COMMAND_KEY,

                ACTION_NAME

        );

        this.putValue(

                Action.SHORT_DESCRIPTION,

                i18n.getTranslation("_Reports")

        );

        this.putValue(

                Action.SMALL_ICON,

                ToolsSwingLocator.getIconThemeManager().getCurrent().get("report-select-reports")

        );

    }

    @Override

    public void actionPerformed(ActionEvent e) {

        try {

//            this.setEnabled(false);

            DataStore store = this.context.getStore();

            if (!(store instanceof FeatureStore)) {

                return;

            }

            List<ReportAction> actions = getReportActions(store, null, context, e);

            JPopupMenu popup = new JPopupMenu();

            for (ReportAction action : actions) {

                popup.add(new JMenuItem(action));

            }

            for (Pair<ReportCustomActionFactory, Boolean> customAction : this.factory.getCustomActions().values()) {

                Action action = customAction.getLeft().createCustomAction(e, context);

                if( action !=null ) {

                    if( customAction.getRight() && popup.getComponentCount()>0 ) {

                        popup.addSeparator();

                    }

                    popup.add(action);

                }

            }

            JComponent button = this.context.getActionButton(ACTION_NAME);

            popup.show(button, 0, button.getHeight());

        } catch (Exception ex) {

            LOGGER.warn("Can't show form", ex);

        } finally {

//            this.setEnabled(true);

        }

    }

    @Override

    public boolean isEnabled() {

        try {

            DataStore store = this.context.getStore();

            if (!(store instanceof FeatureStore)) {

                return false;

            }

            DataSwingManager dataSwingManager = DALSwingLocator.getDataSwingManager();

            List<JsonObject> reports = getReports((FeatureStore) store);

            if (reports != null && !reports.isEmpty()) {

                for (JsonObject json : reports) {

                    for (ReportActionFactory thefactory : dataSwingManager.getReportActionFactories()) {

                        if( thefactory.isApplicable(json,this.context)) {

                            return true;

                        }

                    }

                }

            }

            Map<String, Pair<ReportCustomActionFactory, Boolean>> customActions = this.factory.getCustomActions();

            return customActions!=null && !customActions.isEmpty();

        } catch (Exception ex) {

            LOGGER.warn("Can't check isEnabled", ex);            

        }

        return false;

    }

}