File tree Expand file tree Collapse file tree 2 files changed +15
-4
lines changed
src/main/java/com/ibm/cldk Expand file tree Collapse file tree 2 files changed +15
-4
lines changed Original file line number Diff line number Diff line change @@ -136,11 +136,21 @@ private static JavaCompilationUnit processCompilationUnit(CompilationUnit parseR
136
136
137
137
cUnit .setFilePath (parseResult .getStorage ().map (s -> s .getPath ().toString ()).orElse ("<in-memory>" ));
138
138
139
- com .ibm .cldk .entities .Comment comment = new com .ibm .cldk .entities .Comment ();
139
+ // Set file level comment
140
+ parseResult .getAllComments ().stream ().findFirst ().ifPresent (c -> {
141
+ com .ibm .cldk .entities .Comment fileComment = new com .ibm .cldk .entities .Comment ();
142
+ fileComment .setContent (c .getContent ());
143
+ fileComment .setStartLine (c .getRange ().isPresent () ? c .getRange ().get ().begin .line : -1 );
144
+ fileComment .setEndLine (c .getRange ().isPresent () ? c .getRange ().get ().end .line : -1 );
145
+ fileComment .setStartColumn (c .getRange ().isPresent () ? c .getRange ().get ().begin .column : -1 );
146
+ fileComment .setEndColumn (c .getRange ().isPresent () ? c .getRange ().get ().end .column : -1 );
147
+ fileComment .setJavadoc (c .isJavadocComment ());
148
+ cUnit .getComments ().add (fileComment );
149
+ });
140
150
141
151
// Add class comment
142
152
cUnit .setComments (
143
- parseResult .getAllContainedComments ().stream ().map (c -> {
153
+ parseResult .getAllComments ().stream ().map (c -> {
144
154
com .ibm .cldk .entities .Comment fileComment = new com .ibm .cldk .entities .Comment ();
145
155
fileComment .setContent (c .getContent ());
146
156
fileComment .setStartLine (c .getRange ().isPresent () ? c .getRange ().get ().begin .line : -1 );
@@ -154,7 +164,6 @@ private static JavaCompilationUnit processCompilationUnit(CompilationUnit parseR
154
164
155
165
// Set package name
156
166
cUnit .setPackageName (parseResult .getPackageDeclaration ().map (NodeWithName ::getNameAsString ).orElse ("" ));
157
-
158
167
// Add javadoc comment
159
168
// Add imports
160
169
cUnit .setImports (
Original file line number Diff line number Diff line change 1
1
package com .ibm .cldk .entities ;
2
2
3
3
import lombok .Data ;
4
+
5
+ import java .util .ArrayList ;
4
6
import java .util .List ;
5
7
import java .util .Map ;
6
8
7
9
@ Data
8
10
public class JavaCompilationUnit {
9
11
private String filePath ;
10
12
private String packageName ;
11
- private List <Comment > comments ;
13
+ private List <Comment > comments = new ArrayList <>() ;
12
14
private List <String > imports ;
13
15
private Map <String , Type > typeDeclarations ;
14
16
private boolean isModified ;
You can’t perform that action at this time.
0 commit comments