geocoder-kb 1.2.6

data/lib/geocoder/models/mongoid.rb

@@ -0,0 +1,32 @@
+require 'geocoder/models/base'
+require 'geocoder/models/mongo_base'
+
+module Geocoder
+  module Model
+    module Mongoid
+      include Base
+      include MongoBase
+
+      def self.included(base); base.extend(self); end
+
+      private # --------------------------------------------------------------
+
+      def geocoder_file_name;   "mongoid"; end
+      def geocoder_module_name; "Mongoid"; end
+
+      def geocoder_init(options)
+        super(options)
+        if options[:skip_index] == false
+          # create 2d index
+          if defined?(::Mongoid::VERSION) && ::Mongoid::VERSION >= "3"
+            index({ geocoder_options[:coordinates].to_sym => '2d' }, 
+                  {:min => -180, :max => 180})
+          else
+            index [[ geocoder_options[:coordinates], '2d' ]],
+              :min => -180, :max => 180
+          end
+        end
+      end
+    end
+  end
+end

data/lib/geocoder/query.rb

@@ -0,0 +1,111 @@
+module Geocoder
+  class Query
+    attr_accessor :text, :options
+
+    def initialize(text, options = {})
+      self.text = text
+      self.options = options
+    end
+
+    def execute
+      lookup.search(text, options)
+    end
+
+    def to_s
+      text
+    end
+
+    def sanitized_text
+      if coordinates?
+        if text.is_a?(Array)
+          text.join(',')
+        else
+          text.split(/\s*,\s*/).join(',')
+        end
+      else
+        text
+      end
+    end
+
+    ##
+    # Get a Lookup object (which communicates with the remote geocoding API)
+    # appropriate to the Query text.
+    #
+    def lookup
+      if ip_address?
+        name = options[:ip_lookup] || Configuration.ip_lookup || Geocoder::Lookup.ip_services.first
+      else
+        name = options[:lookup] || Configuration.lookup || Geocoder::Lookup.street_services.first
+      end
+      Lookup.get(name)
+    end
+
+    def url
+      lookup.query_url(self)
+    end
+
+    ##
+    # Is the Query blank? (ie, should we not bother searching?)
+    # A query is considered blank if its text is nil or empty string AND
+    # no URL parameters are specified.
+    #
+    def blank?
+      !params_given? and (
+        (text.is_a?(Array) and text.compact.size < 2) or
+        text.to_s.match(/\A\s*\z/)
+      )
+    end
+
+    ##
+    # Does the Query text look like an IP address?
+    #
+    # Does not check for actual validity, just the appearance of four
+    # dot-delimited numbers.
+    #
+    def ip_address?
+      IpAddress.new(text).valid? rescue false
+    end
+
+    ##
+    # Is the Query text a loopback IP address?
+    #
+    def loopback_ip_address?
+      ip_address? && IpAddress.new(text).loopback?
+    end
+
+    ##
+    # Does the given string look like latitude/longitude coordinates?
+    #
+    def coordinates?
+      text.is_a?(Array) or (
+        text.is_a?(String) and
+        !!text.to_s.match(/\A-?[0-9\.]+, *-?[0-9\.]+\z/)
+      )
+    end
+
+    ##
+    # Return the latitude/longitude coordinates specified in the query,
+    # or nil if none.
+    #
+    def coordinates
+      sanitized_text.split(',') if coordinates?
+    end
+
+    ##
+    # Should reverse geocoding be performed for this query?
+    #
+    def reverse_geocode?
+      coordinates?
+    end
+
+    def language
+      options[:language]
+    end
+
+    private # ----------------------------------------------------------------
+
+    def params_given?
+      !!(options[:params].is_a?(Hash) and options[:params].keys.size > 0)
+    end
+  end
+end

data/lib/geocoder/railtie.rb

@@ -0,0 +1,26 @@
+require 'geocoder/models/active_record'
+
+module Geocoder
+  if defined? Rails::Railtie
+    require 'rails'
+    class Railtie < Rails::Railtie
+      initializer 'geocoder.insert_into_active_record' do
+        ActiveSupport.on_load :active_record do
+          Geocoder::Railtie.insert
+        end
+      end
+      rake_tasks do
+        load "tasks/geocoder.rake"
+        load "tasks/maxmind.rake"
+      end
+    end
+  end
+
+  class Railtie
+    def self.insert
+      if defined?(::ActiveRecord)
+        ::ActiveRecord::Base.extend(Model::ActiveRecord)
+      end
+    end
+  end
+end

data/lib/geocoder/request.rb

@@ -0,0 +1,25 @@
+module Geocoder
+  module Request
+
+    def location
+      @location ||= begin
+        detected_ip = env['HTTP_X_REAL_IP'] || (
+          env['HTTP_X_FORWARDED_FOR'] &&
+          env['HTTP_X_FORWARDED_FOR'].split(",").first.strip
+        )
+        detected_ip = IpAddress.new(detected_ip.to_s)
+        if detected_ip.valid? and !detected_ip.loopback?
+          real_ip = detected_ip.to_s
+        else
+          real_ip = self.ip
+        end
+        Geocoder.search(real_ip).first
+      end
+      @location
+    end
+  end
+end
+
+if defined?(Rack) and defined?(Rack::Request)
+  Rack::Request.send :include, Geocoder::Request
+end

data/lib/geocoder/results/amap.rb

@@ -0,0 +1,85 @@
+require 'geocoder/results/base'
+
+module Geocoder::Result
+  class Amap < Base
+
+    def coordinates
+      @data["roadinters"]['location'].split(",")
+    end
+
+    def address
+      @data['formatted_address']
+    end
+
+    def state
+      province
+    end
+    
+    def province
+      address_components['province']
+    end
+
+    def city
+      address_components['city'] == [] ? province : address_components["city"]
+    end
+
+    def district
+      address_components['district']
+    end
+
+    def street
+      if address_components["neighborhood"]["name"] != []
+        return address_components["neighborhood"]["name"]
+      elsif address_components["streetNumber"]["street"] != []
+        return address_components["streetNumber"]["street"]
+      else
+        return address_components["township"]
+      end
+    end
+
+    def street_number
+      address_components['streetNumber']["number"]
+    end
+
+    def formatted_address
+      @data['formatted_address']
+    end
+
+    def address_components
+      @data['addressComponent']
+    end
+
+    def state_code
+      ""
+    end
+
+    def postal_code
+      ""
+    end
+
+    def country
+      "China"
+    end
+
+    def country_code
+      "CN"
+    end
+
+    ##
+    # Get address components of a given type. Valid types are defined in
+    # Baidu's Geocoding API documentation and include (among others):
+    #
+    #   :business
+    #   :cityCode
+    #
+    def self.response_attributes
+      %w[roads pois roadinters]
+    end
+
+    response_attributes.each do |a|
+      define_method a do
+        @data[a]
+      end
+    end
+  end
+end

data/lib/geocoder/results/baidu.rb

@@ -0,0 +1,79 @@
+require 'geocoder/results/base'
+
+module Geocoder::Result
+  class Baidu < Base
+
+    def coordinates
+      ['lat', 'lng'].map{ |i| @data['location'][i] }
+    end
+
+    def address
+      @data['formatted_address']
+    end
+
+    def state
+      province
+    end
+    
+    def province
+      @data['addressComponent']['province']
+    end
+
+    def city
+      @data['addressComponent']['city']
+    end
+
+    def district
+      @data['addressComponent']['district']
+    end
+
+    def street
+      @data['addressComponent']['street']
+    end
+
+    def street_number
+      @data['addressComponent']['street_number']
+    end
+
+    def formatted_address
+      @data['formatted_address']
+    end
+
+    def address_components
+      @data['addressComponent']
+    end
+
+    def state_code
+      ""
+    end
+
+    def postal_code
+      ""
+    end
+
+    def country
+      "China"
+    end
+
+    def country_code
+      "CN"
+    end
+
+    ##
+    # Get address components of a given type. Valid types are defined in
+    # Baidu's Geocoding API documentation and include (among others):
+    #
+    #   :business
+    #   :cityCode
+    #
+    def self.response_attributes
+      %w[business cityCode]
+    end
+
+    response_attributes.each do |a|
+      define_method a do
+        @data[a]
+      end
+    end
+  end
+end

data/lib/geocoder/results/baidu_ip.rb

@@ -0,0 +1,62 @@
+require 'geocoder/results/base'
+
+module Geocoder::Result
+  class BaiduIp < Base
+    def coordinates
+      [point['y'].to_f, point['x'].to_f]
+    end
+
+    def address
+      @data['address']
+    end
+
+    def state
+      province
+    end
+
+    def province
+      address_detail['province']
+    end
+
+    def city
+      address_detail['city']
+    end
+
+    def district
+      address_detail['district']
+    end
+
+    def street
+      address_detail['street']
+    end
+
+    def street_number
+      address_detail['street_number']
+    end
+
+    def state_code
+      ""
+    end
+
+    def postal_code
+      ""
+    end
+
+    def country
+      "China"
+    end
+
+    def country_code
+      "CN"
+    end
+
+    private
+    def address_detail
+      @data['address_detail']
+    end
+
+    def point
+      @data['point']
+    end
+  end
+end

data/lib/geocoder/results/base.rb

@@ -0,0 +1,67 @@
+module Geocoder
+  module Result
+    class Base
+
+      # data (hash) fetched from geocoding service
+      attr_accessor :data
+
+      # true if result came from cache, false if from request to geocoding
+      # service; nil if cache is not configured
+      attr_accessor :cache_hit
+
+      ##
+      # Takes a hash of data from a parsed geocoding service response.
+      #
+      def initialize(data)
+        @data = data
+        @cache_hit = nil
+      end
+
+      ##
+      # A string in the given format.
+      #
+      def address(format = :full)
+        fail
+      end
+
+      ##
+      # A two-element array: [lat, lon].
+      #
+      def coordinates
+        [@data['latitude'].to_f, @data['longitude'].to_f]
+      end
+
+      def latitude
+        coordinates[0]
+      end
+
+      def longitude
+        coordinates[1]
+      end
+
+      def state
+        fail
+      end
+
+      def province
+        state
+      end
+
+      def state_code
+        fail
+      end
+
+      def province_code
+        state_code
+      end
+
+      def country
+        fail
+      end
+
+      def country_code
+        fail
+      end
+    end
+  end
+end