ManageIQ/manageiq-appliance_console

require 'manageiq/appliance_console/postgres_admin'
require 'fileutils'
require 'linux_admin'

module ManageIQ
module ApplianceConsole
  class DatabaseReplicationStandby < DatabaseReplication
    include ManageIQ::ApplianceConsole::Logging

    REGISTER_CMD    = 'repmgr standby register'.freeze

    attr_accessor :disk, :standby_host, :run_repmgrd_configuration, :resync_data, :force_register

    def initialize
      self.node_number       = nil
      self.database_name     = "vmdb_production"
      self.database_user     = "root"
      self.database_password = nil
      self.primary_host      = nil
      self.standby_host      = LinuxAdmin::NetworkInterface.new(NETWORK_INTERFACE).address
      self.resync_data       = false
    end

    def ask_questions
      clear_screen
      say("Establish Replication Standby Server\n")
      return false if !data_dir_empty? && !confirm_data_resync
      self.disk = ask_for_disk("Standby database disk")
      ask_for_unique_cluster_node_number
      ask_for_database_credentials
      ask_for_standby_host
      ask_for_repmgrd_configuration
      return false unless node_number_valid?
      return false if repmgr_configured? && !confirm_reconfiguration
      confirm
    end

    def confirm
      super
      say(<<-EOS) if disk
        Database Disk:              #{disk.path}
      EOS
      say(<<-EOS)
        Standby Host:               #{standby_host}
        Automatic Failover:         #{run_repmgrd_configuration ? "enabled" : "disabled"}
      EOS
      agree("Apply this Replication Server Configuration? (Y/N): ")
    end

    def ask_for_standby_host
      self.standby_host = ask_for_ip_or_hostname("Standby Server hostname or IP address", standby_host)
    end

    def ask_for_repmgrd_configuration
      self.run_repmgrd_configuration = ask_yn?("Configure Replication Manager (repmgrd) for automatic failover")
    end

    def activate
      say("Configuring Replication Standby Server...")
      stop_postgres
      stop_repmgrd
      initialize_postgresql_disk if disk
      PostgresAdmin.prep_data_directory if disk || resync_data
      relabel_postgresql_dir
      save_database_yml
      create_config_file(standby_host) &&
        write_pgpass_file &&
        clone_standby_server &&
        start_postgres &&
        register_standby_server &&
        (run_repmgrd_configuration ? start_repmgrd : true)
    end

    def data_dir_empty?
      Dir[PostgresAdmin.data_directory.join("*")].empty?
    end

    def confirm_data_resync
      logger.info("Appliance database found under: #{PostgresAdmin.data_directory}")
      say("")
      say("Appliance database found under: #{PostgresAdmin.data_directory}")
      say("Replication standby server can not be configured if the database already exists")
      say("Would you like to remove the existing database before configuring as a standby server?")
      say("  WARNING: This is destructive. This will remove all previous data from this server")
      self.resync_data = ask_yn?("Continue")
    end

    def clone_standby_server
      params = { :h  => primary_host,
                 :U  => database_user,
                 :d  => database_name,
                 :D  => PostgresAdmin.data_directory,
                 nil => %w(standby clone)
               }
      run_repmgr_command("repmgr", params)
    end

    def start_postgres
      LinuxAdmin::Service.new(PostgresAdmin.service_name).enable.start
      true
    end

    def stop_postgres
      LinuxAdmin::Service.new(PostgresAdmin.service_name).stop
      true
    end

    def register_standby_server
      run_repmgr_command(REGISTER_CMD, :force => nil, :wait_sync= => 60)
    end

    def start_repmgrd
      LinuxAdmin::Service.new(repmgr_service_name).enable.start
      true
    rescue AwesomeSpawn::CommandResultError => e
      message = "Failed to start repmgrd: #{e.message}"
      logger.error(message)
      say(message)
      false
    end

    def stop_repmgrd
      LinuxAdmin::Service.new(repmgr_service_name).stop
      true
    end

    def node_number_valid?
      rec = record_for_node_number

      return true if rec.nil?
      node_state = rec["active"] ? "active" : "inactive"

      say("An #{node_state} #{rec["type"]} node (#{rec["node_name"]}) with the node number #{node_number} already exists")
      ask_yn?("Would you like to continue configuration by overwriting the existing node", "N")

    rescue PG::Error => e
      error_msg = "Failed to validate node number #{node_number}. #{e.message}"
      say(error_msg)
      logger.error(error_msg)
      return false
    end

    private

    def save_database_yml
      InternalDatabaseConfiguration.new(:password => database_password).save
    end

    def record_for_node_number
      c = PG::Connection.new(primary_connection_hash)
      c.exec_params(<<-SQL, [node_number]).map_types!(PG::BasicTypeMapForResults.new(c)).first
        SELECT type, node_name, active
        FROM repmgr.nodes where node_id = $1
      SQL
    end

    def initialize_postgresql_disk
      log_and_feedback(__method__) do
        LogicalVolumeManagement.new(:disk                => disk,
                                    :mount_point         => PostgresAdmin.mount_point,
                                    :name                => "pg",
                                    :volume_group_name   => PostgresAdmin.volume_group_name,
                                    :filesystem_type     => PostgresAdmin.database_disk_filesystem,
                                    :logical_volume_path => PostgresAdmin.logical_volume_path).setup

        # if we mounted the disk onto the postgres user's home directory, fix the permissions
        if PostgresAdmin.mount_point.to_s == "/var/lib/pgsql"
          FileUtils.chown(PostgresAdmin.user, PostgresAdmin.group, "/var/lib/pgsql")
          FileUtils.chmod(0o700, "/var/lib/pgsql")
        end
      end
    end

    def relabel_postgresql_dir
      AwesomeSpawn.run!("/sbin/restorecon -R -v #{PostgresAdmin.mount_point}")
    end
  end # class DatabaseReplicationStandby < DatabaseReplication
end # module ApplianceConsole
end