use axum::response::Response;
use lazy_static::lazy_static;
use rand::{distributions::Alphanumeric, Rng};
use tokio::sync::Mutex;

use crate::{
    console::sanatize,
    types::{http::ResponseCode, post::Post, session::Session, user::User, comment::Comment, like::Like},
};

lazy_static! {
    static ref SECRET: Mutex<String> = Mutex::new(String::new());
}

pub fn new_secret() -> String {
    rand::thread_rng()
        .sample_iter(&Alphanumeric)
        .take(32)
        .map(char::from)
        .collect()
}

pub async fn get_secret() -> String {
    let mut secret = SECRET.lock().await;
    if secret.is_empty() {
        *secret = new_secret();
    }
    secret.clone()
}

pub async fn regen_secret() -> String {
    let mut secret = SECRET.lock().await;
    *secret = new_secret();
    secret.clone()
}

pub fn generate_users() -> Response {
    let users = match User::reterieve_all() {
        Ok(users) => users,
        Err(err) => return err,
    };

    let mut html = r#"
        <tr>
            <th>User ID</th>
            <th>First Name</th>
            <th>Last Name</th>
            <th>Email</th>
            <th>Password</th>
            <th>Gender</th>
            <th>Date</th>
            <th>Day</th>
            <th>Month</th>
            <th>Year</th>
        </tr>
    "#
    .to_string();

    for user in users {
        html.push_str(
            &format!("<tr><td>{}</td><td>{}</td><td>{}</td><td>{}</td><td>{}</td><td>{}</td><td>{}</td><td>{}</td><td>{}</td><td>{}</td></tr>",
                user.user_id, sanatize(&user.firstname), sanatize(&user.lastname), sanatize(&user.email), sanatize(&user.password),
                sanatize(&user.gender), user.date, user.day, user.month, user.year
            )
        );
    }

    ResponseCode::Success.text(&html)
}

pub fn generate_posts() -> Response {
    let posts = match Post::reterieve_all() {
        Ok(posts) => posts,
        Err(err) => return err,
    };

    let mut html = r#"
        <tr>
            <th>Post ID</th>
            <th>User ID</th>
            <th>Content</th>
            <th>Date</th>
        </tr>
    "#
    .to_string();

    for post in posts {
        html.push_str(&format!(
            "<tr><td>{}</td><td>{}</td><td>{}</td><td>{}</td></tr>",
            post.post_id,
            post.user_id,
            sanatize(&post.content),
            post.date
        ));
    }

    ResponseCode::Success.text(&html)
}

pub fn generate_sessions() -> Response {
    let sessions = match Session::reterieve_all() {
        Ok(sessions) => sessions,
        Err(err) => return err,
    };

    let mut html = r#"
        <tr>
            <th>User ID</th>
            <th>Token</th>
        </tr>
    "#
    .to_string();

    for session in sessions {
        html.push_str(&format!(
            "<tr><td>{}</td><td>{}</td></tr>",
            session.user_id, session.token
        ));
    }

    ResponseCode::Success.text(&html)
}

pub fn generate_comments() -> Response {
    let comments = match Comment::reterieve_all() {
        Ok(comments) => comments,
        Err(err) => return err,
    };

    let mut html = r#"
        <tr>
            <th>Comment ID</th>
            <th>User ID</th>
            <th>Post ID</th>
            <th>Content</th>
            <th>Date</th>
        </tr>
    "#
    .to_string();

    for comment in comments {
        html.push_str(&format!(
            "<tr><td>{}</td><td>{}</td><td>{}</td><td>{}</td><td>{}</td></tr>",
            comment.comment_id, comment.user_id, comment.post_id, sanatize(&comment.content), comment.date
        ));
    }

    ResponseCode::Success.text(&html)
}

pub fn generate_likes() -> Response {
    let likes = match Like::reterieve_all() {
        Ok(likes) => likes,
        Err(err) => return err,
    };

    let mut html = r#"
        <tr>
            <th>User ID</th>
            <th>Post ID</th>
        </tr>
    "#
    .to_string();

    for like in likes {
        html.push_str(&format!(
            "<tr><td>{}</td><td>{}</td></tr>",
            like.user_id, like.post_id
        ));
    }

    ResponseCode::Success.text(&html)
}