class MMA

# File lib/mma.rb, line 31
def self.fighter_query(fighter)
  sherdog_page = sherdog_page(fighter)

  if sherdog_page == "Fighter not found"
    return "Fighter not found"
  end

  fighter_hash = fighter_template

  fighter_hash[:link] = sherdog_link(fighter)
  fighter_hash[:img_url] = sherdog_page.css('.bio_fighter [itemprop="image"]').attr('src').to_s
  fighter_hash[:name] = sherdog_page.css('h1[itemprop="name"] .fn').text
  fighter_hash[:nickname] = sherdog_page.css('h1[itemprop="name"] .nickname').text.gsub(/"/, '')
  fighter_hash[:age] = sherdog_page.css('.item.birthday strong').text.gsub(/[^\d]/, '')
  fighter_hash[:birthday] = sherdog_page.css('span[itemprop="birthDate"]').text
  fighter_hash[:locality] = sherdog_page.css('span[itemprop="addressLocality"]').text
  fighter_hash[:nationality] = sherdog_page.css('strong[itemprop="nationality"]').text
  fighter_hash[:flag_url] = sherdog_page.css('.birthplace img').attr('src').to_s
  fighter_hash[:association] = sherdog_page.css('.item.association span[itemprop="name"]').text
  fighter_hash[:height] = sherdog_page.css('.item.height strong').text
  fighter_hash[:weight] = sherdog_page.css('.item.weight strong').text
  fighter_hash[:weight_class] = sherdog_page.css('.item.wclass strong').text

  record = sherdog_page.css('.record .count_history')

  wins = record.css('.left_side .bio_graph')[0]
  fighter_hash[:wins][:total] = wins.css('.counter').text
  fighter_hash[:wins][:knockouts] = wins.css('.graph_tag:nth-child(3)').text.to_i.to_s
  fighter_hash[:wins][:submissions] = wins.css('.graph_tag:nth-child(5)').text.to_i.to_s
  fighter_hash[:wins][:decisions] = wins.css('.graph_tag:nth-child(7)').text.to_i.to_s
  fighter_hash[:wins][:others] = wins.css('.graph_tag:nth-child(9)').text.to_i.to_s

  losses = record.css('.left_side .bio_graph')[1]
  fighter_hash[:losses][:total] = losses.css('.counter').text
  fighter_hash[:losses][:knockouts] = losses.css('.graph_tag:nth-child(3)').text.to_i.to_s
  fighter_hash[:losses][:submissions] = losses.css('.graph_tag:nth-child(5)').text.to_i.to_s
  fighter_hash[:losses][:decisions] = losses.css('.graph_tag:nth-child(7)').text.to_i.to_s
  fighter_hash[:losses][:others] = losses.css('.graph_tag:nth-child(9)').text.to_i.to_s

  if record.at('.right_side .bio_graph .card .result:contains("Draws") + span')
    fighter_hash[:draws] = record.at('.right_side .bio_graph .card .result:contains("Draws") + span').text
  end

  if record.at('.right_side .bio_graph .card .result:contains("N/C") + span')
    fighter_hash[:no_contests] = record.at('.right_side .bio_graph .card .result:contains("N/C") + span').text
  end

  fighter_hash[:fights] = build_fights(sherdog_page)

  return fighter_hash
end

# File lib/mma.rb, line 18
def self.sherdog_link(fighter)
  sherdog_links = google_search(fighter).css('.g .r a').select do |link|
    link.to_s.include? SHERDOG_URL
  end
  if sherdog_links != []
    first_link = sherdog_links.first.attribute('href').to_s
    first_link_endtrim = first_link.partition('&').first
    fir_link_trimmed = first_link_endtrim[first_link_endtrim.index(SHERDOG_URL)..-1]
  else
    "Fighter not found"
  end
end

# File lib/mma.rb, line 105
def self.build_fights(sherdog_page)
  fights = []
  sherdog_page.css('.module.fight_history tr:not(.table_head)').each do |row|
    fight = fight_template

    fight[:event] = row.css('td:nth-child(3) a').text
    fight[:date] = row.css('td:nth-child(3) .sub_line').text
    fight[:opponent] = row.css('td:nth-child(2) a').text
    fight[:result] = row.css('td:nth-child(1) .final_result').text

    method = row.css('td:nth-child(4)').text
    if method.split[0].to_s.include? "Draw"
      fight[:method] = "Draw"
    else
      fight[:method] = method.split[0]
    end
    if method[/\(.*?\)/]
      fight[:method_details] = method[/\(.*?\)/][1..-2]
    end

    fight[:round] = row.css('td:nth-child(5)').text
    fight[:time] = row.css('td:nth-child(6)').text
    fight[:referee] = row.css('td:nth-child(4) .sub_line').text

    if fight[:event] != ""
      fights << fight
    end
  end
  fights
end

# File lib/mma.rb, line 171
def self.fight_template
  {
    event: "",
    date: "",
    opponent: "",
    result: "",
    method: "",
    method_details: "",
    round: "",
    time: "",
    referee: ""
  }
end

# File lib/mma.rb, line 136
def self.fighter_template
  {
    link: "",
    img_url: "",
    name: "",
    nickname: "",
    age: "",
    birthday: "",
    locality: "",
    nationality: "",
    flag_url: "",
    association: "",
    height: "",
    weight: "",
    weight_class: "",
    wins: {
      total: 0,
      knockouts: 0,
      submissions: 0,
      decisions: 0,
      others: 0
    },
    losses: {
      total: 0,
      knockouts: 0,
      submissions: 0,
      decisions: 0,
      others: 0
    },
    draws: 0,
    no_contests: 0,
    fights: []
  }
end

# File lib/mma.rb, line 85
def self.google_query(string)
  array = string.split(" ")
  array << "sherdog"
  array.join("+").to_s
end

# File lib/mma.rb, line 91
def self.google_search(fighter)
  url_text = Net::HTTP.get(URI.parse(GOOGLE_URL + google_query(fighter)))
  Nokogiri::HTML(url_text)
end

# File lib/mma.rb, line 96
def self.sherdog_page(fighter)
  if sherdog_link(fighter) != "Fighter not found"
    url_text = Net::HTTP.get(URI.parse(sherdog_link(fighter)))
    return Nokogiri::HTML(url_text)
  else
    return sherdog_link(fighter)
  end
end