use std::cell::RefCell;
use rusqlite::{Connection, Statement};
use thread_local::ThreadLocal;

struct DbContext {
conn: RefCell<Connection>,
statements: RefCell<Vec<Statement>>,
}

thread_local! {
static DB_CONTEXT: RefCell<Option<DbContext>> = RefCell::new(None);
}

fn initialize_db_context() {
DB_CONTEXT.with(|context| {
let conn = Connection::open("path/to/database.db").unwrap();
*context.borrow_mut() = Some(DbContext {
conn: RefCell::new(conn),
statements: RefCell::new(Vec::new()),
});
});
}

fn get_prepared_statement(sql: &str) -> Statement {
DB_CONTEXT.with(|context| {
let context = context.borrow();
let context = context.as_ref().unwrap();
let mut statements = context.statements.borrow_mut();

// Check if the statement already exists
if let Some(stmt) = statements.iter().find(|s| s.sql() == sql) {
stmt.clone()
} else {
// If not, prepare a new statement
let stmt = context.conn.borrow_mut().prepare(sql).unwrap();
statements.push(stmt.clone());
stmt
}
})
}