ReVanced/smali
1
0
Fork 0
mirror of https://github.com/revanced/smali.git synced 2025-05-08 02:14:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Don't modify modifier outside of a write command

Browse Source
This commit is contained in:
Ben Gruver 2014-12-31 14:28:47 -08:00
parent 145cd6d518
commit 2ab03ae212
1 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
smalidea/src/main/java/org/jf/smalidea/psi/impl/SmaliModifierList.java
View File

@ -32,6 +32,7 @@
package org.jf.smalidea.psi.impl;
import com.intellij.lang.ASTNode;
import com.intellij.openapi.command.WriteCommandAction;
import com.intellij.psi.PsiModifier.ModifierConstant;
import com.intellij.psi.PsiModifierList;
import com.intellij.psi.PsiModifierListOwner;
@ -120,21 +121,30 @@ public class SmaliModifierList extends SmaliStubBasedPsiElement<SmaliModifierLis
public void setModifierProperty(@ModifierConstant @NotNull @NonNls String name, boolean addModifier)
throws IncorrectOperationException {
if (addModifier) {
SmaliAccessList accessListNode = findAccessListNode();
final SmaliAccessList accessListNode = findAccessListNode();
if (accessListNode == null) {
throw new IncorrectOperationException("Cannot add modifier: no .class statement");
}
TreeElement leaf = Factory.createSingleLeafElement(SmaliTokens.ACCESS_SPEC, name, null, getManager());
accessListNode.addInternal(leaf, leaf, null, null);
final TreeElement leaf = Factory.createSingleLeafElement(SmaliTokens.ACCESS_SPEC, name, null, getManager());
new WriteCommandAction.Simple(getProject(), getContainingFile()) {
@Override protected void run() throws Throwable {
accessListNode.addInternal(leaf, leaf, null, null);
}
}.execute();
} else {
SmaliAccessList accessListNode = findAccessListNode();
if (accessListNode == null) {
return;
}
ASTNode accessSpec = accessListNode.getAccessFlagNode(name);
final ASTNode accessSpec = accessListNode.getAccessFlagNode(name);
if (accessSpec != null) {
accessSpec.getPsi().delete();
new WriteCommandAction.Simple(getProject(), getContainingFile()) {
@Override protected void run() throws Throwable {
accessSpec.getPsi().delete();
}
}.execute();
}
}
}