/*******************************************************************************
* 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);
}
}