require "mcrain/version"

require 'timeout'
require 'logger'

require 'logger_pipe'

require 'active_support/inflector/inflections'
require 'active_support/core_ext/class/subclasses'
require 'active_support/core_ext/object/blank'
require 'active_support/core_ext/numeric/time'

module Mcrain
  class << self
    def class_names
      @class_names ||= {}
    end

    def class_for(name)
      if class_name = class_names[name]
        class_name.constantize
      else
        if klass = Mcrain::Base.descendants.detect{|c| c.server_name == name}
          class_names[name] = klass.name
        end
        klass
      end
    end

    def register(name, class_name)
      class_names[name] = class_name
    end

    def load_config(file)
      @configuration = Mcrain::Configuration.load_config(file)
    end

    attr_writer :configuration
    def configuration
      @configuration ||= Mcrain::Configuration.new
    end

    def configure
      yield configuration
      configuration
    end

    attr_writer :logger
    def logger
      @logger ||= Logger.new($stderr)
    end

    def wait_port_opened(host, port, options = {})
      logger.debug("wait_port_opened(#{host.inspect}, #{port.inspect}, #{options.inspect})")
      interval = options[:interval] || 10 # second
      Timeout.timeout(options[:timeout] || 60) do
        begin
          s = TCPSocket.open(host, port)
          s.close
          return true
        rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH
          sleep(interval)
          retry
        end
      end
    end

    attr_accessor :before_setup

    attr_writer :home_dir
    def home_dir
      @home_dir ||= (ENV['MCRAIN_HOME'] || File.join(ENV['HOME'], '.mcrain'))
    end
  end

  autoload :Base, 'mcrain/base'
  autoload :Configuration, 'mcrain/configuration'
  autoload :DockerMachine, 'mcrain/docker_machine'
  autoload :ContainerController, 'mcrain/container_controller'
  autoload :ClientProvider, 'mcrain/client_provider'

  autoload :Riak, 'mcrain/riak'
  autoload :Redis, 'mcrain/redis'
  autoload :Rabbitmq, 'mcrain/rabbitmq'
  autoload :Mysql, 'mcrain/mysql'
  autoload :Hbase, 'mcrain/hbase'

  register :riak, "Mcrain::Riak"
  register :redis, "Mcrain::Redis"
  register :rabbitmq, "Mcrain::Rabbitmq"
  register :mysql, "Mcrain::Mysql"
  register :hbase, "Mcrain::Hbase"
end