Application: gvSIG desktop

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

import java.awt.event.ActionEvent;

import java.io.InputStream;

import java.net.URL;

import java.util.ArrayList;

import java.util.Collections;

import java.util.LinkedHashMap;

import java.util.List;

import java.util.Map;

import javax.json.Json;

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.io.IOUtils;

import org.apache.commons.lang3.StringUtils;

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.DALSwingLocator;

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

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.gvsig.tools.resourcesstorage.ResourcesStorage;

import org.gvsig.tools.resourcesstorage.ResourcesStorage.Resource;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

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

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

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.dispose.DisposeUtils;

/**

 *

 * @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")

        );

    }

    private List<JsonObject> getReports(FeatureStore store) {

        List<JsonObject> reports = new ArrayList<>();

        ResourcesStorage resources = null;

        try {

            resources = store.getResourcesStorage();

            if (resources != null) {

                List<Resource> reportsResources = resources.getResources("report");

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

                    for (Resource resource : reportsResources) {

                        InputStream is = null;

                        try {

                            is = resource.asInputStream();

                            JsonObject json = Json.createReader(is).readObject();

                            reports.add(json);

                        } catch (Exception ex) {

                            LOGGER.warn("Can't load report form resource (" + resource.getURL() + ")", ex);

                        } finally {

                            IOUtils.closeQuietly(is);

                        }

                    }

                }

            }

        } finally {

            DisposeUtils.disposeQuietly(resources);

        }

        if (reports.isEmpty()) {

            return null;

        }

        return reports;

    }

    @Override

    public void actionPerformed(ActionEvent e) {

        try {

            DataSwingManager dataSwingManager = DALSwingLocator.getDataSwingManager();

//            this.setEnabled(false);

            DataStore store = this.context.getStore();

            if (!(store instanceof FeatureStore)) {

                return;

            }

            List<ReportAction> actions = new ArrayList<>();

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

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

                for (JsonObject json : reports) {

                    ReportAction action = null;

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

                        if( theFactory.isApplicable(json,this.context, e)) {

                            action = theFactory.createReportAction(

                                    (FeatureStore) store,

                                    this.context.getQuery(),

                                    this.context.getSelecteds(),

                                    json

                            );

                        }

                    }

                    if( action != null ) {

                        String label = (String) action.getValue(Action.NAME);

                        if( StringUtils.isBlank(label) ) {

                            action.putValue(Action.NAME, action.getLabel());

                        }

                        actions.add(action);

                    }

                }

            }

            Collections.sort(

                    actions, 

                    (ReportAction o1, ReportAction o2) -> StringUtils.compare(o1.getLabel(), o2.getLabel())

            );

            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 = this.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;

                        }

                    }

                }

            }

        } catch (Exception ex) {

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

        }

        return false;

    }

}