From 27fbb5aa423ea8c3c4b9a7e4e8a585796053ff09 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Laure=CE=B7t?= <laurentfainsin@protonmail.com>
Date: Sat, 19 Mar 2022 13:09:56 +0100
Subject: [PATCH] feat: basic script to connect, detect and launch the game

---
 src/gen.py | 87 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 87 insertions(+)
 create mode 100644 src/gen.py

diff --git a/src/gen.py b/src/gen.py
new file mode 100644
index 0000000..22d9b62
--- /dev/null
+++ b/src/gen.py
@@ -0,0 +1,87 @@
+from ppadb.client import Client as AdbClient
+import re
+import logging
+
+REGEX_USER = re.compile(r"UserInfo\{([0-9]*):([a-zA-Z ]*):.*")
+PSYONIX_PACKAGE_NAME = "com.Psyonix.RL2D"
+PSYONIX_ACTIVITY_NAME = "com.epicgames.ue4.SplashActivity"
+
+ADB_HOST = "127.0.0.1"
+ADB_PORT = 5037
+
+logging.basicConfig(
+    level=logging.DEBUG,
+    format="%(asctime)s %(name)s %(levelname)-8s  %(message)s",
+    datefmt="(%F %T)",
+)
+
+
+def connect_adb():
+    client = AdbClient(host=ADB_HOST, port=ADB_PORT)
+    devices = client.devices()
+    devices_names = list(map(lambda d: d.serial, devices))
+
+    logging.debug(f"devices detected: {devices_names}")
+
+    if len(devices) == 1:
+        device = devices[0]
+    elif len(devices) == 0:
+        print("No device connected")
+        exit(1)
+    else:
+        print("Select a device :")
+        # TODO
+
+    logging.debug(f"device selected: {device.serial}")
+
+    return device
+
+
+def get_users(device):
+    device_users = device.shell("pm list users")
+    users = REGEX_USER.findall(device_users)
+
+    logging.debug(f"users detected: {users}")
+    return users
+
+
+def detect_game(device, users):
+    users = [
+        (id, name) for (id, name) in users if "package:" in device.shell(f"pm path --user {id} {PSYONIX_PACKAGE_NAME}")
+    ]
+
+    logging.debug(f"playable users: {users}")
+    return users
+
+
+def start_game(device, users):
+    if len(users) == 1:
+        (user_id, user_name) = users[0]
+    elif len(users) == 0:
+        print("No Playable users detected")
+        exit(1)
+    else:
+        print("Select a user to game with :")
+        # TODO
+
+    logging.debug(f"user selected: {user_name}")
+    device.shell(f"am start --user {user_id} -n {PSYONIX_PACKAGE_NAME}/{PSYONIX_ACTIVITY_NAME}")
+    logging.debug("game activity started")
+
+
+def disable_notifications(device):
+    device.shell("adb shell settings put global heads_up_notifications_enabled 0")
+    logging.debug("disabled heads up notificaiton")
+
+
+if __name__ == "__main__":
+
+    device = connect_adb()
+
+    users = get_users(device)
+
+    users = detect_game(device, users)
+
+    start_game(device, users)
+
+    disable_notifications(device)