org.jboss.tools.cdi.ui.marker.AddLocalBeanMarkerResolution.java Source code

Introduction

Here is the source code for org.jboss.tools.cdi.ui.marker.AddLocalBeanMarkerResolution.java

Source

/*******************************************************************************
 * Copyright (c) 2010 Red Hat, Inc.
 * Distributed under license by Red Hat, Inc. All rights reserved.
 * This program is 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:
 *     Red Hat, Inc. - initial API and implementation
 ******************************************************************************/
package org.jboss.tools.cdi.ui.marker;

import java.text.MessageFormat;

import org.eclipse.jdt.core.Flags;
import org.eclipse.jdt.core.IBuffer;
import org.eclipse.jdt.core.ICompilationUnit;
import org.eclipse.jdt.core.IMethod;
import org.eclipse.jdt.core.JavaModelException;
import org.eclipse.jdt.core.Signature;
import org.eclipse.jdt.core.refactoring.CompilationUnitChange;
import org.eclipse.swt.graphics.Image;
import org.eclipse.text.edits.InsertEdit;
import org.eclipse.text.edits.MultiTextEdit;
import org.eclipse.text.edits.ReplaceEdit;
import org.jboss.tools.cdi.core.CDIConstants;
import org.jboss.tools.cdi.core.CDIImages;
import org.jboss.tools.cdi.internal.core.refactoring.CDIMarkerResolutionUtils;
import org.jboss.tools.cdi.ui.CDIUIMessages;
import org.jboss.tools.cdi.ui.CDIUIPlugin;
import org.jboss.tools.common.refactoring.BaseMarkerResolution;
import org.jboss.tools.common.refactoring.MarkerResolutionUtils;

/**
 * @author Daniel Azarov
 */
public class AddLocalBeanMarkerResolution extends BaseMarkerResolution {
    private static final String PUBLIC = "public"; //$NON-NLS-1$
    private static final String PRIVATE = "private"; //$NON-NLS-1$
    private static final String PROTECTED = "protected"; //$NON-NLS-1$
    private static final String SPACE = " "; //$NON-NLS-1$

    private IMethod method;

    public AddLocalBeanMarkerResolution(IMethod method) {
        super(CDIMarkerResolutionUtils.getJavaMember(method).getCompilationUnit());
        this.label = MessageFormat.format(CDIUIMessages.ADD_LOCAL_BEAN_MARKER_RESOLUTION_TITLE,
                new Object[] { method.getDeclaringType().getElementName() });
        this.method = method;
        init();
    }

    @Override
    protected CompilationUnitChange getChange(ICompilationUnit compilationUnit) {
        CompilationUnitChange change = new CompilationUnitChange("", compilationUnit);

        MultiTextEdit edit = new MultiTextEdit();

        change.setEdit(edit);
        try {
            IBuffer buffer = compilationUnit.getBuffer();

            int flag = method.getFlags();

            String text = buffer.getText(method.getSourceRange().getOffset(), method.getSourceRange().getLength());

            // make method public
            int position = method.getSourceRange().getOffset();

            if (!Flags.isPublic(flag)) {
                if (Flags.isPrivate(flag)) {
                    position += text.indexOf(PRIVATE);
                    edit.addChild(new ReplaceEdit(position, PRIVATE.length(), PUBLIC));
                } else if (Flags.isProtected(flag)) {
                    position += text.indexOf(PROTECTED);
                    edit.addChild(new ReplaceEdit(position, PROTECTED.length(), PUBLIC));
                } else {
                    String type = Signature.getSignatureSimpleName(method.getReturnType());
                    position += text.indexOf(type);
                    edit.addChild(new InsertEdit(position, PUBLIC + SPACE));
                }
            }

            // add @LocalBean annotation
            MarkerResolutionUtils.addAnnotation(CDIConstants.LOCAL_BEAN_ANNOTATION_TYPE_NAME, compilationUnit,
                    method.getDeclaringType(), "", edit);
        } catch (JavaModelException e) {
            CDIUIPlugin.getDefault().logError(e);
        }
        return change;
    }

    @Override
    public Image getImage() {
        return CDIImages.getImage(CDIImages.QUICKFIX_ADD);
    }
}