use crate::cache::redis::AppRedis;
use crate::error::{AppError, AppResult};
use crate::service::im::util::PRESENCE_PREFIX;
use ::redis::Cmd;

use super::redis_keys::*;
use super::session::WsSession;

pub fn register_redis_online(redis: &AppRedis, session: &WsSession) -> AppResult<()> {
    let set_key = format!("{WS_ONLINE_PREFIX}{}", session.user_id);
    let conn_id = session.connection_id.to_string();
    let meta_key = format!("{WS_CONNS_PREFIX}{}", session.connection_id);
    let mut conn = redis.get_connection()?;

    Cmd::new()
        .arg("SADD")
        .arg(&set_key)
        .arg(&conn_id)
        .query::<i32>(&mut *conn.inner_mut())?;
    Cmd::new()
        .arg("EXPIRE")
        .arg(&set_key)
        .arg(WS_ONLINE_TTL_SECS)
        .query::<()>(&mut *conn.inner_mut())?;
    Cmd::new()
        .arg("SETEX")
        .arg(&meta_key)
        .arg(WS_ONLINE_TTL_SECS)
        .arg(session.workspace_name.as_str())
        .query::<()>(&mut *conn.inner_mut())?;
    Ok(())
}

pub fn unregister_redis_online(redis: &AppRedis, session: &WsSession) -> AppResult<()> {
    let set_key = format!("{WS_ONLINE_PREFIX}{}", session.user_id);
    let conn_id = session.connection_id.to_string();
    let meta_key = format!("{WS_CONNS_PREFIX}{}", session.connection_id);
    let mut conn = redis.get_connection()?;

    Cmd::new()
        .arg("SREM")
        .arg(&set_key)
        .arg(&conn_id)
        .query::<i32>(&mut *conn.inner_mut())?;

    let remaining: i32 = Cmd::new()
        .arg("SCARD")
        .arg(&set_key)
        .query(&mut *conn.inner_mut())
        .map_err(AppError::Redis)?;
    if remaining == 0 {
        Cmd::new()
            .arg("DEL")
            .arg(&set_key)
            .query::<()>(&mut *conn.inner_mut())?;
        let pk = format!("{PRESENCE_PREFIX}{}", session.user_id);
        let _ = Cmd::new()
            .arg("DEL")
            .arg(&pk)
            .query::<()>(&mut *conn.inner_mut());
    }
    let _ = Cmd::new()
        .arg("DEL")
        .arg(&meta_key)
        .query::<()>(&mut *conn.inner_mut());
    Ok(())
}

pub fn heartbeat_redis(redis: &AppRedis, session: &WsSession) -> AppResult<()> {
    let set_key = format!("{WS_ONLINE_PREFIX}{}", session.user_id);
    let meta_key = format!("{WS_CONNS_PREFIX}{}", session.connection_id);
    let pk = format!("{PRESENCE_PREFIX}{}", session.user_id);
    let mut conn = redis.get_connection()?;

    let _ = Cmd::new()
        .arg("EXPIRE")
        .arg(&set_key)
        .arg(WS_ONLINE_TTL_SECS)
        .query::<()>(&mut *conn.inner_mut());
    let _ = Cmd::new()
        .arg("SETEX")
        .arg(&meta_key)
        .arg(WS_ONLINE_TTL_SECS)
        .arg(session.workspace_name.as_str())
        .query::<()>(&mut *conn.inner_mut());
    let _ = Cmd::new()
        .arg("SETEX")
        .arg(&pk)
        .arg(WS_ONLINE_TTL_SECS)
        .arg("online")
        .query::<()>(&mut *conn.inner_mut());
    Ok(())
}