org.cubictest.exporters.selenium.ui.CustomStepWizard.java Source code

Introduction

Here is the source code for org.cubictest.exporters.selenium.ui.CustomStepWizard.java

Source

/*******************************************************************************
 * Copyright (c) 2005, 2010 Stein K. Skytteren and Christian Schwarz
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 * 
 * Contributors:
 *    Stein K. Skytteren and Christian Schwarz - initial API and implementation
 *******************************************************************************/
package org.cubictest.exporters.selenium.ui;

import org.cubictest.common.utils.ErrorHandler;
import org.cubictest.common.utils.Logger;
import org.cubictest.exporters.selenium.SeleniumBuildPathSupporter;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.NullProgressMonitor;
import org.eclipse.jdt.core.IJavaElement;
import org.eclipse.jdt.core.IJavaProject;
import org.eclipse.jdt.core.JavaCore;
import org.eclipse.jdt.core.JavaModelException;
import org.eclipse.jdt.ui.wizards.NewClassWizardPage;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jface.wizard.Wizard;
import org.eclipse.swt.widgets.Shell;

public class CustomStepWizard extends Wizard {

    private static final String TEST_SUPERCLASS_NAME = "org.cubictest.selenium.custom.ICustomTestStep";
    private NewClassWizardPage classWizard;
    private String name;
    private IProject project;

    public CustomStepWizard() {
        setWindowTitle("Create new CustomTestStep");
    }

    @Override
    public void addPages() {
        classWizard = new NewClassWizardPage();
        IJavaProject javaProject = JavaCore.create(project);
        IResource pageFragmentRoot = null;
        try {
            for (IJavaElement element : javaProject.getChildren())
                if (element.getElementType() == IJavaElement.PACKAGE_FRAGMENT_ROOT) {
                    pageFragmentRoot = element.getResource();
                    break;
                }
        } catch (JavaModelException e) {
            Logger.error(e.getMessage(), e);
        }
        if (pageFragmentRoot != null)
            classWizard.setPackageFragmentRoot(javaProject.getPackageFragmentRoot(pageFragmentRoot), true);
        classWizard.addSuperInterface("org.cubictest.selenium.custom.ICustomTestStep");
        classWizard.setMethodStubSelection(false, false, true, true);
        addPage(classWizard);
    }

    @Override
    public boolean performFinish() {
        name = classWizard.getTypeName();
        if (name != null && name.length() > 0) {
            try {
                IJavaProject javaProject = JavaCore.create(project);
                if (javaProject.findType(TEST_SUPERCLASS_NAME) == null) {
                    addLibToClasspath(javaProject, getShell());
                }
                IProgressMonitor monitor = new NullProgressMonitor();
                classWizard.createType(monitor);
                return true;
            } catch (CoreException e) {
                ErrorHandler.logAndShowErrorDialog(e);
            } catch (InterruptedException e) {
                ErrorHandler.logAndShowErrorDialog(e);
            }
        }
        return false;
    }

    public static boolean addLibToClasspath(IJavaProject javaProject, Shell shell) {
        if (MessageDialog.openQuestion(new Shell(), "Add CubicTest Selenium to classpath",
                "You need to have CubicTest and Selenium libraries on the classpath.\n\nWould you like to add them?")) {
            return SeleniumBuildPathSupporter.putCubicSeleniumLibraryOnClasspath(javaProject, shell);
        }
        return false;
    }

    public String getClassName() {
        String packageText = classWizard.getPackageText();
        String typeTypeName = classWizard.getTypeName();
        return packageText + (packageText.length() > 0 ? "." : "") + typeTypeName;
    }

    public String getPath() {
        return classWizard.getModifiedResource().getFullPath().toPortableString();
    }

    public void setProject(IProject project) {
        this.project = project;
    }

}