class CucumberFormatter

# File lib/cucumber_formatter.rb, line 35
def initialize(step_mother, io, options)
  puts "Initializing CucumberMetrics ...\n\n\n"

  @db_avail = true
end

# File lib/cucumber_formatter.rb, line 124
def after_feature(scenario)
  @scenario_counter = nil
end

# File lib/cucumber_formatter.rb, line 92
def after_feature_element(scenario)
  scenario.failed? ? passed = 0 : passed = 1

  sql = "UPDATE scenario_test_runs SET elapsed_time = #{Time.now - @scenario_time_start}, passed = #{passed},
         updated_at = now() WHERE id = #{@str_id}"

  begin
    Database.query(sql) if @db_avail
  rescue => e
    puts "There was an error saving the elapsed time: #{e.message}"
    @db_avail = false
  end

  #update failure rate if configured
  if OpenStruct.new(YAML.load_file(METRICS_CONFIG_FILE)).failure_rate
    check_past_failures(@scenario.title)
  end

  if scenario.failed?
    begin
      save_links(scenario) if @db_avail
    rescue => e
      puts "There was an error saving the links to the failed scenario: #{e.message}"
      @db_avail = false
    end
  end
  #reset step counter when scenario is finished
  @step_counter = 0
  # clear the background flag - it seems to only happen once
  @background = false
end

# File lib/cucumber_formatter.rb, line 128
def after_features(scenario)
  unless @db_avail
    puts "\n\nThe metrics data may be missing or incomplete because there were problems with the database."
  end
end

# File lib/cucumber_formatter.rb, line 70
def after_step(step)
  step_name = get_step_name(@scenario)
  # background steps seem to be adding an extra after_step call. This is above and beyond
  # the fact that they don't call before_feature_elements until after the background
  # executes
  unless step_name == nil
    step_name = step_name.strip[0..255].gsub('\'', '')
  end
  @start_time = @end_time
  @end_time = Time.now

  sql = "INSERT INTO scenario_steps (scenario_test_run_id, name, elapsed_time, created_at, updated_at)
         VALUES (#{@str_id}, \'#{step_name}\', #{(@end_time - @start_time).round}, now(), now())"
  begin
    Database.query(sql) if @db_avail && step_name != nil
  rescue => e
    puts "There was an error saving the step: #{e.message}"
    @db_avail = false
  end
end

# File lib/cucumber_formatter.rb, line 41
def before_feature(feature)
  @background = true
  @scenario_counter ||= 0

  @scenario = feature.feature_elements[@scenario_counter]
  @scenario_counter += 1
  begin
    save_str_start_time if @db_avail
  rescue => e
    puts "Error saving start time: #{e.message}"
    @db_avail = false
  end
end

# File lib/cucumber_formatter.rb, line 57
def before_feature_element(scenario)
  unless @background
    @scenario = scenario
    begin
      save_str_start_time if @db_avail
    rescue => e
      puts "Error saving start time: #{e.message}"
      @db_avail = false
    end
  end
end

# File lib/cucumber_formatter.rb, line 270
def extract_tags
  @scenario.source_tag_names
end

# File lib/cucumber_formatter.rb, line 234
def get_browser_id
  @browser_id = 0
  sql = "SELECT id FROM browsers WHERE browser_name like \'#{BROWSER}\'"
  results = Database.query(sql)
  results.each do |r|
    @browser_id = r["id"]
  end

  if @browser_id == 0
    sql = "INSERT INTO browsers (browser_name, created_at, updated_at) VALUES(\'#{BROWSER}\', now(), now())"
    Database.query(sql)

    sql = "SELECT MAX(id) as max_id FROM browsers"
    Database.query(sql).each do |r|
      @browser_id = r["max_id"]
    end
  end
end

# File lib/cucumber_formatter.rb, line 215
def get_environment_id
  @env_id = 0
  sql = "SELECT id FROM test_environments WHERE env_name like \'#{TESTENV}\'"
  results = Database.query(sql)
  results.each do |r|
    @env_id = r["id"]
  end

  if @env_id == 0
    sql = "INSERT INTO test_environments (env_name, created_at, updated_at) VALUES(\'#{TESTENV}\', now(), now())"
    Database.query(sql)

    sql = "SELECT MAX(id) as max_id FROM test_environments"
    Database.query(sql).each  do |r|
      @env_id = r["max_id"]
    end
  end
end

# File lib/cucumber_formatter.rb, line 190
def get_scenario_id(scenario)
  # make sure the scenario title will fit in the database
  trimmed_scenario_title = scenario.title.strip[0...255]
  trimmed_scenario_title = trimmed_scenario_title.gsub('\'', '')

  @scenario_id = 0
  sql = "SELECT id FROM scenarios WHERE scenario_name LIKE \'#{trimmed_scenario_title}\'"
  results = Database.query(sql)
  results.each do |r|
    @scenario_id = r["id"]
  end

  # if the scenario isn't in the database, then we need to save it and get its ID
  if @scenario_id == 0
    sql = "INSERT INTO scenarios (scenario_name, created_at, updated_at)
           VALUES (\'#{trimmed_scenario_title}\', now(), now())"
    Database.query(sql)

    sql = "SELECT MAX(id) as max_id FROM scenarios"
    Database.query(sql).each do |r|
      @scenario_id = r["max_id"]
    end
  end
end

# File lib/cucumber_formatter.rb, line 179
def get_step_name(scenario)
  @step_counter ||= 0
  step = scenario.steps.to_a[@step_counter]
  @step_counter += 1
  if step == nil
    step
  else
    step.name
  end
end

# File lib/cucumber_formatter.rb, line 254
def save_links(scenario)
  failed_step = ""
  ((scenario.instance_eval {@steps}).instance_eval {@steps}).each do |step|
    if step.status == :failed
      failed_step = step.name
    end
  end

  sql = "INSERT INTO scenario_failed_links (scenario_test_run_id, failed_step,
            feature, scenario, scenario_file, created_at, updated_at)
         VALUES(#{@str_id}, \'#{failed_step.gsub('\'', '')}\', \'#{scenario.feature.title.gsub('\'', '')}\',
          \'#{scenario.title.gsub('\'', '')}\', \'#{scenario.feature.file.gsub('\'', '')}\', now(), now())"

  Database.query(sql)
end

# File lib/cucumber_formatter.rb, line 136
def save_str_start_time
  @scenario_time_start = Time.now

  @start_time = Time.now
  @end_time = Time.now
  machine_name = Socket.gethostname

  if @scenario == nil
    fail "The scenario did not run"
  end

  get_scenario_id(@scenario)
  get_environment_id
  get_browser_id

  save_test_run(machine_name)
  save_tags
end

# File lib/cucumber_formatter.rb, line 155
def save_tags
  # extract and save the tags
  tags = extract_tags
  tags.each do |t|
    sql = "INSERT INTO scenario_tags (scenario_test_run_id, tag_name, created_at, updated_at)
           VALUES (#{@str_id}, \'#{t}\', now(), now())"
    Database.query(sql)
  end
end

# File lib/cucumber_formatter.rb, line 165
def save_test_run(machine_name)
  # save the test run
  sql = "INSERT INTO scenario_test_runs (scenario_id, test_run_at, test_environment_id,
          browser_id, machine_name, created_at, updated_at)
         VALUES (#{@scenario_id}, now(), #{@env_id}, #{@browser_id}, \'#{machine_name}\', now(), now())"
  Database.query(sql)

  sql = "SELECT MAX(id) as max_id FROM scenario_test_runs"
  Database.query(sql).each do |r|
    @str_id = r["max_id"]
  end
end