ManageIQ/manageiq

$:.push(File.dirname(__FILE__))
require 'evm_application'
require 'evm_rake_helper'

namespace :evm do
  namespace :foreman do
    task :start => [:environment, 'db:seed'] do
      server = MiqServer.my_server

      # Assign and activate the default roles
      server.ensure_default_roles
      server.activate_roles(server.server_role_names)

      # Mark the server as started
      server.update(:status => "started")

      # start the workers using foreman
      exec("foreman start --port=3000")
    end
  end

  desc "Start the ManageIQ EVM Application"
  task :start => :environment do
    EvmApplication.start
  end

  desc "Restart the ManageIQ EVM Application"
  task :restart => :environment do
    EvmApplication.stop
    EvmApplication.start
  end

  desc "Stop the ManageIQ EVM Application"
  task :stop => :environment do
    EvmApplication.stop
  end

  desc "Kill the ManageIQ EVM Application"
  task :kill => :environment do
    EvmApplication.kill
  end

  desc "Report Status of the ManageIQ EVM Application"
  task :status => :environment do
    EvmApplication.status
  end

  desc "Report Status of the ManageIQ EVM Application"
  task :status_full => :environment do
    EvmApplication.status(true)
  end

  desc "Describe inventory of the ManageIQ EVM Application"
  task :inventory => :environment do
    inventory = ExtManagementSystem.inventory_status
    puts inventory.tableize if inventory.present?
  end

  desc "Report overview of queue"
  task :queue => :environment do
    EvmApplication.queue_overview
  end

  desc "Determine if the configured encryption key is valid"
  task :validate_encryption_key => :environment do
    raise "Invalid encryption key" unless EvmApplication.encryption_key_valid?

    puts "Encryption key valid"
  end

  desc "Write a remote region id to this server's REGION file"
  task :join_region => :environment do
    configured_region = ApplicationRecord.region_number_from_sequence.to_i
    EvmApplication.set_region_file(Rails.root.join("REGION"), configured_region)
  end

  desc "Determine the deployment scenario"
  task :deployment_status => :environment do
    status_to_code = {
      "new_deployment" => 3,
      "new_replica"    => 4,
      "redeployment"   => 5,
      "upgrade"        => 6,
      "no_database"    => 7
    }

    status = EvmApplication.deployment_status
    if status == "new_replica"
      puts "Database contains the following server records:"
      MiqServer.all.each { |server| puts "  #{server.guid}: id: #{server.id}, name: #{server.name}, zone: #{server.zone&.description}" }
    end

    puts "Deployment status is #{status}"
    exit status_to_code[status]
  end

  task :compile_assets => 'evm:assets:compile'
  namespace :assets do
    desc "Compile assets (clobber and precompile)"
    task :compile do
      EvmRakeHelper.with_dummy_database_url_configuration do
        Rake::Task["assets:clobber"].invoke
        Rake::Task["assets:precompile"].invoke
      end
    end
  end

  desc "Compile STI inheritance relationship cache"
  task :compile_sti_loader do
    EvmRakeHelper.with_dummy_database_url_configuration do
      Rake::Task["environment"].invoke
      DescendantLoader.instance.class_inheritance_relationships
    end
  end

  # Example usage:
  #  bin/rake evm:raise_server_event -- --event db_failover_executed
  desc 'Raise evm event'
  task :raise_server_event => :environment do
    require 'optimist'
    opts = Optimist.options(EvmRakeHelper.extract_command_options) do
      opt :event, "Server Event", :type => :string, :required => true
    end
    EvmDatabase.raise_server_event(opts[:event])
  end
end