78 lines
2.4 KiB
Java
78 lines
2.4 KiB
Java
package fr.pandacube.lib.chat;
|
||
|
||
import java.util.ArrayList;
|
||
import java.util.List;
|
||
|
||
/**
|
||
* A tree structure of {@link Chat} component intended to be rendered in chat using {@link #render(boolean)}.
|
||
*/
|
||
public class ChatTreeNode {
|
||
|
||
private static final String TREE_MIDDLE_CONNECTED = "├";
|
||
private static final String TREE_END_CONNECTED = "└";
|
||
private static final String TREE_MIDDLE_OPEN = "│§0`§r";
|
||
private static final String TREE_END_OPEN = "§0```§r";
|
||
private static final String TREE_MIDDLE_OPEN_CONSOLE = "│";
|
||
private static final String TREE_END_OPEN_CONSOLE = " "; // nbsp
|
||
|
||
|
||
/**
|
||
* The component for the current node.
|
||
*/
|
||
public final Chat component;
|
||
|
||
/**
|
||
* Children nodes.
|
||
*/
|
||
public final List<ChatTreeNode> children = new ArrayList<>();
|
||
|
||
/**
|
||
* Construct a new {@link ChatTreeNode}.
|
||
* @param cmp the component for the current node.
|
||
*/
|
||
public ChatTreeNode(Chat cmp) {
|
||
component = cmp;
|
||
}
|
||
|
||
/**
|
||
* Adds a child to the current node.
|
||
* @param child the child to add.
|
||
* @return this.
|
||
*/
|
||
public ChatTreeNode addChild(ChatTreeNode child) {
|
||
children.add(child);
|
||
return this;
|
||
}
|
||
|
||
/**
|
||
* Generate a tree view based on this tree structure.
|
||
* <p>
|
||
* Each element in the returned list represent 1 line of this tree view.
|
||
* Thus, the caller may send each line separately or at once, depending on the quantity of data.
|
||
* @param console true to render for console, false otherwise.
|
||
* @return an array of component, each element being a single line.
|
||
*/
|
||
public List<Chat> render(boolean console) {
|
||
List<Chat> ret = new ArrayList<>();
|
||
|
||
ret.add(ChatStatic.chat()
|
||
.then(component));
|
||
|
||
for (int i = 0; i < children.size(); i++) {
|
||
List<Chat> childComponents = children.get(i).render(console);
|
||
boolean last = i == children.size() - 1;
|
||
for (int j = 0; j < childComponents.size(); j++) {
|
||
|
||
String prefix = last ? (j == 0 ? TREE_END_CONNECTED : (console ? TREE_END_OPEN_CONSOLE : TREE_END_OPEN))
|
||
: (j == 0 ? TREE_MIDDLE_CONNECTED : (console ? TREE_MIDDLE_OPEN_CONSOLE : TREE_MIDDLE_OPEN));
|
||
|
||
ret.add(ChatStatic.text(prefix)
|
||
.then(childComponents.get(j)));
|
||
}
|
||
}
|
||
|
||
|
||
return ret;
|
||
}
|
||
}
|