Application: gvSIG desktop » gvSIG Document Layout

/*

 * 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.app.extension.reportfromlayout;

import java.awt.Dimension;

import java.awt.event.ActionEvent;

import java.io.File;

import java.io.FileOutputStream;

import java.io.InputStream;

import java.util.List;

import java.util.function.Predicate;

import javax.json.JsonObject;

import javax.swing.DefaultComboBoxModel;

import javax.swing.JOptionPane;

import javax.swing.SwingUtilities;

import javax.swing.event.ChangeEvent;

import javax.swing.event.TreeSelectionEvent;

import org.apache.commons.io.IOUtils;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.app.ApplicationLocator;

import org.gvsig.app.project.documents.layout.LayoutManager;

import org.gvsig.app.project.documents.layout.report.LayoutReportActionImpl;

import org.gvsig.app.project.documents.layout.report.LayoutReportActionFactory;

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

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.storesrepository.StoresRepositoryController;

import org.gvsig.i18n.Messages;

import org.gvsig.json.Json;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dispose.DisposeUtils;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.tools.resourcesstorage.ResourcesStorage;

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

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

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

import org.gvsig.tools.swing.api.pickercontroller.FilePickerController;

import org.gvsig.tools.swing.api.threadsafedialogs.ThreadSafeDialogsManager;

import org.gvsig.tools.swing.api.windowmanager.Dialog;

import org.gvsig.tools.swing.api.windowmanager.WindowManager_v2;

import org.gvsig.tools.util.LabeledValue;

import org.gvsig.tools.util.LabeledValueImpl;

import org.gvsig.utils.GenericFileFilter;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 *

 * @author fdiaz

 */

public class ExportReportToLayout extends ExportReportToLayoutPanelView {

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

    private StoresRepositoryController treeSourceController;

    private FeatureStore source;

    private Dialog dialog;

    private FilePickerController pickerFile;

    public ExportReportToLayout() {

        this.initComponents();

    }

    private void initComponents() {

        DataSwingManager dataSwingManager = DALSwingLocator.getSwingManager();

        translate();

//        fillCboLayout();

        this.treeSourceController = dataSwingManager.createStoresRepositoryController(this.treeSource);

        this.treeSource.addTreeSelectionListener((TreeSelectionEvent e) -> {

            doChangeSource();

            doUpdateComponents();

        });

        ToolsSwingManager toolsSwingManager = ToolsSwingLocator.getToolsSwingManager();

        this.pickerFile = toolsSwingManager.createFilePickerController(txtFile, btnFile);

        GenericFileFilter layoutMainExtensionFilter = new GenericFileFilter(

                LayoutManager.TEMPLATE_FILE_POINTEXT,

                Messages.getText("_Layout_template") + " (*"

                + LayoutManager.TEMPLATE_FILE_POINTEXT + ")");

//        this.pickerFile.addChoosableFileFilter(layoutMainExtensionFilter);

        this.pickerFile.setFileFilter(layoutMainExtensionFilter);

        this.pickerFile.addChangeListener((ChangeEvent e) -> {

            doUpdateComponents();

        });

        this.cboReport.addActionListener((ActionEvent e) -> {

            doChangeReport();

        });

        Dimension screensize = ApplicationLocator.getApplicationManager().getUIManager().getInternalSize();

        ToolsSwingUtils.ensureHeightWitdh(

                this, 

                Math.max(3*(screensize.height/4), ToolsSwingUtils.rows2px(80)),

                Math.max(2*(screensize.width/4), ToolsSwingUtils.cols2px(35)), 

                7*(screensize.height/8), 

                3*(screensize.width/4)

        );        

        SwingUtilities.invokeLater(this::doUpdateComponents);

    }

    private void translate() {

        ToolsSwingManager manager = ToolsSwingLocator.getToolsSwingManager();

        manager.translate(lblSource);

        manager.translate(lblReport);

        manager.translate(lblFile);

    }

    private void doChangeSource() {

        FeatureStore theSource = this.treeSourceController.getSelectedFeatureStore();

        if( theSource==null ) {

            return;

        }

        if(this.source != null){

            DisposeUtils.disposeQuietly(this.source);

        }

        this.source = theSource;

        fillCboReport();

    }

    private void doUpdateComponents() {

        this.treeSourceController.setEnabled(true);

        this.cboReport.setEnabled(this.source != null);

        if(this.dialog != null){

            boolean enabled = (this.source != null) && (this.pickerFile.get() != null);

            this.dialog.setButtonEnabled(WindowManager_v2.BUTTON_OK, enabled);

        }

    }

    private void fillCboReport() {

        DataSwingManager dataSwingManager = DALSwingLocator.getDataSwingManager();

        DefaultComboBoxModel<LabeledValue<ReportAction>> model = new DefaultComboBoxModel();

        if(this.source == null) {

            this.cboReport.setModel(model);

            return;

        }

        List<ReportAction> reports = dataSwingManager.getReportActions(source, (Predicate<JsonObject>) (JsonObject t) -> LayoutReportActionFactory.isJsonApplicable(t));

        for (ReportAction report : reports) {

            model.addElement(new LabeledValueImpl<>(report.getReportLabel(), report));

        }

        cboReport.setModel(model);

    }

//    List<Document> getLayouts() {

//

//        List<Document> layouts = ApplicationLocator.getProjectManager().getCurrentProject().getDocuments(LayoutDocument.class);

//

//        return layouts;

//

//    }

//    private void fillCboLayout() {

//        DefaultComboBoxModel<LabeledValue<LayoutDocument>> model = new DefaultComboBoxModel();

//        List<Document> layouts = getLayouts();

//        

//        if(layouts.size() > 1){

//            Collections.sort(layouts, new CompareUtils.NullSafeComparator<Document>() {

//                @Override

//                public int safeCompare(Document o1, Document o2) {

//                    return StringUtils.compare(o1.getName(), o2.getName());

//                }

//            });

//        }

//        

//        for (Document layout : layouts) {

//            if(layout instanceof LayoutDocument){

//                model.addElement(new LabeledValueImpl<>(layout.getName(), (LayoutDocument)layout));

//            }

//        }

//        cboLayout.setModel(model);

//

//    }

    private ReportAction getReportAction(){

        return (ReportAction) ((LabeledValue)cboReport.getSelectedItem()).getValue();

    }

//    private LayoutDocument getLayoutDocument(){

//        return (LayoutDocument) ((LabeledValue)cboLayout.getSelectedItem()).getValue();

//    }

    public void fetch(ExportReportToLayoutParameters params){

        params.setSource(this.source);

        params.setReportAction(this.getReportAction());

        params.setFile(this.pickerFile.get());

    }

    public boolean doExportReportToLayout(ExportReportToLayoutParameters params) {

        FeatureStore source = params.getSource();

        ReportAction report = params.getReportAction();

        File file = params.getFile();

        ThreadSafeDialogsManager dialogs = ToolsSwingLocator.getThreadSafeDialogsManager();

        I18nManager i18n = ToolsLocator.getI18nManager();

        if(report == null) {

            dialogs.messageDialog(

                    i18n.getTranslation("_Report_cant_be_null"),

                    null,

                    "_Export_report_to_layout",

                    JOptionPane.WARNING_MESSAGE);

            return false;

        }

        if(source == null){

            dialogs.messageDialog(

                    i18n.getTranslation("_Source_cant_be_null"),

                    null,

                    "_Export_report_to_layout",

                    JOptionPane.WARNING_MESSAGE);

            return false;

        }

        if(file == null){

            dialogs.messageDialog(

                    i18n.getTranslation("_File_cant_be_null"),

                    null,

                    "_Export_report_to_layout",

                    JOptionPane.WARNING_MESSAGE);

            return false;

        }

        return doExportReportToLayout(source, report, file);

    }

    public boolean doExportReportToLayout(FeatureStore store, ReportAction report, File templateFile) {

        ResourcesStorage.Resource resourceReport = null;

        ResourcesStorage.Resource resourceLayout = null;

        FileOutputStream templateOs = null;

        ThreadSafeDialogsManager dialogs = ToolsSwingLocator.getThreadSafeDialogsManager();

        I18nManager i18n = ToolsLocator.getI18nManager();

        try {

            ResourcesStorage resources = store.getResourcesStorage();

            String resourceReportName = getResourceName(resources, report.getReportName());

            if (resourceReportName == null) {

                dialogs.messageDialog(

                        i18n.getTranslation("_Cant_find_resource_XreportnameX",

                                new String[]{

                                                resourceReportName,

                                        }

                        ),

                        null,

                        i18n.getTranslation("_Export_report_to_layout"),

                        JOptionPane.WARNING_MESSAGE);

                return false;

            }

            resourceReport = resources.getResource(resourceReportName);

            JsonObject jsonReport = Json.createObject(resourceReport.asInputStream());

            String templateName = jsonReport.getString(LayoutReportActionImpl.LAYOUT_REPORT_TEMPLATE_NAME, LayoutReportActionImpl.LAYOUT_REPORT_DEFAULT_TEMPLATE_NAME);

            if(StringUtils.isBlank(templateName)){

                dialogs.messageDialog(

                        i18n.getTranslation("_Template_name_is_blank"//,

//                                new String[]{

//                                                resourceReportName,

//                                        }

                        ),

                        null,

                        i18n.getTranslation("_Export_report_to_layout"),

                        JOptionPane.WARNING_MESSAGE);

                return false;

            }

            resourceLayout = resources.getResource(templateName);

            templateOs = new FileOutputStream(templateFile);

            IOUtils.copy(resourceLayout.asInputStream(), templateOs);

            return true;

        } catch (Exception ex) {

            dialogs.messageDialog(

                    i18n.getTranslation("_Cant_export_report_to_layout"),

                    null,

                    i18n.getTranslation("_Export_report_to_layout"),

                    JOptionPane.WARNING_MESSAGE);

            return false;

        } finally {

            IOUtils.closeQuietly(templateOs);

            IOUtils.closeQuietly(resourceReport);

            IOUtils.closeQuietly(resourceLayout);

        }

    }

//    private String getNextResourceName(ResourcesStorage resources, String resourceName){

//        List<ResourcesStorage.Resource> reports = resources.getResources(resourceName);

//        if(reports == null || reports.isEmpty()){

//            return resourceName;

//        }

//        int n = -1;

//        for (ResourcesStorage.Resource report : reports) {

//            if(StringUtils.contains(report.getName(), resources.getSeparator())){

//                String[] s = StringUtils.split(report.getName(), resources.getSeparator());

//                n = Math.max(n, Integer.parseInt(s[0]));

//            } else {

//                n=0;

//            }

//        }

//        

//        if (n < 0) {

//            return resourceName;

//        }

//        

//        n++;

//        return String.valueOf(n) + resources.getSeparator() + resourceName;

//

//        

//    }

//    

    private String getResourceName(ResourcesStorage resources, String reportName) {

        try {

            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 = javax.json.Json.createReader(is).readObject();

                            String name = json.getString(LayoutReportActionImpl.LAYOUT_REPORT_NAME, null);

                            if(StringUtils.equalsIgnoreCase(name, reportName)){

                                return resource.getName();

                            }

                        } 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);

        }

        return null;

    }

    private void doChangeReport() {

        ReportAction report = getReportAction();

        this.pickerFile.set(null);

//        if(report == null){

//            this.txtIdentifier.setText("");

//            this.txtLabel.setText("");

//        } else {

//            this.txtIdentifier.setText(report.getReportName());

//            this.txtLabel.setText(report.getReportLabel());

//            

//        }

    }

    void setDialog(Dialog dialog) {

        this.dialog = dialog;

    }

}