Application: gvSIG desktop » gvSIG tools: Catalog

import java.awt.Dimension;

import java.net.MalformedURLException;

import java.net.URL;

import javax.swing.JLabel;

import javax.swing.JOptionPane;

import javax.swing.JPanel;

import javax.swing.JPasswordField;

import org.apache.commons.dbcp.SQLNestedException;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.fmap.dal.resource.db.DBParameters;

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

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

	public PostgisLayerLoader(Resource resource) {

		super(resource);

	}

	protected boolean hasSpecificLayer() {

		return false;

	}

	protected String getLayerName() {        

		return getResource().getName();

	}   

	protected DataStoreParameters createDataStoreParameters()

			throws InitializeException, ProviderNotRegisteredException {

		String jdbcUrl = getResource().getLinkage();

		String table = getResource().getName();

		DataStoreParameters dataStoreParameters = DATA_MANAGER.createStoreParameters("PostgreSQL");

		parseParams(dataStoreParameters, jdbcUrl, table);

		return dataStoreParameters;

	}

	private void parseParams(DataStoreParameters dataStoreParameters, String jdbcUrl, String table){

		StringTokenizer sti = new StringTokenizer(jdbcUrl,"?");

		String dbURL = sti.nextToken();

		String user = "";

		String pwd = "";

		if(sti.hasMoreTokens()){

			String p = sti.nextToken();

			TreeMap map = separateParams(p);

			user = (String) map.get((String) "USER");

			pwd = (String) map.get((String) "PASSWORD");

			map = Strings.separateParams(table);

			dataStoreParameters.setDynValue("USER", user);

			dataStoreParameters.setDynValue("PASSWORD",pwd);        

			dataStoreParameters.setDynValue("WHERECLAUSE","");

			dataStoreParameters.setDynValue("DBURL",dbURL);   

		}else{

			if(dbURL.startsWith("jdbc:postgresql://")){

				//jdbc:postgresql://test.gvsig.net:7432/gvsig:****@test

				try {

					URL url = new URL(dbURL.replace("jdbc:postgresql://", "http://"));

					if(url != null) {

						dataStoreParameters.setDynValue(JDBCConnectionParameters.HOST_PARAMTER_NAME, url.getHost());

						dataStoreParameters.setDynValue(JDBCConnectionParameters.PORT_PARAMTER_NAME, url.getPort()+"");

						//						/gvsig:****@test

						String path = url.getPath();

						user = path.substring(1, path.indexOf(":"));

						pwd = getCredentials(user);

						String db = path.substring(path.indexOf("@")+1);

						dataStoreParameters.setDynValue(DBParameters.USER_PARAMTER_NAME, user);

						dataStoreParameters.setDynValue(DBParameters.PASSWORD_PARAMTER_NAME, pwd);        

						dataStoreParameters.setDynValue(JDBCConnectionParameters.DBNAME_PARAMTER_NAME, db);     

					}

				} catch (MalformedURLException e) {

					// TODO Auto-generated catch block

					e.printStackTrace();

				}

			}

		}

		dataStoreParameters.setDynValue("Table", table); 

	}

	private String getCredentials(String user) {

		JPanel panel = new JPanel();

		JLabel label = new JLabel(Messages.getText("password_needed")+" ("+user+"):");

		JPasswordField pass = new JPasswordField();

		pass.setPreferredSize(new Dimension(150, 25));

		panel.add(label);

		panel.add(pass);

		String[] options = new String[]{Messages.getText("connectButton"), Messages.getText("cancelSearchButton")};

		int option = JOptionPane.showOptionDialog(null, panel, Messages.getText("password_needed_title"),

				JOptionPane.NO_OPTION, JOptionPane.PLAIN_MESSAGE,

				null, options, options[1]);

		if(option == 0){

			return new String(pass.getPassword());

		}

		return "";

	}

//	public void loadLayer() throws LayerLoaderException {

//		FLayer layer;

//		try{

//			if (hasSpecificLayer()){

//				layer = createLayerFromExtensionPoint();       

//			}else{

//				layer = createLayerFromParameters();                 

//			}

//			if (layer != null){

//				addLayerToView(layer);

//			}

//		} catch (Exception e) {

//			if(e. instanceof SQLNestedException){

//				JPanel panel = new JPanel();

//				JLabel label = new JLabel("Enter a password:");

//				JPasswordField pass = new JPasswordField();

//				panel.add(label);

//				panel.add(pass);

//				String[] options = new String[]{"OK", "Cancel"};

//				int option = JOptionPane.showOptionDialog(null, panel, "The title",

//						JOptionPane.NO_OPTION, JOptionPane.PLAIN_MESSAGE,

//						null, options, options[1]);

//				if(option == 0 && !StringUtils.isBlank(new String(pass.getPassword()))){

//					String pwd =  new String(pass.getPassword());

//					try{

//						DataStoreParameters dataStoreParameters = createDataStoreParameters(); 

//						dataStoreParameters.setDynValue(DBParameters.PASSWORD_PARAMTER_NAME, pwd);

//						MAP_CONTEXT_MANAGER.createLayer(getLayerName(), dataStoreParameters);

//					}catch (Exception ex) {

//						throw new LayerLoaderException(ex.getMessage(), ex, getWindowMessage());

//					} 

//				}else{

//					throw new LayerLoaderException(e.getMessage(), e, getWindowMessage());

//				}

//			}else{

//				e.printStackTrace();

//				throw new LayerLoaderException(e.getMessage(), e, getWindowMessage());

//			}

//		} 

//

//	}	

	private TreeMap separateParams(String pairValues){

		TreeMap map = new TreeMap(); 

		String[] params = pairValues.split("&");

		for (int i = 0; i < params.length; i++) {

			String[] nameValue = params[i].split("=");

			map.put(nameValue[0].toUpperCase(), nameValue[1]);

		}

		return map;

	}

	/*

	 *  (non-Javadoc)

	 * @see es.gva.cit.gvsig.catalogClient.loaders.LayerLoader#getErrorMessage()

	 */

	protected String getErrorMessage() {

		return Messages.getText("postgisError") + ".\n" +

				Messages.getText("server") + ": " + 

				getResource().getLinkage() + "\n" +

				Messages.getText("parameters") + ": " +

				getResource().getName();

	}

	/*

	 *  (non-Javadoc)

	 * @see es.gva.cit.gvsig.catalogClient.loaders.LayerLoader#getWindowMessage()

	 */

	protected String getWindowMessage() {

		return Messages.getText("postgisLoad");

	}