打包时支持选择编译后APK的权限

Author: TonyJiangWJ

apkbuilder/build.gradle

@@ -28,7 +28,6 @@ dependencies {
     })
     testImplementation 'junit:junit:4.13.2'
     api fileTree(dir: 'libs', include: ['*.jar'])
-    api files('libs/tiny-sign-0.9.jar') 
     api files('libs/commons-io-2.5.jar')
     implementation "androidx.core:core-ktx:1.8.0"
 }

apkbuilder/libs/tiny-sign-0.9.jar

@@ -13,8 +13,6 @@
 import java.util.zip.ZipEntry;
 import java.util.zip.ZipInputStream;
 
-import pxb.android.tinysign.TinySign;
-
 /**
  * Created by Stardust on 2017/10/23.
  */
@@ -52,12 +50,6 @@ public void unzip() throws IOException {
         }
     }
 
-    public void repackage(String newApkPath) throws Exception {
-        FileOutputStream fos = new FileOutputStream(newApkPath);
-        TinySign.sign(new File(mWorkspacePath), fos);
-        fos.close();
-    }
-
     public void cleanWorkspace() {
 
     }

apkbuilder/src/main/java/com/stardust/autojs/apkbuilder/ApkBuilder.java

@@ -1,5 +1,7 @@
 package com.stardust.autojs.apkbuilder;
 
+import android.util.Log;
+
 import org.apache.commons.io.IOUtils;
 
 import java.io.IOException;
@@ -94,28 +96,44 @@ public void onAttr(AxmlWriter.Attr attr) {
         }
     }
 
+    public boolean filterPermission(String permissionName) {
+        return true;
+    }
+
 
-    private class MutableAxmlWriter extends AxmlWriter {
-        private class MutableNodeImpl extends AxmlWriter.NodeImpl {
+    public class MutableAxmlWriter extends AxmlWriter {
+        public class MutableNodeImpl extends AxmlWriter.NodeImpl {
+            private boolean ignore;
+            private final String name;
 
             MutableNodeImpl(String ns, String name) {
                 super(ns, name);
+                this.name = name;
             }
 
             @Override
             protected void onAttr(AxmlWriter.Attr a) {
+                if ("uses-permission".equals(this.name) && "name".equals(a.name.data) && a.value instanceof StringItem) {
+                    if (ManifestEditor.this.filterPermission(((StringItem) a.value).data)) {
+                        ignore = true;
+                    }
+                }
                 ManifestEditor.this.onAttr(a);
                 super.onAttr(a);
             }
 
 
             @Override
             public NodeVisitor child(String ns, String name) {
+                Log.d("MutableAxmlWriter", "child: " + ns + " name: " + name);
                 NodeImpl child = new MutableNodeImpl(ns, name);
                 this.children.add(child);
                 return child;
             }
 
+            public boolean isIgnore() {
+                return ignore;
+            }
         }
 
         @Override

apkbuilder/src/main/java/com/stardust/autojs/apkbuilder/ApkPackager.java

@@ -15,6 +15,10 @@
  */
 package pxb.android.axml;
 
+import android.util.Log;
+
+import com.stardust.autojs.apkbuilder.ManifestEditor;
+
 import pxb.android.StringItem;
 import pxb.android.StringItems;
 
@@ -31,6 +35,7 @@
  * @author <a href="mailto:[email protected]">Panxiaobo</a>
  */
 public class AxmlWriter extends AxmlVisitor {
+    private static final String TAG = "AxmlWriter";
     static final Comparator<Attr> ATTR_CMP = new Comparator<Attr>() {
 
         @Override
@@ -91,6 +96,12 @@ private int prepare() throws IOException {
         int size = 0;
 
         for (NodeImpl first : firsts) {
+            if (first instanceof ManifestEditor.MutableAxmlWriter.MutableNodeImpl) {
+                if (((ManifestEditor.MutableAxmlWriter.MutableNodeImpl)first).isIgnore()) {
+                    Log.d(TAG, "prepare: first is ignore: " + first.name);
+                    continue;
+                }
+            }
             size += first.prepare(this);
         }
         {
@@ -162,6 +173,12 @@ public byte[] toByteArray() throws IOException {
         }
 
         for (NodeImpl first : firsts) {
+            if (first instanceof ManifestEditor.MutableAxmlWriter.MutableNodeImpl) {
+                if (((ManifestEditor.MutableAxmlWriter.MutableNodeImpl)first).isIgnore()) {
+                    Log.d(TAG, "toByteArray: first is ignore: " + first.name);
+                    continue;
+                }
+            }
             first.write(out);
         }
 
@@ -342,6 +359,12 @@ public int prepare(AxmlWriter axmlWriter) {
             int size = 24 + 36 + attrs.size() * 20;// 24 for end tag,36+x*20 for
             // start tag
             for (NodeImpl child : children) {
+                if (child instanceof ManifestEditor.MutableAxmlWriter.MutableNodeImpl) {
+                    if (((ManifestEditor.MutableAxmlWriter.MutableNodeImpl)child).isIgnore()) {
+                        Log.d(TAG, "prepare: child is ignore: " + child.name);
+                        continue;
+                    }
+                }
                 size += child.prepare(axmlWriter);
             }
             if (text != null) {
@@ -398,6 +421,12 @@ void write(ByteBuffer out) throws IOException {
 
             // children
             for (NodeImpl child : children) {
+                if (child instanceof ManifestEditor.MutableAxmlWriter.MutableNodeImpl) {
+                    if (((ManifestEditor.MutableAxmlWriter.MutableNodeImpl)child).isIgnore()) {
+                        Log.d(TAG, "white: child is ignore: " + child.name);
+                        continue;
+                    }
+                }
                 child.write(out);
             }
 

apkbuilder/src/main/java/com/stardust/autojs/apkbuilder/ManifestEditor.java

@@ -29,7 +29,9 @@
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Collections;
+import java.util.HashSet;
 import java.util.List;
+import java.util.Set;
 import java.util.concurrent.Callable;
 import java.util.concurrent.atomic.AtomicBoolean;
 
@@ -74,6 +76,7 @@ public static class AppConfig {
         Boolean usePaddleOcr = false;
         Boolean useMlKitOcr = false;
         Boolean useTessTwo = false;
+        Set<String> enabledPermission = new HashSet<>();
 
         public static AppConfig fromProjectConfig(String projectDir, ProjectConfig projectConfig) {
             String icon = projectConfig.getIcon();
@@ -183,6 +186,14 @@ public Boolean getUseTessTwo() {
         public void setUseTessTwo(Boolean useTessTwo) {
             this.useTessTwo = useTessTwo;
         }
+
+        public Set<String> getEnabledPermission() {
+            return enabledPermission;
+        }
+
+        public void setEnabledPermission(Set<String> enabledPermission) {
+            this.enabledPermission = enabledPermission;
+        }
     }
 
     private ProgressCallback mProgressCallback;
@@ -550,7 +561,13 @@ public void onAttr(AxmlWriter.Attr attr) {
             } else {
                 super.onAttr(attr);
             }
+        }
 
+        @Override
+        public boolean filterPermission(String permissionName) {
+            Log.d(TAG, "filterPermission: " + permissionName);
+            boolean persist = mAppConfig.getEnabledPermission().contains(permissionName);
+            return !persist;
         }
     }
 }