fitnesse.wikitext.parser.HtmlWriter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of fitnesse Show documentation
The fully integrated standalone wiki, and acceptance testing framework.
There is a newer version: 20181217
package fitnesse.wikitext.parser;

import fitnesse.html.HtmlElement;
import java.util.Stack;

public class HtmlWriter {
    private StringBuilder html = new StringBuilder();
    private int lastEndLine = -1;
    private Tags tags = new Tags();

    public void startTag(String tag) {
        closeCurrentTag(false);
        indentTag();
        tags.push(tag, false);
        writeTag(tag);
    }

    public void startTagInline(String tag) {
        closeCurrentTag(true);
        tags.push(tag, true);
        writeTag(tag);
    }

    public void putText(String text) {
        closeCurrentTag(true);
        html.append(text);
    }

    private void writeTag(String tag) {
        html.append('<');
        html.append(tag);
    }

    private void indentTag() {
        for (int i = 0; i < tags.size(); i++) html.append('\t');
    }

    private void closeCurrentTag(boolean newTagInline) {
        if (tags.size() > 0) {
            if (tags.isOpen()) {
                html.append('>');
                tags.close();
            }
            if (!newTagInline) writeEndLine();
        }
    }

    public void endTag() {
        Tag tag = tags.pop();
        if (tag.open) {
            html.append(" />");
        }
        else {
            if (html.length() == lastEndLine) {
                indentTag();
            }
            html.append("');
        }
        if (!tag.inline) writeEndLine();
    }

    private void writeEndLine() {
        if (html.length() == lastEndLine) return;
        html.append(HtmlElement.endl);
        lastEndLine = html.length();
    }

    public void putTag(String tag) {
        startTag(tag);
        endTag();
    }

    public void putTagInline(String tag) {
        startTagInline(tag);
        endTag();
    }

    public void putAttribute(String name, String value) {
        html.append(' ');
        html.append(name);
        html.append("=\"");
        html.append(value);
        html.append('"');
    }

    public String toHtml() { return html.toString(); }

    private class Tags {
        private Stack tags = new Stack<>();
        private Tag top = new Tag(null, false);
        private boolean isEmpty = true;

        public void push(String name, boolean inline) {
            if (!isEmpty) {
                tags.push(top);
                top = new Tag(name, inline);
            }
            else {
                top.name = name;
                top.inline = inline;
                isEmpty = false;
            }
        }

        public Tag pop() {
            Tag result = top;
            if (tags.empty()) {
                isEmpty = true;
            }
            else {
                top = tags.pop();
            }
            return result;
        }

        public int size() { return tags.size() + (isEmpty ? 0 : 1); }

        public boolean isOpen() { return top.open; }

        public void close() { top.open = false; }
    }

    private class Tag {
        public String name;
        public boolean open;
        public boolean inline;

        public Tag(String name, boolean inline) {
            this.name = name;
            this.inline = inline;
            open = true;
        }
    }
}