bundles/org.eclipse.jst.ws.jaxws.ui/src/org/eclipse/jst/ws/internal/jaxws/ui/wizards/ConfigureHandlerWizard.java

/*******************************************************************************
 * Copyright (c) 2010 Shane Clarke.
 * 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:
 *    Shane Clarke - initial API and implementation
 *******************************************************************************/
package org.eclipse.jst.ws.internal.jaxws.ui.wizards;

import java.lang.reflect.InvocationTargetException;
import java.util.ArrayList;
import java.util.List;

import javax.jws.HandlerChain;

import org.eclipse.core.resources.IContainer;
import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.resources.ResourcesPlugin;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IPath;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.Path;
import org.eclipse.core.runtime.jobs.ISchedulingRule;
import org.eclipse.core.runtime.jobs.Job;
import org.eclipse.emf.common.util.URI;
import org.eclipse.jdt.core.ICompilationUnit;
import org.eclipse.jdt.core.IType;
import org.eclipse.jdt.core.dom.AST;
import org.eclipse.jdt.core.dom.ASTParser;
import org.eclipse.jdt.core.dom.Annotation;
import org.eclipse.jdt.core.dom.CompilationUnit;
import org.eclipse.jdt.core.dom.MemberValuePair;
import org.eclipse.jface.wizard.IWizardPage;
import org.eclipse.jface.wizard.Wizard;
import org.eclipse.jst.ws.annotations.core.AnnotationsCore;
import org.eclipse.jst.ws.annotations.core.utils.AnnotationUtils;
import org.eclipse.jst.ws.internal.jaxws.core.utils.JAXWSHandlerUtils;
import org.eclipse.jst.ws.internal.jaxws.ui.JAXWSUIMessages;
import org.eclipse.jst.ws.internal.jaxws.ui.JAXWSUIPlugin;
import org.eclipse.ui.PlatformUI;
import org.eclipse.ui.actions.WorkspaceModifyOperation;

public class ConfigureHandlerWizard extends Wizard {
    private AddHandlerChainPage addHandlerChainPage;
    private OrderHandlerChainPage orderHandlerChainPage;

    private boolean fileCreated;
    private boolean addAnnotation;

    private IPath handlerChainPath;
    private IType type;

    public ConfigureHandlerWizard(IType type) {
        this.type = type;
        setWindowTitle(JAXWSUIMessages.JAXWS_CONFIGURE_HANDLER_WIZARD_TITLE);
        //TODO replace with dedicated handler wizban
        setDefaultPageImageDescriptor(JAXWSUIPlugin.getImageDescriptor("$nl$/icons/wizban/new_wiz.png"));  //$NON-NLS-1$
        handlerChainPath = getHandlerChainPath(type);
    }

    @Override
    public void addPages() {
        if (addHandlerChainPage == null) {
            addHandlerChainPage = new AddHandlerChainPage(type);
        }
        if (orderHandlerChainPage == null) {
            orderHandlerChainPage = new OrderHandlerChainPage(handlerChainPath, type.getJavaProject());
        }
        if (handlerChainPath != null && handlerChainPath.isEmpty()) {
            addPage(addHandlerChainPage);
        }
        addPage(orderHandlerChainPage);
    }

    @Override
    public IWizardPage getNextPage(IWizardPage page) {
        if (page == addHandlerChainPage && addHandlerChainPage.isPageComplete()) {
            if (addHandlerChainPage.isCreateHandlerChain()) {
                handlerChainPath = new Path(addHandlerChainPage.getNewHandlerChainPath());
                orderHandlerChainPage.setHandlerChainPath(handlerChainPath);
                addAnnotation = true;
                fileCreated = true;
                addHandlerChainPage.setFileCreated(fileCreated);
            }
            if (addHandlerChainPage.isEditHandlerChain()) {
                handlerChainPath = new Path(addHandlerChainPage.getExistingHandlerChainPath());
                orderHandlerChainPage.setHandlerChainPath(handlerChainPath);
                addAnnotation = true;
                fileCreated = false;
                addHandlerChainPage.setFileCreated(fileCreated);
            }
        }
        return super.getNextPage(page);
    }

    private IPath getHandlerChainPath(IType type) {
        if (type != null) {
            Annotation handlerChain = AnnotationUtils.getAnnotation(type, HandlerChain.class);
            if (handlerChain != null) {
                String file = AnnotationUtils.getStringValue(handlerChain, "file"); //$NON-NLS-1$
                if (file != null) {
                    if (!file.startsWith("../")) { //$NON-NLS-1$
                        return type.getPackageFragment().getPath().append(new Path(file));
                    } else {
                        return findHandlerChainPath(file);
                    }
                }
            }
        }
        return Path.EMPTY;
    }

    private IPath findHandlerChainPath(String file) {
        IContainer container = type.getResource().getParent();
        if (container != null) {
            while(file.startsWith("../")) { //$NON-NLS-1$
                file = file.substring(3);
                if (container != null) {
                    container = container.getParent();
                }
            }
            if (container != null) {
                IResource handlerChainFile = container.findMember(file);
                if (handlerChainFile != null) {
                    return handlerChainFile.getFullPath();
                }
            }
        }
        return Path.EMPTY;
    }

    @Override
    public boolean performFinish() {
        try {

            ISchedulingRule rule = null;
            Job job = Job.getJobManager().currentJob();
            if (job != null) {
                rule = job.getRule();
            } else {
                rule = ResourcesPlugin.getWorkspace().getRoot();
            }
            WorkspaceModifyOperation workspaceModifyOperation = new WorkspaceModifyOperation(rule) {

                @Override
                protected void execute(IProgressMonitor monitor) throws CoreException, InvocationTargetException, InterruptedException {
                    if (addAnnotation) {
                        IFile handlerChainFile = ResourcesPlugin.getWorkspace().getRoot().getFile(handlerChainPath);

                        URI relativeURI = URI.createPlatformResourceURI(handlerChainFile.getLocation().toOSString(),
                                false).deresolve(
                                        URI.createPlatformResourceURI(type.getResource().getLocation().toOSString(),
                                                false));

                        ICompilationUnit compilationUnit = type.getCompilationUnit();
                        ASTParser parser = ASTParser.newParser(AST.JLS3);
                        parser.setSource(compilationUnit);
                        CompilationUnit cu = (CompilationUnit)parser.createAST(null);
                        List<MemberValuePair> memberValuePairs = new ArrayList<MemberValuePair>();
                        MemberValuePair filePair = AnnotationsCore.createStringMemberValuePair(cu.getAST(),
                                "file", relativeURI.toFileString());  //$NON-NLS-1$
                        memberValuePairs.add(filePair);
                        Annotation handlerChainAnnotation = AnnotationsCore.createNormalAnnotation(cu.getAST(),
                                HandlerChain.class.getSimpleName(), memberValuePairs);
                        AnnotationUtils.addAnnotation(type, handlerChainAnnotation);
                        AnnotationUtils.addImport(type, HandlerChain.class.getCanonicalName());
                    }
                    JAXWSHandlerUtils.writeDocumentToFile(handlerChainPath, orderHandlerChainPage.getDocument());
                }
            };
            PlatformUI.getWorkbench().getProgressService().runInUI(getContainer(), workspaceModifyOperation, rule);
        } catch (InvocationTargetException ite) {
            JAXWSUIPlugin.log(ite);
            return false;
        } catch  (InterruptedException ie) {
            JAXWSUIPlugin.log(ie);
            return false;
        }
        return true;
    }

    public void deleteFile(IPath filePath) {
        IFile handlerFile = ResourcesPlugin.getWorkspace().getRoot().getFile(filePath);
        if (handlerFile.exists()) {
            try {
                handlerFile.delete(true, null);
            } catch (CoreException ce) {
                JAXWSUIPlugin.log(ce.getStatus());
            }
        }
    }
    
    @Override
    public boolean performCancel() {
        if (fileCreated) {
            IPath filePath = type.getPackageFragment().getPath().append(new Path("handler-chain.xml")); //$NON-NLS-1$
            deleteFile(filePath);
        }
        return super.performCancel();
    }

}