Restoring code structure with jadx

[YarinOmesi]

Hi
I want to write a jadx plugin to restore the code structure of apps that uses libraries, for example google protobuf.

I what to add enums, methods and classes that where inlined or changed to jadx engine/classes.

I am thinking the flow will look like:

1. understand/find protobuf messages/classes.
2. add enums and method according to protobuf source
3. ...more logic...

Is such thing possible ? Can I add classes when jadx ia already "loaded" ?

[skylot]

Can I add classes when jadx ia already "loaded" ?

@YarinOmesi it depends on what data you need.
The best approach is to use context.addCodeInput(...) and create classes from the scratch, but this way you will work only with input files and not have access to app classes.
Another approach is adding classes during init in pre-decompilation pass (implements JadxPreparePass), add with rootNode.addClassNode, you will have access to classes (not methods code), but this way is not tested and not used for now, so you may encounter some issues.