net.nemerosa.ontrack.model.support.MessageAnnotationUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ontrack-model Show documentation
Ontrack module: ontrack-model
There is a newer version: 4.4.5
package net.nemerosa.ontrack.model.support;

import net.nemerosa.ontrack.model.support.tree.DefaultNodeFactory;
import net.nemerosa.ontrack.model.support.tree.Node;
import net.nemerosa.ontrack.model.support.tree.NodeFactory;
import net.nemerosa.ontrack.model.support.tree.NodeVisitor;
import net.nemerosa.ontrack.model.support.tree.support.Markup;
import org.apache.commons.lang3.StringUtils;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Map;

import static org.apache.commons.lang3.StringEscapeUtils.escapeHtml4;

public final class MessageAnnotationUtils {

    public static String annotate(String text, List messageAnnotators) {
        Node root = annotateAsNode(text, messageAnnotators);
        final StringBuilder html = new StringBuilder();
        root.visit(new NodeVisitor() {

            @Override
            public void start(Node node) {
                Markup m = node.getData();
                if (m != null) {
                    if (m.isOnlyText()) {
                        html.append(escapeHtml4(m.getText()));
                    } else {
                        html.append("<").append(m.getType());
                        Map attributes = m.getAttributes();
                        for (Map.Entry attr : attributes.entrySet()) {
                            html.append(" ").append(escapeHtml4(attr.getKey())).append("=\"").append(escapeHtml4(attr.getValue())).append("\"");
                        }
                        if (node.isLeaf()) {
                            html.append("/>");
                        } else {
                            html.append(">");
                        }
                    }
                }
            }

            @Override
            public void end(Node node) {
                Markup m = node.getData();
                if (m != null && !m.isOnlyText() && !node.isLeaf()) {
                    html.append("");
                }
            }
        });
        return html.toString();
    }

    public static Node annotateAsNode(String text, List messageAnnotators) {
        final NodeFactory factory = new DefaultNodeFactory<>();
        Node root = factory.leaf(Markup.text(text));
        if (messageAnnotators != null) {
            for (final MessageAnnotator messageAnnotator : messageAnnotators) {
                if (messageAnnotator != null) {
                    root = root.transform(node -> {
                        if (node.isLeaf()) {
                            String text1 = node.getData().getText();
                            if (StringUtils.isNotBlank(text1)) {
                                Collection annotations = messageAnnotator.annotate(text1);
                                Collection> nodes = new ArrayList<>();
                                for (MessageAnnotation annotation : annotations) {
                                    if (annotation.isText()) {
                                        nodes.add(factory.leaf(Markup.text(annotation.getText())));
                                    } else {
                                        Node child = factory.leaf(Markup.of(annotation.getType(), annotation.getAttributes()));
                                        if (annotation.hasText()) {
                                            child.append(
                                                    factory.leaf(Markup.text(annotation.getText()))
                                            );
                                        }
                                        nodes.add(child);
                                    }
                                }
                                return factory.node(null, nodes);
                            }
                        }
                        return node;
                    }, factory);
                }
            }
        }
        return root;
    }
}