require 'mkmf'
require 'date'

# Allow for custom compiler to be specified.
RbConfig::MAKEFILE_CONFIG['CC'] = ENV['CC'] if ENV['CC']

# All instances of mysql_config on PATH ...
def mysql_config_paths
  ENV['PATH'].split(File::PATH_SEPARATOR).collect do |path|
    %W(#{path}/mysql_config #{path}/mysql_config5)
      .detect { |bin| File.exist?(bin) }
  end
end

# The first mysql_config binary on PATH ...
def default_mysql_config_path
  mysql_config_paths.compact.first
end

def mysql_config(type)
  IO.popen("#{default_mysql_config_path} --#{type}").readline.chomp
rescue
  nil
end

def default_prefix
  if (mc = default_mysql_config_path)
    File.dirname(mc, 2)
  else
    '/usr/local'
  end
end

# Allow overriding path to mysql_config on command line using:
# ruby extconf.rb --with-mysql-config=/path/to/mysql_config
if with_config('mysql-config', default_mysql_config_path)
  includes = mysql_config('include').split(/\s+/).map do |dir|
    dir.gsub(/^-I/, '')
  end.uniq
  libs = mysql_config('libs').split(/\s+/).grep(/^-L/).map do |dir|
    dir.gsub(/^-L/, '')
  end.uniq

  linked = mysql_config('libs').split(/\s+/).grep(/^-l/).map do |dir|
    dir.gsub(/^-l/, '')
  end.uniq

  dir_config('mysql', includes, libs)
  linked.each do |link|
    have_library link
  end
else
  _, lib = dir_config('mysql', default_prefix)
  lib_dirs =
    [lib, '/usr/lib', '/usr/local/lib', '/opt/local/lib'].collect do |path|
      [path, "#{path}/mysql", "#{path}/mysql5/mysql"]
    end
  find_library('mysqlclient', 'mysql_query', *lib_dirs.flatten) || exit(1)
  find_header('mysql.h', *lib_dirs.flatten.map { |p| p.gsub('/lib', '/include') })
end

have_func('localtime_r')
have_func('gmtime_r')

have_header 'mysql.h'
have_const 'MYSQL_TYPE_STRING', 'mysql.h'
have_const 'MYSQL_TYPE_BIT', 'mysql.h'
have_const 'MYSQL_TYPE_NEWDECIMAL', 'mysql.h'
have_func 'mysql_query', 'mysql.h'
have_func 'mysql_ssl_set', 'mysql.h'
have_func 'mysql_sqlstate', 'mysql.h'
have_func 'mysql_get_ssl_cipher', 'mysql.h'
have_func 'mysql_set_character_set', 'mysql.h'
have_func 'mysql_get_server_version', 'mysql.h'
have_func 'mysql_real_escape_string_quote', 'mysql.h'
have_struct_member 'MYSQL_FIELD', 'charsetnr', 'mysql.h'

have_func('rb_thread_fd_select')

# rubocop:disable Style/GlobalVars
$CFLAGS << ' -DHAVE_NO_DATETIME_NEWBANG' unless DateTime.respond_to?(:new!)

$CFLAGS << ' -Wall '

# rubocop:enable Style/GlobalVars

create_makefile('do_mysql/do_mysql')