org.fife.ui.rsyntaxtextarea.folding.YamlFoldParser Maven / Gradle / Ivy
/*
*
* YamlFoldParser.java - Fold parser for YAML.
*
* This library is distributed under a modified BSD license. See the included
* LICENSE file for details.
*/
package org.fife.ui.rsyntaxtextarea.folding;
import org.fife.ui.rsyntaxtextarea.RSyntaxTextArea;
import org.fife.ui.rsyntaxtextarea.Token;
import javax.swing.text.BadLocationException;
import java.util.ArrayList;
import java.util.List;
import java.util.Stack;
/**
* Fold parser for YAML. Based on org.fife.ui.rsyntaxtextarea.folding.HtmlFoldParser.
*
* Fold is started by each line.
* Each fold contains all following lines with indent higher than the current line.
*/
public class YamlFoldParser implements FoldParser {
/**
* Constructor.
*
*/
public YamlFoldParser() {
}
private static boolean isSpaces(Token t) {
String lexeme = t.getLexeme();
return lexeme.trim().isEmpty();
}
/**
* {@inheritDoc}
*/
@Override
public List getFolds(RSyntaxTextArea textArea) {
List folds = new ArrayList<>();
Stack indentStack = new Stack<>();
Fold currentFold = null;
int lineCount = textArea.getLineCount();
// closing needs to be done with offset of the previous line
int lastOffset = 0;
try {
for (int line=0; line= indent && currentFold != null) {
currentFold.setEndOffset(lastOffset);
Fold parentFold = currentFold.getParent();
// Don't add fold markers for single-line blocks
if (currentFold.isOnSingleLine()) {
removeFold(currentFold, folds);
}
currentFold = parentFold;
indentStack.pop();
} else {
break;
}
}
// scan until the end of the line
while (t != null && t.isPaintable()) {
offset = t.getOffset();
t = t.getNextToken();
}
lastOffset = offset;
if (currentFold==null) {
currentFold = new Fold(FoldType.CODE, textArea, startLine.getOffset());
folds.add(currentFold);
} else {
currentFold = currentFold.createChild(FoldType.CODE, startLine.getOffset());
}
indentStack.push(indent);
}
} catch (BadLocationException ble) { // Should never happen
ble.printStackTrace();
}
return folds;
}
/**
* If this fold has a parent fold, this method removes it from its parent.
* Otherwise, it's assumed to be the most recent (top-level) fold in the
* folds list, and is removed from that.
*
* @param fold The fold to remove.
* @param folds The list of top-level folds.
*/
private static void removeFold(Fold fold, List folds) {
if (!fold.removeFromParent()) {
folds.remove(folds.size()-1);
}
}
}