redbubble/kumo_dockercloud_gem

require 'csv'

module KumoDockerCloud
  class HaproxyContainer
    def initialize(container_id, client)
      @container_id = container_id
      @client = client
    end

    def disable_server(server_name)
      haproxy_server_name = haproxy_server_name(server_name)
      HaproxyCommand.new(@container_id, @client).execute("disable server #{haproxy_server_name}")
    end

    def enable_server(server_name)
      haproxy_server_name = haproxy_server_name(server_name)
      HaproxyCommand.new(@container_id, @client).execute("enable server #{haproxy_server_name}")
    end

    private

    def stats
      haproxy_command = HaproxyCommand.new(@container_id, @client)
      command_output = ''
      retry_counter = 0
      while command_output.empty? && retry_counter < 3
        command_output = haproxy_command.execute('show stat')
        retry_counter += 1
      end
      raise HAProxyStateError.new("Could not get stats from HAProxy backend from container id: #{@container_id}") if command_output.empty?
      CSV.parse(command_output, headers: true)
    end

    def haproxy_server_name(server_name)
      current_stats = stats
      haproxy_server_record = current_stats.find { |stat| prefix_match? stat, server_name }

      raise HAProxyStateError.new("Unable to map #{server_name} to a HAProxy backend, I saw #{ get_server_names(current_stats) }") unless haproxy_server_record

      "#{haproxy_server_record['# pxname']}/#{haproxy_server_record['svname']}"
    end

    def prefix_match?(stat_record, server_name)
      stat_record["svname"].downcase.start_with? server_name.downcase.gsub('-', '_')
    end

    def get_server_names(stats_object)
      stats_object.select { |record| record['# pxname'] == 'default_service' }.map { |record| record['svname']}.join(', ')
    end
  end
end