1 files changed, 28 insertions, 2 deletions
diff --git a/app/services/packages/rpm/repository_metadata/base_builder.rb b/app/services/packages/rpm/repository_metadata/base_builder.rb
index 9d76336d764..2c0a11457ec 100644
--- a/app/services/packages/rpm/repository_metadata/base_builder.rb
+++ b/app/services/packages/rpm/repository_metadata/base_builder.rb
@@ -3,17 +3,43 @@ module Packages
   module Rpm
     module RepositoryMetadata
       class BaseBuilder
+        def initialize(xml: nil, data: {})
+          @xml = Nokogiri::XML(xml) if xml.present?
+          @data = data
+        end
+
         def execute
-          build_empty_structure
+          return build_empty_structure if xml.blank?
+
+          update_xml_document
+          update_package_count
+          xml.to_xml
         end
 
         private
 
+        attr_reader :xml, :data
+
         def build_empty_structure
           Nokogiri::XML::Builder.new(encoding: 'UTF-8') do |xml|
-            xml.public_send(self.class::ROOT_TAG, self.class::ROOT_ATTRIBUTES) # rubocop:disable GitlabSecurity/PublicSend
+            xml.method_missing(self.class::ROOT_TAG, self.class::ROOT_ATTRIBUTES)
           end.to_xml
         end
+
+        def update_xml_document
+          # Add to the root xml element a new package metadata node
+          xml.at(self.class::ROOT_TAG).add_child(build_new_node)
+        end
+
+        def update_package_count
+          packages_count = xml.css("//#{self.class::ROOT_TAG}/package").count
+
+          xml.at(self.class::ROOT_TAG).attributes["packages"].value = packages_count.to_s
+        end
+
+        def build_new_node
+          raise NotImplementedError, "#{self.class} should implement #{__method__}"
+        end
       end
     end
   end