require 'multi_json'
require 'httparty'

class WarmerInterface
  def initialize(options = {})
    puts options.inspect
    @host = options[:host]
    @port = options[:port]
    @query = MultiJson.dump(options[:query])
    @warmer_name = options[:warmer_name]
    @index = options[:index]
    @action = options[:action]
    @warmer_state = nil
  end

  def add_warmer
    url = "http://#{@host}:#{@port}/#{@index}/_warmer/#{@warmer_name}"
    puts url
    puts @query
    HTTParty.put(url, {:body => @query})
  end

  def remove_warmer
    puts "removing warmer #{@warmer_name}"
    `curl -s -XDELETE #{@host}:#{@port}/#{@index}/_warmer/#{@warmer_name}`
  end

  def enable_warmer
    puts "closing #{@index}"
    `curl -s -XPOST '#{@host}:#{@port}/#{@index}/_close'`
    puts "enabling warmer"
    `curl -s -XPUT '#{@host}:#{@port}/#{@index}/_settings?pretty=true' -d '{"index.warmer.enabled":"true"}'`
    puts "opening #{@index}"
    `curl -s -XPOST '#{@host}:#{@port}/#{@index}/_open'`
  end

  def disable_warmer
    puts "closing #{@index}"
    `curl -s -XPOST '#{@host}:#{@port}/#{@index}/_close'`
    puts "disabling warmer"
    `curl -s -XPUT '#{@host}:#{@port}/#{@index}/_settings?pretty=true' -d '{"index.warmer.enabled":"false"}'`
    puts "opening #{@index}"
    `curl -s -XPOST '#{@host}:#{@port}/#{@index}/_open'`
  end

  def determine_interaction
    unless @index.nil? || @host.nil? || @port.nil?
      case command = @action.to_sym
        when :add_warmer then add_warmer
        when :remove_warmer then remove_warmer
        when :enable_warmer then enable_warmer
        when :disable_warmer then disable_warmer
        else abort "#{command} is not a recognized action for determine_interaction from warmers_interface"
      end
    else
      puts "index, host and port are required to interact with the warmers"
    end
  end
end