cmd-koji-upload: Refactor get_rpm_list

 - Refactor get_rpm_list to allow packages from the
new meta.json and extensions.json from the node-image

Signed-off-by: Renata Ravanelli <rravanel@redhat.com>

Author: ravanelli

src/cmd-koji-upload

@@ -105,7 +105,8 @@ class Build(_Build):
         # new kind of image here, but _Build is geared towards that and has a
         # image_name_base() member that wants a platform string to name stuff
         self.platform = "koji"
-        self._tmpdir = tempfile.mkdtemp(prefix="koji-build", dir=kwargs['buildroot'])
+        self.buildroot = kwargs['buildroot']
+        self._tmpdir = tempfile.mkdtemp(prefix="koji-build", dir=self.buildroot)
         self._state_file_tpl = self._tmpdir + "/cosa-cmd-koji-upload-{name}-{version}-{release}"
         kwargs.update({
             "require_commit": True,
@@ -116,7 +117,7 @@ class Build(_Build):
         self.name = f"{self.build_name}-{self.basearch}"
         self.source = self.get_meta_key(
             "meta", self.ckey("container-config-git"))
-        self.host_rpms = self.get_rpm_list('host')
+        self.host_rpms = self.get_rpm_list(host=True)
     def __del__(self):
         try:
             shutil.rmtree(self._tmpdir)
@@ -195,20 +196,41 @@ class Build(_Build):
 
         return fname, True
 
-    def get_rpm_list(self, host=None):
+    def get_rpm_list(self, **kwargs):
         """
         Translate commitmeta.json/HOST OS rpms into a json list
         Returns the json rpms list
         """
         components = []
-        if host is None:
+        if "meta" in kwargs:
+            file_path = f"{self.buildroot}/meta.json"
+            with open(file_path) as f:
+                data = json.load(f)
+                rpms = data['rpmdb.pkglist']
+        if "extensions" in kwargs:
+            file_path = f"{self.buildroot}/extensions.json"
+            with open(file_path) as f:
+                 data = json.load(f)
+                 rpms = []
+                 # Extensions in node image are stored different
+                 for name, full_version in data.items():
+                     version_release, arch = full_version.rsplit('.', 1)
+                     version_str, release = version_release.rsplit('-', 1)
+                     if '.' in version_str:
+                         epoch, version = version_str.split('.', 1)
+                     else:
+                         epoch = '0'
+                         version = version_str
+                     rpm_list = [name, epoch, version, release, arch]
+                     rpms.append(rpm_list)
+        if "commitmeta" in kwargs:
             rpms = self.commit["rpmostree.rpmdb.pkglist"]
-        else:
+        if "host" in kwargs:
             host_rpms = subprocess.check_output('rpm -qa --qf="%{NAME}:%{EPOCH}:%{RELEASE}:%{VERSION}:%{ARCH}:%{SIGMD5}:%{SIGPGP} \n"', shell=True).strip()
             rpms = (host_rpms.decode('utf-8')).split("\n")
 
         for rpm in rpms:
-            if host is None:
+            if any(key in kwargs for key in ("meta", "commitmeta", "extensions")):
                 name, epoch, version, release, arch = rpm
                 sigmd5, sigpgp, epoch = None, None, None
             else:
@@ -648,6 +670,10 @@ class Upload(_KojiBase):
 
     @property
     def image_files(self):
+         """ Backward-compatible property to maintain existing behavior. """
+         return self.get_image_files(meta=False)
+
+    def get_image_files(self, meta=False):
         """ Generate outputs prepares the output listing. """
         if self._image_files is not None:
             return self._image_files
@@ -657,7 +683,11 @@ class Upload(_KojiBase):
             file_output = self.get_file_meta(value)
             if file_output is not None:
                 if "commitmeta.json" in value['upload_path']:
-                    file_output["components"] = self.build.get_rpm_list()
+                    file_output["components"] = self.build.get_rpm_list(commitmeta=True)
+                if "meta.json" in value['upload_path'] and meta is not False:
+                    file_output["components"] = self.build.get_rpm_list(meta=True)
+                if "extensions.json" in value['upload_path'] and meta is not False:
+                    file_output["components"] = self.build.get_rpm_list(extensions=True)
                 outputs.append(file_output)
         self._image_files = outputs
         return self._image_files