PropertyBeanReferenceContentAssistCalculator.java

/*******************************************************************************
 * Copyright (c) 2008, 2016 Spring IDE Developers
 * 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:
 *     Spring IDE Developers - initial API and implementation
 *******************************************************************************/
package org.springframework.ide.eclipse.beans.ui.editor.contentassist;

import java.util.Arrays;
import java.util.Collections;
import java.util.List;

import org.eclipse.core.resources.IFile;
import org.eclipse.jdt.core.IMethod;
import org.eclipse.jdt.core.IType;
import org.eclipse.jdt.core.JavaModelException;
import org.springframework.ide.eclipse.beans.ui.editor.util.BeansCompletionUtils;
import org.springframework.ide.eclipse.beans.ui.editor.util.BeansEditorUtils;
import org.springframework.ide.eclipse.core.java.Introspector;
import org.springframework.ide.eclipse.core.java.JdtUtils;

/**
 * {@link IContentAssistCalculator} that extends {@link BeanReferenceContentAssistCalculator} in
 * order to give bean reference proposals that have a matching types a higher proposal priority.
 * 
 * @author Christian Dupuis
 * @since 2.0.4
 */
public class PropertyBeanReferenceContentAssistCalculator extends
		BeanReferenceContentAssistCalculator {

	@Override
	public void computeProposals(IContentAssistContext context,
			IContentAssistProposalRecorder recorder) {
		BeansCompletionUtils.addBeanReferenceProposals(context, recorder, showExternal,
				calculateType(context, recorder));
	}

	protected List<String> calculateType(IContentAssistContext context,
			IContentAssistProposalRecorder recorder) {
		if (context.getParentNode() != null
				&& "bean".equals(context.getParentNode().getLocalName())) {
			IFile file = context.getFile();
			String className = BeansEditorUtils.getClassNameForBean(file, context
					.getDocument(), context.getParentNode());
			if (file != null && file.exists()) {
				IType type = JdtUtils.getJavaType(context.getFile().getProject(), className);
				String propertyName = BeansEditorUtils.getAttribute(context.getNode(), "name");
				if (propertyName != null) {
					try {
						IMethod method = Introspector.getWritableProperty(type, propertyName);
						if (method != null) {
							// It is safe to assume that we have one parameter since
							// this is a property setter
							String parameterTypeName = JdtUtils.resolveClassNameBySignature(method
									.getParameterTypes()[0], type);
							if (parameterTypeName != null) {
								return Arrays.asList(new String[] { parameterTypeName });
							}
						}
					}
					catch (JavaModelException e) {
					}
				}
			}
		}
		return Collections.emptyList();
	}

}