Application: gvSIG desktop

package org.gvsig.h2spatial;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.io.File;

import java.net.BindException;

import java.sql.SQLException;

import java.text.MessageFormat;

import java.util.ArrayList;

import java.util.List;

import org.apache.commons.collections.CollectionUtils;

import org.apache.commons.io.FilenameUtils;

import org.apache.commons.lang3.ArrayUtils;

import org.apache.commons.lang3.StringUtils;

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

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

import org.gvsig.fmap.dal.serverexplorer.filesystem.FilesystemStoreParameters;

import org.gvsig.fmap.dal.store.jdbc.JDBCServerExplorer;

import org.gvsig.fmap.dal.store.jdbc.JDBCServerExplorerParameters;

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

import org.gvsig.fmap.dal.swing.jdbc.JDBCConnectionPanel;

import org.gvsig.fmap.mapcontext.MapContextLocator;

import org.gvsig.fmap.mapcontext.MapContextManager;

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

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

import org.gvsig.tools.ToolsLocator;

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

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

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

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

import org.gvsig.webbrowser.WebBrowserFactory;

import org.gvsig.webbrowser.WebBrowserPanel;

import org.h2.tools.Server;

public class H2SpatialExtension extends Extension {

    @Override

    public void initialize() {

    }

    @Override

    public boolean canQueryByAction() {

        return true;

    }

    @Override

    public void postInitialize() {

        MapContextManager mapContextMgr = MapContextLocator.getMapContextManager();

        IconThemeHelper.registerIcon("layer", "layer-icon-h2", this);

        mapContextMgr.registerIconLayer("H2Spatial", "layer-icon-h2");

    }

    @Override

    public void execute(String actionCommand) {

        if( "tools-h2spatial-show-admintool".equalsIgnoreCase(actionCommand) ) {

            startserver();

            startbrowser();

            return;

        }

        if( "tools-h2spatial-linklayer".equalsIgnoreCase(actionCommand) ) {

            linkActiveLayers();

        }

    }

    private void startserver()  {

        try {

            Server server = Server.createWebServer("-webPort","8082");

            server.start();

        } catch (SQLException ex) {

            if( ex.getCause() instanceof BindException ) {

                logger.warn("Can't start H2 web server, ? Already running ?");

            } else {

                logger.warn("Can't start H2 web server.",ex);

            }

        }

    }

    private List<FLyrVect> getSelectedLayers() {

        ApplicationManager application = ApplicationLocator.getManager();

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

        if( viewdoc == null ) {

            return null;

        }

        FLayer[] layers = viewdoc.getMapContext().getLayers().getActives();

        if( ArrayUtils.isEmpty(layers) ) {

            return null;

        }

        List<FLyrVect> r = new ArrayList<>();

        for( FLayer layer : layers ) {

            if( layer instanceof FLyrVect ) {

                FeatureStore store = ((FLyrVect)layer).getFeatureStore();

                if( store != null ) {

                    if( store.getParameters() instanceof FilesystemStoreParameters &&

                        StringUtils.equalsIgnoreCase(store.getProviderName(),"Shape") ) {

                        r.add((FLyrVect) layer);

                    }

                }

            }

        }

        if( r.isEmpty() ) {

            return null;

        }

        return r;

    }

    @Override

    public boolean isEnabled() {

        return true;

    }

    @Override

    public boolean isVisible() {

        return true;

    }

    @Override

    public boolean isEnabled(String action) {

        if( "tools-h2spatial-show-admintool".equalsIgnoreCase(action) ) {

            return true;

        } 

        if( "tools-h2spatial-linklayer".equalsIgnoreCase(action) ) {

            return ! CollectionUtils.isEmpty(this.getSelectedLayers());

        }

        return false;

    }

    @Override

    public boolean isVisible(String action) {

        return true;

    }

    private void startbrowser() {

//        try {

//            DesktopOpen desktop = ToolsUtilLocator.getToolsUtilManager().createDesktopOpen();

//            desktop.browse(new URI("http://localhost:8082"));

//        } catch (URISyntaxException ex) {

//        }

        WindowManager winmanager = ToolsSwingLocator.getWindowManager();

        WebBrowserPanel webbrowser = WebBrowserFactory.createWebBrowserPanel();

        winmanager.showWindow(

            webbrowser.asJComponent(), 

            ToolsLocator.getI18nManager().getTranslation("_H2_administration_tool"), 

            WindowManager.MODE.WINDOW

        );

        webbrowser.setPage("http://localhost:8082");

    }

    private void linkActiveLayers() {

        final List<FLyrVect> layers = this.getSelectedLayers();

        if( CollectionUtils.isEmpty(layers) ) {

            return;

        }

        final JDBCConnectionPanel jdbcConnectionPanel = DALSwingLocator.getSwingManager().createJDBCConnectionPanel();

        jdbcConnectionPanel.setConnectorFilter(new JDBCConnectionPanel.ConnectorFilter() {

            @Override

            public boolean accept(JDBCServerExplorerParameters parameters) {

                return StringUtils.equalsIgnoreCase(parameters.getExplorerName(), "H2Spatial");

            }

        });

        WindowManager_v2 winmanager = (WindowManager_v2) ToolsSwingLocator.getWindowManager();

        final Dialog dlg = winmanager.createDialog(

            jdbcConnectionPanel.asJComponent(),

            "Select database",

            null, 

            WindowManager_v2.BUTTONS_OK_CANCEL

        );

        dlg.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                if( dlg.getAction()==WindowManager_v2.BUTTONS_OK ) {

                    JDBCServerExplorerParameters params = jdbcConnectionPanel.getServerExplorerParameters();

                    linkLayers(layers, params);

                }

            }

        });

        dlg.show(WindowManager.MODE.WINDOW);

    }

    private void linkLayers(List<FLyrVect> layers, JDBCServerExplorerParameters params) {

        try {

            JDBCServerExplorer explorer = (JDBCServerExplorer) DALLocator.getDataManager().openServerExplorer(

                params.getExplorerName(),

                params

            );

            for( FLyrVect layer : layers ) {

                FilesystemStoreParameters layerParameters = (FilesystemStoreParameters) 

                    layer.getFeatureStore().getParameters();

                File f = layerParameters.getFile();

                String sql = MessageFormat.format(

                    "CALL FILE_TABLE(''{0}'', ''{1}'')",

                    FilenameUtils.normalize(f.getAbsolutePath(), true),

                    layer.getName()

                );

                explorer.execute(sql);

            }

        } catch (Exception ex) {

        }

    }

}