com.intellij.ide.impl.SelectInEditorManagerImpl.java Source code

Introduction

Here is the source code for com.intellij.ide.impl.SelectInEditorManagerImpl.java

Source

/*
 * Copyright 2000-2014 JetBrains s.r.o.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.intellij.ide.impl;

import com.intellij.ide.SelectInEditorManager;
import com.intellij.injected.editor.DocumentWindow;
import com.intellij.injected.editor.VirtualFileWindow;
import com.intellij.openapi.Disposable;
import com.intellij.openapi.editor.Editor;
import com.intellij.openapi.editor.colors.EditorColors;
import com.intellij.openapi.editor.colors.EditorColorsManager;
import com.intellij.openapi.editor.event.CaretEvent;
import com.intellij.openapi.editor.event.CaretListener;
import com.intellij.openapi.editor.ex.DocumentEx;
import com.intellij.openapi.editor.markup.HighlighterLayer;
import com.intellij.openapi.editor.markup.HighlighterTargetArea;
import com.intellij.openapi.editor.markup.RangeHighlighter;
import com.intellij.openapi.editor.markup.TextAttributes;
import com.intellij.openapi.fileEditor.FileEditorManager;
import com.intellij.openapi.fileEditor.OpenFileDescriptor;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.util.ProperTextRange;
import com.intellij.openapi.util.TextRange;
import com.intellij.openapi.vfs.VirtualFile;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import javax.swing.*;
import java.awt.event.FocusEvent;
import java.awt.event.FocusListener;

/**
 * @author MYakovlev
 * Date: Jul 1, 2002
 */
public class SelectInEditorManagerImpl extends SelectInEditorManager
        implements Disposable, FocusListener, CaretListener {
    private final Project myProject;
    private RangeHighlighter mySegmentHighlighter;
    private Editor myEditor;

    public SelectInEditorManagerImpl(Project project) {
        myProject = project;
    }

    @Override
    public void dispose() {
        releaseAll();
    }

    @Override
    public void selectInEditor(VirtualFile file, final int startOffset, final int endOffset,
            final boolean toSelectLine, final boolean toUseNormalSelection) {
        releaseAll();
        final TextRange textRange;
        if (file instanceof VirtualFileWindow) {
            DocumentWindow documentWindow = ((VirtualFileWindow) file).getDocumentWindow();
            textRange = documentWindow.injectedToHost(new TextRange(startOffset, endOffset));
            file = ((VirtualFileWindow) file).getDelegate();
        } else {
            textRange = new ProperTextRange(startOffset, endOffset);
        }
        openEditor(file, endOffset);
        final Editor editor = openEditor(file, textRange.getStartOffset());

        SwingUtilities.invokeLater(new Runnable() { // later to let focus listener chance to handle events
            @Override
            public void run() {
                if (editor != null && !editor.isDisposed()) {
                    doSelect(toUseNormalSelection, editor, toSelectLine, textRange);
                }
            }
        });
    }

    private void doSelect(final boolean toUseNormalSelection, @NotNull final Editor editor,
            final boolean toSelectLine, final TextRange textRange) {
        int startOffset = textRange.getStartOffset();
        int endOffset = textRange.getEndOffset();
        if (toUseNormalSelection) {
            DocumentEx doc = (DocumentEx) editor.getDocument();
            if (toSelectLine) {
                int lineNumber = doc.getLineNumber(startOffset);
                if (lineNumber >= 0 && lineNumber < doc.getLineCount()) {
                    editor.getSelectionModel().setSelection(doc.getLineStartOffset(lineNumber),
                            doc.getLineEndOffset(lineNumber) + doc.getLineSeparatorLength(lineNumber));
                }
            } else {
                editor.getSelectionModel().setSelection(startOffset, endOffset);
            }
            return;
        }

        TextAttributes selectionAttributes = EditorColorsManager.getInstance().getGlobalScheme()
                .getAttributes(EditorColors.SEARCH_RESULT_ATTRIBUTES);

        releaseAll();

        if (toSelectLine) {
            DocumentEx doc = (DocumentEx) editor.getDocument();
            int lineNumber = doc.getLineNumber(startOffset);
            if (lineNumber >= 0 && lineNumber < doc.getLineCount()) {
                mySegmentHighlighter = editor.getMarkupModel().addRangeHighlighter(
                        doc.getLineStartOffset(lineNumber),
                        doc.getLineEndOffset(lineNumber) + doc.getLineSeparatorLength(lineNumber),
                        HighlighterLayer.LAST + 1, selectionAttributes, HighlighterTargetArea.EXACT_RANGE);
            }
        } else {
            mySegmentHighlighter = editor.getMarkupModel().addRangeHighlighter(startOffset, endOffset,
                    HighlighterLayer.LAST + 1, selectionAttributes, HighlighterTargetArea.EXACT_RANGE);
        }
        myEditor = editor;
        myEditor.getContentComponent().addFocusListener(this);
        myEditor.getCaretModel().addCaretListener(this);
    }

    @Override
    public void focusGained(FocusEvent e) {
        releaseAll();
    }

    @Override
    public void focusLost(FocusEvent e) {
    }

    @Override
    public void caretPositionChanged(CaretEvent e) {
        releaseAll();
    }

    @Override
    public void caretAdded(CaretEvent e) {
    }

    @Override
    public void caretRemoved(CaretEvent e) {
    }

    private void releaseAll() {
        if (mySegmentHighlighter != null && myEditor != null) {
            mySegmentHighlighter.dispose();
            myEditor.getContentComponent().removeFocusListener(this);
            myEditor.getCaretModel().removeCaretListener(this);
            mySegmentHighlighter = null;
            myEditor = null;
        }
    }

    @Nullable
    private Editor openEditor(VirtualFile file, int textOffset) {
        if (file == null || !file.isValid()) {
            return null;
        }
        OpenFileDescriptor descriptor = new OpenFileDescriptor(myProject, file, textOffset);
        return FileEditorManager.getInstance(myProject).openTextEditor(descriptor, false);
    }
}