-
Notifications
You must be signed in to change notification settings - Fork 4.9k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat(script): add methods for apply renames and reload tabs (#2008)
- Loading branch information
Showing
23 changed files
with
351 additions
and
144 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
28 changes: 28 additions & 0 deletions
28
jadx-core/src/main/java/jadx/api/plugins/pass/impl/SimpleAfterLoadPass.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,28 @@ | ||
| package jadx.api.plugins.pass.impl; | ||
|
|
||
| import java.util.function.Consumer; | ||
|
|
||
| import jadx.api.JadxDecompiler; | ||
| import jadx.api.plugins.pass.JadxPassInfo; | ||
| import jadx.api.plugins.pass.types.JadxAfterLoadPass; | ||
|
|
||
| public class SimpleAfterLoadPass implements JadxAfterLoadPass { | ||
|
|
||
| private final JadxPassInfo info; | ||
| private final Consumer<JadxDecompiler> init; | ||
|
|
||
| public SimpleAfterLoadPass(String name, Consumer<JadxDecompiler> init) { | ||
| this.info = new SimpleJadxPassInfo(name); | ||
| this.init = init; | ||
| } | ||
|
|
||
| @Override | ||
| public JadxPassInfo getInfo() { | ||
| return info; | ||
| } | ||
|
|
||
| @Override | ||
| public void init(JadxDecompiler decompiler) { | ||
| init.accept(decompiler); | ||
| } | ||
| } |
178 changes: 178 additions & 0 deletions
178
jadx-gui/src/main/java/jadx/gui/events/services/RenameService.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,178 @@ | ||
| package jadx.gui.events.services; | ||
|
|
||
| import java.util.ArrayList; | ||
| import java.util.Collections; | ||
| import java.util.HashSet; | ||
| import java.util.List; | ||
| import java.util.Objects; | ||
| import java.util.Set; | ||
| import java.util.function.Consumer; | ||
| import java.util.stream.Collectors; | ||
|
|
||
| import org.jetbrains.annotations.NotNull; | ||
| import org.slf4j.Logger; | ||
| import org.slf4j.LoggerFactory; | ||
|
|
||
| import jadx.api.JadxDecompiler; | ||
| import jadx.api.JavaNode; | ||
| import jadx.api.data.ICodeRename; | ||
| import jadx.api.data.impl.JadxCodeData; | ||
| import jadx.api.plugins.events.JadxEvents; | ||
| import jadx.api.plugins.events.types.NodeRenamedByUser; | ||
| import jadx.core.utils.Utils; | ||
| import jadx.core.utils.exceptions.JadxRuntimeException; | ||
| import jadx.gui.jobs.TaskStatus; | ||
| import jadx.gui.settings.JadxProject; | ||
| import jadx.gui.treemodel.JClass; | ||
| import jadx.gui.treemodel.JNode; | ||
| import jadx.gui.treemodel.JRenameNode; | ||
| import jadx.gui.ui.MainWindow; | ||
| import jadx.gui.ui.TabbedPane; | ||
| import jadx.gui.ui.codearea.ClassCodeContentPanel; | ||
| import jadx.gui.ui.codearea.CodeArea; | ||
| import jadx.gui.ui.panel.ContentPanel; | ||
| import jadx.gui.utils.CacheObject; | ||
| import jadx.gui.utils.JNodeCache; | ||
| import jadx.gui.utils.NLS; | ||
| import jadx.gui.utils.UiUtils; | ||
|
|
||
| /** | ||
| * Rename service listen for user rename events. | ||
| * For each event: | ||
| * - add/update rename entry in project code data | ||
| * - update code and/or invalidate cache for related classes | ||
| * - apply all needed UI updates (tabs, classes tree) | ||
| */ | ||
| public class RenameService { | ||
| private static final Logger LOG = LoggerFactory.getLogger(RenameService.class); | ||
|
|
||
| public static void init(MainWindow mainWindow) { | ||
| RenameService renameService = new RenameService(mainWindow); | ||
| mainWindow.events().addListener(JadxEvents.NODE_RENAMED_BY_USER, renameService::process); | ||
| } | ||
|
|
||
| private final MainWindow mainWindow; | ||
|
|
||
| private RenameService(MainWindow mainWindow) { | ||
| this.mainWindow = mainWindow; | ||
| } | ||
|
|
||
| private void process(NodeRenamedByUser event) { | ||
| try { | ||
| LOG.debug("Applying rename event: {}", event); | ||
| JRenameNode node = getRenameNode(event); | ||
| updateCodeRenames(set -> processRename(node, event.getNewName(), set)); | ||
| refreshState(node); | ||
| } catch (Exception e) { | ||
| LOG.error("Rename failed", e); | ||
| UiUtils.errorMessage(mainWindow, "Rename failed:\n" + Utils.getStackTrace(e)); | ||
| } | ||
| } | ||
|
|
||
| private @NotNull JRenameNode getRenameNode(NodeRenamedByUser event) { | ||
| Object renameNode = event.getRenameNode(); | ||
| if (renameNode instanceof JRenameNode) { | ||
| return (JRenameNode) renameNode; | ||
| } | ||
| JadxDecompiler decompiler = mainWindow.getWrapper().getDecompiler(); | ||
| JavaNode javaNode = decompiler.getJavaNodeByRef(event.getNode()); | ||
| if (javaNode != null) { | ||
| JNode node = mainWindow.getCacheObject().getNodeCache().makeFrom(javaNode); | ||
| if (node instanceof JRenameNode) { | ||
| return (JRenameNode) node; | ||
| } | ||
| } | ||
| throw new JadxRuntimeException("Failed to resolve node: " + event.getNode()); | ||
| } | ||
|
|
||
| private void processRename(JRenameNode node, String newName, Set<ICodeRename> renames) { | ||
| ICodeRename rename = node.buildCodeRename(newName, renames); | ||
| renames.remove(rename); | ||
| if (newName.isEmpty()) { | ||
| node.removeAlias(); | ||
| } else { | ||
| renames.add(rename); | ||
| } | ||
| } | ||
|
|
||
| private void updateCodeRenames(Consumer<Set<ICodeRename>> updater) { | ||
| JadxProject project = mainWindow.getProject(); | ||
| JadxCodeData codeData = project.getCodeData(); | ||
| if (codeData == null) { | ||
| codeData = new JadxCodeData(); | ||
| } | ||
| Set<ICodeRename> set = new HashSet<>(codeData.getRenames()); | ||
| updater.accept(set); | ||
| List<ICodeRename> list = new ArrayList<>(set); | ||
| Collections.sort(list); | ||
| codeData.setRenames(list); | ||
| project.setCodeData(codeData); | ||
| } | ||
|
|
||
| private void refreshState(JRenameNode node) { | ||
| List<JavaNode> toUpdate = new ArrayList<>(); | ||
| node.addUpdateNodes(toUpdate); | ||
|
|
||
| JNodeCache nodeCache = mainWindow.getCacheObject().getNodeCache(); | ||
| Set<JClass> updatedTopClasses = toUpdate | ||
| .stream() | ||
| .map(JavaNode::getTopParentClass) | ||
| .map(nodeCache::makeFrom) | ||
| .filter(Objects::nonNull) | ||
| .collect(Collectors.toSet()); | ||
|
|
||
| LOG.debug("Classes to update: {}", updatedTopClasses); | ||
| if (updatedTopClasses.isEmpty()) { | ||
| return; | ||
| } | ||
| mainWindow.getBackgroundExecutor().execute("Refreshing", | ||
| () -> { | ||
| mainWindow.getWrapper().reloadCodeData(); | ||
| UiUtils.uiRunAndWait(() -> refreshTabs(mainWindow.getTabbedPane(), updatedTopClasses)); | ||
| refreshClasses(updatedTopClasses); | ||
| }, | ||
| (status) -> { | ||
| if (status == TaskStatus.CANCEL_BY_MEMORY) { | ||
| mainWindow.showHeapUsageBar(); | ||
| UiUtils.errorMessage(mainWindow, NLS.str("message.memoryLow")); | ||
| } | ||
| node.reload(mainWindow); | ||
| }); | ||
| } | ||
|
|
||
| private void refreshClasses(Set<JClass> updatedTopClasses) { | ||
| CacheObject cache = mainWindow.getCacheObject(); | ||
| if (updatedTopClasses.size() < 10) { | ||
| // small batch => reload | ||
| LOG.debug("Classes to reload: {}", updatedTopClasses.size()); | ||
| for (JClass cls : updatedTopClasses) { | ||
| try { | ||
| cls.reload(cache); | ||
| } catch (Exception e) { | ||
| LOG.error("Failed to reload class: {}", cls.getFullName(), e); | ||
| } | ||
| } | ||
| } else { | ||
| // big batch => unload | ||
| LOG.debug("Classes to unload: {}", updatedTopClasses.size()); | ||
| for (JClass cls : updatedTopClasses) { | ||
| try { | ||
| cls.unload(cache); | ||
| } catch (Exception e) { | ||
| LOG.error("Failed to unload class: {}", cls.getFullName(), e); | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| private void refreshTabs(TabbedPane tabbedPane, Set<JClass> updatedClasses) { | ||
| for (ContentPanel tab : tabbedPane.getTabs()) { | ||
| JClass rootClass = tab.getNode().getRootClass(); | ||
| if (updatedClasses.remove(rootClass)) { | ||
| ClassCodeContentPanel contentPanel = (ClassCodeContentPanel) tab; | ||
| CodeArea codeArea = (CodeArea) contentPanel.getJavaCodePanel().getCodeArea(); | ||
| codeArea.refreshClass(); | ||
| } | ||
| } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.