use dashmap::DashMap;
use once_cell::sync::Lazy;
use rand::Rng;
use rand::seq::SliceRandom;
use std::sync::Arc;

use crate::anonymous_clients::anonymous_client_connection::AnonymousClientConnection;

static ANONYMOUS_USERS: Lazy<DashMap<u64, Arc<AnonymousClientConnection>>> =
    Lazy::new(|| DashMap::new());

#[allow(dead_code)]
pub async fn add_anonymous_user(connection: Arc<AnonymousClientConnection>) {
    ANONYMOUS_USERS.insert(connection.get_user_id(), connection);
}

#[allow(dead_code)]
pub async fn remove_anonymous_user(user_id: u64) {
    ANONYMOUS_USERS.remove(&user_id);
}

pub async fn get_anonymous_user(user_id: u64) -> Option<Arc<AnonymousClientConnection>> {
    ANONYMOUS_USERS.get(&user_id).map(|c| c.clone())
}

pub async fn get_anonymous_user_by_name(
    username: String,
) -> Option<Arc<AnonymousClientConnection>> {
    let users: Vec<_> = ANONYMOUS_USERS
        .iter()
        .map(|ref_multi| ref_multi.value().clone())
        .collect();

    for user_conn in users {
        if user_conn.get_user_name().await == username {
            return Some(user_conn);
        }
    }

    return None;
}

// TODO: implement check if taken
pub fn generate_username() -> String {
    let adjectives = ["Swift", "Clever", "Brave", "Sneaky", "Fierce"];
    let nouns = ["Tiger", "Eagle", "Shark", "Wolf", "Dragon"];

    let mut rng = rand::thread_rng();
    let adj = adjectives.choose(&mut rng).unwrap();
    let noun = nouns.choose(&mut rng).unwrap();

    let number: u16 = rng.gen_range(0..10000);

    format!("{}{}{}", adj, noun, number)
}