org.openrewrite.xml.internal.XmlPrinter Maven / Gradle / Ivy
Show all versions of rewrite-xml Show documentation
/*
* Copyright 2020 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.openrewrite.xml.internal;
import org.openrewrite.Cursor;
import org.openrewrite.PrintOutputCapture;
import org.openrewrite.marker.Marker;
import org.openrewrite.xml.XmlVisitor;
import org.openrewrite.xml.tree.Xml;
import java.util.function.UnaryOperator;
public class XmlPrinter
extends XmlVisitor> {
@Override
public Xml visitDocument(Xml.Document document, PrintOutputCapture p) {
beforeSyntax(document, p);
document = (Xml.Document) super.visitDocument(document, p);
afterSyntax(document, p);
p.append(document.getEof());
return document;
}
@Override
public Xml visitProlog(Xml.Prolog prolog, PrintOutputCapture
p) {
beforeSyntax(prolog, p);
super.visitProlog(prolog, p);
afterSyntax(prolog, p);
return prolog;
}
@Override
public Xml visitXmlDecl(Xml.XmlDecl xmlDecl, PrintOutputCapture
p) {
beforeSyntax(xmlDecl, p);
p.append("");
afterSyntax(xmlDecl, p);
return xmlDecl;
}
@Override
public Xml visitTag(Xml.Tag tag, PrintOutputCapture
p) {
beforeSyntax(tag, p);
p.append('<');
boolean isJspDirective = getCursor().getParentOrThrow().getValue() instanceof Xml.Prolog;
if (isJspDirective) {
p.append("%@");
}
p.append(tag.getName());
visit(tag.getAttributes(), p);
p.append(tag.getBeforeTagDelimiterPrefix());
if (tag.getClosing() == null) {
p.append("/>");
} else {
p.append('>');
visit(tag.getContent(), p);
visit(tag.getClosing(), p);
}
afterSyntax(tag, p);
return tag;
}
@Override
public Xml visitTagClosing(Xml.Tag.Closing closing, PrintOutputCapture
p) {
beforeSyntax(closing, p);
p.append("");
afterSyntax(closing, p);
return closing;
}
@Override
public Xml visitAttribute(Xml.Attribute attribute, PrintOutputCapture
p) {
beforeSyntax(attribute, p);
p.append(attribute.getKey().getPrefix())
.append(attribute.getKeyAsString())
.append(attribute.getBeforeEquals())
.append('=');
visit(attribute.getValue(), p);
afterSyntax(attribute, p);
return attribute;
}
@Override
public Xml visitAttributeValue(Xml.Attribute.Value value, PrintOutputCapture
p) {
beforeSyntax(value, p);
char valueDelim;
if (Xml.Attribute.Value.Quote.Double.equals(value.getQuote())) {
valueDelim = '"';
} else {
valueDelim = '\'';
}
p.append(valueDelim)
.append(value.getValue())
.append(valueDelim);
afterSyntax(value, p);
return value;
}
@Override
public Xml visitComment(Xml.Comment comment, PrintOutputCapture
p) {
beforeSyntax(comment, p);
p.append("");
afterSyntax(comment, p);
return comment;
}
@Override
public Xml visitProcessingInstruction(Xml.ProcessingInstruction processingInstruction, PrintOutputCapture
p) {
beforeSyntax(processingInstruction, p);
p.append("");
afterSyntax(processingInstruction, p);
return processingInstruction;
}
@Override
public Xml visitCharData(Xml.CharData charData, PrintOutputCapture
p) {
beforeSyntax(charData, p);
if (charData.isCdata()) {
p.append("");
} else {
p.append(charData.getText());
}
p.append(charData.getAfterText());
afterSyntax(charData, p);
return charData;
}
@Override
public Xml visitDocTypeDecl(Xml.DocTypeDecl docTypeDecl, PrintOutputCapture
p) {
beforeSyntax(docTypeDecl, p);
p.append("');
afterSyntax(docTypeDecl, p);
return docTypeDecl;
}
@Override
public Xml visitDocTypeDeclExternalSubsets(Xml.DocTypeDecl.ExternalSubsets externalSubsets, PrintOutputCapture
p) {
beforeSyntax(externalSubsets, p);
p.append('[');
visit(externalSubsets.getElements(), p);
p.append(']');
afterSyntax(externalSubsets, p);
return externalSubsets;
}
@Override
public Xml visitElement(Xml.Element element, PrintOutputCapture
p) {
beforeSyntax(element, p);
visit(element.getSubset(), p);
p.append(element.getBeforeTagDelimiterPrefix());
afterSyntax(element, p);
return element;
}
@Override
public Xml visitIdent(Xml.Ident ident, PrintOutputCapture
p) {
beforeSyntax(ident, p);
p.append(ident.getName());
afterSyntax(ident, p);
return ident;
}
@Override
public Xml visitJspDirective(Xml.JspDirective jsp, PrintOutputCapture
p) {
beforeSyntax(jsp, p);
p.append("<%@");
p.append(jsp.getBeforeTypePrefix());
p.append(jsp.getType());
visit(jsp.getAttributes(), p);
p.append(jsp.getBeforeDirectiveEndPrefix());
p.append("%>");
afterSyntax(jsp, p);
return jsp;
}
private static final UnaryOperator XML_MARKER_WRAPPER =
out -> "";
private void beforeSyntax(Xml x, PrintOutputCapture p) {
for (Marker marker : x.getMarkers().getMarkers()) {
p.append(p.getMarkerPrinter().beforePrefix(marker, new Cursor(getCursor(), marker), XML_MARKER_WRAPPER));
}
p.append(x.getPrefix());
visitMarkers(x.getMarkers(), p);
for (Marker marker : x.getMarkers().getMarkers()) {
p.append(p.getMarkerPrinter().beforeSyntax(marker, new Cursor(getCursor(), marker), XML_MARKER_WRAPPER));
}
}
private void afterSyntax(Xml x, PrintOutputCapture
p) {
for (Marker marker : x.getMarkers().getMarkers()) {
p.append(p.getMarkerPrinter().afterSyntax(marker, new Cursor(getCursor(), marker), XML_MARKER_WRAPPER));
}
}
}