Merge branch 'db/javac-ast'
[idea/community.git] / platform / vcs-impl / src / com / intellij / openapi / vcs / actions / AnnotationPresentation.java
1 /*
2  * Copyright 2000-2010 JetBrains s.r.o.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package com.intellij.openapi.vcs.actions;
17
18 import com.intellij.openapi.actionSystem.AnAction;
19 import com.intellij.openapi.editor.colors.ColorKey;
20 import com.intellij.openapi.editor.colors.EditorFontType;
21 import com.intellij.openapi.localVcs.UpToDateLineNumberProvider;
22 import com.intellij.openapi.vcs.annotate.*;
23 import com.intellij.openapi.vcs.history.VcsRevisionNumber;
24 import com.intellij.util.Consumer;
25 import com.intellij.util.ObjectUtils;
26 import org.jetbrains.annotations.NotNull;
27 import org.jetbrains.annotations.Nullable;
28
29 import java.util.ArrayList;
30 import java.util.List;
31
32 class AnnotationPresentation implements TextAnnotationPresentation {
33   @NotNull private final FileAnnotation myFileAnnotation;
34   @NotNull private final UpToDateLineNumberProvider myUpToDateLineNumberProvider;
35   @Nullable private final AnnotationSourceSwitcher mySwitcher;
36   private final ArrayList<AnAction> myActions = new ArrayList<>();
37
38   AnnotationPresentation(@NotNull FileAnnotation fileAnnotation,
39                          @NotNull UpToDateLineNumberProvider upToDateLineNumberProvider,
40                          @Nullable final AnnotationSourceSwitcher switcher) {
41     myUpToDateLineNumberProvider = upToDateLineNumberProvider;
42     myFileAnnotation = fileAnnotation;
43     mySwitcher = switcher;
44   }
45
46   public EditorFontType getFontType(final int line) {
47     VcsRevisionNumber revision = myFileAnnotation.originalRevision(line);
48     VcsRevisionNumber currentRevision = myFileAnnotation.getCurrentRevision();
49     return currentRevision != null && currentRevision.equals(revision) ? EditorFontType.BOLD : EditorFontType.PLAIN;
50   }
51
52   public ColorKey getColor(final int line) {
53     if (mySwitcher == null) return AnnotationSource.LOCAL.getColor();
54     return mySwitcher.getAnnotationSource(line).getColor();
55   }
56
57   public List<AnAction> getActions(int line) {
58     int correctedNumber = myUpToDateLineNumberProvider.getLineNumber(line);
59     for (AnAction action : myActions) {
60       UpToDateLineNumberListener upToDateListener = ObjectUtils.tryCast(action, UpToDateLineNumberListener.class);
61       if (upToDateListener != null) upToDateListener.consume(correctedNumber);
62
63       LineNumberListener listener = ObjectUtils.tryCast(action, LineNumberListener.class);
64       if (listener != null) listener.consume(line);
65     }
66
67     return myActions;
68   }
69
70   @NotNull
71   public List<AnAction> getActions() {
72     return myActions;
73   }
74
75   public void addAction(AnAction action) {
76     myActions.add(action);
77   }
78
79   public void addAction(AnAction action, int index) {
80     myActions.add(index, action);
81   }
82 }