Is there an equivalent library for VSCODE plugin like PSI library for Intellij?

[akashsivaram]

We are currently developing a VSCODE Plugin to add Open API 3/ Swagger Annotations for Springboot Rest Controllers and associated model classes. We already implemented this for IntelliJ using PSI which was quite seamless, but we are struggling to find a effective way to parse Java Code in VS Code. I have explored various approaches including LSP and different parsers, but most solutions are based on Regex patterns to match Java identifiers and then finding the line number and column position and then inserting the required Annotation at the specific position. What I need is a robust way to do the following,
Parse Java files to identify methods and parameters
Add Open API 3 / Swagger annotations to the code
Support regenerating annotations multiple times without duplication

As mentioned above I was able to do the above seamlessly in IntelliJ without having to worry about which line number or column position my method or parameter is present, this was all abstracted away by the IntelliJ's PSI library. Looking for a similar option in VSCODE, is there something available for VSCODE already that I can make use of?

[alefragnani]

Hi @akashsivaram ,

No, there is no equivalent to IntelliJ PSI in VS Code. If you need syntactic or semantic info about a source code, you must develop it in your extension, or rely on others extension providing such info for yours.

But, depending on your needs, you may use VS Commands to provide you structured info about the source code, abstracting the programming language. For instance, you may use vscode.executeDocumentSymbolProvider in order to retrieve all symbols (methods, classes, interfaces, variables, etc) from a file. With these symbols, you may navigate and/or update the source, because it contains informations like its position. It’s the same info you see in the Outline view, or when you use the Go to Symbol command.

Something like this would provide you all symbols in a file

    const docSymbols = await commands.executeCommand(
        'vscode.executeDocumentSymbolProvider',
        window.activeTextEditor.document.uri
    ) as DocumentSymbol[]

I use this on my Separators extension. Feel free to look at its source code to better understand how it works.

Hope this helps

[akashsivaram]

Yeah, we tried this approach as well but when I generate the annotations for more than one time using the extension, it not identifying the recent changes correctly and adding the code redundantly after the first try. And even after writing methods to remove the existing annotations its not identifying the recent methods and annotations after first generation of the code with extension. Is there any robust option to achieve this

[alefragnani]

Are using the TextEditor.edit API to update the document (add and remove annotations)? If yes, are you trying to apply all annotations at once, or one for each method?