Fixes #38267 - rake ovirt:drop

lib/tasks/ovirt.rake

@@ -0,0 +1,122 @@
+namespace :ovirt do
+  desc <<~END_DESC
+    Remove all oVirt / RHV compute resources & profiles from Foreman.
+    Hosts associated with these compute resources will be disassociated and kept.
+    Host groups with oVirt compute resources & profiles will have their compute resource and compute profile removed.
+    Compute profiles, images, and other resources will be removed.
+
+    Use OVIRT_PRETEND=true to simulate the operation without making any changes.
+    Use OVIRT_ASSUME_YES=true to skip the confirmation prompt.
+
+    Examples:
+      # foreman-rake ovirt:drop
+      # OVIRT_ASSUME_YES=true foreman-rake ovirt:drop
+      # OVIRT_PRETEND=true foreman-rake ovirt:drop
+  END_DESC
+
+  task :drop => :environment do
+    compute_resources = ComputeResource.unscoped.where(type: 'Foreman::Model::Ovirt')
+
+    nothing_to_do if compute_resources.empty?
+
+    pretend_mode = Foreman::Cast.to_bool(ENV['OVIRT_PRETEND'])
+    assume_yes = Foreman::Cast.to_bool(ENV['OVIRT_ASSUME_YES'])
+
+    if pretend_mode
+      User.as_anonymous_admin do
+        run_in_pretend_mode(compute_resources)
+      end
+    end
+
+    ask_user(compute_resources) unless assume_yes
+
+    User.as_anonymous_admin do
+      run_for_real(compute_resources)
+    end
+
+    the_end
+  end
+
+  def run_in_pretend_mode(compute_resources)
+    puts ''
+    puts "Running in pretend mode. Following actions would be performed:"
+
+    compute_resources.each do |cr|
+      puts "Remove '#{cr.name} [#{cr.id}]' compute resource"
+
+      puts "\nDissacociate the following hosts:"
+      cr.hosts.each do |host|
+        puts "- #{host.name} [#{host.id}]"
+      end
+
+      puts "\nUpdate the following host groups:"
+      cr.hostgroups.each do |hg|
+        puts "- '#{hg.name}' [#{hg.id}]"
+      end
+
+      puts "\nRemove the following compute profiles:"
+      cr.compute_profiles.each do |cp|
+        puts "- '#{cp.name}' [#{cp.id}]"
+      end
+
+      puts "\nRemove the following images:"
+      cr.images.each do |image|
+        puts "- '#{image.name}' [#{image.id}]"
+      end
+    end
+
+    puts ""
+    exit 0
+  end
+
+  def run_for_real(compute_resources)
+    puts 'Starting Ovirt drop ...'
+
+    compute_resources.each do |cr|
+      puts "Processing data for '#{cr.name}' compute resource."
+
+      cr.hosts.each do |host|
+        puts "Disassociating host '#{host.name}' from the compute resource."
+        host.disassociate!
+      end
+
+      puts "Removing compute resources and compute profiles from host groups."
+      cr.hostgroups.update_all(compute_resource_id: nil, compute_profile_id: nil)
+
+      puts "Removing '#{cr.name}' compute resource."
+      cr.destroy
+    end
+  end
+
+  def ask_user(compute_resources)
+    puts "!!! WARNING !!! This operation is irreversible."
+    puts "Affected compute resource(s): #{compute_resources.map(&:name).join(', ')}."
+    puts ""
+    puts " - This task will remove all oVirt / RHV compute resources from Foreman."
+    puts " - All hosts associated with these compute resources will be disassociated and kept."
+    puts " - All host groups with compute resources will have their compute resource and compute profile removed."
+    puts " - Compute profiles, images and other resources associated with the compute resources will be removed."
+    puts ""
+    puts "Are you sure you want to continue? (yes/no)"
+
+    answer = $stdin.gets.chomp.downcase
+
+    unless answer == "yes"
+      puts "Aborting task."
+      exit 1
+    end
+  end
+
+  def nothing_to_do
+    puts "No 'Foreman::Model::Ovirt' compute resources found."
+    puts "Nothing to do, exiting."
+    exit 0
+  end
+
+  def the_end
+    puts ''
+    puts 'oVirt cleanup was successfully completed.'
+    puts 'Exiting.'
+    exit 0
+  end
+end