i2bskn/mycmd

# coding: utf-8

module Mycmd
  class Configuration
    CONFIG_FILE = ".mycmd.yml"

    VALID_OPTIONS_KEYS = [
      :host,
      :username,
      :password,
      :port,
      :database,
      :socket,
      :flags,
      :encoding,
      :read_timeout,
      :write_timeout,
      :connect_timeout,
      :reconnect,
      :local_infile,
    ].freeze

    attr_accessor *VALID_OPTIONS_KEYS
    attr_accessor :tasks
    attr_reader :path

    def initialize
      reset
    end

    def merge(params)
      default
      params.each do |k,v|
        self.send("#{k.to_s}=", v)
      end
    end

    def to_hash
      VALID_OPTIONS_KEYS.inject({}) do |c,k|
        c.store(k.to_sym, self.send(k.to_s)) unless self.send(k.to_s).nil?
        c
      end
    end

    def connect
      Mysql2::Client.new(to_hash)
    end

    def reset
      @path = Configuration.config_find
      if @path
        merge YAML.load_file(@path)
      else
        default
      end
    end

    def default
      self.host = "localhost"
      self.port = 3306
      self.username = "root"
    end

    class << self
      def connect
        conf = Configuration.new
        conf.connect
      end

      def get_variables
        sql = "SELECT GV.* FROM INFORMATION_SCHEMA.GLOBAL_VARIABLES AS GV"
        client = self.connect
        variables = {}
        client.query(sql).each do |row|
          variables.store(row["VARIABLE_NAME"].downcase.to_sym, row["VARIABLE_VALUE"])
        end
        variables
      end

      def config_find(path = File.expand_path("."))
        file = File.join(path, CONFIG_FILE)
        if File.exists?(file)
          file
        else
          if path == "/"
            file = File.join(ENV["HOME"], CONFIG_FILE)
            File.exists?(file) ? file : nil
          else
            config_find(File.expand_path("..", path))
          end
        end
      end
    end
  end
end