com.liferay.ide.project.ui.modules.NewLiferayComponentWizard.java Source code

Introduction

Here is the source code for com.liferay.ide.project.ui.modules.NewLiferayComponentWizard.java

Source

/*******************************************************************************
 * Copyright (c) 2000-present Liferay, Inc. All rights reserved.
 *
 * This library is free software; you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License as published by the Free
 * Software Foundation; either version 2.1 of the License, or (at your option)
 * any later version.
 *
 * This library is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
 * details.
 *
 *******************************************************************************/

package com.liferay.ide.project.ui.modules;

import com.liferay.ide.core.IBundleProject;
import com.liferay.ide.core.LiferayCore;
import com.liferay.ide.core.util.CoreUtil;
import com.liferay.ide.project.core.modules.NewLiferayComponentOp;
import com.liferay.ide.project.ui.ProjectUI;

import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.IResource;
import org.eclipse.jdt.core.IJavaElement;
import org.eclipse.jdt.core.IJavaProject;
import org.eclipse.jdt.core.IPackageFragment;
import org.eclipse.jdt.core.IPackageFragmentRoot;
import org.eclipse.jdt.core.IType;
import org.eclipse.jdt.core.JavaCore;
import org.eclipse.jface.dialogs.IMessageProvider;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.wizard.IWizardPage;
import org.eclipse.sapphire.ui.def.DefinitionLoader;
import org.eclipse.sapphire.ui.forms.swt.SapphireWizard;
import org.eclipse.sapphire.ui.forms.swt.SapphireWizardPage;
import org.eclipse.swt.widgets.Display;
import org.eclipse.ui.INewWizard;
import org.eclipse.ui.IWorkbench;
import org.eclipse.ui.IWorkbenchPage;
import org.eclipse.ui.IWorkbenchWizard;
import org.eclipse.ui.PlatformUI;
import org.eclipse.ui.ide.IDE;

/**
 * @author Simon Jiang
 */
public class NewLiferayComponentWizard extends SapphireWizard<NewLiferayComponentOp>
        implements IWorkbenchWizard, INewWizard {

    private boolean firstErrorMessageRemoved = false;
    private IProject initialProject;
    private IPackageFragment initialPackage;
    private IPackageFragmentRoot initialPackageRoot;

    public NewLiferayComponentWizard() {
        super(createDefaultOp(), DefinitionLoader.sdef(NewLiferayComponentWizard.class).wizard());
    }

    @Override
    public IWizardPage[] getPages() {
        final IWizardPage[] wizardPages = super.getPages();

        if (!firstErrorMessageRemoved && wizardPages != null) {
            final SapphireWizardPage wizardPage = (SapphireWizardPage) wizardPages[0];

            final String message = wizardPage.getMessage();
            final int messageType = wizardPage.getMessageType();

            if (messageType == IMessageProvider.ERROR && !CoreUtil.isNullOrEmpty(message)) {
                wizardPage.setMessage("Please enter a project name.", SapphireWizardPage.NONE); //$NON-NLS-1$
                firstErrorMessageRemoved = true;
            }
        }

        return wizardPages;
    }

    @Override
    public void init(IWorkbench workbench, IStructuredSelection selection) {
        if (selection != null && !selection.isEmpty()) {
            final Object element = selection.getFirstElement();

            if (element instanceof IResource) {
                initialProject = ((IResource) element).getProject();
            } else if (element instanceof IJavaProject) {
                initialProject = ((IJavaProject) element).getProject();
            } else if (element instanceof IPackageFragment) {
                initialPackage = ((IPackageFragment) element);
                initialProject = ((IJavaElement) element).getResource().getProject();
            } else if (element instanceof IPackageFragmentRoot) {
                initialPackageRoot = (IPackageFragmentRoot) element;
                initialProject = initialPackageRoot.getJavaProject().getProject();
            } else if (element instanceof IJavaElement) {
                initialProject = ((IJavaElement) element).getResource().getProject();
            }

            if (initialProject != null) {
                final IBundleProject bundleProject = LiferayCore.create(IBundleProject.class, initialProject);

                if (bundleProject != null && "jar".equals(bundleProject.getBundleShape())
                        && !bundleProject.isFragmentBundle()) {
                    element().setProjectName(initialProject.getName());

                    if (initialPackage != null) {
                        element().setPackageName(initialPackage.getElementName());
                    }
                }
            }
        }
    }

    private static NewLiferayComponentOp createDefaultOp() {
        return NewLiferayComponentOp.TYPE.instantiate();
    }

    @Override
    public void performPostFinish() {
        final NewLiferayComponentOp op = element().nearest(NewLiferayComponentOp.class);
        IProject currentProject = CoreUtil.getProject(op.getProjectName().content());

        if (currentProject == null) {
            return;
        }

        IJavaProject javaProject = JavaCore.create(currentProject);

        if (javaProject == null) {
            return;
        }

        try {
            String componentClass = op.getComponentClassName().content();
            String packageName = op.getPackageName().text();

            final IType type = javaProject.findType(packageName, componentClass);

            if (type == null) {
                return;
            }

            final IFile classFile = (IFile) type.getResource();

            if (classFile != null) {
                Display.getCurrent().asyncExec(new Runnable() {

                    public void run() {
                        try {
                            IWorkbenchPage page = PlatformUI.getWorkbench().getActiveWorkbenchWindow()
                                    .getActivePage();
                            IDE.openEditor(page, classFile, true);
                        } catch (Exception e) {
                            ProjectUI.logError(e);
                        }
                    }
                });
            }
        } catch (Exception e) {
            ProjectUI.logError(e);
        }
    }
}