huginn/app/models/agents/adioso_agent.rb

module Agents
  class AdiosoAgent < Agent
    cannot_receive_events!

    default_schedule "every_1d"

    description <<~MD
      The Adioso Agent will tell you the minimum airline prices between a pair of cities, and within a certain period of time.

      The currency is USD. Please make sure that the difference between `start_date` and `end_date` is less than 150 days. You will need to contact [Adioso](http://adioso.com/) for a `username` and `password`.
    MD

    event_description <<~MD
      If flights are present then events look like:

          {
            "cost": 75.23,
            "date": "June 25, 2013",
            "route": "New York to Chicago"
          }

      otherwise

          {
            "nonetodest": "No flights found to the specified destination"
          }
    MD

    def default_options
      {
        'start_date' => Date.today.httpdate[0..15],
        'end_date' => Date.today.plus_with_duration(100).httpdate[0..15],
        'from' => "New York",
        'to' => "Chicago",
        'username' => "xx",
        'password' => "xx",
        'expected_update_period_in_days' => "1"
      }
    end

    def working?
      event_created_within?(options['expected_update_period_in_days']) && !recent_error_logs?
    end

    def validate_options
      unless %w[
        start_date end_date from to username password expected_update_period_in_days
      ].all? { |field| options[field].present? }
        errors.add(:base, "All fields are required")
      end
    end

    def date_to_unix_epoch(date)
      date.to_time.to_i
    end

    def check
      auth_options = {
        basic_auth: {
          username: interpolated[:username],
          password: interpolated[:password]
        }
      }
      parse_response = HTTParty.get(
        "http://api.adioso.com/v2/search/parse?#{{ q: "#{interpolated[:from]} to #{interpolated[:to]}" }.to_query}",
        auth_options
      )
      fare_request = parse_response["search_url"].gsub(
        /(end=)(\d*)([^\d]*)(\d*)/,
        "\\1#{date_to_unix_epoch(interpolated['end_date'])}\\3#{date_to_unix_epoch(interpolated['start_date'])}"
      )
      fare = HTTParty.get fare_request, auth_options

      if fare["warnings"]
        create_event payload: fare["warnings"]
      else
        event = fare["results"].min_by { |x| x["cost"] }
        event["date"]  = Time.at(event["date"]).to_date.httpdate[0..15]
        event["route"] = "#{interpolated['from']} to #{interpolated['to']}"
        create_event payload: event
      end
    end
  end
end