#!/usr/bin/env ruby -Ku

require 'ftools'
require 'rubygems'

gem 'addressable', '~> 2.1'
gem 'faker',       '~> 0.3.1'
gem 'ruby-prof',   '~> 0.7.3'

require 'addressable/uri'
require 'faker'
require 'ruby-prof'

require File.expand_path(File.join(File.dirname(__FILE__), '..', 'lib', 'dm-core'))

TEXT_OUTPUT = DataMapper.root / 'profile_results.txt'
HTML_OUTPUT = DataMapper.root / 'profile_results.html'
CALL_OUTPUT = DataMapper.root / 'profile_results.prof'

SOCKET_FILE = Pathname.glob(%w[
  /opt/local/var/run/mysql5/mysqld.sock
  /tmp/mysqld.sock
  /tmp/mysql.sock
  /var/mysql/mysql.sock
  /var/run/mysqld/mysqld.sock
]).find { |path| path.socket? }

configuration_options = {
  :adapter  => 'mysql',
  :database => 'dm_core_test',
  :host     => 'localhost',
  :username => 'root',
  :password => '',
  :socket   => SOCKET_FILE,
}

DataMapper::Logger.new(DataMapper.root / 'log' / 'dm.log', :debug)
adapter = DataMapper.setup(:default, configuration_options)

if configuration_options[:adapter]
  sqlfile       = File.join(File.dirname(__FILE__), '..', 'tmp', 'performance.sql')
  mysql_bin     = %w[ mysql mysql5 ].select { |bin| `which #{bin}`.length > 0 }
  mysqldump_bin = %w[ mysqldump mysqldump5 ].select { |bin| `which #{bin}`.length > 0 }
end

class User
  include DataMapper::Resource

  property :id,         Serial
  property :name,       String
  property :email,      String
  property :about,      Text,   :lazy => false
  property :created_on, Date
end

class Exhibit
  include DataMapper::Resource

  property :id,         Serial
  property :name,       String
  property :zoo_id,     Integer
  property :user_id,    Integer
  property :notes,      Text,    :lazy => false
  property :created_on, Date

  belongs_to :user
end

DataMapper.auto_migrate!

def touch_attributes(*exhibits)
  exhibits.flatten.each do |exhibit|
    exhibit.id
    exhibit.name
    exhibit.created_on
  end
end

def touch_relationships(*exhibits)
  exhibits.flatten.each do |exhibit|
    exhibit.id
    exhibit.name
    exhibit.created_on
    exhibit.user
  end
end

# RubyProf, making profiling Ruby pretty since 1899!
def profile(&b)
  results = RubyProf.profile(&b)

  TEXT_OUTPUT.open('w+') do |file|
    RubyProf::FlatPrinter.new(results).print(file)
  end

  HTML_OUTPUT.open('w+') do |file|
    RubyProf::GraphHtmlPrinter.new(results).print(file)
  end

  CALL_OUTPUT.open('w+') do |file|
    RubyProf::CallTreePrinter.new(results).print(file)
  end
end

c = configuration_options

if sqlfile && File.exists?(sqlfile)
  puts "Found data-file. Importing from #{sqlfile}"
  #adapter.execute("LOAD DATA LOCAL INFILE '#{sqlfile}' INTO TABLE exhibits")
  `#{mysql_bin} -u #{c[:username]} #{"-p#{c[:password]}" unless c[:password].blank?} #{c[:database]} < #{sqlfile}`
else
  puts 'Generating data for benchmarking...'

  # pre-compute the insert statements and fake data compilation,
  # so the benchmarks below show the actual runtime for the execute
  # method, minus the setup steps

  # Using the same paragraph for all exhibits because it is very slow
  # to generate unique paragraphs for all exhibits.
  notes = Faker::Lorem.paragraphs.join($/)
  today = Date.today

  puts 'Inserting 10,000 users and exhibits...'
  10_000.times do
    user = User.create(
      :created_on => today,
      :name       => Faker::Name.name,
      :email      => Faker::Internet.email
    )

    Exhibit.create(
      :created_on => today,
      :name       => Faker::Company.name,
      :user       => user,
      :notes      => notes,
      :zoo_id     => rand(10).ceil
    )
  end

  if sqlfile
    answer = nil
    until answer && answer[/\A(?:y(?:es)?|no?)\b/i]
      print('Would you like to dump data into tmp/performance.sql (for faster setup)? [Yn]');
      STDOUT.flush
      answer = gets
    end

    if %w[ y yes ].include?(answer.downcase)
      File.makedirs(File.dirname(sqlfile))
      #adapter.execute("SELECT * INTO OUTFILE '#{sqlfile}' FROM exhibits;")
      `#{mysqldump_bin} -u #{c[:username]} #{"-p#{c[:password]}" unless c[:password].blank?} #{c[:database]} exhibits users > #{sqlfile}`
      puts "File saved\n"
    end
  end
end

TIMES = 10_000

exhibits = Exhibit.all.to_a

profile do
#  dm_obj = Exhibit.get(1)
#  puts 'Model#id'
#  (TIMES * 100).times { dm_obj.id }
#
#  puts 'Model.new (instantiation)'
#  TIMES.times { Exhibit.new }
#
#  puts 'Model.new (setting attributes)'
#  TIMES.times { Exhibit.new(:name => 'sam', :zoo_id => 1) }
#
#  puts 'Model.get specific (not cached)'
#  TIMES.times { touch_attributes(Exhibit.get(1)) }
#
#  puts 'Model.get specific (cached)'
#  repository(:default) do
#    TIMES.times { touch_attributes(Exhibit.get(1)) }
#  end

  puts 'Model.first'
  TIMES.times { touch_attributes(Exhibit.first) }

#  puts 'Model.all limit(100)'
#  (TIMES / 10).ceil.times { touch_attributes(Exhibit.all(:limit => 100)) }
#
#  puts 'Model.all limit(100) with relationship'
#  (TIMES / 10).ceil.times { touch_relationships(Exhibit.all(:limit => 100)) }
#
#  puts 'Model.all limit(10,000)'
#  (TIMES / 1000).ceil { touch_attributes(Exhibit.all(:limit => 10_000)) }
#
#  exhibit = {
#    :name       => Faker::Company.name,
#    :zoo_id     => rand(10).ceil,
#    :notes      => Faker::Lorem.paragraphs.join($/),
#    :created_on => Date.today
#  }
#
#  puts 'Model.create'
#  TIMES.times { Exhibit.create(exhibit) }
#
#  attrs_first  = { :name => 'sam', :zoo_id => 1 }
#  attrs_second = { :name => 'tom', :zoo_id => 1 }
#
#  puts 'Resource#attributes='
#  TIMES.times { exhibit = Exhibit.new(attrs_first); exhibit.attributes = attrs_second }
#
#  puts 'Resource#update'
#  TIMES.times { |index| exhibit = exhibits[index]; exhibit.name = 'bob'; exhibit.save }
#
#  puts 'Resource#destroy'
#  TIMES.times { |index| exhibits[index].destroy }
#
#  puts 'Model.transaction'
#  TIMES.times { Exhibit.transaction { Exhibit.new } }
end

puts "Done!"