class Geocoder::Lookup::Esri

def base_query_url(query)
  action = query.reverse_geocode? ? "reverseGeocode" : "find"
  "#{protocol}://geocode.arcgis.com/arcgis/rest/services/World/GeocodeServer/#{action}?"
end

def create_and_save_token!(query)
  token_instance = create_token
  if query.options[:token]
    query.options[:token] = token_instance
  else
    Geocoder.merge_into_lookup_config(:esri, token: token_instance)
  end
  token_instance
end

def create_token
  Geocoder::EsriToken.generate_token(*configuration.api_key)
end

def for_storage(query)
  if query.options.has_key?(:for_storage)
    query.options[:for_storage]
  else
    configuration[:for_storage]
  end
end

def name
  "Esri"
end

def query_url_params(query)
  params = {
    :f => "pjson",
    :outFields => "*"
  }
  if query.reverse_geocode?
    params[:location] = query.coordinates.reverse.join(',')
  else
    params[:text] = query.sanitized_text
  end
  params[:token] = token(query)
  if for_storage_value = for_storage(query)
    params[:forStorage] = for_storage_value
  end
  params[:sourceCountry] = configuration[:source_country] if configuration[:source_country]
  params[:preferredLabelValues] = configuration[:preferred_label_values] if configuration[:preferred_label_values]
  params.merge(super)
end

def results(query)
  return [] unless doc = fetch_data(query)
  if (doc['error'].nil?)
    if (!query.reverse_geocode?)
      return [] if !doc['locations'] || doc['locations'].empty?
    end
    return [ doc ]
  else
    case [ doc['error']['code'] ]
    when [498]
      raise_error(Geocoder::InvalidApiKey, doc['error']['message']) ||
        Geocoder.log(:warn, "#{self.name} Geocoding API error: #{doc['error']['message']}")
    when [ 403 ]
      raise_error(Geocoder::RequestDenied, 'ESRI request denied') ||
        Geocoder.log(:warn, "#{self.name} ESRI request denied: #{doc['error']['message']}")
    when [ 500 ], [501]
      raise_error(Geocoder::ServiceUnavailable, 'ESRI service unavailable') ||
        Geocoder.log(:warn, "#{self.name} ESRI service error: #{doc['error']['message']}")
    else
      raise_error(Geocoder::Error, doc['error']['message']) ||
        Geocoder.log(:warn, "#{self.name} Geocoding error: #{doc['error']['message']}")
    end
  end
  return []
end

def supported_protocols
  [:https]
end

def token(query)
  token_instance = if query.options[:token]
                     query.options[:token]
                   else
                     configuration[:token]
                   end
  if !valid_token_configured?(token_instance) && configuration.api_key
    token_instance = create_and_save_token!(query)
  end
  token_instance.to_s unless token_instance.nil?
end

def valid_token_configured?(token_instance)
  !token_instance.nil? && token_instance.active?
end