use cache::CacheConn;
use config::redis::RedisConfig;

use crate::error::InternalError;

#[derive(Clone)]
pub struct NonceStore {
    conn: CacheConn,
}

impl NonceStore {
    pub fn from_conn(conn: CacheConn) -> Self {
        Self { conn }
    }

    pub async fn connect(config: &RedisConfig, _prefix: &str) -> Result<Self, InternalError> {
        let conn = CacheConn::init(config)
            .await
            .map_err(|error| InternalError::Config(format!("redis: {error}")))?;
        Ok(Self::from_conn(conn))
    }

    /// `SET key 1 NX EX ttl`：第一次出现返回 true，重放返回 false。
    pub async fn accept(&self, key: &str, ttl_secs: u64) -> Result<bool, InternalError> {
        let mut conn = self.conn.clone();
        let result: Option<String> = redis::cmd("SET")
            .arg(key)
            .arg("1")
            .arg("NX")
            .arg("EX")
            .arg(ttl_secs.max(1))
            .query_async(&mut conn)
            .await?;
        Ok(result.is_some())
    }
}

pub fn key(audience: &str, nonce: &str) -> String {
    format!("event:identity:nonce:{audience}:{nonce}")
}