org.swixml.examples.wizard.GenerateBeanPage Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of swixml-examples-wizard Show documentation
The newest version!
package org.swixml.examples.wizard;


import java.awt.Component;
import java.beans.PropertyDescriptor;
import java.io.File;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import javax.swing.JFileChooser;
import javax.swing.JOptionPane;

import org.apache.ddlutils.Platform;
import org.apache.ddlutils.PlatformFactory;
import org.apache.ddlutils.model.Database;
import org.apache.ddlutils.model.Table;
import org.jdesktop.application.Action;
import org.jdesktop.application.Application;
import org.jdesktop.observablecollections.ObservableCollections;
import org.netbeans.spi.wizard.ResultProgressHandle;
import org.netbeans.spi.wizard.Wizard;
import org.netbeans.spi.wizard.WizardPage;
import org.netbeans.spi.wizard.WizardPanelNavResult;
import org.swixml.jsr295.BindingUtils;


@SuppressWarnings("serial")
public class GenerateBeanPage extends WizardPage implements DDLWizardConstants{

	class GenerateBeanTask extends WizardPanelNavResult {

		
            @Override
            public void start(Map wizardData, ResultProgressHandle progress) {
                progress.setBusy("Generating...");


                try {
                    progress.setBusy("Generating Beans!");
                    //BeanGeneratorUtils.generateBean(generateBeanList, getPackageName(), outputDir);

                    //progress.setBusy("Generating BeanInfo!");
                    //BeanGeneratorUtils.generateBeanInfo(generateBeanList, getPackageName(), outputDir);

                    progress.finished(WizardPanelNavResult.PROCEED);
                } catch (Exception e) {
                	e.printStackTrace();
                    progress.failed(String.format("Error on Bean creation\n%s", e.getMessage()), true /*canNavigateBack*/);
                }

            }

	}

	File outputDir;
	final List generateBeanList = ObservableCollections.observableList( new ArrayList() );
	
	public GenerateBeanPage() {
		super("generate", "Generate Bean&BeanInfo");

		
		setGenerateBean(false);


		try {
			
			Map map = BindingUtils.getPropertyMap(GenerateBean.class);
			
			{
			PropertyDescriptor pd = map.get("selected"); 
			BindingUtils.setTableColumnEditable(pd, true);
			BindingUtils.setTableColumnIndex(pd, 1);
			BindingUtils.setTableColumnIsBound(pd, true);
			pd.setDisplayName("");
			}
			{
			PropertyDescriptor pd = map.get("tableName"); 
			BindingUtils.setTableColumnEditable(pd, false);
			BindingUtils.setTableColumnIndex(pd, 2);
			BindingUtils.setTableColumnIsBound(pd, true);
			}
			{
			PropertyDescriptor pd = map.get("beanName"); 
			BindingUtils.setTableColumnEditable(pd, true);
			BindingUtils.setTableColumnIndex(pd, 3);
			BindingUtils.setTableColumnIsBound(pd, true);
			}
		} catch (Exception e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
		
	}

	@Override
	public WizardPanelNavResult allowFinish(String stepName, Map settings,	Wizard wizard) {
		return new GenerateBeanTask();
	}

	/**
	 * 
	 * @param wizardData
	 * @return
	 */
	private Database createFormLiveDatabase(Map wizardData) throws Exception {
		final String driverClass = (String)wizardData.get(DRIVERCLASS);
                final String connectionUrl = (String)wizardData.get(CONNECTIONURL);

		
		Platform platform = PlatformFactory.createNewPlatformInstance(driverClass, connectionUrl);
		
		java.sql.Connection connection = null;
		
		try {
			connection = DDLWizardApplication.getConnection(wizardData);

			Database result = platform.readModelFromDatabase( connection, "dbName");
			
			return result; 
		}
		finally {
			DDLWizardApplication.closeConnection(connection);
		}
		
	}
	
	/**
	 * 
	 * @param wizardData
	 * @return
	 */
	private Database getCreatedDatabase(Map wizardData) {
            Database db = (Database) wizardData.get(DATABASE_MODEL);

            return db;
	}
	

	@SuppressWarnings("unchecked")
	@Override
	protected void renderingPage() {
            super.renderingPage();

            Database db = null;
            try {
                db = (Boolean.TRUE.equals(getWizardData(CREATEDB))) ? getCreatedDatabase(getWizardDataMap()) : createFormLiveDatabase(getWizardDataMap());

                getGenerateBeanList().clear();
                for (Table table : db.getTables()) {

                    getGenerateBeanList().add(new GenerateBean(table));
                }


            } catch (Exception e) {
            	e.printStackTrace();
                JOptionPane.showMessageDialog(null, "Fatal Error on render component! \n" + e.getMessage(), "Error", JOptionPane.ERROR_MESSAGE);
                Application.getInstance().exit();
            }



	}

	@Override
	protected String validateContents(Component component, Object event) {
		if( !isGenerateBean() ) return null;
		if( DDLWizardApplication.isEmpty( getOutputDir() )) return "Select Output directory";
		
		return null;
	}

	public final List getGenerateBeanList() {
		return generateBeanList;
	}
	
	public final Class getGenerateBeanClass() {
		return GenerateBean.class;
	}

	public final String getPackageName() {
		return (String) getWizardData(PACKAGE_NAME);
	}

	public final void setPackageName(String packageName) {
		super.putWizardData(PACKAGE_NAME, packageName);
		firePropertyChange(PACKAGE_NAME, null, null);
	}

	public final boolean isGenerateBean() {
		return Boolean.TRUE.equals(getWizardData(GENERATE_BEAN));
	}

	public final void setGenerateBean(boolean generateBean) {
		putWizardData(GENERATE_BEAN, generateBean);
		firePropertyChange(GENERATE_BEAN, null, null);
	}

	public final String getOutputDir() {
		return (String) getWizardData("outputDir");
	}

	public final void setOutputDir(String outputDir) {
		putWizardData("outputDir", outputDir);
		firePropertyChange("outputDir", null, null);
	}

	@Action
	public void selectOutputDir() {
		 JFileChooser chooser = new JFileChooser();
		    chooser.setCurrentDirectory( new File(System.getProperty("user.home")));
		    chooser.setFileSelectionMode(JFileChooser.DIRECTORIES_ONLY);
		    int returnVal = chooser.showOpenDialog(this);
		    if(returnVal == JFileChooser.APPROVE_OPTION) {
		           
		    	outputDir = chooser.getSelectedFile();
		    	setOutputDir(outputDir.getPath());
				
		    }


	}

}