EditorTools.java

package com.proalpha.java.oea.plugins.querygenerator;
 
import java.util.Scanner;
 
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jface.text.BadLocationException;
import org.eclipse.jface.text.IDocument;
import org.eclipse.jface.text.ITextOperationTarget;
import org.eclipse.jface.text.ITextSelection;
import org.eclipse.jface.text.ITextViewer;
import org.eclipse.jface.text.link.LinkedModeModel;
import org.eclipse.jface.text.link.LinkedModeUI;
import org.eclipse.jface.text.link.LinkedPosition;
import org.eclipse.jface.text.link.LinkedPositionGroup;
import org.eclipse.jface.viewers.ISelection;
import org.eclipse.jface.viewers.ISelectionProvider;
import org.eclipse.ui.IEditorPart;
import org.eclipse.ui.IWorkbenchPage;
import org.eclipse.ui.IWorkbenchWindow;
import org.eclipse.ui.PlatformUI;
import org.eclipse.ui.texteditor.IDocumentProvider;
import org.eclipse.ui.texteditor.ITextEditor;
 
public class EditorTools {
    
    private static EditorTools editorTools = null;
    
    private EditorTools() {
        super();
    }
    
    public static EditorTools getInstance() {
        
        if (editorTools == null)
            editorTools = new EditorTools();
        
        return editorTools;
        
    }
    
    public void insertText(String insertString) {
        
        this.insertText(getActiveEditor(), insertString);
        
    }
    
    public String getSelectionText() {
        
        return getSelectionText(getActiveEditor());
        
    }
    
    public String getSelectionText(ITextEditor editor) {
        
        ISelectionProvider selectionProvider = editor.getSelectionProvider();
        ISelection selection = selectionProvider.getSelection();
        
        if (selection instanceof ITextSelection) {
            
            ITextSelection textSelection = (ITextSelection) selection;
            return textSelection.getText();
            
        }
        
        return "";
 
    }
    
    public void insertText(ITextEditor editor, String insertString) {
        
        if (editor != null) {
            
            IDocumentProvider documentProvider = editor.getDocumentProvider();
            IDocument document = documentProvider.getDocument(editor.getEditorInput());
            
            ISelectionProvider selectionProvider = editor.getSelectionProvider();
            ISelection selection = selectionProvider.getSelection();
            
            if (selection instanceof ITextSelection) {
                
                try {
                    
                    // Get selected text
                    ITextSelection textSelection = (ITextSelection) selection;
 
                    int line = textSelection.getStartLine();
                    int selectionOffset = textSelection.getOffset();
 
                    int indentation = selectionOffset - document.getLineOffset(line);
                    insertString = indent(insertString, indentation, 1);
 
                    // Parse for variables in text
                    VariableParser variableParser = new VariableParser(insertString);
                    variableParser.parse();
 
                    // Insert text at current cursor position.
                    insertString = variableParser.getResultText();
                    document.replace(selectionOffset, textSelection.getLength(), insertString);
 
                    // Add linked positions for parsed variables
                    Variable[] variables = variableParser.getVariables();
 
                    if (variables.length > 0) {
 
                        ITextViewer viewer = (ITextViewer) editor.getAdapter(ITextOperationTarget.class);
                        LinkedModeModel model= new LinkedModeModel();
 
                        int posCount = 0;
                        int exitPosition = selectionOffset + insertString.length();
 
                        for (int i = 0; i < variables.length; i++) {
 
                            switch (variables[i].getVariableType()) {
 
                            case CURSOR_POS:
                                exitPosition = selectionOffset + variables[i].getOffset();
                                break;
 
                            default:
 
                                LinkedPositionGroup group = new LinkedPositionGroup();
 
                                group.addPosition(new LinkedPosition(document, selectionOffset + variables[i].getOffset(), variables[i].getLength()));
                                model.addGroup(group);
 
                                posCount++;
 
                            }
 
                        }
 
                        if (posCount == 0)
                            editor.selectAndReveal(exitPosition, 0);
 
                        else {
 
                            model.forceInstall();
 
                            LinkedModeUI ui = new LinkedModeUI(model, viewer);
                            ui.setExitPosition(viewer, exitPosition, 0, Integer.MAX_VALUE);
                            ui.enter();
 
                        }
 
                    }
 
                } catch (BadLocationException e) {
                    IWorkbenchWindow window = PlatformUI.getWorkbench().getActiveWorkbenchWindow();
                    MessageDialog.openInformation(window.getShell(), "QueryGenerator", e.getMessage());
                }
                
                /*
                OETextEditor oeEditor = (OETextEditor) editorPart.getAdapter(OETextEditor.class);
                
                if (oeEditor != null) {
                    
                    ContextTypeRegistry registry = OETemplateSettings.getContextTypeRegistry();
                    TemplateContextType type = registry.getContextType(OETemplateContextType.OE_CONTEXT_TYPE);
                    
                    Template template = new Template(
                            "QueryTemplate",
                            "QueryTemplate",
                            type.getId(),
                            insertString,
                            true);
                    
                    OETemplateContext context = new OETemplateContext(type, document, selectionOffset, 0);
                    
                    Region region = new Region(selectionOffset, 0);
                    OETemplateProposal proposal = new OETemplateProposal(template, context, region, null);
                    
                    ITextViewer viewer = (ITextViewer) editor.getAdapter(ITextOperationTarget.class);
                    proposal.apply(viewer, ' ', 0, selectionOffset);
                    
                }
                */
                
                /*
                ITextViewer viewer = (ITextViewer) editor.getAdapter(ITextOperationTarget.class);
                
                LinkedModeModel model= new LinkedModeModel();
                
                LinkedPositionGroup group= new LinkedPositionGroup();
                group.addPosition(new LinkedPosition(document, selectionOffset + 5, 5));
                model.addGroup(group);
                
                model.forceInstall();
                
                LinkedModeUI ui= new LinkedModeUI(model, viewer);
                ui.setExitPosition(viewer, selectionOffset + insertString.length(), 0, Integer.MAX_VALUE);
                ui.enter();
                */
                
                
                // Place cursor at end of inserted text.
//              editor.selectAndReveal(selectionOffset, insertString.length());
//              editor.setHighlightRange(selectionOffset, insertString.length(), true);
                
            }
            
        }
        
    }
    
    private ITextEditor getActiveEditor() {
        
        IWorkbenchWindow window = PlatformUI.getWorkbench().getActiveWorkbenchWindow();
        IWorkbenchPage page = window.getActivePage();
        IEditorPart editorPart = page.getActiveEditor();
        
        return (ITextEditor) editorPart.getAdapter(ITextEditor.class);
 
    }
    
    private String indent(String text, int spaceCount, int skipLines) {
        
        String lineSeparator = System.getProperty("line.separator");
        
        StringBuilder stringBuilder = new StringBuilder();
        Scanner scanner = new Scanner(text);
        
        for (int lineNo = 1; scanner.hasNextLine(); lineNo++) {
            
            if (lineNo > 1)
                stringBuilder.append(lineSeparator);
            
            if (lineNo > skipLines)
                for (int i = 0; i < spaceCount; i++)
                    stringBuilder.append(" ");
            
            stringBuilder.append(scanner.nextLine());
            
        }
        
        scanner.close();
        
        return stringBuilder.toString();
        
    }
    
 
}