Upstream/SKCraft-Launcher
1
0
Fork 0
mirror of https://github.com/SKCraft/Launcher.git synced 2024-11-28 12:57:07 +01:00
Code Issues Projects Releases Wiki Activity

Merge pull request #82 from seebs/master

Browse Source 
Avoid pathological failure mode when trimming lines.
This commit is contained in:
sk89q 2015-08-03 13:40:54 -07:00
commit 40555cce59
2 changed files with 45 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
launcher/src/main/java/com/skcraft/launcher/swing/MessageLog.java
View File

@ -111,22 +111,30 @@ public class MessageLog extends JPanel {
* @param line line * @param line line
* @param attributes attribute set, or null for none * @param attributes attribute set, or null for none
*/ */
public void log(String line, AttributeSet attributes) { public void log(final String line, AttributeSet attributes) {
final Document d = document;
final JTextComponent t = textComponent;
if (colorEnabled) { if (colorEnabled) {
if (line.startsWith("(!!)")) { if (line.startsWith("(!!)")) {
attributes = highlightedAttributes; attributes = highlightedAttributes;
} }
} }
final AttributeSet a = attributes;
SwingUtilities.invokeLater(new Runnable() {
@Override
public void run() {
try { try {
int offset = document.getLength(); int offset = d.getLength();
document.insertString(offset, line, d.insertString(offset, line,
(attributes != null && colorEnabled) ? attributes : defaultAttributes); (a != null && colorEnabled) ? a : defaultAttributes);
textComponent.setCaretPosition(document.getLength()); t.setCaretPosition(d.getLength());
} catch (BadLocationException ble) { } catch (BadLocationException ble) {
} }
} }
});
}
/** /**
* Get an output stream that can be written to. * Get an output stream that can be written to.

22
launcher/src/main/java/com/skcraft/launcher/util/LimitLinesDocumentListener.java
View File

@ -21,6 +21,7 @@ import javax.swing.text.Element;
public class LimitLinesDocumentListener implements DocumentListener { public class LimitLinesDocumentListener implements DocumentListener {
private int maximumLines; private int maximumLines;
private boolean isRemoveFromStart; private boolean isRemoveFromStart;
private volatile boolean isRemoving;
/** /**
* Specify the number of lines to be stored in the Document. Extra lines * Specify the number of lines to be stored in the Document. Extra lines
@ -34,6 +35,7 @@ public class LimitLinesDocumentListener implements DocumentListener {
boolean isRemoveFromStart) { boolean isRemoveFromStart) {
setLimitLines(maximumLines); setLimitLines(maximumLines);
this.isRemoveFromStart = isRemoveFromStart; this.isRemoveFromStart = isRemoveFromStart;
this.isRemoving = false;
} }
/** /**
@ -54,6 +56,8 @@ public class LimitLinesDocumentListener implements DocumentListener {
// Changes to the Document can not be done within the listener // Changes to the Document can not be done within the listener
// so we need to add the processing to the end of the EDT // so we need to add the processing to the end of the EDT
if (!this.isRemoving) {
this.isRemoving = true;
SwingUtilities.invokeLater(new Runnable() { SwingUtilities.invokeLater(new Runnable() {
@Override @Override
public void run() { public void run() {
@ -61,6 +65,7 @@ public class LimitLinesDocumentListener implements DocumentListener {
} }
}); });
} }
}
@Override @Override
public void removeUpdate(DocumentEvent e) { public void removeUpdate(DocumentEvent e) {
@ -74,20 +79,25 @@ public class LimitLinesDocumentListener implements DocumentListener {
// The root Element of the Document will tell us the total number // The root Element of the Document will tell us the total number
// of line in the Document. // of line in the Document.
try {
Document document = e.getDocument(); Document document = e.getDocument();
Element root = document.getDefaultRootElement(); Element root = document.getDefaultRootElement();
int excess = root.getElementCount() - maximumLines;
while (root.getElementCount() > maximumLines) { if (excess > 0) {
if (isRemoveFromStart) { if (isRemoveFromStart) {
removeFromStart(document, root); removeFromStart(document, root, excess);
} else { } else {
removeFromEnd(document, root); removeFromEnd(document, root);
} }
} }
} finally {
this.isRemoving = false;
}
} }
private void removeFromStart(Document document, Element root) { private void removeFromStart(Document document, Element root, int excess) {
Element line = root.getElement(0); Element line = root.getElement(excess - 1);
int end = line.getEndOffset(); int end = line.getEndOffset();
try { try {
@ -101,9 +111,9 @@ public class LimitLinesDocumentListener implements DocumentListener {
// We use start minus 1 to make sure we remove the newline // We use start minus 1 to make sure we remove the newline
// character of the previous line // character of the previous line
Element line = root.getElement(root.getElementCount() - 1); Element line = root.getElement(maximumLines);
int start = line.getStartOffset(); int start = line.getStartOffset();
int end = line.getEndOffset(); int end = root.getEndOffset();
try { try {
document.remove(start - 1, end - start); document.remove(start - 1, end - start);