assistassigncompletionproposal.java

Python Development Environment (Python IDE plugin for Eclipse). Features editor, code completion, re

package org.python.pydev.editor.correctionassist.heuristics;

import org.eclipse.jface.text.IDocument;
import org.eclipse.jface.text.contentassist.ICompletionProposal;
import org.eclipse.jface.text.contentassist.IContextInformation;
import org.eclipse.jface.text.link.LinkedModeModel;
import org.eclipse.jface.text.link.LinkedModeUI;
import org.eclipse.jface.text.link.LinkedPositionGroup;
import org.eclipse.jface.text.link.ProposalPosition;
import org.eclipse.jface.text.source.ISourceViewer;
import org.eclipse.swt.graphics.Image;
import org.eclipse.swt.graphics.Point;
import org.eclipse.ui.texteditor.link.EditorLinkedModeUI;
import org.python.pydev.core.uiutils.RunInUiThread;
import org.python.pydev.editor.PyEdit;
import org.python.pydev.editor.codecompletion.PyCompletionProposal;

public class AssistAssignCompletionProposal extends PyCompletionProposal{

	private PyEdit pyEdit;

	public AssistAssignCompletionProposal(String replacementString, int replacementOffset, int replacementLength, int cursorPosition, Image image, String displayString, IContextInformation contextInformation, String additionalProposalInfo, int priority, PyEdit edit) {
		super(replacementString, replacementOffset, replacementLength, cursorPosition, image, displayString, contextInformation, additionalProposalInfo, priority);
		this.pyEdit = edit;
	}
	
	@Override
	public void apply(IDocument document) {
		try {
			//default apply
			int lineOfOffset = document.getLineOfOffset(fReplacementOffset);
			document.replace(fReplacementOffset, fReplacementLength, fReplacementString);
			int lineOffset = document.getLineOffset(lineOfOffset);
			int lineLength = document.getLineLength(lineOfOffset);
			String lineDelimiter = document.getLineDelimiter(lineOfOffset);
			int lineDelimiterLen = lineDelimiter != null? lineDelimiter.length():0;
			
			ISourceViewer viewer= pyEdit.getPySourceViewer();
			
			LinkedModeModel model= new LinkedModeModel();
			LinkedPositionGroup group= new LinkedPositionGroup();
			
			//the len-3 is because of the end of the string: " = " because the replacement string is
			//something like "xxx = " 
            ProposalPosition proposalPosition = new ProposalPosition(document, fReplacementOffset, fReplacementString.length()-3, 0 , new ICompletionProposal[0]);
        	group.addPosition(proposalPosition);
			
        	model.addGroup(group);
			model.forceInstall();
			
			final LinkedModeUI ui= new EditorLinkedModeUI(model, viewer);
			ui.setExitPosition(viewer, lineOffset+ lineLength-lineDelimiterLen, 0, Integer.MAX_VALUE);
			Runnable r = new Runnable(){
				public void run() {
					ui.enter();
				}
			};
			RunInUiThread.async(r);

		} catch (Throwable x) {
			// ignore
			x.printStackTrace();
		}
	}

	@Override
	public Point getSelection(IDocument document) {
		return null;
	}
}