require "logstash/devutils/rspec/spec_helper"
require "ftw"
require "logstash/plugin"
require "logstash/json"
require "stud/try"
require "longshoreman"
require "logstash/outputs/elasticsearch"
require 'elasticsearch'

CONTAINER_NAME = "logstash-output-elasticsearch-#{rand(999).to_s}"
CONTAINER_IMAGE = "elasticsearch"
CONTAINER_TAG = "2.0"

DOCKER_INTEGRATION = ENV["DOCKER_INTEGRATION"]

module ESHelper
  def get_host_port
    addr = DOCKER_INTEGRATION ? Longshoreman.new.get_host_ip : "127.0.0.1"
    "#{addr}:#{get_port}"
  end

  def get_port
    return 9200 unless DOCKER_INTEGRATION

    container = Longshoreman::Container.new
    container.get(CONTAINER_NAME)
    container.rport(9200)
  end

  def get_client
    Elasticsearch::Client.new(:hosts => [get_host_port])
  end
end


RSpec.configure do |config|
  config.include ESHelper

  if DOCKER_INTEGRATION
    # this :all hook gets run before every describe block that is tagged with :integration => true.
    config.before(:all, :integration => true) do


      # check if container exists already before creating new one.
      begin
        ls = Longshoreman::new
        ls.container.get(CONTAINER_NAME)
      rescue Docker::Error::NotFoundError
        scriptDir = File.expand_path File.dirname(__FILE__) + '/fixtures/scripts'
        Longshoreman.new("#{CONTAINER_IMAGE}:#{CONTAINER_TAG}", CONTAINER_NAME, {
          'Cmd' => [ "-Des.script.inline=on", "-Des.script.indexed=on" ],
          'HostConfig' => {
            'Binds' => ["#{scriptDir}:/usr/share/elasticsearch/config/scripts"],
            'PublishAllPorts' => true
          }
        })
        # TODO(talevy): verify ES is running instead of static timeout
        sleep 10
      end
    end

    # we want to do a final cleanup after all :integration runs,
    # but we don't want to clean up before the last block.
    # This is a final blind check to see if the ES docker container is running and
    # needs to be cleaned up. If no container can be found and/or docker is not
    # running on the system, we do nothing.
    config.after(:suite) do
      # only cleanup docker container if system has docker and the container is running
      begin
        ls = Longshoreman::new
        ls.container.get(CONTAINER_NAME)
        ls.cleanup
      rescue Docker::Error::NotFoundError, Excon::Errors::SocketError
        # do nothing
      end
    end
  end
end