Java tutorial
/******************************************************************************* * 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.core.resources.IFile; import org.eclipse.core.resources.IMarker; import org.eclipse.core.runtime.CoreException; import org.eclipse.core.runtime.NullProgressMonitor; 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.IType; import org.eclipse.jdt.core.JavaModelException; import org.eclipse.jdt.core.Signature; import org.eclipse.jface.text.IDocument; import org.eclipse.jface.text.contentassist.IContextInformation; import org.eclipse.swt.graphics.Image; import org.eclipse.swt.graphics.Point; 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.EclipseUtil; import org.jboss.tools.common.model.util.EclipseJavaUtil; import org.jboss.tools.common.quickfix.IQuickFix; /** * @author Daniel Azarov */ public class MakeMethodBusinessMarkerResolution implements IQuickFix { 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 String label; private IMethod method; private IType localInterface; private IFile file; public MakeMethodBusinessMarkerResolution(IMethod method, IType localInterface) { this.label = MessageFormat.format(CDIUIMessages.MAKE_METHOD_BUSINESS_MARKER_RESOLUTION_TITLE, new Object[] { method.getElementName(), localInterface.getElementName() }); this.method = method; this.localInterface = localInterface; try { this.file = (IFile) method.getUnderlyingResource(); } catch (JavaModelException e) { CDIUIPlugin.getDefault().logError(e); } } @Override public String getLabel() { return label; } @Override public void run(IMarker marker) { internal_run(); } private void internal_run() { try { ICompilationUnit original = EclipseUtil.getCompilationUnit(file); if (original == null) { return; } ICompilationUnit compilationUnit = original.getWorkingCopy(new NullProgressMonitor()); 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); buffer.replace(position, PRIVATE.length(), PUBLIC); } else if (Flags.isProtected(flag)) { position += text.indexOf(PROTECTED); buffer.replace(position, PROTECTED.length(), PUBLIC); } else { String type = Signature.getSignatureSimpleName(method.getReturnType()); position += text.indexOf(type); buffer.replace(position, 0, PUBLIC + SPACE); } } compilationUnit.commitWorkingCopy(false, new NullProgressMonitor()); compilationUnit.discardWorkingCopy(); // add method to interface original = localInterface.getCompilationUnit(); compilationUnit = original.getWorkingCopy(new NullProgressMonitor()); IType interfaceType = compilationUnit.getType(localInterface.getElementName()); StringBuffer content = new StringBuffer(); content.append(PUBLIC + SPACE); String simpleType = Signature.getSignatureSimpleName(method.getReturnType()); content.append(simpleType); content.append(SPACE); content.append(method.getElementName()); content.append("("); //$NON-NLS-1$ IType originalType = method.getDeclaringType(); addImport(originalType, simpleType, compilationUnit); String[] types = method.getParameterTypes(); String[] names = method.getParameterNames(); for (int i = 0; i < method.getNumberOfParameters(); i++) { if (i > 0) content.append(", "); //$NON-NLS-1$ simpleType = Signature.getSignatureSimpleName(types[i]); addImport(originalType, simpleType, compilationUnit); content.append(simpleType); content.append(SPACE); content.append(names[i]); } content.append(");"); //$NON-NLS-1$ interfaceType.createMethod(content.toString(), null, false, new NullProgressMonitor()); compilationUnit.commitWorkingCopy(false, new NullProgressMonitor()); compilationUnit.discardWorkingCopy(); } catch (CoreException ex) { CDIUIPlugin.getDefault().logError(ex); } } private void addImport(IType originalType, String simpleName, ICompilationUnit compilationUnit) throws JavaModelException { String qualifiedName = EclipseJavaUtil.resolveType(originalType, simpleName); CDIMarkerResolutionUtils.addImport(qualifiedName, compilationUnit); } @Override public String getDescription() { return label; } @Override public Image getImage() { return CDIImages.getImage(CDIImages.QUICKFIX_ADD); } @Override public int getRelevance() { return 100; } @Override public void apply(IDocument document) { internal_run(); } @Override public Point getSelection(IDocument document) { return null; } @Override public String getAdditionalProposalInfo() { return label; } @Override public String getDisplayString() { return label; } @Override public IContextInformation getContextInformation() { return null; } }