valkey-rb 0.3.5

data/lib/valkey/request_type.rb

@@ -0,0 +1,436 @@
+# frozen_string_literal: true
+
+class Valkey
+  #
+  # this module defines constants for request types used in Valkey.
+  # Each constant represents a specific command or operation that can be performed.
+  #
+  module RequestType
+    INVALID_REQUEST = 0
+    CUSTOM_COMMAND = 1
+
+    # Bitmap commands
+    BIT_COUNT = 101
+    BIT_FIELD = 102
+    BIT_FIELD_READ_ONLY = 103
+    BIT_OP = 104
+    BIT_POS = 105
+    GET_BIT = 106
+    SET_BIT = 107
+
+    # Cluster commands
+    ASKING = 201
+    CLUSTER_ADD_SLOTS = 202
+    CLUSTER_ADD_SLOTS_RANGE = 203
+    CLUSTER_BUMP_EPOCH = 204
+    CLUSTER_COUNT_FAILURE_REPORTS = 205
+    CLUSTER_COUNT_KEYS_IN_SLOT = 206
+    CLUSTER_DEL_SLOTS = 207
+    CLUSTER_DEL_SLOTS_RANGE = 208
+    CLUSTER_FAILOVER = 209
+    CLUSTER_FLUSH_SLOTS = 210
+    CLUSTER_FORGET = 211
+    CLUSTER_GET_KEYS_IN_SLOT = 212
+    CLUSTER_INFO = 213
+    CLUSTER_KEY_SLOT = 214
+    CLUSTER_LINKS = 215
+    CLUSTER_MEET = 216
+    CLUSTER_MY_ID = 217
+    CLUSTER_MY_SHARD_ID = 218
+    CLUSTER_NODES = 219
+    CLUSTER_REPLICAS = 220
+    CLUSTER_REPLICATE = 221
+    CLUSTER_RESET = 222
+    CLUSTER_SAVE_CONFIG = 223
+    CLUSTER_SET_CONFIG_EPOCH = 224
+    CLUSTER_SETSLOT = 225
+    CLUSTER_SHARDS = 226
+    CLUSTER_SLAVES = 227
+    CLUSTER_SLOTS = 228
+    READ_ONLY = 229
+    READ_WRITE = 230
+
+    # Connection Management commands
+    AUTH = 301
+    CLIENT_CACHING = 302
+    CLIENT_GET_NAME = 303
+    CLIENT_GET_REDIR = 304
+    CLIENT_ID = 305
+    CLIENT_INFO = 306
+    CLIENT_KILL_SIMPLE = 307
+    CLIENT_KILL = 308
+    CLIENT_LIST = 309
+    CLIENT_NO_EVICT = 310
+    CLIENT_NO_TOUCH = 311
+    CLIENT_PAUSE = 312
+    CLIENT_REPLY = 313
+    CLIENT_SET_INFO = 314
+    CLIENT_SET_NAME = 315
+    CLIENT_TRACKING = 316
+    CLIENT_TRACKING_INFO = 317
+    CLIENT_UNBLOCK = 318
+    CLIENT_UNPAUSE = 319
+    ECHO = 320
+    HELLO = 321
+    PING = 322
+    QUIT = 323 # deprecated in 7.2.0
+    RESET = 324
+    SELECT = 325
+
+    # Generic commands
+    COPY = 401
+    DEL = 402
+    DUMP = 403
+    EXISTS = 404
+    EXPIRE = 405
+    EXPIRE_AT = 406
+    EXPIRE_TIME = 407
+    KEYS = 408
+    MIGRATE = 409
+    MOVE = 410
+    OBJECT_ENCODING = 411
+    OBJECT_FREQ = 412
+    OBJECT_IDLE_TIME = 413
+    OBJECT_REF_COUNT = 414
+    PERSIST = 415
+    PEXPIRE = 416
+    PEXPIRE_AT = 417
+    PEXPIRE_TIME = 418
+    PTTL = 419
+    RANDOM_KEY = 420
+    RENAME = 421
+    RENAME_NX = 422
+    RESTORE = 423
+    SCAN = 424
+    SORT = 425
+    SORT_READ_ONLY = 426
+    TOUCH = 427
+    TTL = 428
+    TYPE = 429
+    UNLINK = 430
+    WAIT = 431
+    WAIT_AOF = 432
+
+    # Geospatial indices commands
+    GEO_ADD = 501
+    GEO_DIST = 502
+    GEO_HASH = 503
+    GEO_POS = 504
+    GEO_RADIUS = 505 # deprecated in 6.2.0
+    GEO_RADIUS_READ_ONLY = 506 # deprecated in 6.2.0
+    GEO_RADIUS_BY_MEMBER = 507 # deprecated in 6.2.0
+    GEO_RADIUS_BY_MEMBER_READ_ONLY = 508 # deprecated in 6.2.0
+    GEO_SEARCH = 509
+    GEO_SEARCH_STORE = 510
+
+    # Hash commands
+    HDEL = 601
+    HEXISTS = 602
+    HGET = 603
+    HGET_ALL = 604
+    HINCR_BY = 605
+    HINCR_BY_FLOAT = 606
+    HKEYS = 607
+    HLEN = 608
+    HMGET = 609
+    HMSET = 610
+    HRAND_FIELD = 611
+    HSCAN = 612
+    HSET = 613
+    HSET_NX = 614
+    HSTRLEN = 615
+    HVALS = 616
+    HSETEX = 617
+    HGETEX = 618
+    HEXPIRE = 619
+    HEXPIREAT = 620
+    HPEXPIRE = 621
+    HPEXPIREAT = 622
+    HPERSIST = 623
+    HTTL = 624
+    HPTTL = 625
+    HEXPIRETIME = 626
+    HPEXPIRETIME = 627
+
+    # HyperLogLog commands
+    PFADD = 701
+    PFCOUNT = 702
+    PFMERGE = 703
+
+    # List commands
+    BLMOVE = 801
+    BLMPOP = 802
+    BLPOP = 803
+    BRPOP = 804
+    BRPOPLPUSH = 805 # deprecated in 6.2.0
+    LINDEX = 806
+    LINSERT = 807
+    LLEN = 808
+    LMOVE = 809
+    LMPOP = 810
+    LPOP = 811
+    LPOS = 812
+    LPUSH = 813
+    LPUSHX = 814
+    LRANGE = 815
+    LREM = 816
+    LSET = 817
+    LTRIM = 818
+    RPOP = 819
+    RPOPLPUSH = 820 # deprecated in 6.2.0
+    RPUSH = 821
+    RPUSHX = 822
+
+    # Pub/Sub commands
+    PSUBSCRIBE = 901
+    PUBLISH = 902
+    PUBSUB_CHANNELS = 903
+    PUBSUB_NUM_PAT = 904
+    PUBSUB_NUM_SUB = 905
+    PUBSUB_SHARD_CHANNELS = 906
+    PUBSUB_SHARD_NUM_SUB = 907
+    PUNSUBSCRIBE = 908
+    SPUBLISH = 909
+    SSUBSCRIBE = 910
+    SUBSCRIBE = 911
+    SUNSUBSCRIBE = 912
+    UNSUBSCRIBE = 913
+
+    # Scripting and Functions commands
+    EVAL = 1001
+    EVAL_READ_ONLY = 1002
+    EVAL_SHA = 1003
+    EVAL_SHA_READ_ONLY = 1004
+    FCALL = 1005
+    FCALL_READ_ONLY = 1006
+    FUNCTION_DELETE = 1007
+    FUNCTION_DUMP = 1008
+    FUNCTION_FLUSH = 1009
+    FUNCTION_KILL = 1010
+    FUNCTION_LIST = 1011
+    FUNCTION_LOAD = 1012
+    FUNCTION_RESTORE = 1013
+    FUNCTION_STATS = 1014
+    SCRIPT_DEBUG = 1015
+    SCRIPT_EXISTS = 1016
+    SCRIPT_FLUSH = 1017
+    SCRIPT_KILL = 1018
+    SCRIPT_LOAD = 1019
+    SCRIPT_SHOW = 1020
+
+    # Server management commands
+    ACL_CAT = 1101
+    ACL_DEL_USER = 1102
+    ACL_DRY_RUN = 1103
+    ACL_GEN_PASS = 1104
+    ACL_GET_USER = 1105
+    ACL_LIST = 1106
+    ACL_LOAD = 1107
+    ACL_LOG = 1108
+    ACL_SAVE = 1109
+    ACL_SET_USER = 1110
+    ACL_USERS = 1111
+    ACL_WHOAMI = 1112
+    BG_REWRITE_AOF = 1113
+    BG_SAVE = 1114
+    COMMAND_ = 1115 # Command - renamed to avoid collisions
+    COMMAND_COUNT = 1116
+    COMMAND_DOCS = 1117
+    COMMAND_GET_KEYS = 1118
+    COMMAND_GET_KEYS_AND_FLAGS = 1119
+    COMMAND_INFO = 1120
+    COMMAND_LIST = 1121
+    CONFIG_GET = 1122
+    CONFIG_RESET_STAT = 1123
+    CONFIG_REWRITE = 1124
+    CONFIG_SET = 1125
+    DB_SIZE = 1126
+    FAIL_OVER = 1127
+    FLUSH_ALL = 1128
+    FLUSH_DB = 1129
+    INFO = 1130
+    LAST_SAVE = 1131
+    LATENCY_DOCTOR = 1132
+    LATENCY_GRAPH = 1133
+    LATENCY_HISTOGRAM = 1134
+    LATENCY_HISTORY = 1135
+    LATENCY_LATEST = 1136
+    LATENCY_RESET = 1137
+    LOLWUT = 1138
+    MEMORY_DOCTOR = 1139
+    MEMORY_MALLOC_STATS = 1140
+    MEMORY_PURGE = 1141
+    MEMORY_STATS = 1142
+    MEMORY_USAGE = 1143
+    MODULE_LIST = 1144
+    MODULE_LOAD = 1145
+    MODULE_LOAD_EX = 1146
+    MODULE_UNLOAD = 1147
+    MONITOR = 1148
+    PSYNC = 1149
+    REPL_CONF = 1150
+    REPLICA_OF = 1151
+    RESTORE_ASKING = 1152
+    ROLE = 1153
+    SAVE = 1154
+    SHUT_DOWN = 1155
+    SLAVE_OF = 1156
+    SLOWLOG_GET = 1157
+    SLOWLOG_LEN = 1158
+    SLOWLOG_RESET = 1159
+    SWAP_DB = 1160
+    SYNC = 1161
+    TIME = 1162
+
+    # Set commands
+    SADD = 1201
+    SCARD = 1202
+    SDIFF = 1203
+    SDIFF_STORE = 1204
+    SINTER = 1205
+    SINTER_CARD = 1206
+    SINTER_STORE = 1207
+    SISMEMBER = 1208
+    SMEMBERS = 1209
+    SMISMEMBER = 1210
+
+    # Set commands
+    S_MOVE = 1211
+    S_POP = 1212
+    S_RAND_MEMBER = 1213
+    S_REM = 1214
+    S_SCAN = 1215
+    S_UNION = 1216
+    S_UNION_STORE = 1217
+
+    # Sorted set commands
+    BZ_MPOP = 1301
+    BZ_POP_MAX = 1302
+    BZ_POP_MIN = 1303
+    Z_ADD = 1304
+    Z_CARD = 1305
+    Z_COUNT = 1306
+    Z_DIFF = 1307
+    Z_DIFF_STORE = 1308
+    Z_INCR_BY = 1309
+    Z_INTER = 1310
+    Z_INTER_CARD = 1311
+    Z_INTER_STORE = 1312
+    Z_LEX_COUNT = 1313
+    Z_MPOP = 1314
+    Z_MSCORE = 1315
+    Z_POP_MAX = 1316
+    Z_POP_MIN = 1317
+    Z_RAND_MEMBER = 1318
+    Z_RANGE = 1319
+    Z_RANGE_BY_LEX = 1320
+    Z_RANGE_BY_SCORE = 1321
+    Z_RANGE_STORE = 1322
+    Z_RANK = 1323
+    Z_REM = 1324
+    Z_REM_RANGE_BY_LEX = 1325
+    Z_REM_RANGE_BY_RANK = 1326
+    Z_REM_RANGE_BY_SCORE = 1327
+    Z_REV_RANGE = 1328
+    Z_REV_RANGE_BY_LEX = 1329
+    Z_REV_RANGE_BY_SCORE = 1330
+    Z_REV_RANK = 1331
+    Z_SCAN = 1332
+    Z_SCORE = 1333
+    Z_UNION = 1334
+    Z_UNION_STORE = 1335
+
+    # Stream commands
+    X_ACK = 1401
+    X_ADD = 1402
+    X_AUTO_CLAIM = 1403
+    X_CLAIM = 1404
+    X_DEL = 1405
+    X_GROUP_CREATE = 1406
+    X_GROUP_CREATE_CONSUMER = 1407
+    X_GROUP_DEL_CONSUMER = 1408
+    X_GROUP_DESTROY = 1409
+    X_GROUP_SET_ID = 1410
+    X_INFO_CONSUMERS = 1411
+    X_INFO_GROUPS = 1412
+    X_INFO_STREAM = 1413
+    X_LEN = 1414
+    X_PENDING = 1415
+    X_RANGE = 1416
+    X_READ = 1417
+    X_READ_GROUP = 1418
+    X_REV_RANGE = 1419
+    X_SET_ID = 1420
+    X_TRIM = 1421
+
+    # String commands
+    APPEND = 1501
+    DECR = 1502
+    DECR_BY = 1503
+    GET = 1504
+    GET_DEL = 1505
+    GET_EX = 1506
+    GET_RANGE = 1507
+    GET_SET = 1508 # deprecated in 6.2.0
+    INCR = 1509
+    INCR_BY = 1510
+    INCR_BY_FLOAT = 1511
+    LCS = 1512
+    MGET = 1513
+    MSET = 1514
+    MSET_NX = 1515
+    PSET_EX = 1516 # deprecated in 2.6.12
+    SET = 1517
+    SET_EX = 1518 # deprecated in 2.6.12
+    SET_NX = 1519 # deprecated in 2.6.12
+    SET_RANGE = 1520
+    STRLEN = 1521
+    SUBSTR = 1522
+
+    # Transaction commands
+    DISCARD = 1601
+    EXEC = 1602
+    MULTI = 1603
+    UNWATCH = 1604
+    WATCH = 1605
+
+    # JSON commands
+    JSON_ARR_APPEND = 2001
+    JSON_ARR_INDEX = 2002
+    JSON_ARR_INSERT = 2003
+    JSON_ARR_LEN = 2004
+    JSON_ARR_POP = 2005
+    JSON_ARR_TRIM = 2006
+    JSON_CLEAR = 2007
+    JSON_DEBUG = 2008
+    JSON_DEL = 2009
+    JSON_FORGET = 2010
+    JSON_GET = 2011
+    JSON_MGET = 2012
+    JSON_NUM_INCR_BY = 2013
+    JSON_NUM_MULT_BY = 2014
+    JSON_OBJ_KEYS = 2015
+    JSON_OBJ_LEN = 2016
+    JSON_RESP = 2017
+    JSON_SET = 2018
+    JSON_STR_APPEND = 2019
+    JSON_STR_LEN = 2020
+    JSON_TOGGLE = 2021
+    JSON_TYPE = 2022
+
+    # Vector Search commands
+    FT_LIST = 2101
+    FT_AGGREGATE = 2102
+    FT_ALIAS_ADD = 2103
+    FT_ALIAS_DEL = 2104
+    FT_ALIAS_LIST = 2105
+    FT_ALIAS_UPDATE = 2106
+    FT_CREATE = 2107
+    FT_DROP_INDEX = 2108
+    FT_EXPLAIN = 2109
+    FT_EXPLAIN_CLI = 2110
+    FT_INFO = 2111
+    FT_PROFILE = 2112
+    FT_SEARCH = 2113
+  end
+end

data/lib/valkey/response_type.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+class Valkey
+  #
+  # this module defines constants for response types used in Valkey.
+  # Each constant represents a specific type of response that can be received from the server.
+  #
+  module ResponseType
+    NULL = 0
+    INT = 1
+    FLOAT = 2
+    BOOL = 3
+    STRING = 4
+    ARRAY = 5
+    MAP = 6
+    SETS = 7
+    OK = 8
+    ERROR = 9
+  end
+end

data/lib/valkey/utils.rb

@@ -0,0 +1,253 @@
+# frozen_string_literal: true
+
+class Valkey
+  # Valkey Utils module
+  #
+  # This module provides utility functions for transforming and processing
+  # data structures commonly used in Valkey commands.
+  #
+  # It includes methods for converting values to boolean, hash, or float,
+  # as well as methods for handling specific Valkey command responses.
+  #
+  module Utils
+    Boolify = lambda { |value|
+      return value if value.is_a?(TrueClass) || value.is_a?(FalseClass)
+
+      value != 0 unless value.nil?
+    }
+
+    BoolifySet = lambda { |value|
+      case value
+      when "OK"
+        true
+      when nil
+        false
+      else
+        value
+      end
+    }
+
+    Hashify = lambda { |value|
+      if value.respond_to?(:each_slice)
+        value.each_slice(2).to_h
+      else
+        value
+      end
+    }
+
+    Pairify = lambda { |value|
+      if value.respond_to?(:each_slice)
+        value.each_slice(2).to_a
+      else
+        value
+      end
+    }
+
+    Floatify = lambda { |value|
+      case value
+      when "inf"
+        Float::INFINITY
+      when "-inf"
+        -Float::INFINITY
+      when String
+        Float(value)
+      else
+        value
+      end
+    }
+
+    FloatifyPair = lambda { |(first, score)|
+      [first, Floatify.call(score)]
+    }
+
+    FloatifyPairs = lambda { |value|
+      return value unless value.respond_to?(:each_slice)
+
+      value.each_slice(2).map(&FloatifyPair)
+    }
+
+    HashifyInfo = lambda { |reply|
+      lines = reply.split("\r\n").grep_v(/^(#|$)/)
+      lines.map! { |line| line.split(':', 2) }
+      lines.compact!
+      lines.to_h
+    }
+
+    HashifyStreams = lambda { |reply|
+      case reply
+      when nil
+        {}
+      else
+        reply.transform_values { |entries| HashifyStreamEntries.call(entries) }
+      end
+    }
+
+    EMPTY_STREAM_RESPONSE = [nil].freeze
+    private_constant :EMPTY_STREAM_RESPONSE
+
+    HashifyStreamEntries = lambda { |reply|
+      return [] if reply.nil?
+
+      return [] if !reply.is_a?(Array) || reply.empty?
+
+      # Reply format: [[entry_id, [field1, value1, field2, value2, ...]], ...]
+      # Match redis-rb: return flat arrays [["id", ["field", "value", ...]], ...]
+      # Check if first element is a pair [entry_id, values_array]
+      first_elem = reply.first
+      if first_elem.is_a?(Array) && first_elem.length == 2
+        # Already in pair format: [[entry_id, [fields...]], ...]
+        reply.compact.map do |entry_id, values|
+          # Return flat array format like redis-rb, not hash
+          values_array = if values.nil?
+                           []
+                         elsif values.is_a?(Array)
+                           values
+                         else
+                           []
+                         end
+          [entry_id, values_array]
+        end
+      else
+        # Flat array format: [entry_id1, [field1, value1, ...], entry_id2, [field2, value2, ...], ...]
+        reply.compact.each_slice(2).map do |entry_id, values|
+          # Return flat array format like redis-rb, not hash
+          values_array = if values.nil?
+                           []
+                         elsif values.is_a?(Array)
+                           values
+                         else
+                           []
+                         end
+          [entry_id, values_array]
+        end
+      end
+    }
+
+    HashifyStreamAutoclaim = lambda { |reply|
+      {
+        'next' => reply[0],
+        'entries' => if reply[1].nil?
+                       []
+                     elsif reply[1].is_a?(Array)
+                       # Reply[1] is already an array of entries: [[id, [field, value, ...]], ...]
+                       # Use HashifyStreamEntries to convert them properly
+                       HashifyStreamEntries.call(reply[1])
+                     else
+                       []
+                     end
+      }
+    }
+
+    HashifyStreamAutoclaimJustId = lambda { |reply|
+      {
+        'next' => reply[0],
+        'entries' => reply[1]
+      }
+    }
+
+    HashifyStreamPendings = lambda { |reply|
+      {
+        'size' => reply[0],
+        'min_entry_id' => reply[1],
+        'max_entry_id' => reply[2],
+        'consumers' => reply[3].nil? ? {} : reply[3].to_h
+      }
+    }
+
+    HashifyStreamPendingDetails = lambda { |reply|
+      reply.map do |arr|
+        {
+          'entry_id' => arr[0],
+          'consumer' => arr[1],
+          'elapsed' => arr[2],
+          'count' => arr[3]
+        }
+      end
+    }
+
+    HashifyClusterNodeInfo = lambda { |str|
+      arr = str.split(' ')
+      {
+        'node_id' => arr[0],
+        'ip_port' => arr[1],
+        'flags' => arr[2].split(','),
+        'master_node_id' => arr[3],
+        'ping_sent' => arr[4],
+        'pong_recv' => arr[5],
+        'config_epoch' => arr[6],
+        'link_state' => arr[7],
+        'slots' => arr[8].nil? ? nil : Range.new(*arr[8].split('-'))
+      }
+    }
+
+    HashifyClusterSlots = lambda { |reply|
+      reply.map do |arr|
+        first_slot, last_slot = arr[0..1]
+        master = { 'ip' => arr[2][0], 'port' => arr[2][1], 'node_id' => arr[2][2] }
+        replicas = arr[3..].map { |r| { 'ip' => r[0], 'port' => r[1], 'node_id' => r[2] } }
+        {
+          'start_slot' => first_slot,
+          'end_slot' => last_slot,
+          'master' => master,
+          'replicas' => replicas
+        }
+      end
+    }
+
+    HashifyClusterNodes = lambda { |reply|
+      reply.split(/[\r\n]+/).map { |str| HashifyClusterNodeInfo.call(str) }
+    }
+
+    HashifyClusterSlaves = lambda { |reply|
+      reply.map { |str| HashifyClusterNodeInfo.call(str) }
+    }
+
+    Noop = ->(reply) { reply }
+
+    # Parse Redis URL format: redis://[:password@]host[:port][/db]
+    # Also supports: rediss:// (SSL)
+    #
+    # @param [String] url Redis URL to parse
+    # @return [Hash] Parsed connection options with keys: :host, :port, :password, :username, :db, :ssl
+    # @example
+    #   parse_redis_url('redis://:secret@localhost:6379/15')
+    #   # => { host: 'localhost', port: 6379, password: 'secret', db: 15, ssl: false }
+    #
+    #   parse_redis_url('rediss://user:secret@localhost:6380/0')
+    #   # => { host: 'localhost', port: 6380, username: 'user', password: 'secret', db: 0, ssl: true }
+    def self.parse_redis_url(url)
+      return {} unless url.is_a?(String) && !url.empty?
+
+      # Match redis:// or rediss:// URLs
+      # Format: redis[s]://[username:password@]host[:port][/db][?param=value]
+      # Supports: redis://host, redis://user:pass@host, redis://:pass@host
+      # The regex handles:
+      # - No auth: redis://host...
+      # - Username and password: redis://user:pass@host...
+      # - Password only: redis://:pass@host...
+      match = url.match(%r{\A(redis|rediss)://(?:([^:@]*):([^@]+)@)?([^:/]+)(?::(\d+))?(?:/(\d+))?(?:\?.*)?\z})
+
+      return {} unless match
+
+      scheme = match[1]
+      username = match[2]
+      password = match[3]
+      host = match[4]
+      port = match[5]&.to_i
+      db = match[6]&.to_i
+      ssl = scheme == "rediss"
+
+      result = {
+        host: host,
+        port: port || 6379,
+        ssl: ssl
+      }
+
+      result[:username] = username if username && !username.empty?
+      result[:password] = password if password && !password.empty?
+      result[:db] = db if db
+
+      result
+    end
+  end
+end

data/lib/valkey/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+class Valkey
+  VERSION = "0.3.5"
+end