/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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.imaginea.kodebeagle.base.util;
import com.imaginea.kodebeagle.base.object.WindowObjects;
import com.intellij.codeInsight.highlighting.HighlightUsagesHandler;
import com.intellij.openapi.editor.CaretModel;
import com.intellij.openapi.editor.Document;
import com.intellij.openapi.editor.Editor;
import com.intellij.openapi.editor.EditorFactory;
import com.intellij.openapi.editor.LogicalPosition;
import com.intellij.openapi.editor.ScrollType;
import com.intellij.openapi.editor.ScrollingModel;
import com.intellij.openapi.editor.SelectionModel;
import com.intellij.openapi.editor.markup.EffectType;
import com.intellij.openapi.editor.markup.HighlighterLayer;
import com.intellij.openapi.editor.markup.HighlighterTargetArea;
import com.intellij.openapi.editor.markup.MarkupModel;
import com.intellij.openapi.editor.markup.TextAttributes;
import com.intellij.openapi.fileEditor.FileEditorManager;
import com.intellij.openapi.util.Pair;
import com.intellij.openapi.util.io.FileUtil;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.openapi.vfs.LocalFileSystem;
import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.psi.PsiDocumentManager;
import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiFile;
import com.intellij.ui.Gray;
import com.intellij.ui.JBColor;
import java.awt.*;
import java.io.File;
import java.io.IOException;
import java.security.NoSuchAlgorithmException;
import java.util.List;
import java.util.Set;
import java.util.TreeSet;
public class EditorDocOps {
private static final String JAVA_IO_TMP_DIR = "java.io.tmpdir";
private WindowObjects windowObjects = WindowObjects.getInstance();
private WindowEditorOps windowEditorOps = new WindowEditorOps();
private static final Color HIGHLIGHTING_COLOR =
new JBColor(new Color(255, 250, 205), Gray._100);
public final Pair<Integer, Integer> getLineOffSets(final Editor projectEditor,
final int distance) {
Document document = projectEditor.getDocument();
SelectionModel selectionModel = projectEditor.getSelectionModel();
int head = 0;
int tail = document.getLineCount() - 1;
if (selectionModel.hasSelection()) {
head = document.getLineNumber(selectionModel.getSelectionStart());
tail = document.getLineNumber(selectionModel.getSelectionEnd());
/*Selection model gives one more line if line is selected completely.
By Checking if complete line is slected and decreasing tail*/
if ((document.getLineStartOffset(tail) == selectionModel.getSelectionEnd())) {
tail--;
}
} else {
int currentLine = document.getLineNumber(projectEditor.getCaretModel().getOffset());
if (currentLine - distance >= 0) {
head = currentLine - distance;
}
if (currentLine + distance <= document.getLineCount() - 1) {
tail = currentLine + distance;
}
}
int start = document.getLineStartOffset(head);
int end = document.getLineEndOffset(tail);
Pair<Integer, Integer> pair = new Pair<>(start, end);
return pair;
}
public final VirtualFile getVirtualFile(final String fileName,
final String displayFileName,
final String contents)
throws IOException, NoSuchAlgorithmException {
final String tempDir = System.getProperty(JAVA_IO_TMP_DIR);
final String trimmedFileName =
FileUtil.sanitizeFileName(StringUtil.trimEnd(fileName, displayFileName));
final String digest = Utils.getInstance().getDigestAsString(trimmedFileName);
final String fullFilePath = Utils.getInstance()
.createFileWithContents(displayFileName, contents, tempDir, digest);
// Refreshing File System is required so that it is aware of newly created files
final VirtualFile virtualFile = LocalFileSystem.getInstance()
.refreshAndFindFileByIoFile(new File(fullFilePath));
if (virtualFile == null) {
throw new IllegalArgumentException("Virtual file should not be null."
+ " Can be an issue with FileSystem.");
}
windowEditorOps.setWriteStatus(virtualFile, false);
return virtualFile;
}
public final void addHighlighting(final List<Integer> linesForHighlighting,
final Document document) {
TextAttributes attributes = new TextAttributes();
JBColor color = JBColor.GREEN;
attributes.setEffectColor(color);
attributes.setEffectType(EffectType.SEARCH_MATCH);
attributes.setBackgroundColor(HIGHLIGHTING_COLOR);
Editor projectEditor =
FileEditorManager.getInstance(windowObjects.getProject()).getSelectedTextEditor();
if (projectEditor != null) {
PsiFile psiFile =
PsiDocumentManager.getInstance(windowObjects.getProject()).
getPsiFile(projectEditor.getDocument());
MarkupModel markupModel = projectEditor.getMarkupModel();
if (markupModel != null) {
markupModel.removeAllHighlighters();
for (int line : linesForHighlighting) {
line = line - 1;
if (line < document.getLineCount()) {
int startOffset = document.getLineStartOffset(line);
int endOffset = document.getLineEndOffset(line);
String lineText =
document.getCharsSequence().
subSequence(startOffset, endOffset).toString();
int lineStartOffset =
startOffset + lineText.length() - lineText.trim().length();
markupModel.addRangeHighlighter(lineStartOffset, endOffset,
HighlighterLayer.ERROR, attributes,
HighlighterTargetArea.EXACT_RANGE);
if (psiFile != null && psiFile.findElementAt(lineStartOffset) != null) {
HighlightUsagesHandler.doHighlightElements(projectEditor,
new PsiElement[]{psiFile.findElementAt(lineStartOffset)},
attributes, false);
}
}
}
}
}
}
public final void gotoLine(final int pLineNumber, final Document document) {
int lineNumber = pLineNumber;
Editor projectEditor =
FileEditorManager.getInstance(windowObjects.getProject()).getSelectedTextEditor();
if (projectEditor != null) {
CaretModel caretModel = projectEditor.getCaretModel();
//document is 0-indexed
if (lineNumber > document.getLineCount()) {
lineNumber = document.getLineCount() - 1;
} else {
lineNumber = lineNumber - 1;
}
caretModel.moveToLogicalPosition(new LogicalPosition(lineNumber, 0));
ScrollingModel scrollingModel = projectEditor.getScrollingModel();
scrollingModel.scrollToCaret(ScrollType.CENTER);
}
}
public final String getContentsInLines(final String fileContents,
final List<Integer> lineNumbersList) {
Document document = EditorFactory.getInstance().createDocument(fileContents);
Set<Integer> lineNumbersSet = new TreeSet<Integer>(lineNumbersList);
StringBuilder stringBuilder = new StringBuilder();
int prev = lineNumbersSet.iterator().next();
for (int line : lineNumbersSet) {
//Document is 0 indexed
line = line - 1;
if (line < document.getLineCount() - 1) {
if (prev != line - 1) {
stringBuilder.append(System.lineSeparator());
prev = line;
}
int startOffset = document.getLineStartOffset(line);
int endOffset = document.getLineEndOffset(line)
+ document.getLineSeparatorLength(line);
String code = document.getCharsSequence().
subSequence(startOffset, endOffset).
toString().trim()
+ System.lineSeparator();
stringBuilder.append(code);
}
}
return stringBuilder.toString();
}
}
