module PrettySql

# File lib/pretty_sql.rb, line 7
def prettify_sql(sql)
  pygmentize_sql(tidy_sql(sql))
end

# File lib/pretty_sql.rb, line 11
def pygmentize_sql(sql)
  # Colorize SQL by piping to pygmentize if it exists.
  
  return sql unless pygmentize_path

  IO.popen("#{pygmentize_path} -l sql", 'r+') do |process|
    process.write(sql)
    process.close_write
    process.read
  end
end

# File lib/pretty_sql.rb, line 23
def tidy_sql(sql)
  # Auto-indent SQL

  rule = AnbtSql::Rule.new
  rule.keyword = AnbtSql::Rule::KEYWORD_UPPER_CASE
  %w(count sum substr date).each do |func_name|
    rule.function_names << func_name.upcase
  end
  rule.indent_string = "    "
  formatter = AnbtSql::Formatter.new(rule)
  formatter.format(sql)
end

# File lib/pretty_sql.rb, line 42
def get_pygmentize_executable
  # Find executable for pygmentize
  
  return ENV['PYGMENTIZE_PATH'] if ENV['PYGMENTIZE_PATH'] 
    
  exts = ENV['PATHEXT'] ? ENV['PATHEXT'].split(';') : ['']
  ENV['PATH'].split(File::PATH_SEPARATOR).each do |path|
    exts.each { |ext|
      exe = File.join(path, "#{PYGMENTIZE_EXEC_NAME}#{ext}")
      return exe if File.executable?(exe) && !File.directory?(exe)
    }
  end

  return nil
end

# File lib/pretty_sql.rb, line 38
def pygmentize_path
  @pygmentize_path ||= get_pygmentize_executable
end