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
136136
137137 cUnit .setFilePath (parseResult .getStorage ().map (s -> s .getPath ().toString ()).orElse ("<in-memory>" ));
138138
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+ });
140150
141151 // Add class comment
142152 cUnit .setComments (
143- parseResult .getAllContainedComments ().stream ().map (c -> {
153+ parseResult .getAllComments ().stream ().map (c -> {
144154 com .ibm .cldk .entities .Comment fileComment = new com .ibm .cldk .entities .Comment ();
145155 fileComment .setContent (c .getContent ());
146156 fileComment .setStartLine (c .getRange ().isPresent () ? c .getRange ().get ().begin .line : -1 );
@@ -154,7 +164,6 @@ private static JavaCompilationUnit processCompilationUnit(CompilationUnit parseR
154164
155165 // Set package name
156166 cUnit .setPackageName (parseResult .getPackageDeclaration ().map (NodeWithName ::getNameAsString ).orElse ("" ));
157-
158167 // Add javadoc comment
159168 // Add imports
160169 cUnit .setImports (
Original file line number Diff line number Diff line change 11package com .ibm .cldk .entities ;
22
33import lombok .Data ;
4+
5+ import java .util .ArrayList ;
46import java .util .List ;
57import java .util .Map ;
68
79@ Data
810public class JavaCompilationUnit {
911 private String filePath ;
1012 private String packageName ;
11- private List <Comment > comments ;
13+ private List <Comment > comments = new ArrayList <>() ;
1214 private List <String > imports ;
1315 private Map <String , Type > typeDeclarations ;
1416 private boolean isModified ;
You can’t perform that action at this time.
0 commit comments