capacitor-dex-editor 0.0.59 → 0.0.61

package/android/src/main/java/com/aetherlink/dexeditor/DexEditorPluginPlugin.java

@@ -601,10 +601,32 @@ public class DexEditorPluginPlugin extends Plugin {
                 ));
                 break;
 
+            case "deleteFileFromApk":
+                result.put("data", dexManager.deleteFileFromApk(
+                    params.getString("apkPath"),
+                    params.getString("filePath")
+                ));
+                break;
+
+            case "addFileToApk":
+                result.put("data", dexManager.addFileToApk(
+                    params.getString("apkPath"),
+                    params.getString("filePath"),
+                    params.getString("content"),
+                    params.optBoolean("isBase64", false)
+                ));
+                break;
+
             case "listSessions":
                 result.put("data", dexManager.listAllSessions());
                 break;
 
+            case "precompileSmaliCache":
+                result.put("data", dexManager.precompileSmaliCache(
+                    params.getString("sessionId")
+                ));
+                break;
+
             // ==================== XML/资源操作 ====================
             case "getManifest":
                 result.put("data", dexManager.getManifestFromApk(

package/android/src/main/java/com/aetherlink/dexeditor/DexManager.java

@@ -151,6 +151,8 @@ public class DexManager {
         String apkPath;
         Map<String, DexBackedDexFile> dexFiles;
         Map<String, ClassDef> modifiedClasses;
+        Map<String, String> smaliCache;  // className -> smali 代码缓存
+        boolean smaliCacheReady = false;
         boolean modified = false;
 
         MultiDexSession(String sessionId, String apkPath) {
@@ -158,6 +160,7 @@ public class DexManager {
             this.apkPath = apkPath;
             this.dexFiles = new HashMap<>();
             this.modifiedClasses = new HashMap<>();
+            this.smaliCache = new HashMap<>();
         }
 
         void addDex(String dexName, DexBackedDexFile dexFile) {
@@ -1830,8 +1833,8 @@ public class DexManager {
                         
                     case "string":
                     case "code":
-                        // 需要反编译 smali 搜索
-                        String smali = getSmaliForClass(dexFile, classDef);
+                        // 使用缓存的 smali 搜索
+                        String smali = getCachedSmali(session, className, dexFile, classDef);
                         String smaliMatch = caseSensitive ? smali : smali.toLowerCase();
                         if (smaliMatch.contains(queryMatch)) {
                             JSObject item = new JSObject();
@@ -1853,8 +1856,8 @@ public class DexManager {
                         break;
                         
                     case "int":
-                        // 搜索整数常量
-                        String smaliForInt = getSmaliForClass(dexFile, classDef);
+                        // 搜索整数常量（使用缓存）
+                        String smaliForInt = getCachedSmali(session, className, dexFile, classDef);
                         if (smaliForInt.contains("0x" + query) || smaliForInt.contains(" " + query + "\n") || 
                             smaliForInt.contains(" " + query + " ")) {
                             JSObject item = new JSObject();
@@ -1893,6 +1896,62 @@ public class DexManager {
         }
     }
 
+    /**
+     * 获取缓存的 Smali 代码（用于搜索优化）
+     */
+    private String getCachedSmali(MultiDexSession session, String className, DexBackedDexFile dexFile, ClassDef classDef) {
+        // 先查缓存
+        String cached = session.smaliCache.get(className);
+        if (cached != null) {
+            return cached;
+        }
+        // 缓存未命中，反编译并缓存
+        String smali = getSmaliForClass(dexFile, classDef);
+        session.smaliCache.put(className, smali);
+        return smali;
+    }
+
+    /**
+     * 预编译会话中所有类的 Smali（用于加速搜索）
+     */
+    public JSObject precompileSmaliCache(String sessionId) throws Exception {
+        MultiDexSession session = multiDexSessions.get(sessionId);
+        if (session == null) {
+            throw new IllegalArgumentException("Session not found: " + sessionId);
+        }
+        
+        if (session.smaliCacheReady) {
+            JSObject result = new JSObject();
+            result.put("cached", true);
+            result.put("count", session.smaliCache.size());
+            return result;
+        }
+        
+        int count = 0;
+        long startTime = System.currentTimeMillis();
+        
+        for (Map.Entry<String, DexBackedDexFile> entry : session.dexFiles.entrySet()) {
+            DexBackedDexFile dexFile = entry.getValue();
+            for (ClassDef classDef : dexFile.getClasses()) {
+                String className = convertTypeToClassName(classDef.getType());
+                if (!session.smaliCache.containsKey(className)) {
+                    String smali = getSmaliForClass(dexFile, classDef);
+                    session.smaliCache.put(className, smali);
+                    count++;
+                }
+            }
+        }
+        
+        session.smaliCacheReady = true;
+        long elapsed = System.currentTimeMillis() - startTime;
+        
+        JSObject result = new JSObject();
+        result.put("success", true);
+        result.put("count", count);
+        result.put("elapsedMs", elapsed);
+        return result;
+    }
+
     /**
      * 从多 DEX 会话获取类的 Smali 代码
      */
@@ -2332,6 +2391,156 @@ public class DexManager {
         return result;
     }
 
+    /**
+     * 从 APK 中删除指定文件
+     */
+    public JSObject deleteFileFromApk(String apkPath, String filePath) throws Exception {
+        JSObject result = new JSObject();
+        
+        java.io.File apkFile = new java.io.File(apkPath);
+        java.io.File tempApkFile = new java.io.File(apkPath + ".tmp");
+        
+        java.util.zip.ZipInputStream zis = new java.util.zip.ZipInputStream(new java.io.FileInputStream(apkFile));
+        java.util.zip.ZipOutputStream zos = new java.util.zip.ZipOutputStream(new java.io.FileOutputStream(tempApkFile));
+        
+        java.util.zip.ZipEntry entry;
+        boolean found = false;
+        String normalizedPath = filePath.replaceFirst("^/+", "");
+        
+        while ((entry = zis.getNextEntry()) != null) {
+            String entryName = entry.getName();
+            
+            if (entryName.equals(filePath) || entryName.equals(normalizedPath)) {
+                // 跳过要删除的文件
+                found = true;
+                continue;
+            }
+            
+            // 复制其他文件
+            java.util.zip.ZipEntry newEntry = new java.util.zip.ZipEntry(entryName);
+            if (entry.getMethod() == java.util.zip.ZipEntry.STORED) {
+                newEntry.setMethod(java.util.zip.ZipEntry.STORED);
+                newEntry.setSize(entry.getSize());
+                newEntry.setCrc(entry.getCrc());
+            }
+            zos.putNextEntry(newEntry);
+            
+            byte[] buffer = new byte[8192];
+            int len;
+            while ((len = zis.read(buffer)) > 0) {
+                zos.write(buffer, 0, len);
+            }
+            zos.closeEntry();
+        }
+        
+        zis.close();
+        zos.close();
+        
+        if (!found) {
+            tempApkFile.delete();
+            result.put("success", false);
+            result.put("error", "文件未找到: " + filePath);
+            return result;
+        }
+        
+        // 替换原文件
+        if (!apkFile.delete()) {
+            tempApkFile.delete();
+            result.put("success", false);
+            result.put("error", "无法删除原 APK");
+            return result;
+        }
+        
+        if (!tempApkFile.renameTo(apkFile)) {
+            copyFile(tempApkFile, apkFile);
+            tempApkFile.delete();
+        }
+        
+        result.put("success", true);
+        result.put("message", "文件已删除: " + filePath);
+        result.put("needSign", true);
+        return result;
+    }
+
+    /**
+     * 向 APK 中添加或替换文件
+     */
+    public JSObject addFileToApk(String apkPath, String filePath, String content, boolean isBase64) throws Exception {
+        JSObject result = new JSObject();
+        
+        // 解码内容
+        byte[] contentBytes;
+        if (isBase64) {
+            contentBytes = android.util.Base64.decode(content, android.util.Base64.DEFAULT);
+        } else {
+            contentBytes = content.getBytes("UTF-8");
+        }
+        
+        java.io.File apkFile = new java.io.File(apkPath);
+        java.io.File tempApkFile = new java.io.File(apkPath + ".tmp");
+        
+        java.util.zip.ZipInputStream zis = new java.util.zip.ZipInputStream(new java.io.FileInputStream(apkFile));
+        java.util.zip.ZipOutputStream zos = new java.util.zip.ZipOutputStream(new java.io.FileOutputStream(tempApkFile));
+        
+        java.util.zip.ZipEntry entry;
+        String normalizedPath = filePath.replaceFirst("^/+", "");
+        boolean replaced = false;
+        
+        while ((entry = zis.getNextEntry()) != null) {
+            String entryName = entry.getName();
+            
+            if (entryName.equals(filePath) || entryName.equals(normalizedPath)) {
+                // 跳过要替换的文件，稍后添加新版本
+                replaced = true;
+                continue;
+            }
+            
+            // 复制其他文件
+            java.util.zip.ZipEntry newEntry = new java.util.zip.ZipEntry(entryName);
+            if (entry.getMethod() == java.util.zip.ZipEntry.STORED) {
+                newEntry.setMethod(java.util.zip.ZipEntry.STORED);
+                newEntry.setSize(entry.getSize());
+                newEntry.setCrc(entry.getCrc());
+            }
+            zos.putNextEntry(newEntry);
+            
+            byte[] buffer = new byte[8192];
+            int len;
+            while ((len = zis.read(buffer)) > 0) {
+                zos.write(buffer, 0, len);
+            }
+            zos.closeEntry();
+        }
+        
+        // 添加新文件
+        java.util.zip.ZipEntry newEntry = new java.util.zip.ZipEntry(normalizedPath);
+        newEntry.setSize(contentBytes.length);
+        zos.putNextEntry(newEntry);
+        zos.write(contentBytes);
+        zos.closeEntry();
+        
+        zis.close();
+        zos.close();
+        
+        // 替换原文件
+        if (!apkFile.delete()) {
+            tempApkFile.delete();
+            result.put("success", false);
+            result.put("error", "无法删除原 APK");
+            return result;
+        }
+        
+        if (!tempApkFile.renameTo(apkFile)) {
+            copyFile(tempApkFile, apkFile);
+            tempApkFile.delete();
+        }
+        
+        result.put("success", true);
+        result.put("message", replaced ? "文件已替换: " + filePath : "文件已添加: " + filePath);
+        result.put("needSign", true);
+        return result;
+    }
+
     /**
      * 保存多 DEX 会话的修改到 APK
      */

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "capacitor-dex-editor",
-  "version": "0.0.59",
+  "version": "0.0.61",
   "description": "Capacitor-plugin-for-editing-DEX-files-in-APK",
   "main": "dist/plugin.cjs.js",
   "module": "dist/esm/index.js",