build: include size-limits to device-metadata

[openwrt/staging/blocktrron.git] / scripts / json_add_image_info.py

diff --git a/scripts/json_add_image_info.py b/scripts/json_add_image_info.py
index 915e5f61812578ec9e4e92c5aead2da190a7d8b5..f4b3336e3269d99cb96c5ac7e86051ffe86e6b9e 100755 (executable)
--- a/scripts/json_add_image_info.py
+++ b/scripts/json_add_image_info.py
@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
 
-from os import getenv
+from os import getenv, path
 from pathlib import Path
 from sys import argv
 import hashlib
@@ -35,6 +35,17 @@ def get_titles():
     return titles
 
 
+def get_numerical_size(image_size):
+    if image_size.endswith("g"):
+        return int(image_size[:-1]) * 1024 * 1024 * 1024
+    elif image_size.endswith("m"):
+        return int(image_size[:-1]) * 1024 * 1024
+    elif image_size.endswith("k"):
+        return int(image_size[:-1]) * 1024
+    else:
+        return int(image_size)
+
+
 device_id = getenv("DEVICE_ID")
 
 sha256_hash = hashlib.sha256()
@@ -52,6 +63,8 @@ if file_path.with_suffix(file_path.suffix + ".sha256sum").exists():
 else:
     hash_unsigned = hash_file
 
+file_size = path.getsize(file_path)
+
 file_info = {
     "metadata_version": 1,
     "target": "{}/{}".format(getenv("TARGET"), getenv("SUBTARGET")),
@@ -67,6 +80,7 @@ file_info = {
                     "name": getenv("FILE_NAME"),
                     "sha256": hash_file,
                     "sha256_unsigned": hash_unsigned,
+                    "size": file_size,
                 }
             ],
             "device_packages": getenv("DEVICE_PACKAGES").split(),
@@ -76,6 +90,17 @@ file_info = {
     },
 }
 
+if getenv("IMAGE_SIZE") or getenv("KERNEL_SIZE"):
+    file_info["profiles"][device_id]["limits"] = {}
+    if getenv("IMAGE_SIZE"):
+        file_info["profiles"][device_id]["limits"]["image_size"] = get_numerical_size(
+            getenv("IMAGE_SIZE")
+        )
+    if getenv("KERNEL_SIZE"):
+        file_info["profiles"][device_id]["limits"]["kernel_size"] = get_numerical_size(
+            getenv("KERNEL_SIZE")
+        )
+
 if getenv("FILE_FILESYSTEM"):
     file_info["profiles"][device_id]["images"][0]["filesystem"] = getenv(
         "FILE_FILESYSTEM"