module AppleSystemStatus
  require "capybara"
  require 'capybara/poltergeist'
  require "active_support/core_ext/object/blank"

  class Crawler
    USER_AGENT = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_9_5) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/43.0.2357.134 Safari/537.36"

    MAX_RETRY_COUNT = 5

    def initialize
      Capybara.register_driver :poltergeist do |app|
        Capybara::Poltergeist::Driver.new(app, js_errors: false, window_size: [1280, 800])
      end
      @session = Capybara::Session.new(:poltergeist)
      @session.driver.headers = { "User-Agent" => USER_AGENT }
    end

    def quit!
      @session.driver.quit if @session
    end

    # crawl apple system status page
    # @param country [String] country code. (e.g. jp, ca, fr. default. us)
    # @param title   [String] If specified, narrow the service title
    # @return [Hash]
    # @example response format
    #   {
    #     title: ,
    #     services: [
    #       { title: , description: , status:  }
    #     ]
    #   }
    def perform(country: nil, title: nil)
      @session.visit(apple_url(country))

      response = {
        title:    @session.find(".section-date .date-copy").text.strip,
        services: [],
      }

      MAX_RETRY_COUNT.times do
        services = fetch_services

        if services.empty?
          # wait until the page is fully loaded
          sleep 1
        else
          response[:services] = services
          break
        end
      end

      raise "Not found services" if response[:services].empty?

      unless title.blank?
        response[:services].select! { |service| service[:title] == title }
      end

      response[:services].sort_by! { |service| service[:title] }

      response
    end

    def apple_url(country)
      if country.blank? || country == "us"
        "https://www.apple.com/support/systemstatus/"
      else
        "https://www.apple.com/#{country}/support/systemstatus/"
      end
    end

    # crawl apple system status page. When finished crawling, clear capybara session
    # @param country [String] country code. (e.g. jp, ca, fr. default. us)
    # @param title   [String] If specified, narrow the service title
    # @return [Hash]
    # @example response format
    #   {
    #     title: ,
    #     services: [
    #       { title: , description: , status:  }
    #     ]
    #   }
    # @link https://github.com/teampoltergeist/poltergeist#memory-leak
    def self.perform(country: nil, title: nil)
      crawler = AppleSystemStatus::Crawler.new
      crawler.perform(country: country, title: title)
    ensure
      crawler.quit!
    end

    private

    def fetch_services
      @session.all("#ssp-lights-table td").each_with_object([]) do |td, services|
        begin
          names = td.find(".light-container .light-content.light-name").text.split(/[-:]/).map(&:strip)
          light_image = td.find(".light-container .light-content.light-image div")["class"]

          services << {
            title:       names[0],
            description: names[1],
            status:      light_image.gsub("light-", ""),
          }
        rescue Capybara::ElementNotFound
          # suppress error (for blank cell)
          # NOTE: Capybara::Node::Matchers#has_css? is very slow!
        end
      end
    end
  end
end