Function to check if the device is supported by the image

Author: tsterbak

openandroidinstaller/openandroidinstaller.py

@@ -30,11 +30,11 @@
 from installer_config import InstallerConfig, Step, _load_config
 from loguru import logger
 from tool_utils import call_tool_with_command, search_device
-from utils import AppState, get_download_link
+from utils import AppState, get_download_link, image_recovery_works_with_device
 from widgets import call_button, confirm_button, get_title, link_button
 
 # Toggle to True for development purposes
-DEVELOPMENT = False
+DEVELOPMENT = True
 DEVELOPMENT_CONFIG = "yuga"  # "a3y17lte"  # "sargo"
 
 
@@ -250,7 +250,7 @@ def __init__(
 
     def build(self):
         self.download_link = get_download_link(
-            self.state.config.metadata.get("devicecode", "test")
+            self.state.config.metadata.get("devicecode", "ERROR")
         )
         self.confirm_button = confirm_button(self.on_confirm)
         self.confirm_button.disabled = True
@@ -264,6 +264,8 @@ def build(self):
         # add title and progressbar
         self.right_view.controls.append(get_title("Pick image and recovery files:"))
         self.right_view.controls.append(self.progressbar)
+        # text row to show infos during the process
+        self.info_field = Row()
         # if there is an available download, show the button to the page
         if self.download_link:
             self.right_view.controls.append(
@@ -323,15 +325,25 @@ def build(self):
                 ),
                 self.selected_recovery,
                 Divider(),
-                Text("If you selected both files you can continue."),
+                Text("If you selected both files and they work for your device you can continue."),
+                self.info_field,
                 Row([self.confirm_button]),
             ]
         )
         return self.view
 
     def enable_button_if_ready(self, e):
         """Enable the confirm button if both files have been selected."""
+
         if (".zip" in self.selected_image.value) and (".img" in self.selected_recovery.value):
+            if not image_recovery_works_with_device(
+                device_code=self.state.config.metadata.get("devicecode"), image_path=self.state.image_path
+                ):
+                # if image and recovery work for device allow to move on, otherwise display message
+                self.info_field.controls.append(Text("Image and recovery don't work with the device. Please select different ones."))
+                self.right_view.update()
+                return
+            self.info_field.controls = []
             self.confirm_button.disabled = False
             self.right_view.update()
         else:
@@ -465,6 +477,7 @@ def pick_image_result(self, e: FilePickerResultEvent):
         )
         if e.files:
             self.image_path = e.files[0].path
+            self.state.image_path = e.files[0].path
             logger.info(f"Selected image from {self.image_path}")
         else:
             logger.info("No image selected.")
@@ -476,6 +489,7 @@ def pick_recovery_result(self, e: FilePickerResultEvent):
         )
         if e.files:
             self.recovery_path = e.files[0].path
+            self.state.recovery_path = e.files[0].path
             logger.info(f"Selected recovery from {self.recovery_path}")
         else:
             logger.info("No image selected.")

openandroidinstaller/utils.py

@@ -15,6 +15,7 @@
 
 from typing import Optional
 
+import zipfile
 import requests
 from loguru import logger
 
@@ -38,6 +39,20 @@ def get_download_link(devicecode: str) -> Optional[str]:
         return
 
 
+def image_recovery_works_with_device(device_code: str, image_path: str) -> bool:
+    """Determine if an image and recovery works for the given device."""
+    with zipfile.ZipFile(image_path) as image_zip:
+        with image_zip.open("META-INF/com/android/metadata", mode="r") as image_metadata:
+            metadata = image_metadata.readlines()
+            supported_devices = str(metadata[-1]).split("=")[-1][:-3].split(",")
+            logger.info(f"Image works with device: {supported_devices}")
+
+            if device_code in supported_devices:
+                logger.info("Device supported by the image.")
+                return True
+    return False
+
+
 class AppState:
     """Container class to store all kinds of state."""