Script for ~/.irbrc to enable SQL query output for Rails 3

gistfile1.sh

extend_console 'rails3', defined?(ActiveSupport::Notifications), false do
    $odd_or_even_queries = false
    ActiveSupport::Notifications.subscribe('sql.active_record') do |*args|
      $odd_or_even_queries = !$odd_or_even_queries
      color = $odd_or_even_queries ? ANSI[:CYAN] : ANSI[:MAGENTA]
      event = ActiveSupport::Notifications::Event.new(*args)
      time = "%.1fms" % event.duration
      name = event.payload[:name]
      sql = event.payload[:sql].gsub("\n", " ").squeeze(" ")
      puts " #{ANSI[:UNDERLINE]}#{color}#{name} (#{time})#{ANSI[:RESET]} #{sql}"
    end
  end