use rand::RngExt;

use rusqlite::{Connection, params};

use serde::{Deserialize, Serialize};

use std::fmt;

use std::sync::{Arc, Mutex};

pub type Db = Arc<Mutex<Connection>>;

#[derive(Debug)]

pub enum DbError {

    Sqlite(rusqlite::Error),

    LockPoisoned,

}

impl fmt::Display for DbError {

    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {

        match self {

            DbError::Sqlite(e) => write!(f, "Database error: {}", e),

            DbError::LockPoisoned => write!(f, "Database lock poisoned"),

        }

    }

}

impl std::error::Error for DbError {}

impl From<rusqlite::Error> for DbError {

    fn from(e: rusqlite::Error) -> Self {

        DbError::Sqlite(e)

    }

}

#[derive(Serialize, Deserialize)]

pub struct Snippet {

    pub id: i64,

    pub short_id: String,

    pub content: String,

    pub name: String,

}

const ALPHABET: &[u8] = b"0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz";

fn generate_short_id() -> String {

    let mut rng = rand::rng();

    (0..10)

        .map(|_| ALPHABET[rng.random_range(0..ALPHABET.len())] as char)

        .collect()

}

pub fn init_db() -> Result<Db, DbError> {

    let conn = Connection::open("sipp.sqlite")?;

    conn.execute(

        "CREATE TABLE IF NOT EXISTS snippets (

            id INTEGER PRIMARY KEY AUTOINCREMENT,

            short_id TEXT NOT NULL UNIQUE,

            content TEXT NOT NULL,

            name TEXT NOT NULL

        )",

        [],

    )?;

    Ok(Arc::new(Mutex::new(conn)))

}

pub fn create_snippet(db: &Db, name: &str, content: &str) -> Result<Snippet, DbError> {

    let conn = db.lock().map_err(|_| DbError::LockPoisoned)?;

    let short_id = generate_short_id();

    conn.execute(

        "INSERT INTO snippets (short_id, content, name) VALUES (?1, ?2, ?3)",

        params![short_id, content, name],

    )?;

    let id = conn.last_insert_rowid();

    Ok(Snippet {

        id,

        short_id,

        content: content.to_string(),

        name: name.to_string(),

    })

}

pub fn get_snippet_by_short_id(db: &Db, short_id: &str) -> Result<Option<Snippet>, DbError> {

    let conn = db.lock().map_err(|_| DbError::LockPoisoned)?;

    match conn.query_row(

        "SELECT id, short_id, content, name FROM snippets WHERE short_id = ?1",

        params![short_id],

        |row| {

            Ok(Snippet {

                id: row.get(0)?,

                short_id: row.get(1)?,

                content: row.get(2)?,

                name: row.get(3)?,

            })

        },

    ) {

        Ok(snippet) => Ok(Some(snippet)),

        Err(rusqlite::Error::QueryReturnedNoRows) => Ok(None),

        Err(e) => Err(DbError::Sqlite(e)),

    }

}

pub fn get_all_snippets(db: &Db) -> Result<Vec<Snippet>, DbError> {

    let conn = db.lock().map_err(|_| DbError::LockPoisoned)?;

    let mut stmt = conn

        .prepare("SELECT id, short_id, content, name FROM snippets ORDER BY id DESC")?;

    let snippets = stmt.query_map([], |row| {

        Ok(Snippet {

            id: row.get(0)?,

            short_id: row.get(1)?,

            content: row.get(2)?,

            name: row.get(3)?,

        })

    })?

    .filter_map(|r| r.ok())

    .collect();

    Ok(snippets)

}

pub fn delete_snippet_by_short_id(db: &Db, short_id: &str) -> Result<bool, DbError> {

    let conn = db.lock().map_err(|_| DbError::LockPoisoned)?;

    let rows_affected = conn.execute(

        "DELETE FROM snippets WHERE short_id = ?1",

        params![short_id],

    )?;

    Ok(rows_affected > 0)

}

pub fn update_snippet_by_short_id(

    db: &Db,

    short_id: &str,

    name: &str,

    content: &str,

) -> Result<Option<Snippet>, DbError> {

    let conn = db.lock().map_err(|_| DbError::LockPoisoned)?;

    let rows_affected = conn.execute(

        "UPDATE snippets SET name = ?1, content = ?2 WHERE short_id = ?3",

        params![name, content, short_id],

    )?;

    if rows_affected == 0 {

        return Ok(None);

    }

    match conn.query_row(

        "SELECT id, short_id, content, name FROM snippets WHERE short_id = ?1",

        params![short_id],

        |row| {

            Ok(Snippet {

                id: row.get(0)?,

                short_id: row.get(1)?,

                content: row.get(2)?,

                name: row.get(3)?,

            })

        },

    ) {

        Ok(snippet) => Ok(Some(snippet)),

        Err(rusqlite::Error::QueryReturnedNoRows) => Ok(None),

        Err(e) => Err(DbError::Sqlite(e)),

    }

}