DBConnectWizard.java

/*********************************************************************************************
 * Copyright (c) 2014-2015  Software Behaviour Analysis Lab, Concordia University, Montreal, Canada
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of Eclipse Public License v1.0 License which
 * accompanies this distribution, and is available at http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *    Syed Shariyar Murtaza -- Initial design and implementation
 **********************************************************************************************/

package org.eclipse.tracecompass.totalads.ui.models.dbconnect;

import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jface.wizard.Wizard;
import org.eclipse.tracecompass.totalads.dbms.DBMSFactory;
import org.eclipse.tracecompass.totalads.ui.models.dbconnect.AdvanceDBConfigurationPage;
import org.eclipse.tracecompass.totalads.ui.models.dbconnect.DBConfigurationPage;
import org.eclipse.tracecompass.totalads.ui.models.dbconnect.Messages;
import org.eclipse.ui.PlatformUI;

/**
 * This class creates a wizard dialog box to connect to a database
 *
 * @author <p>
 *         Syed Shariyar Murtaza justsshary@hotmail.com
 *         </p>
 *
 */
public class DBConnectWizard extends Wizard {
    private DBConfigurationPage basicConfigurationPage;
    private AdvanceDBConfigurationPage advanceDBConfigurationPage;

    /**
     * Constructor
     */
    public DBConnectWizard() {
        super();
        setNeedsProgressMonitor(true);

    }

    /*
     * (non-Javadoc)
     *
     * @see org.eclipse.jface.wizard.Wizard#getWindowTitle()
     */
    @Override
    public String getWindowTitle() {
        return Messages.DBConnectWizard_DBConnectionWizard;
    }

    /*
     * (non-Javadoc)
     *
     * @see org.eclipse.jface.wizard.Wizard#addPages()
     */
    @Override
    public void addPages() {
        basicConfigurationPage = new DBConfigurationPage();
        advanceDBConfigurationPage = new AdvanceDBConfigurationPage();
        addPage(basicConfigurationPage);
        addPage(advanceDBConfigurationPage);

    }

    /*
     * Creates a model in the database when the Finish button is clicked
     * (non-Javadoc)
     *
     * @see org.eclipse.jface.wizard.Wizard#performFinish()
     */
    @Override
    public boolean performFinish() {

        String exception = ""; //$NON-NLS-1$
        String msgTitle = "TotalADS"; //$NON-NLS-1$
        try {
            // DBMSFactory.INSTANCE.init();

            String portNum = basicConfigurationPage.getPort();
            String host = basicConfigurationPage.getHost();
            Integer port = Integer.parseInt(portNum);

            if (getContainer().getCurrentPage().equals(advanceDBConfigurationPage)) {

                String userName = advanceDBConfigurationPage.getUserName();
                String password = advanceDBConfigurationPage.getPassword();
                String database = advanceDBConfigurationPage.getDatabase();

                if (userName.isEmpty() || password.isEmpty() || database.isEmpty()) {
                    exception = Messages.DBConnectWizard_NoEmptyFields;
                } else {
                    exception = DBMSFactory.INSTANCE.openConnection(host, port, userName, password, database);
                }
            } else {
                exception = DBMSFactory.INSTANCE.openConnection(host, port);
            }

        } catch (NumberFormatException ex) {
            exception = Messages.DBConnectWizard_InvalidPor;
            getContainer().showPage(basicConfigurationPage);

        } catch (Exception ex) {
            if (ex.getMessage() != null) {
                exception = ex.getMessage();
            } else {
                exception = Messages.DBConnectWizard_NoConnection;
            }
        }

        if (!exception.isEmpty()) {
            MessageDialog.openError(PlatformUI.getWorkbench().getActiveWorkbenchWindow().getShell(),
                    msgTitle, exception);
            return false;
        }
        return true;

    }

    /*
     * (non-Javadoc)
     *
     * @see org.eclipse.jface.wizard.Wizard#canFinish()
     */
    @Override
    public boolean canFinish() {
        return true;
    }
    /*
     * @Override protected void createButtonsForButtonBar(Composite parent) {
     * super.createButtonsForButtonBar(parent); Button OKButton =
     * getButton(IDialogConstants.OK_ID); OKButton.setText("Connect"); }
     */

}