diff options
Diffstat (limited to 'javaparser-core/src/main/java/com/github/javaparser/printer/concretesyntaxmodel/CsmList.java')
-rw-r--r-- | javaparser-core/src/main/java/com/github/javaparser/printer/concretesyntaxmodel/CsmList.java | 120 |
1 files changed, 120 insertions, 0 deletions
diff --git a/javaparser-core/src/main/java/com/github/javaparser/printer/concretesyntaxmodel/CsmList.java b/javaparser-core/src/main/java/com/github/javaparser/printer/concretesyntaxmodel/CsmList.java new file mode 100644 index 000000000..30fae7ed4 --- /dev/null +++ b/javaparser-core/src/main/java/com/github/javaparser/printer/concretesyntaxmodel/CsmList.java @@ -0,0 +1,120 @@ +/* + * Copyright (C) 2007-2010 JĂșlio Vilmar Gesser. + * Copyright (C) 2011, 2013-2016 The JavaParser Team. + * + * This file is part of JavaParser. + * + * JavaParser can be used either under the terms of + * a) the GNU Lesser General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * b) the terms of the Apache License + * + * You should have received a copy of both licenses in LICENCE.LGPL and + * LICENCE.APACHE. Please refer to those files for details. + * + * JavaParser is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU Lesser General Public License for more details. + */ + +package com.github.javaparser.printer.concretesyntaxmodel; + +import com.github.javaparser.ast.Node; +import com.github.javaparser.ast.NodeList; +import com.github.javaparser.ast.observer.ObservableProperty; +import com.github.javaparser.printer.ConcreteSyntaxModel; +import com.github.javaparser.printer.SourcePrinter; + +import java.util.Collection; +import java.util.Iterator; + +public class CsmList implements CsmElement { + private final ObservableProperty property; + private final CsmElement separatorPost; + private final CsmElement separatorPre; + private final CsmElement preceeding; + private final CsmElement following; + + public ObservableProperty getProperty() { + return property; + } + + public CsmElement getSeparatorPost() { + return separatorPost; + } + + public CsmElement getSeparatorPre() { + return separatorPre; + } + + public CsmElement getPreceeding() { + return preceeding; + } + + public CsmElement getFollowing() { + return following; + } + + public CsmList(ObservableProperty property, CsmElement separator) { + this(property, new CsmNone(), separator, new CsmNone(), new CsmNone()); + } + + public CsmList(ObservableProperty property) { + this(property, new CsmNone(), new CsmNone(), new CsmNone(), new CsmNone()); + } + + public CsmList(ObservableProperty property, CsmElement separatorPre, CsmElement separatorPost, CsmElement preceeding, CsmElement following) { + this.property = property; + this.separatorPre = separatorPre; + this.separatorPost = separatorPost; + this.preceeding = preceeding; + this.following = following; + } + + @Override + public void prettyPrint(Node node, SourcePrinter printer) { + if (property.isAboutNodes()) { + NodeList nodeList = property.getValueAsMultipleReference(node); + if (nodeList == null) { + return; + } + if (!nodeList.isEmpty() && preceeding != null) { + preceeding.prettyPrint(node, printer); + } + for (int i = 0; i < nodeList.size(); i++) { + if (separatorPre != null && i != 0) { + separatorPre.prettyPrint(node, printer); + } + ConcreteSyntaxModel.genericPrettyPrint(nodeList.get(i), printer); + if (separatorPost != null && i != (nodeList.size() - 1)) { + separatorPost.prettyPrint(node, printer); + } + } + if (!nodeList.isEmpty() && following != null) { + following.prettyPrint(node, printer); + } + } else { + Collection<?> values = property.getValueAsCollection(node); + if (values == null) { + return; + } + if (!values.isEmpty() && preceeding != null) { + preceeding.prettyPrint(node, printer); + } + for (Iterator it = values.iterator(); it.hasNext(); ) { + if (separatorPre != null && it.hasNext()) { + separatorPre.prettyPrint(node, printer); + } + printer.print(PrintingHelper.printToString(it.next())); + if (separatorPost != null && it.hasNext()) { + separatorPost.prettyPrint(node, printer); + } + } + if (!values.isEmpty() && following != null) { + following.prettyPrint(node, printer); + } + } + } +} |