org.eclipse.jdt.debug.ui/ui/org/eclipse/jdt/internal/debug/ui/CodeSnippetCompletionProcessor.java

/*******************************************************************************
 * Copyright (c) 2000, 2005 IBM Corporation and others.
 * 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:
 *     IBM Corporation - initial API and implementation
 *******************************************************************************/
package org.eclipse.jdt.internal.debug.ui;

import org.eclipse.jdt.core.IJavaProject;
import org.eclipse.jdt.core.IType;
import org.eclipse.jdt.core.JavaModelException;
import org.eclipse.jdt.internal.debug.ui.display.DisplayCompletionProcessor;
import org.eclipse.jdt.internal.ui.text.template.contentassist.TemplateEngine;
import org.eclipse.jdt.internal.ui.text.template.contentassist.TemplateProposal;
import org.eclipse.jdt.ui.text.java.IJavaCompletionProposal;
import org.eclipse.jface.text.ITextSelection;
import org.eclipse.jface.text.ITextViewer;
import org.eclipse.jface.text.contentassist.ICompletionProposal;

public class CodeSnippetCompletionProcessor extends DisplayCompletionProcessor {
	
	public interface ITypeProvider {
		IType getType();
	}

	/**
	 * The dialog with which this processor is associated.
	 */
	private ITypeProvider fTypeProvider;
		
	public CodeSnippetCompletionProcessor(ITypeProvider detailFormatDialog) {
		fTypeProvider= detailFormatDialog;
	}

	/**
	 * @see org.eclipse.jface.text.contentassist.IContentAssistProcessor#computeCompletionProposals(ITextViewer, int)
	 */
	public ICompletionProposal[] computeCompletionProposals(ITextViewer viewer, int documentOffset) {
		try {
	        setErrorMessage(null);
			IType receivingType= fTypeProvider.getType();
			if (receivingType == null) {
	            setErrorMessage(DebugUIMessages.CodeSnippetCompletionProcessor_0); //$NON-NLS-1$
				return new ICompletionProposal[0];
			}
			
			IJavaProject project= receivingType.getJavaProject();
			try {
				// Generate selections from the compilation unit
				ITextSelection textSelection= (ITextSelection)viewer.getSelectionProvider().getSelection();			
				configureResultCollector(project, textSelection);	
				receivingType.codeComplete(viewer.getDocument().get().toCharArray(), -1, documentOffset,
					 new char[0][], new char[0][],
					 new int[0], false, getCollector());
					 
				IJavaCompletionProposal[] results= getCollector().getJavaCompletionProposals();
				
				// Generate selections from the template engine
				TemplateEngine templateEngine= getTemplateEngine();
				if (templateEngine != null) {
					templateEngine.reset();
					templateEngine.complete(viewer, documentOffset, null);
					TemplateProposal[] templateResults= templateEngine.getResults();
	
					// concatenate arrays
					IJavaCompletionProposal[] total= new IJavaCompletionProposal[results.length + templateResults.length];
					System.arraycopy(templateResults, 0, total, 0, templateResults.length);
					System.arraycopy(results, 0, total, templateResults.length, results.length);
					results= total;					
				}	 
				 //Order here and not in result collector to make sure that the order
				 //applies to all proposals and not just those of the compilation unit. 
				return order(results);	
			} catch (JavaModelException x) {
				handle(viewer, x);
			}
			return null;
		} finally {
			releaseCollector();
		}
	}
}