Added OnesideTextDiffViewer case

JetBrains-Research · Feb 13, 2023 · 355d25e · 355d25e
1 parent 9922adc
commit 355d25e
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/plugin/src/main/java/org/jetbrains/research/refactorinsight/ui/windows/DiffWindow.java b/plugin/src/main/java/org/jetbrains/research/refactorinsight/ui/windows/DiffWindow.java
@@ -9,6 +9,7 @@
 import com.intellij.diff.tools.simple.*;
 import com.intellij.diff.tools.util.base.DiffViewerListener;
 import com.intellij.diff.tools.util.base.IgnorePolicy;
+import com.intellij.diff.tools.util.side.OnesideTextDiffViewer;
 import com.intellij.diff.tools.util.side.TwosideTextDiffViewer;
 import com.intellij.diff.util.DiffUserDataKeysEx;
 import com.intellij.ide.highlighter.JavaClassFileType;
@@ -285,6 +286,13 @@ private void putCommitIdData(FrameDiffTool.DiffViewer viewer, @NotNull DiffReque
                 twosideTextDiffViewer.getEditor1().getVirtualFile()
                         .putUserData(Keys.CHILD_COMMIT_ID, change.getAfterRevision().getRevisionNumber().asString());
             }
+        } else if (viewer instanceof OnesideTextDiffViewer) {
+            Change change = request.getUserData(ChangeDiffRequestProducer.CHANGE_KEY);
+            if (change != null && change.getType() == Change.Type.NEW &&  change.getAfterRevision() != null) {
+                ((OnesideTextDiffViewer) viewer).getEditor().getVirtualFile().putUserData(Keys.COMMIT_ID, change.getAfterRevision().getRevisionNumber().asString());
+            } else if (change != null && change.getType() == Change.Type.DELETED &&  change.getBeforeRevision() != null) {
+                ((OnesideTextDiffViewer) viewer).getEditor().getVirtualFile().putUserData(Keys.COMMIT_ID, change.getBeforeRevision().getRevisionNumber().asString());
+            }
         }
     }