• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * Copyright (C) 2007-2010 Júlio Vilmar Gesser.
3  * Copyright (C) 2011, 2013-2016 The JavaParser Team.
4  *
5  * This file is part of JavaParser.
6  *
7  * JavaParser can be used either under the terms of
8  * a) the GNU Lesser General Public License as published by
9  *     the Free Software Foundation, either version 3 of the License, or
10  *     (at your option) any later version.
11  * b) the terms of the Apache License
12  *
13  * You should have received a copy of both licenses in LICENCE.LGPL and
14  * LICENCE.APACHE. Please refer to those files for details.
15  *
16  * JavaParser is distributed in the hope that it will be useful,
17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
19  * GNU Lesser General Public License for more details.
20  */
21 
22 package com.github.javaparser.printer.concretesyntaxmodel;
23 
24 import com.github.javaparser.ast.Node;
25 import com.github.javaparser.ast.NodeList;
26 import com.github.javaparser.ast.observer.ObservableProperty;
27 import com.github.javaparser.printer.ConcreteSyntaxModel;
28 import com.github.javaparser.printer.SourcePrinter;
29 
30 import java.util.Collection;
31 import java.util.Iterator;
32 
33 public class CsmList implements CsmElement {
34     private final ObservableProperty property;
35     private final CsmElement separatorPost;
36     private final CsmElement separatorPre;
37     private final CsmElement preceeding;
38     private final CsmElement following;
39 
getProperty()40     public ObservableProperty getProperty() {
41         return property;
42     }
43 
getSeparatorPost()44     public CsmElement getSeparatorPost() {
45         return separatorPost;
46     }
47 
getSeparatorPre()48     public CsmElement getSeparatorPre() {
49         return separatorPre;
50     }
51 
getPreceeding()52     public CsmElement getPreceeding() {
53         return preceeding;
54     }
55 
getFollowing()56     public CsmElement getFollowing() {
57         return following;
58     }
59 
CsmList(ObservableProperty property, CsmElement separator)60     public CsmList(ObservableProperty property, CsmElement separator) {
61         this(property, new CsmNone(), separator, new CsmNone(), new CsmNone());
62     }
63 
CsmList(ObservableProperty property)64     public CsmList(ObservableProperty property) {
65         this(property, new CsmNone(), new CsmNone(), new CsmNone(), new CsmNone());
66     }
67 
CsmList(ObservableProperty property, CsmElement separatorPre, CsmElement separatorPost, CsmElement preceeding, CsmElement following)68     public CsmList(ObservableProperty property, CsmElement separatorPre, CsmElement separatorPost, CsmElement preceeding, CsmElement following) {
69         this.property = property;
70         this.separatorPre = separatorPre;
71         this.separatorPost = separatorPost;
72         this.preceeding = preceeding;
73         this.following = following;
74     }
75 
76     @Override
prettyPrint(Node node, SourcePrinter printer)77     public void prettyPrint(Node node, SourcePrinter printer) {
78         if (property.isAboutNodes()) {
79             NodeList nodeList = property.getValueAsMultipleReference(node);
80             if (nodeList == null) {
81                 return;
82             }
83             if (!nodeList.isEmpty() && preceeding != null) {
84                 preceeding.prettyPrint(node, printer);
85             }
86             for (int i = 0; i < nodeList.size(); i++) {
87                 if (separatorPre != null && i != 0) {
88                     separatorPre.prettyPrint(node, printer);
89                 }
90                 ConcreteSyntaxModel.genericPrettyPrint(nodeList.get(i), printer);
91                 if (separatorPost != null && i != (nodeList.size() - 1)) {
92                     separatorPost.prettyPrint(node, printer);
93                 }
94             }
95             if (!nodeList.isEmpty() && following != null) {
96                 following.prettyPrint(node, printer);
97             }
98         } else {
99             Collection<?> values = property.getValueAsCollection(node);
100             if (values == null) {
101                 return;
102             }
103             if (!values.isEmpty() && preceeding != null) {
104                 preceeding.prettyPrint(node, printer);
105             }
106             for (Iterator it = values.iterator(); it.hasNext(); ) {
107                 if (separatorPre != null && it.hasNext()) {
108                     separatorPre.prettyPrint(node, printer);
109                 }
110                 printer.print(PrintingHelper.printToString(it.next()));
111                 if (separatorPost != null && it.hasNext()) {
112                     separatorPost.prettyPrint(node, printer);
113                 }
114             }
115             if (!values.isEmpty() && following != null) {
116                 following.prettyPrint(node, printer);
117             }
118         }
119     }
120 }
121