db.redis: add optional password support with edis.connect(password: some_pass) (#25371)

Author: davlgd

vlib/db/redis/README.md

@@ -20,8 +20,10 @@ import db.redis
 
 fn main() {
 	// Connect to Redis
+	// Uncomment passwod line if authentication is needed
 	mut db := redis.connect(redis.Config{
 		host: 'localhost'
+		// password: 'your_password'
 		port: 6379
 	})!
 

vlib/db/redis/redis.v

@@ -33,27 +33,49 @@ mut:
 @[params]
 pub struct Config {
 pub mut:
-	host    string = '127.0.0.1' // Redis server host
-	port    u16    = 6379        // Redis server port
-	version int    = 2           // RESP protocol version (default: v2)
+	host     string = '127.0.0.1' // Redis server host
+	port     u16    = 6379        // Redis server port
+	password string // Redis server password (optional)
+	version  int = 2 // RESP protocol version (default: v2)
 }
 
 // connect establishes a connection to a Redis server
 pub fn connect(config Config) !DB {
-	conn := net.dial_tcp('${config.host}:${config.port}')!
-	return DB{
-		conn:     conn
+	mut db := DB{
+		conn:     net.dial_tcp('${config.host}:${config.port}')!
 		version:  config.version
 		cmd_buf:  []u8{cap: cmd_buf_pre_allocate_len}
 		resp_buf: []u8{cap: resp_buf_pre_allocate_len}
 	}
+
+	// Authenticate if password is provided
+	if config.password.len > 0 {
+		db.auth(config.password)!
+	}
+
+	return db
 }
 
 // close terminates the connection to Redis server
 pub fn (mut db DB) close() ! {
 	db.conn.close()!
 }
 
+// auth sends an AUTH command to the server with the given password.
+pub fn (mut db DB) auth(password string) ! {
+	resp := db.cmd('AUTH', password)!
+	match resp {
+		string {
+			if resp != 'OK' {
+				return error('Authentication failed: ${resp}')
+			}
+		}
+		else {
+			return error('Authentication failed: unexpected response type')
+		}
+	}
+}
+
 // ping sends a PING command to verify server responsiveness
 pub fn (mut db DB) ping() !string {
 	db.conn.write_string('*1\r\n$4\r\nPING\r\n')!

vlib/db/redis/redis_test.v

@@ -1,9 +1,12 @@
 // vtest build: started_redis?
-import db.redis
+import os
 import time
+import db.redis
+
+const redis_password = os.getenv('VREDIS_PASSWORD')
 
 fn test_redis() {
-	mut db := redis.connect() or { panic(err) }
+	mut db := redis.connect(password: redis_password)!
 	assert db.ping()! == 'PONG'
 
 	// delete all keys first
@@ -57,7 +60,7 @@ fn test_redis() {
 }
 
 fn test_redis_pipeline() {
-	mut db := redis.connect() or { panic(err) }
+	mut db := redis.connect(password: redis_password)!
 	assert db.ping()! == 'PONG'
 
 	// start pipleline mode