0.5.0

kb_lib/src/db/schema.rs

@@ -0,0 +1,39 @@
+// file: kb_lib/src/db/schema.rs
+
+//! Database schema initialization.
+
+/// Ensures that the database schema exists.
+pub(super) async fn ensure_schema(
+    database: &crate::KbDatabase,
+) -> Result<(), crate::KbError> {
+    match database.connection() {
+        crate::KbDatabaseConnection::Sqlite(pool) => {
+            let metadata_table_result = sqlx::query(
+                r#"
+CREATE TABLE IF NOT EXISTS kb_db_metadata (
+    key TEXT NOT NULL PRIMARY KEY,
+    value TEXT NOT NULL,
+    updated_at TEXT NOT NULL
+)
+                "#,
+            )
+            .execute(pool)
+            .await;
+            if let Err(error) = metadata_table_result {
+                return Err(crate::KbError::Db(format!(
+                    "cannot create table kb_db_metadata on sqlite: {}",
+                    error
+                )));
+            }
+            let schema_version = crate::KbDbMetadataDto::new(
+                "schema_version".to_string(),
+                env!("CARGO_PKG_VERSION").to_string(),
+            );
+            let upsert_result = crate::upsert_db_metadata(database, &schema_version).await;
+            if let Err(error) = upsert_result {
+                return Err(error);
+            }
+            Ok(())
+        },
+    }
+}