extract_utils: add support for generating boot jars list

Change-Id: I699a2165a2a9280fcbb175b7e7cec907b0e1e9df

Author: luk1337

extract_utils/file.py

@@ -43,6 +43,7 @@
 class FileArgs(str, Enum):
     AB = 'AB'
     CERTIFICATE = 'CERTIFICATE'
+    BOOT_JAR = 'BOOT_JAR'
     DUMMY_SHARED_LIB = 'DUMMY_SHARED_LIB'
     EXTRACT_ONLY = 'EXTRACT_ONLY'
     MAKE_COPY_RULE = 'MAKE_COPY_RULE'
@@ -65,6 +66,7 @@ class FileArgs(str, Enum):
 FILE_ARGS_TYPE_MAP = {
     FileArgs.AB: True,
     FileArgs.CERTIFICATE: str,
+    FileArgs.BOOT_JAR: True,
     FileArgs.DUMMY_SHARED_LIB: True,
     FileArgs.EXTRACT_ONLY: True,
     FileArgs.MAKE_COPY_RULE: True,
@@ -482,6 +484,7 @@ def __init__(
         # packages_files is a FileTree to help with performance while grouping
         # multiple files of the same type together
         self.package_files = FileTree()
+        self.boot_jars = FileTree()
         self.dummy_shared_libs = FileTree()
         self.package_symlinks = SimpleFileList()
         self.copy_files = SimpleFileList()
@@ -522,6 +525,9 @@ def __add_file(self, file: File, section: Optional[str]):
         if FileArgs.SYMLINK in file.args:
             self.package_symlinks.add(file)
 
+        if FileArgs.BOOT_JAR in file.args:
+            self.boot_jars.add(file)
+
         if FileArgs.DUMMY_SHARED_LIB in file.args:
             self.dummy_shared_libs.add(file)
             return

extract_utils/makefiles.py

@@ -434,6 +434,23 @@ def write_packages_inclusion(package_names: List[str], out: TextIO):
     out.write('\n')
 
 
+def write_boot_jars(
+    ctx: MakefilesCtx,
+    base_file_tree: FileTree,
+):
+    if not list(base_file_tree):
+        return
+
+    ctx.product_mk_out.write('\nPRODUCT_BOOT_JARS +=')
+
+    for file in base_file_tree:
+        _, package_name = file_stem_package_name(file, can_have_stem=True)
+        line = f' \\\n    {package_name}'
+        ctx.product_mk_out.write(line)
+
+    ctx.product_mk_out.write('\n')
+
+
 def write_product_packages(
     ctx: MakefilesCtx,
     packages_ctx: ProductPackagesCtx,

extract_utils/module.py

@@ -25,6 +25,7 @@
     MakefilesCtx,
     ProductPackagesCtx,
     write_board_info_file,
+    write_boot_jars,
     write_bp_header,
     write_bp_soong_namespaces,
     write_dummy_shared_libs,
@@ -210,6 +211,11 @@ def write_makefiles(self, module: ExtractUtilsModule, ctx: MakefilesCtx):
             self.file_list.package_symlinks,
         )
 
+        write_boot_jars(
+            ctx,
+            self.file_list.boot_jars,
+        )
+
         write_dummy_shared_libs(
             ctx,
             packages_ctx,