java-parser: New entrypoint lexAndParse to return both tokens and CST.

jhipster · Nov 26, 2023 · e53f3a4 · e53f3a4
1 parent ae03d4a
commit e53f3a4
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 4 deletions.
diff --git a/packages/java-parser/api.d.ts b/packages/java-parser/api.d.ts
@@ -30,6 +30,14 @@ export declare type CstChildrenDictionary = {
   [identifier: string]: CstElement[];
 };
 
+export function lexAndParse(
+  text: string,
+  startProduction?: string
+): {
+  tokens: IToken[];
+  cst: CstNode;
+};
+
 export function parse(text: string, startProduction?: string): CstNode;
 
 export const BaseJavaCstVisitor: JavaCstVisitorConstructor<any, any>;

diff --git a/packages/java-parser/src/index.js b/packages/java-parser/src/index.js
@@ -9,7 +9,7 @@ const BaseJavaCstVisitor = parser.getBaseCstVisitorConstructor();
 const BaseJavaCstVisitorWithDefaults =
   parser.getBaseCstVisitorConstructorWithDefaults();
 
-function parse(inputText, entryPoint = "compilationUnit") {
+function lexAndParse(inputText, entryPoint = "compilationUnit") {
   // Lex
   const lexResult = JavaLexer.tokenize(inputText);
 
@@ -25,7 +25,8 @@ function parse(inputText, entryPoint = "compilationUnit") {
     );
   }
 
-  parser.input = lexResult.tokens;
+  const tokens = lexResult.tokens;
+  parser.input = tokens;
   parser.mostEnclosiveCstNodeByStartOffset = {};
   parser.mostEnclosiveCstNodeByEndOffset = {};
 
@@ -51,16 +52,21 @@ function parse(inputText, entryPoint = "compilationUnit") {
   }
 
   attachComments(
-    lexResult.tokens,
+    tokens,
     lexResult.groups.comments,
     parser.mostEnclosiveCstNodeByStartOffset,
     parser.mostEnclosiveCstNodeByEndOffset
   );
 
-  return cst;
+  return { cst, tokens };
+}
+
+function parse(inputText, entryPoint = "compilationUnit") {
+  return lexAndParse(inputText, entryPoint).cst;
 }
 
 module.exports = {
+  lexAndParse,
   parse,
   BaseJavaCstVisitor,
   BaseJavaCstVisitorWithDefaults