/*******************************************************************************
* Copyright (c) 2011 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 org.eclipse.jdt.core.IAnnotation;
import org.eclipse.jdt.core.ICompilationUnit;
import org.eclipse.jdt.core.JavaModelException;
import org.eclipse.jdt.core.refactoring.CompilationUnitChange;
import org.eclipse.osgi.util.NLS;
import org.eclipse.swt.graphics.Image;
import org.eclipse.text.edits.MultiTextEdit;
import org.eclipse.text.edits.ReplaceEdit;
import org.eclipse.text.edits.TextEdit;
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;
public class ChangeAnnotationMarkerResolution extends BaseMarkerResolution {
private IAnnotation annotation;
protected String sourceString = "";
protected String changeString = CDIMarkerResolutionUtils.AT;
private boolean useBraces = true;
private String[] qualifiedNames = new String[0];
public ChangeAnnotationMarkerResolution(IAnnotation annotation){
super(CDIMarkerResolutionUtils.getJavaMember(annotation).getCompilationUnit());
this.annotation = annotation;
changeString += annotation.getElementName();
try {
sourceString = annotation.getSource();
} catch (JavaModelException e) {
CDIUIPlugin.getDefault().logError(e);
}
label = NLS.bind(CDIUIMessages.CHANGE_ANNOTATION_MARKER_RESOLUTION_TITLE, sourceString, changeString);
init();
}
public ChangeAnnotationMarkerResolution(IAnnotation annotation, String parameter){
this(annotation);
qualifiedNames = new String[]{parameter};
String shortName = CDIMarkerResolutionUtils.getShortName(parameter);
changeString += "("+shortName+")";
label = NLS.bind(CDIUIMessages.CHANGE_ANNOTATION_MARKER_RESOLUTION_TITLE, sourceString, changeString);
init();
}
public ChangeAnnotationMarkerResolution(IAnnotation annotation, String[] typeNames){
this(annotation);
this.qualifiedNames = typeNames;
String[] shortNames = CDIMarkerResolutionUtils.getShortNames(qualifiedNames);
String totalList = CDIMarkerResolutionUtils.getTotalList(shortNames);
if(useBraces)
totalList = CDIMarkerResolutionUtils.OPEN_BRACE+totalList+CDIMarkerResolutionUtils.CLOSE_BRACE;
changeString += "("+totalList+")";
label = NLS.bind(CDIUIMessages.CHANGE_ANNOTATION_MARKER_RESOLUTION_TITLE, sourceString, changeString);
init();
}
public ChangeAnnotationMarkerResolution(IAnnotation annotation, String[] typeNames, boolean useBraces){
this(annotation, typeNames);
this.useBraces = useBraces;
label = NLS.bind(CDIUIMessages.CHANGE_ANNOTATION_MARKER_RESOLUTION_TITLE, sourceString, changeString);
init();
}
@Override
protected CompilationUnitChange getChange(ICompilationUnit compilationUnit){
CompilationUnitChange change = new CompilationUnitChange("", compilationUnit);
MultiTextEdit edit = new MultiTextEdit();
change.setEdit(edit);
try{
for(String qualifiedName : qualifiedNames){
CDIMarkerResolutionUtils.addImport(qualifiedName, compilationUnit, true, edit);
}
IAnnotation workingCopyAnnotation = CDIMarkerResolutionUtils.findWorkingCopy(compilationUnit, annotation);
TextEdit re = new ReplaceEdit(workingCopyAnnotation.getSourceRange().getOffset(), workingCopyAnnotation.getSourceRange().getLength(), changeString);
edit.addChild(re);
} catch (JavaModelException e) {
CDIUIPlugin.getDefault().logError(e);
}
return change;
}
@Override
public Image getImage() {
return CDIImages.getImage(CDIImages.QUICKFIX_CHANGE);
}
}