Examples with AnnotationSourceSwitcher com.intellij.openapi.vcs.annotate.AnnotationSourceSwitcher used on opensource projects

Example 1 with AnnotationSourceSwitcher

use of com.intellij.openapi.vcs.annotate.AnnotationSourceSwitcher in project intellij-community by JetBrains.

the class AnnotateToggleAction method doAnnotate.

private static void doAnnotate(@NotNull final Editor editor, @NotNull final Project project, @Nullable final VirtualFile currentFile, @NotNull final FileAnnotation fileAnnotation, @NotNull final AbstractVcs vcs, @NotNull final UpToDateLineNumberProvider upToDateLineNumbers, final boolean warnAboutSuspiciousAnnotations) {
    if (warnAboutSuspiciousAnnotations) {
        int expectedLines = Math.max(upToDateLineNumbers.getLineCount(), 1);
        int actualLines = Math.max(fileAnnotation.getLineCount(), 1);
        if (Math.abs(expectedLines - actualLines) > 1) {
            // 1 - for different conventions about files ending with line separator
            editor.setHeaderComponent(new MyEditorNotificationPanel(editor, vcs, () -> {
                doAnnotate(editor, project, currentFile, fileAnnotation, vcs, upToDateLineNumbers, false);
            }));
            return;
        }
    }
    Disposable disposable = new Disposable() {

        @Override
        public void dispose() {
            fileAnnotation.dispose();
        }
    };
    if (fileAnnotation.getFile() != null && fileAnnotation.getFile().isInLocalFileSystem()) {
        VcsAnnotationLocalChangesListener changesListener = ProjectLevelVcsManager.getInstance(project).getAnnotationLocalChangesListener();
        changesListener.registerAnnotation(fileAnnotation.getFile(), fileAnnotation);
        Disposer.register(disposable, new Disposable() {

            @Override
            public void dispose() {
                changesListener.unregisterAnnotation(fileAnnotation.getFile(), fileAnnotation);
            }
        });
    }
    editor.getGutter().closeAllAnnotations();
    fileAnnotation.setCloser(() -> {
        UIUtil.invokeLaterIfNeeded(() -> {
            if (project.isDisposed())
                return;
            editor.getGutter().closeAllAnnotations();
        });
    });
    fileAnnotation.setReloader(newFileAnnotation -> {
        if (editor.getGutter().isAnnotationsShown()) {
            assert Comparing.equal(fileAnnotation.getFile(), newFileAnnotation.getFile());
            doAnnotate(editor, project, currentFile, newFileAnnotation, vcs, upToDateLineNumbers, false);
        }
    });
    final EditorGutterComponentEx editorGutter = (EditorGutterComponentEx) editor.getGutter();
    final List<AnnotationFieldGutter> gutters = new ArrayList<>();
    final AnnotationSourceSwitcher switcher = fileAnnotation.getAnnotationSourceSwitcher();
    final AnnotationPresentation presentation = new AnnotationPresentation(fileAnnotation, upToDateLineNumbers, switcher, disposable);
    if (currentFile != null && vcs.getCommittedChangesProvider() != null) {
        presentation.addAction(new ShowDiffFromAnnotation(fileAnnotation, vcs, currentFile));
    }
    presentation.addAction(new CopyRevisionNumberFromAnnotateAction(fileAnnotation));
    presentation.addAction(Separator.getInstance());
    final Couple<Map<VcsRevisionNumber, Color>> bgColorMap = computeBgColors(fileAnnotation, editor);
    final Map<VcsRevisionNumber, Integer> historyIds = computeLineNumbers(fileAnnotation);
    if (switcher != null) {
        switcher.switchTo(switcher.getDefaultSource());
        final LineAnnotationAspect revisionAspect = switcher.getRevisionAspect();
        final CurrentRevisionAnnotationFieldGutter currentRevisionGutter = new CurrentRevisionAnnotationFieldGutter(fileAnnotation, revisionAspect, presentation, bgColorMap);
        final MergeSourceAvailableMarkerGutter mergeSourceGutter = new MergeSourceAvailableMarkerGutter(fileAnnotation, presentation, bgColorMap);
        SwitchAnnotationSourceAction switchAction = new SwitchAnnotationSourceAction(switcher, editorGutter);
        presentation.addAction(switchAction);
        switchAction.addSourceSwitchListener(currentRevisionGutter);
        switchAction.addSourceSwitchListener(mergeSourceGutter);
        currentRevisionGutter.consume(switcher.getDefaultSource());
        mergeSourceGutter.consume(switcher.getDefaultSource());
        gutters.add(currentRevisionGutter);
        gutters.add(mergeSourceGutter);
    }
    final LineAnnotationAspect[] aspects = fileAnnotation.getAspects();
    for (LineAnnotationAspect aspect : aspects) {
        gutters.add(new AspectAnnotationFieldGutter(fileAnnotation, aspect, presentation, bgColorMap));
    }
    if (historyIds != null) {
        gutters.add(new HistoryIdColumn(fileAnnotation, presentation, bgColorMap, historyIds));
    }
    gutters.add(new HighlightedAdditionalColumn(fileAnnotation, presentation, bgColorMap));
    final AnnotateActionGroup actionGroup = new AnnotateActionGroup(gutters, editorGutter, bgColorMap);
    presentation.addAction(actionGroup, 1);
    gutters.add(new ExtraFieldGutter(fileAnnotation, presentation, bgColorMap, actionGroup));
    presentation.addAction(new AnnotateCurrentRevisionAction(fileAnnotation, vcs));
    presentation.addAction(new AnnotatePreviousRevisionAction(fileAnnotation, vcs));
    addActionsFromExtensions(presentation, fileAnnotation);
    for (AnnotationFieldGutter gutter : gutters) {
        final AnnotationGutterLineConvertorProxy proxy = new AnnotationGutterLineConvertorProxy(upToDateLineNumbers, gutter);
        if (gutter.isGutterAction()) {
            editor.getGutter().registerTextAnnotation(proxy, proxy);
        } else {
            editor.getGutter().registerTextAnnotation(proxy);
        }
    }
}