use std::mem::replace;

use color_eyre::eyre::{eyre, Result};
use sqlx::{Connection, SqliteConnection};

#[derive(Debug)]
pub struct Model<T> {
    pub items: Vec<T>,
    pub db: SqliteConnection,
}
impl<T> Model<T> {
    pub fn add_item(&mut self, item: T) -> Result<()> {
        self.items.push(item);
        Ok(())
    }

    pub fn add_to_db(&mut self, item: T) -> Result<()> {
        todo!()
    }

    pub fn update_item(&mut self, item: T, index: i32) -> Result<()> {
        if let Some(current_item) = self.items.get_mut(index as usize)
        {
            let _old_item = replace(current_item, item);
            Ok(())
        } else {
            Err(eyre!(
                "Item doesn't exist in model. Id was {}",
                index
            ))
        }
    }

    pub fn remove_item(&mut self, index: i32) -> Result<()> {
        self.items.remove(index as usize);
        Ok(())
    }

    pub fn get_item(&self, index: i32) -> Option<&T>
    {
        self.items.get(index as usize)
    }

    pub fn find<P>(&self, f: P) -> Option<&T>
    where
        P: FnMut(&&T) -> bool,
    {
        self.items.iter().find(f)
    }

    pub fn insert_item(&mut self, item: T, index: i32) -> Result<()> {
        self.items.insert(index as usize, item);
        Ok(())
    }
}

impl<T> Default for Model<T> {
    fn default() -> Self {
        Self {
            items: vec![],
            db: get_db(),
        }
    }
}

pub fn get_db() -> SqliteConnection {
    let rt = tokio::runtime::Runtime::new().unwrap();
    let mut data = dirs::data_local_dir().unwrap();
    data.push("lumina");
    data.push("library-db.sqlite3");
    let mut db_url = String::from("sqlite://");
    db_url.push_str(data.to_str().unwrap());
    rt.block_on(async {
        SqliteConnection::connect(&db_url)
            .await
            .expect("problems")
    })
}

pub trait Modeling {
    type Item;

    fn setup_db() -> SqliteConnection {
        let rt = tokio::runtime::Runtime::new().unwrap();
        let mut data = dirs::data_local_dir().unwrap();
        data.push("lumina");
        data.push("library-db.sqlite3");
        let mut db_url = String::from("sqlite://");
        db_url.push_str(data.to_str().unwrap());
        rt.block_on(async {
            SqliteConnection::connect(&db_url)
                .await
                .expect("problems")
        })
    }
}

#[cfg(test)]
mod test {}