Add a few geo commands

- GEOADD
- GEOPOS
- GEOSEARCH (only FROMLONLAT + BYRADIUS)

src/commands.cr

@@ -4,6 +4,7 @@ require "./commands/list"
 require "./commands/set"
 require "./commands/sorted_set"
 require "./commands/stream"
+require "./commands/geo"
 
 module Redis
   # All Redis commands are defined in this module. Any paradigm that needs to
@@ -17,6 +18,7 @@ module Redis
     include Commands::Set
     include Commands::SortedSet
     include Commands::Stream
+    include Commands::Geo
 
     # Execute the given command and return the result from the server. Commands
     # must be an `Enumerable` and its `size` method must be re-entrant.

src/commands/geo.cr

@@ -0,0 +1,67 @@
+# Namespace for types that support `Redis::Commands::Geo` commands.
+module Redis::Geo
+  record Radius, magnitude : String, unit : Unit do
+    def self.new(magnitude : Int, unit : Unit)
+      new magnitude.to_s, unit
+    end
+
+    def to_tuple
+      {magnitude, unit.to_s}
+    end
+
+    enum Unit
+      M
+      KM
+      FT
+      MI
+    end
+  end
+
+  enum Sort
+    ASC
+    DESC
+  end
+end
+
+module Redis::Commands::Geo
+  def geoadd(key : String, *entries : String, nx = nil, xx = nil, ch = nil)
+    command = {"geoadd", key}
+    command += {"nx"} if nx
+    command += {"xx"} if xx
+    command += {"ch"} if ch
+    command += entries
+
+    run command
+  end
+
+  def geopos(key : String, *members : String)
+    run({"geopos", key, *members})
+  end
+
+  def geosearch(
+    key : String,
+    *,
+    fromlonlat lonlat : Tuple(String, String),
+    byradius radius : Redis::Geo::Radius,
+    sort : Redis::Geo::Sort? = nil,
+    count : Int | String | Nil = nil,
+    withcoord : Bool = false,
+    withdist : Bool = false,
+  )
+    command = {
+      "geosearch",
+      key,
+      "fromlonlat",
+      *lonlat,
+      "byradius",
+      *radius.to_tuple,
+    }
+
+    command += {sort.to_s} if sort
+    command += {"count", count.to_s} if count
+    command += {"withcoord"} if withcoord
+    command += {"withdist"} if withdist
+
+    run command
+  end
+end

src/connection.cr

@@ -212,7 +212,7 @@ module Redis
     # :nodoc:
     macro override_return_types(methods)
       {% for method, return_type in methods %}
-        {% for methods in [Commands, Commands::Hash, Commands::List, Commands::Set, Commands::SortedSet, Commands::Stream].map(&.methods.select { |m| m.name == method }).reject(&.nil?) %}
+        {% for methods in [Commands, Commands::Hash, Commands::List, Commands::Set, Commands::SortedSet, Commands::Stream, Commands::Geo].map(&.methods.select { |m| m.name == method }).reject(&.nil?) %}
           {% for m in methods %}
             # :nodoc:
             def {{method.id}}(
@@ -336,6 +336,9 @@ module Redis
         xreadgroup: Array(Value)?,
         xrevrange:  Array,
         xtrim:      Int64,
+
+        geopos:    Array,
+        geosearch: Array,
       })
     end