# Copyright (c) 2011, SoundCloud Ltd., Rany Keddo, Tobias Bielohlawek, Tobias
# Schmidt

module Lhm
  module SqlHelper
    extend self

    def annotation
      "/* large hadron migration */"
    end

    def idx_name(table_name, cols)
      column_names = column_definition(cols).map(&:first)
      "index_#{ table_name }_on_#{ column_names.join("_and_") }"
    end

    def idx_spec(cols)
      column_definition(cols).map do |name, length|
        "`#{ name }`#{ length }"
      end.join(', ')
    end

    def table?(table_name)
      connection.table_exists?(table_name)
    end

    def sql(statements)
      [statements].flatten.each do |statement|
        connection.execute(tagged(statement))
      end
    rescue ActiveRecord::StatementInvalid => e
      error e.message
    end

    def update(statements)
      [statements].flatten.inject(0) do |memo, statement|
        memo += connection.update(tagged(statement))
      end
    rescue ActiveRecord::StatementInvalid => e
      error e.message
    end

    def version_string
      connection.select_one("show variables like 'version'")["Value"]
    end

  private

    def tagged(statement)
      "#{ statement } #{ SqlHelper.annotation }"
    end

    def column_definition(cols)
      Array(cols).map do |column|
        column.to_s.match(/`?([^\(]+)`?(\([^\)]+\))?/).captures
      end
    end

    # Older versions of MySQL contain an atomic rename bug affecting bin
    # log order. Affected versions extracted from bug report:
    #
    #   http://bugs.mysql.com/bug.php?id=39675
    #
    # More Info: http://dev.mysql.com/doc/refman/5.5/en/metadata-locking.html
    def supports_atomic_switch?
      major, minor, tiny = version_string.split('.').map(&:to_i)

      case major
      when 4 then return false if minor and minor < 2
      when 5
        case minor
        when 0 then return false if tiny and tiny < 52
        when 1 then return false
        when 4 then return false if tiny and tiny < 4
        when 5 then return false if tiny and tiny < 3
        end
      when 6
        case minor
        when 0 then return false if tiny and tiny < 11
        end
      end
      return true
    end
  end
end