Execute commands on waydroid shell with pipes

tools/helper.py

@@ -8,13 +8,19 @@ import requests
 from tools.logger import Logger
 from tqdm import tqdm
 import hashlib
+from typing import Optional
+
 
 def get_download_dir():
     download_loc = ""
     if os.environ.get("XDG_CACHE_HOME", None) is None:
-        download_loc = os.path.join('/', "home", os.environ.get("SUDO_USER", os.environ["USER"]), ".cache", "waydroid-script", "downloads")
+        download_loc = os.path.join('/', "home", os.environ.get(
+            "SUDO_USER", os.environ["USER"]), ".cache", "waydroid-script", "downloads"
+        )
     else:
-        download_loc = os.path.join(os.environ["XDG_CACHE_HOME"], "waydroid-script", "downloads")
+        download_loc = os.path.join(
+            os.environ["XDG_CACHE_HOME"], "waydroid-script", "downloads"
+        )
     if not os.path.exists(download_loc):
         os.makedirs(download_loc)
     return download_loc
@@ -23,21 +29,65 @@ def get_download_dir():
 def get_data_dir():
     return os.path.join('/', "home", os.environ.get("SUDO_USER", os.environ["USER"]), ".local", "share", "waydroid", "data")
 
-def run(args, env=None, ignore=""):
-    result = subprocess.run(args=args, env=env, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+# execute on host
+def run(args: list, env: Optional[str] = None, ignore: Optional[str] = None):
+    result = subprocess.run(
+        args=args, 
+        env=env, 
+        stdout=subprocess.PIPE, 
+        stderr=subprocess.PIPE
+    )
+
     # print(result.stdout.decode())
     if result.stderr:
         error = result.stderr.decode("utf-8")
-        if re.match(ignore, error):
+        if ignore and re.match(ignore, error):
             return result
         Logger.error(error)
         raise subprocess.CalledProcessError(
-                    returncode = result.returncode,
-                    cmd = result.args,
-                    stderr = result.stderr
+            returncode=result.returncode,
+            cmd=result.args,
+            stderr=result.stderr
         )
     return result
 
+# execute on waydroid shell
+def shell(arg: str, env: Optional[str] = None):
+    a = subprocess.Popen(
+        args=["sudo", "waydroid", "shell"],
+        stdin=subprocess.PIPE,
+        stdout=subprocess.PIPE,
+        stderr=subprocess.PIPE
+    )
+    subprocess.Popen(
+        args=["echo", "export BOOTCLASSPATH=/apex/com.android.art/javalib/core-oj.jar:/apex/com.android.art/javalib/core-libart.jar:/apex/com.android.art/javalib/core-icu4j.jar:/apex/com.android.art/javalib/okhttp.jar:/apex/com.android.art/javalib/bouncycastle.jar:/apex/com.android.art/javalib/apache-xml.jar:/system/framework/framework.jar:/system/framework/ext.jar:/system/framework/telephony-common.jar:/system/framework/voip-common.jar:/system/framework/ims-common.jar:/system/framework/framework-atb-backward-compatibility.jar:/apex/com.android.conscrypt/javalib/conscrypt.jar:/apex/com.android.media/javalib/updatable-media.jar:/apex/com.android.mediaprovider/javalib/framework-mediaprovider.jar:/apex/com.android.os.statsd/javalib/framework-statsd.jar:/apex/com.android.permission/javalib/framework-permission.jar:/apex/com.android.sdkext/javalib/framework-sdkextensions.jar:/apex/com.android.wifi/javalib/framework-wifi.jar:/apex/com.android.tethering/javalib/framework-tethering.jar"],
+        stdout=a.stdin,
+        stdin=subprocess.PIPE
+    ).communicate()
+
+    if env:
+        subprocess.Popen(
+            args=["echo", env],
+            stdout=a.stdin,
+            stdin=subprocess.PIPE
+        ).communicate()
+
+    subprocess.Popen(
+        args=["echo", arg],
+        stdout=a.stdin,
+        stdin=subprocess.PIPE
+    ).communicate()
+
+    a.stdin.close()
+    if a.stderr.read():
+        Logger.error(a.stderr.read().decode('utf-8'))
+        raise subprocess.CalledProcessError(
+            returncode=a.returncode,
+            cmd=a.args,
+            stderr=a.stderr
+        )
+    return a.stdout.read().decode("utf-8")
+
 def download_file(url, f_name):
     md5 = ""
     response = requests.get(url, stream=True)