akar-main 0.2.1

//! Database — the main entry point for Akar.

use akar_catalog::Catalog;
use akar_common::file_system::VirtualFileSystemRegistry;
use akar_common::memory::MemoryManager;
use akar_common::task_system::TaskSystem;
use akar_extension::{ExtensionContext, ExtensionRegistry};
use akar_function::FunctionRegistry;
use akar_storage::StorageManager;
use akar_storage::stats::StatsStore;
use akar_storage::table::ColumnDefinition;
use akar_transaction::TransactionManager;
use std::collections::HashMap;
use std::path::{Path, PathBuf};
use std::sync::atomic::{AtomicBool, AtomicU64, Ordering};
use std::sync::{Arc, Mutex, OnceLock};

/// Name of the file that holds the serialized system catalog.
///
/// The catalog file is the source of truth for DDL (schema changes survive
/// restarts via this file); WAL records only carry DML.
pub const CATALOG_FILE_NAME: &str = "catalog.json";

/// Name of the file holding the cross-process lock.
///
/// The lock file is created with an exclusive lock by the first process to
/// open a database directory and prevents a second process from opening the
/// same directory concurrently (P45.4). Read-only opens take a shared lock.
pub const LOCK_FILE_NAME: &str = "akar.lock";

/// Configuration for the database.
#[derive(Debug, Clone)]
pub struct SystemConfig {
    pub buffer_pool_size: u64,
    pub max_num_threads: u64,
    pub enable_compression: bool,
    pub read_only: bool,
    pub max_db_size: u64,
    pub auto_checkpoint: bool,
    pub checkpoint_threshold: i64,
    /// When true, multiple write transactions can run concurrently.
    /// When false, only one write transaction at a time is allowed.
    pub concurrent_writes: bool,
    /// Memory threshold (in bytes) for triggering disk spilling during
    /// bulk ingest (COPY FROM, large batch inserts).
    ///
    /// When a NodeGroup's estimated in-memory size exceeds this value,
    /// its data is spilled to a temp file and the in-memory buffer is
    /// cleared. After all rows are ingested, spilled files are merged
    /// back into persistent storage.
    ///
    /// Default: 80% of `buffer_pool_size`. Set to 0 to disable spilling.
    pub spill_threshold: u64,
}

impl Default for SystemConfig {
    fn default() -> Self {
        Self {
            buffer_pool_size: 0,
            max_num_threads: 0,
            enable_compression: true,
            read_only: false,
            max_db_size: u64::from(u32::MAX),
            auto_checkpoint: true,
            checkpoint_threshold: -1,
            concurrent_writes: true,
            // Default: 80% of buffer_pool_size, or 0 if not set
            spill_threshold: 0,
        }
    }
}

/// The main database instance.
///
/// Manages the storage engine, catalog, transaction manager, and all subsystem
/// instances. Create via [`Database::new`] with a path and [`SystemConfig`].
///
/// # Examples
///
/// ```no_run
/// use akar_main::database::{Database, SystemConfig};
/// use akar_main::connection::Connection;
///
/// let db = std::sync::Arc::new(Database::new("./my_db", SystemConfig::default())?);
/// let conn = Connection::new(&db);
/// conn.query("CREATE NODE TABLE Person(name STRING, age INT64, PRIMARY KEY(name))")?;
/// # Ok::<(), String>(())
/// ```
/// Cross-process path lock, reentrant within this process (P53.35, E3).
///
/// The first `Database` on a path takes the OS-level lock and keeps the file
/// handle here; later opens on the *same path in this process* share that
/// handle (refcount) instead of failing — this is what allows the kairos
/// harness pattern of a fixture store and a fresh store on one path in the
/// same process. The last `Database` to drop removes the entry, closing the
/// handle and releasing the OS lock. Cross-process exclusion is unchanged:
/// every process owns a private registry, so a second process still fails to
/// acquire the OS lock while the first holds it.
static PROCESS_PATH_LOCKS: OnceLock<Mutex<HashMap<PathBuf, (std::fs::File, u32)>>> = OnceLock::new();

fn process_path_locks() -> &'static Mutex<HashMap<PathBuf, (std::fs::File, u32)>> {
    PROCESS_PATH_LOCKS.get_or_init(|| Mutex::new(HashMap::new()))
}

/// Slot guard in `PROCESS_PATH_LOCKS`. On drop it decrements the refcount for
/// the path and removes the entry (dropping the shared OS lock handle) once
/// the count reaches zero.
struct PathLock {
    key: PathBuf,
}

impl Drop for PathLock {
    fn drop(&mut self) {
        let mut reg = process_path_locks().lock().unwrap();
        if let Some((_, count)) = reg.get_mut(&self.key) {
            *count -= 1;
            if *count == 0 {
                reg.remove(&self.key);
            }
        }
    }
}

/// Resolve the canonical lock-file path for a database directory so that
/// alternative spellings of the same directory share one registry slot.
fn lock_key(db_path: &Path) -> PathBuf {
    std::fs::canonicalize(db_path)
        .unwrap_or_else(|_| db_path.to_path_buf())
        .join(LOCK_FILE_NAME)
}

#[allow(dead_code)]
pub struct Database {
    pub(crate) storage_manager: Arc<StorageManager>,
    pub(crate) catalog: Arc<Mutex<Catalog>>,
    pub(crate) transaction_manager: Arc<TransactionManager>,
    pub(crate) function_registry: Arc<Mutex<FunctionRegistry>>,
    pub(crate) task_system: Arc<TaskSystem>,
    pub(crate) memory_manager: Arc<MemoryManager>,
    pub(crate) extension_registry: Mutex<ExtensionRegistry>,
    pub(crate) stats_store: Arc<Mutex<StatsStore>>,
    pub(crate) vfs: Arc<VirtualFileSystemRegistry>,
    /// Configuration used at database creation time.
    pub(crate) config: SystemConfig,
    /// Runtime-overridable spill threshold via `SET spill_threshold`.
    /// The override is tracked explicitly so `SET spill_threshold=0` disables
    /// spilling instead of falling back to the config default (P52.50).
    spill_threshold_override: AtomicU64,
    spill_threshold_overridden: AtomicBool,
    /// Registry slot holding this process's share of the cross-process path
    /// lock (see `PROCESS_PATH_LOCKS` / `PathLock`).
    _lock: Option<PathLock>,
}

impl Database {
    /// Override the spill threshold at runtime (via `SET spill_threshold`).
    /// A value of `0` explicitly disables spilling.
    pub fn set_spill_threshold(&self, bytes: u64) {
        self.spill_threshold_override.store(bytes, Ordering::Relaxed);
        self.spill_threshold_overridden.store(true, Ordering::Relaxed);
        // Propagate the runtime override so bulk ingest on existing node
        // tables spills at the new threshold (P51.44).
        self.storage_manager.set_spiller(self.spiller());
    }

    /// Get the effective spill threshold in bytes.
    ///
    /// Priority:
    /// 1. Runtime override (via `SET spill_threshold`; `0` = explicitly disabled)
    /// 2. `config.spill_threshold`
    /// 3. 80% of `buffer_pool_size`
    /// 4. 0 (disabled)
    pub fn effective_spill_threshold(&self) -> u64 {
        if self.spill_threshold_overridden.load(Ordering::Relaxed) {
            return self.spill_threshold_override.load(Ordering::Relaxed);
        }
        if self.config.spill_threshold > 0 {
            return self.config.spill_threshold;
        }
        if self.config.buffer_pool_size > 0 {
            return (self.config.buffer_pool_size as f64 * 0.8) as u64;
        }
        0
    }

    /// Create a `Spiller` instance using the current database configuration.
    ///
    /// Returns `None` if spilling is disabled (threshold is 0).
    pub fn spiller(&self) -> Option<Arc<akar_storage::Spiller>> {
        let threshold = self.effective_spill_threshold();
        if threshold == 0 {
            return None;
        }
        let spill_dir = self.storage_manager.db_path().join("spill");
        Some(Arc::new(akar_storage::Spiller::new(spill_dir, threshold)))
    }

    /// Create a [`StorageDriver`] for programmatic storage-level access
    /// (page counts, buffer stats, file sizes, table counts) without Cypher.
    pub fn storage_driver(&self) -> crate::storage_driver::StorageDriver {
        crate::storage_driver::StorageDriver::new(self.storage_manager.clone(), self.catalog.clone(), self.vfs.clone())
    }

    /// Get a reference to the schema catalog for programmatic metadata access.
    pub fn catalog(&self) -> Arc<Mutex<Catalog>> {
        self.catalog.clone()
    }

    /// Get the data table catalog for programmatic data access.
    ///
    /// Prefer using the unified DDL methods on `Database` instead of
    /// accessing the table catalog directly.
    pub fn table_catalog(&self) -> Arc<akar_storage::TableCatalog> {
        self.storage_manager.table_catalog()
    }

    // ── Unified DDL operations ──────────────────────────────────────────
    //
    // These methods ensure storage-level table creation/deletion is atomic.
    // Schema entries are managed by the binder (via `Catalog`) during the
    // bind phase.  These methods handle the data-level side:
    // storage table creation, serial sequences, ART indexes.

    /// Create a node table: data table + serial sequences + ART index.
    ///
    /// The schema entry is created by the binder during `bind()`.
    /// This method creates the storage-level table and associated resources.
    pub fn create_node_table(&self, name: String, columns: Vec<akar_catalog::CatalogColumn>) -> Result<u64, String> {
        // 1. Create the data-level table
        let storage_columns: Vec<ColumnDefinition> = columns
            .iter()
            .map(|c| ColumnDefinition {
                name: c.name.clone(),
                logical_type: c.logical_type,
                is_primary_key: c.is_primary_key,
                compression: c.compression,
            })
            .collect();
        let node_table = self.storage_manager.create_node_table(name.clone(), storage_columns);
        let table_id = node_table.table_id;

        // 2. Auto-create backing sequences for SERIAL columns
        {
            let mut cat = self.catalog.lock().map_err(|e| format!("Lock poisoned: {e}"))?;
            for col in &columns {
                if col.logical_type == akar_common::types::LogicalTypeID::Serial {
                    if let akar_catalog::CatalogResult::Created { .. } = cat.create_serial_sequence(&name, &col.name) {
                        tracing::info!("Created serial sequence for {name}.{}", col.name);
                    }
                }
            }
        }

        // 3. Auto-create ART index for primary key
        if columns.iter().any(|c| c.is_primary_key) {
            let index_name = format!("{name}_pk_idx");
            self.storage_manager
                .create_art_index(&name, &index_name)
                .map_err(|e| format!("Failed to create ART PK index for table '{name}': {e}"))?;
        }

        tracing::info!("Created node table '{name}'");
        Ok(table_id)
    }

    /// Create a rel table: data table.
    ///
    /// The schema entry is created by the binder during `bind()`.
    /// This method creates the storage-level table.
    pub fn create_rel_table(
        &self,
        name: String,
        src_table_id: u64,
        dst_table_id: u64,
        columns: Vec<akar_catalog::CatalogColumn>,
    ) -> Result<u64, String> {
        // 1. Create the data-level table
        let storage_columns: Vec<ColumnDefinition> = columns
            .iter()
            .map(|c| ColumnDefinition {
                name: c.name.clone(),
                logical_type: c.logical_type,
                is_primary_key: c.is_primary_key,
                compression: c.compression,
            })
            .collect();
        let rel_table =
            self.storage_manager
                .create_rel_table(name.clone(), src_table_id, dst_table_id, storage_columns);
        let table_id = rel_table.table_id;

        tracing::info!("Created rel table '{name}'");
        Ok(table_id)
    }

    /// Drop a table: serial sequences + data table + schema entry.
    pub fn drop_table(&self, name: &str) -> Result<(), String> {
        // 1. Drop auto-created serial sequences. Sequences are named
        // `{table}_{column}_serial`, so a prefix match on `{name}_` would also
        // drop sequences owned by tables sharing the prefix (dropping `person`
        // would remove `person_x`'s `person_x_id_serial`). Enumerate the
        // table's own SERIAL columns and drop exactly their sequences (P51.28).
        {
            let mut cat = self.catalog.lock().map_err(|e| format!("Lock poisoned: {e}"))?;
            let node_cols: Vec<String> = cat
                .node_tables()
                .into_iter()
                .filter(|t| t.name == name)
                .flat_map(|t| t.columns.iter())
                .filter(|c| c.logical_type == akar_common::types::LogicalTypeID::Serial)
                .map(|c| c.name.clone())
                .collect();
            let rel_cols: Vec<String> = cat
                .rel_tables()
                .into_iter()
                .filter(|t| t.name == name)
                .flat_map(|t| t.columns.iter())
                .filter(|c| c.logical_type == akar_common::types::LogicalTypeID::Serial)
                .map(|c| c.name.clone())
                .collect();
            for col in node_cols.into_iter().chain(rel_cols) {
                let seq_name = akar_catalog::SequenceEntry::get_serial_name(name, &col);
                if let akar_catalog::CatalogResult::Dropped { .. } = cat.drop_sequence(&seq_name) {
                    tracing::info!("Dropped serial sequence '{seq_name}'");
                }
            }
            // Drop schema entry
            cat.drop_table(name);
        }

        // 2. Drop the data table
        let table_catalog = self.storage_manager.table_catalog();
        let node_tid = table_catalog.get_node_table_by_name(name).map(|t| t.table_id);
        let rel_tid = table_catalog.get_rel_table_by_name(name).map(|t| t.table_id);
        table_catalog.drop_node_table(name);
        table_catalog.drop_rel_table(name);
        if let Some(tid) = node_tid {
            self.storage_manager.drop_table_persistence(tid);
        }
        if let Some(tid) = rel_tid {
            self.storage_manager.drop_table_persistence(tid);
        }

        tracing::info!("Dropped table '{name}'");
        Ok(())
    }

    /// Create a vector index: data index + auto-populate.
    ///
    /// The schema entry is created by the binder during `bind()`.
    #[cfg(feature = "vector-extension")]
    pub fn create_vector_index(
        &self,
        index_name: String,
        table_name: String,
        column_name: String,
        metric: akar_vector::hnsw::DistanceMetric,
        dimensions: u32,
    ) -> Result<(), String> {
        // 1. Create the data-level index
        self.storage_manager.create_vector_index(
            index_name.clone(),
            table_name.clone(),
            column_name.clone(),
            metric,
            dimensions,
        );

        // 2. Auto-populate from existing table data
        let table_catalog = self.storage_manager.table_catalog();
        if let Some(table) = table_catalog.get_node_table_by_name(&table_name) {
            let col_idx = table.columns.iter().position(|c| c.name == column_name);
            if let Some(col_idx) = col_idx {
                for row_id in 0..table.num_rows as usize {
                    if let Some(val) = table.get_value(row_id, col_idx) {
                        if let Ok(vec) = akar_storage::extract_f64_list_from_value(val) {
                            if let Some(mut vi) = table_catalog.get_vector_index_by_name_mut(&index_name) {
                                vi.hnsw_mut().insert(vec, row_id);
                            }
                        }
                    }
                }
            }
        }

        tracing::info!("Created vector index '{index_name}'");
        Ok(())
    }

    /// Rebuild the HNSW graph of every vector index on the given tables.
    ///
    /// The index was only populated during `CREATE VECTOR INDEX`; without this
    /// hook the graph served stale/positional row ids after INSERT/DELETE
    /// (P52.38).
    #[cfg(feature = "vector-extension")]
    pub fn refresh_vector_indexes(&self, table_ids: &[u64]) {
        self.storage_manager
            .table_catalog()
            .refresh_vector_indexes_for_tables(table_ids);
    }

    /// No-op when the vector extension is not compiled in.
    #[cfg(not(feature = "vector-extension"))]
    pub fn refresh_vector_indexes(&self, _table_ids: &[u64]) {}

    /// Create an ART index on a node table: data index.
    ///
    /// The schema entry is created by the binder during `bind()`.
    pub fn create_art_index(&self, table_name: &str, index_name: &str) -> Result<(), String> {
        self.storage_manager.create_art_index(table_name, index_name)?;
        Ok(())
    }

    /// Drop an ART index from a node table: data index + schema entry.
    pub fn drop_art_index(&self, table_name: &str, _index_name: &str) -> Result<(), String> {
        self.storage_manager.drop_art_index(table_name, _index_name)?;

        // Update the schema entry
        {
            let mut cat = self.catalog.lock().map_err(|e| format!("Lock poisoned: {e}"))?;
            if let Some(entry) = cat.get_entry_by_name_mut(table_name) {
                if let akar_catalog::CatalogEntry::NodeTable(t) = entry {
                    t.index_type = None;
                    t.index_name = None;
                }
            }
        }

        Ok(())
    }

    /// Get the number of rows in a table by name.
    pub fn table_num_rows(&self, name: &str) -> u64 {
        self.storage_manager.table_catalog().node_table_num_rows(name)
    }

    /// Get table IDs for write operations (used by transaction locking).
    ///
    /// Propagates a poisoned catalog lock instead of silently treating it as a
    /// missing table (Audit 2 NIT).
    pub fn get_table_id(&self, name: &str) -> Result<Option<u64>, String> {
        let catalog = self.catalog.lock().map_err(|e| format!("Lock poisoned: {e}"))?;
        Ok(catalog.get_table_id(name))
    }

    /// Path of the persisted catalog file for this database.
    pub fn catalog_file_path(&self) -> PathBuf {
        self.storage_manager.db_path().join(CATALOG_FILE_NAME)
    }

    /// Connect to a remote Akar server (embedded server mode, P47).
    ///
    /// The server process owns the [`Database`] instance and its exclusive file
    /// lock; remote clients only talk to the server over TCP and never open the
    /// database directory themselves. Multiple processes can therefore access
    /// the same database concurrently: one writer via the server plus any
    /// number of read-only clients (and additional writers when
    /// `concurrent_writes` is enabled).
    ///
    /// See [`crate::remote::RemoteDatabase`] for the returned handle's API.
    pub fn connect_tcp(addr: impl Into<String>) -> Result<crate::remote::RemoteDatabase, String> {
        crate::remote::RemoteDatabase::connect_tcp(addr)
    }

    /// Returns `true` when the database runs fully in-memory (`:memory:`),
    /// in which case catalog persistence is skipped.
    pub fn is_in_memory(&self) -> bool {
        self.storage_manager.db_path().to_string_lossy() == ":memory:"
    }

    /// Persist the system catalog to disk.
    ///
    /// Called after every DDL statement so schema changes survive restarts.
    /// The write is atomic (temp file + rename) and no-ops for `:memory:`
    /// databases.
    pub fn persist_catalog(&self) -> Result<(), String> {
        if self.is_in_memory() {
            return Ok(());
        }
        let catalog = self.catalog.lock().map_err(|e| format!("Lock poisoned: {e}"))?;
        catalog
            .save_to_path(&self.catalog_file_path())
            .map_err(|e| format!("Failed to persist catalog: {e}"))
    }

    /// Restore storage-level tables from the loaded catalog.
    ///
    /// Called during `Database::new` after the catalog file is loaded so that
    /// WAL DML replay and subsequent queries reference the same table IDs that
    /// were in use when the database was shut down.
    fn restore_storage_from_catalog(&self) {
        let catalog = match self.catalog.lock() {
            Ok(c) => c,
            Err(_) => return,
        };
        for entry in catalog.all_entries() {
            match entry {
                akar_catalog::CatalogEntry::NodeTable(t) => {
                    let columns: Vec<_> = t.columns.iter().map(ColumnDefinition::from).collect();
                    let index_name = if t.has_art_index() {
                        t.index_name.as_deref()
                    } else {
                        None
                    };
                    self.storage_manager
                        .restore_node_table(t.table_id, t.name.clone(), columns, index_name);
                }
                akar_catalog::CatalogEntry::RelTable(t) => {
                    let columns: Vec<_> = t.columns.iter().map(ColumnDefinition::from).collect();
                    self.storage_manager.restore_rel_table(
                        t.table_id,
                        t.name.clone(),
                        t.src_table_id,
                        t.dst_table_id,
                        columns,
                    );
                }
                _ => {}
            }
        }
    }

    /// Open or create a database at the given path.
    ///
    /// # Arguments
    /// * `db_path` — filesystem path where database files are stored.
    /// * `config` — buffer pool size, thread count, etc.
    ///
    /// # Errors
    /// Returns `Err` if the path is not writable or if existing data is corrupt.
    pub fn new(db_path: impl Into<PathBuf>, config: SystemConfig) -> Result<Self, String> {
        let db_path = db_path.into();
        let is_memory = db_path.to_string_lossy() == ":memory:";

        // Multi-process guard: take a file lock on <db_path>/akar.lock so two
        // processes cannot open the same database directory concurrently.
        // Read-only opens take a shared lock (multiple readers allowed); write
        // opens take an exclusive lock. In-process reopens of the same path
        // share the held lock via PROCESS_PATH_LOCKS (P53.35, E3).
        let lock = if is_memory {
            None
        } else {
            std::fs::create_dir_all(&db_path)
                .map_err(|e| format!("Failed to create database directory '{}': {e}", db_path.display()))?;
            let key = lock_key(&db_path);
            let mut reg = process_path_locks().lock().unwrap();
            match reg.get_mut(&key) {
                // Already open in this process: share the held OS lock.
                Some((_, count)) => {
                    *count += 1;
                }
                None => {
                    let file = std::fs::OpenOptions::new()
                        .create(true)
                        .read(true)
                        .write(true)
                        .truncate(false)
                        .open(&key)
                        .map_err(|e| format!("Failed to open lock file '{}': {e}", key.display()))?;
                    let result = if config.read_only {
                        file.try_lock_shared()
                    } else {
                        file.try_lock()
                    };
                    result
                        .map_err(|_| format!("Database '{}' is already open by another process", db_path.display()))?;
                    reg.insert(key.clone(), (file, 1));
                }
            }
            Some(PathLock { key })
        };

        let memory_manager = Arc::new(MemoryManager::new(config.max_db_size));
        let task_system = Arc::new(TaskSystem::new(config.max_num_threads as usize));

        // Load a previously-persisted catalog (if any). Fresh databases and
        // databases created before catalog persistence start with an empty
        // catalog, preserving backward compatibility.
        let catalog_file = db_path.join(CATALOG_FILE_NAME);
        let catalog = Arc::new(Mutex::new(
            Catalog::load_from_path(&catalog_file)
                .map_err(|e| format!("Failed to load persisted catalog: {e}"))?
                .unwrap_or_default(),
        ));
        let transaction_manager = {
            let tx_config = akar_transaction::TransactionManagerConfig {
                concurrent_writes: config.concurrent_writes,
            };
            Arc::new(TransactionManager::new_with_config(tx_config))
        };
        let function_registry = Arc::new(Mutex::new(FunctionRegistry::new()));
        let storage_manager = Arc::new(StorageManager::new(db_path.clone(), memory_manager.clone()));
        let stats_store = Arc::new(Mutex::new(StatsStore::new()));
        let vfs = Arc::new(VirtualFileSystemRegistry::new());

        let mut db = Self {
            storage_manager,
            catalog,
            transaction_manager,
            function_registry,
            task_system,
            memory_manager,
            extension_registry: Mutex::new(ExtensionRegistry::new()),
            stats_store,
            vfs,
            spill_threshold_override: AtomicU64::new(0),
            spill_threshold_overridden: AtomicBool::new(false),
            _lock: lock,
            config,
        };

        // Recreate storage-level tables from the restored catalog (if any) so
        // WAL DML replay below operates on the same table IDs.
        db.restore_storage_from_catalog();

        // Propagate the configured spiller (if any) so bulk ingest on the
        // restored tables spills to disk once a NodeGroup exceeds the memory
        // threshold (P51.44).
        db.storage_manager.set_spiller(db.spiller());

        // Load built-in extensions
        db.register_builtin_extensions();

        // Load all registered extensions
        {
            let mut ext_registry = db
                .extension_registry
                .lock()
                .map_err(|e| format!("Lock poisoned: {e}"))?;
            let context = ExtensionContext::new(db.function_registry.clone(), db.catalog.clone(), db.vfs.clone());
            for result in ext_registry.load_all(&context) {
                match result {
                    (name, Ok(())) => tracing::info!("Extension '{name}' loaded successfully"),
                    (name, Err(e)) => tracing::warn!("Extension '{name}' failed to load: {e}"),
                }
            }
        }

        // Register built-in sequence functions (nextval/currval)
        {
            let mut reg = db.function_registry.lock().map_err(|e| format!("Lock poisoned: {e}"))?;
            crate::connection::utils::register_sequence_scalars(&mut reg, db.catalog.clone());
        }

        // Recovery (P45.4, amended P60.2): `recover()` loads the durable
        // column mirrors FIRST — the state at the last checkpoint — then
        // replays the WAL's typed Insert/Delete/Update records (including
        // those decoded from LocalWALData blobs) on top. This reconstructs
        // full state even when no checkpoint ever ran, without double-
        // applying rows.
        //
        // (P61.3) Recovery MUST NOT fall back to a fresh, empty database on
        // failure: that silently destroys every committed write in the WAL.
        // On replay/persist error the database fails to open and the WAL file
        // is left untouched, so an operator can still repair it.
        if let Err(e) = db.storage_manager.recover() {
            return Err(format!(
                "WAL recovery failed (database may need manual repair): {e}. \
                     Refusing to start with an empty database — check the WAL."
            ));
        }

        Ok(db)
    }

    /// Register built-in extensions (JSON, FTS, Vector, HTTPFS, DuckDB).
    fn register_builtin_extensions(&mut self) {
        #[cfg(feature = "json-extension")]
        {
            let ext = Box::new(akar_json::JsonExtension::new());
            if let Ok(mut reg) = self.extension_registry.lock() {
                reg.register(ext);
            }
        }
        #[cfg(feature = "fts-extension")]
        {
            let ext = Box::new(akar_fts::FtsExtension::new());
            if let Ok(mut reg) = self.extension_registry.lock() {
                reg.register(ext);
            }
        }
        #[cfg(feature = "vector-extension")]
        {
            let ext = Box::new(akar_vector::VectorExtension::new());
            if let Ok(mut reg) = self.extension_registry.lock() {
                reg.register(ext);
            }
        }
        #[cfg(all(feature = "httpfs-extension", not(akar_wasm)))]
        {
            let ext = Box::new(akar_httpfs::HttpfsExtension::new());
            if let Ok(mut reg) = self.extension_registry.lock() {
                reg.register(ext);
            }
        }
        #[cfg(all(feature = "duckdb-extension", not(akar_wasm)))]
        {
            let ext = Box::new(akar_duckdb::DuckDbExtension::new());
            if let Ok(mut reg) = self.extension_registry.lock() {
                reg.register(ext);
            }
        }
        #[cfg(feature = "algo-extension")]
        {
            let ext = Box::new(akar_algo::AlgoExtension::new());
            if let Ok(mut reg) = self.extension_registry.lock() {
                reg.register(ext);
            }
        }
        #[cfg(feature = "neo4j-extension")]
        {
            let ext = Box::new(akar_neo4j::Neo4jExtension::new());
            if let Ok(mut reg) = self.extension_registry.lock() {
                reg.register(ext);
            }
        }
        #[cfg(feature = "llm-extension")]
        {
            let ext = Box::new(akar_llm::LlmExtension::new());
            if let Ok(mut reg) = self.extension_registry.lock() {
                reg.register(ext);
            }
        }
        #[cfg(all(feature = "ml-extension", not(akar_wasm)))]
        {
            let ext = Box::new(akar_ml::extension::MlExtension::new());
            if let Ok(mut reg) = self.extension_registry.lock() {
                reg.register(ext);
            }
        }
        #[cfg(all(feature = "sqlite-extension", not(akar_wasm)))]
        {
            let ext = Box::new(akar_sqlite::SqliteExtension::new());
            if let Ok(mut reg) = self.extension_registry.lock() {
                reg.register(ext);
            }
        }
        #[cfg(any(feature = "delta-extension", feature = "delta-native"))]
        {
            let ext = Box::new(akar_delta::DeltaExtension::new());
            if let Ok(mut reg) = self.extension_registry.lock() {
                reg.register(ext);
            }
        }
        #[cfg(any(feature = "iceberg-extension", feature = "iceberg-native"))]
        {
            let ext = Box::new(akar_iceberg::IcebergExtension::new());
            if let Ok(mut reg) = self.extension_registry.lock() {
                reg.register(ext);
            }
        }
        #[cfg(any(feature = "azure-extension", feature = "azure-native"))]
        {
            let ext = Box::new(akar_azure::AzureExtension::new());
            if let Ok(mut reg) = self.extension_registry.lock() {
                reg.register(ext);
            }
        }
        #[cfg(all(feature = "postgres-extension", not(akar_wasm)))]
        {
            let ext = Box::new(akar_postgres::PostgresExtension::new());
            if let Ok(mut reg) = self.extension_registry.lock() {
                reg.register(ext);
            }
        }
        #[cfg(any(feature = "unity-catalog-extension", feature = "unity-catalog-native"))]
        {
            let ext = Box::new(akar_unity_catalog::UnityCatalogExtension::new());
            if let Ok(mut reg) = self.extension_registry.lock() {
                reg.register(ext);
            }
        }
    }
}