Merge pull request #1471 from ankane/maxmind_local_country

Adds support for country file lookup to MaxMind Local
alexreisner · Dec 1, 2020 · 920ab3b · 920ab3b
2 parents c804871 + 1bac82c
commit 920ab3b
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/lib/geocoder/lookups/maxmind_local.rb b/lib/geocoder/lookups/maxmind_local.rb
@@ -30,7 +30,13 @@ def required_api_key_parts
     def results(query)
       if configuration[:file]
         geoip_class = RUBY_PLATFORM == "java" ? JGeoIP : GeoIP
-        result = geoip_class.new(configuration[:file]).city(query.to_s)
+        geoip_instance = geoip_class.new(configuration[:file])
+        result =
+          if configuration[:package] == :country
+            geoip_instance.country(query.to_s)
+          else
+            geoip_instance.city(query.to_s)
+          end
         result.nil? ? [] : [encode_hash(result.to_hash)]
       elsif configuration[:package] == :city
         addr = IPAddr.new(query.text).to_i