use rusqlite::Connection;

pub fn get_setting(key: String, connection: &Connection) -> Option<String> {
    let mut stmt = connection
        .prepare("SELECT value FROM settings WHERE id = ?1")
        .unwrap();
    let mut iter = stmt.query_map((key,), |row| row.get(0)).unwrap();

    if let Some(row) = iter.next()
        && let Ok(value) = row
    {
        return Some(value);
    }

    None
}

pub fn set_setting(key: String, value: String, connection: &Connection) {
    let current = get_settings_list(connection);
    if current.contains(&key) {
        update_settings(key, value, connection);
    } else {
        create_settings(key, value, connection);
    }
}

pub fn delete_settings(key: String, connection: &Connection) {
    connection
        .execute("DELETE FROM settings WHERE id = ?1", (&key,))
        .unwrap();
}

fn create_settings(key: String, value: String, connection: &Connection) {
    connection
        .execute(
            "INSERT into settings (id, value) VALUES (?1, ?2)",
            (key, value),
        )
        .unwrap();
}

fn update_settings(key: String, value: String, connection: &Connection) {
    connection
        .execute(
            "update settings
set value = ?2
where id = ?1",
            (key, value),
        )
        .unwrap();
}

fn get_settings_list(connection: &Connection) -> Vec<String> {
    let mut stmt = connection.prepare("SELECT id FROM settings").unwrap();
    let iter = stmt.query_map((), |row| row.get(0)).unwrap();
    iter.map(|row| row.unwrap()).collect()
}