Application: gvSIG desktop

/*

 * To change this license header, choose License Headers in Project Properties.

 * To change this template file, choose Tools | Templates

 * and open the template in the editor.

 */

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

import java.awt.event.ActionEvent;

import java.io.InputStream;

import java.util.ArrayList;

import java.util.Collections;

import java.util.List;

import java.util.function.Predicate;

import javax.json.Json;

import javax.json.JsonObject;

import javax.swing.Action;

import org.apache.commons.io.IOUtils;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.fmap.dal.DataStore;

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

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

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

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

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

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

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

import org.gvsig.tools.dispose.DisposeUtils;

import org.gvsig.tools.resourcesstorage.ResourcesStorage;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 *

 * @author fdiaz

 */

public class ReportUtils {

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

    private ReportUtils(){

    }

    public static List<JsonObject> getReports(FeatureStore store) {

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

        ResourcesStorage resources = null;

        try {

            resources = store.getResourcesStorage();

            if (resources != null) {

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

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

                    for (ResourcesStorage.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);

                        }

                        DisposeUtils.disposeQuietly(resource);

                    }

                }

            }

        } finally {

            DisposeUtils.disposeQuietly(resources);

        }

        if (reports.isEmpty()) {

            return null;

        }

        return reports;

    }

    public static List<ReportAction> getReportActions(DataStore store) {

        return getReportActions(store, null, null, null);

    }

    public static List<ReportAction> getReportActions(DataStore store, Predicate filter) {

        return getReportActions(store, filter, null, null);

    }

    public static List<ReportAction> getReportActions(DataStore store, Predicate<JsonObject> filter, DALActionFactory.DALActionContext context, ActionEvent e) {

        DataSwingManager dataSwingManager = DALSwingLocator.getDataSwingManager();

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

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

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

            FeatureSelection selecteds = null;

            if(context == null){

                selecteds = context.getSelecteds();

            }

            for (JsonObject json : reports) {

                if(filter == null || filter.test(json)){

                    ReportAction action = null;

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

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

                            if(context == null){

                                action = theFactory.createReportAction(

                                        (FeatureStore) store,

                                        null,

                                        null,

                                        json

                                );

                            } else {

                                action = theFactory.createReportAction(

                                        (FeatureStore) store,

                                        context.getQuery(),

                                        selecteds,

                                        json

                                );

                            }

                        }

                    }

                    if( action != null ) {

                        String name = action.getReportName();

                        if( StringUtils.isBlank(name) ) {

                            action.setReportName(store.getName());

                        }

                        String label = action.getReportLabel();

                        if( StringUtils.isBlank(label) ) {

                            action.setReportLabel(action.getReportName());

                        }

                        actions.add(action);

                    }

                }

            }

        }

        Collections.sort(

                actions,

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

        );

        return actions;

    }

}