generalbots/src/basic/keywords/set_current_context.rs

use std::sync::Arc;
use log::{error, info, trace};
use crate::shared::state::AppState;
use crate::shared::models::UserSession;
use rhai::Engine;
use rhai::Dynamic;

/// Registers the `SET_CURRENT_CONTEXT` keyword which stores a context value in Redis
/// and marks the context as active.
///
/// # Arguments
///
/// * `state` – Shared application state (Arc<AppState>).
/// * `user` – The current user session (provides user_id and session id).
/// * `engine` – The script engine where the custom syntax will be registered.
pub fn set_current_context_keyword(state: Arc<AppState>, user: UserSession, engine: &mut Engine) {
    // Clone the Redis client (if any) for use inside the async task.
    let cache = state.cache.clone();

    engine
        .register_custom_syntax(
            &["SET_CURRENT_CONTEXT", "$expr$", "AS", "$expr$"],
            true,
            move |context, inputs| {
                // First expression is the context name, second is the value.
                let context_name = context.eval_expression_tree(&inputs[0])?.to_string();
                let context_value = context.eval_expression_tree(&inputs[1])?.to_string();

                info!(
                    "SET_CURRENT_CONTEXT command executed - name: {}, value: {}",
                    context_name,
                    context_value
                );

                // Build a Redis key that is unique per user and session.
                let redis_key = format!(
                    "context:{}:{}:{}",
                    user.user_id,
                    user.id,
                    context_name
                );

                trace!(
                    target: "app::set_current_context",
                    "Constructed Redis key: {} for user {}, session {}, context {}",
                    redis_key,
                    user.user_id,
                    user.id,
                    context_name
                );

                // If a Redis client is configured, perform the SET operation asynchronously.
                if let Some(cache_client) = &cache {
                    trace!("Redis client is available, preparing to set context value");

                    // Clone values needed inside the async block.
                    let cache_client = cache_client.clone();
                    let redis_key = redis_key.clone();
                    let context_value = context_value.clone();
                    let context_name = context_name.clone();

                    trace!(
                        "Cloned cache_client, redis_key ({}) and context_value (len={}) for async task",
                        redis_key,
                        context_value.len()
                    );

                    // Spawn a background task so we don't need an async closure here.
                    tokio::spawn(async move {
                        trace!("Async task started for SET_CURRENT_CONTEXT operation");

                        // Acquire an async Redis connection.
                        let mut conn = match cache_client.get_multiplexed_async_connection().await {
                            Ok(conn) => {
                                trace!("Successfully acquired async Redis connection");
                                conn
                            }
                            Err(e) => {
                                error!("Failed to connect to cache: {}", e);
                                trace!("Aborting SET_CURRENT_CONTEXT task due to connection error");
                                return;
                            }
                        };

                        // Perform the SET command for the context value.
                        trace!(
                            "Executing Redis SET command with key: {} and value length: {}",
                            redis_key,
                            context_value.len()
                        );
                        let set_result: Result<(), redis::RedisError> = redis::cmd("SET")
                            .arg(&redis_key)
                            .arg(&context_value)
                            .query_async(&mut conn)
                            .await;

                        match set_result {
                            Ok(_) => {
                                trace!("Successfully set context in Redis for key {}", redis_key);
                            }
                            Err(e) => {
                                error!("Failed to set cache value: {}", e);
                                trace!("SET_CURRENT_CONTEXT Redis SET command failed");
                                return;
                            }
                        }

                        // Mark the context as active in a separate hash.
                        let active_key = format!("active_context:{}:{}", user.user_id, user.id);
                        trace!("Setting active flag for context {} in hash {}", context_name, active_key);
                        let hset_result: Result<i64, redis::RedisError> = redis::cmd("HSET")
                            .arg(&active_key)
                            .arg(&context_name)
                            .arg("active")
                            .query_async(&mut conn)
                            .await;

                        match hset_result {
                            Ok(fields_added) => {
                                trace!("Active flag set for context {} (fields added: {})", context_name, fields_added);
                            }
                            Err(e) => {
                                error!("Failed to set active flag for context {}: {}", context_name, e);
                            }
                        }
                    });
                } else {
                    trace!("No Redis client configured; SET_CURRENT_CONTEXT will not persist to cache");
                }

                Ok(Dynamic::UNIT)
            },
        )
        .unwrap();
}