0.5.4

2026-04-24 08:53:40 +02:00
parent a7030d7d0f
commit 54778373b8
31 changed files with 1706 additions and 164 deletions
--- a/kb_lib/src/db/queries/dex.rs
+++ b/kb_lib/src/db/queries/dex.rs
@@ -0,0 +1,113 @@
+// file: kb_lib/src/db/queries/dex.rs
+
+//! Queries for `kb_dexes`.
+
+/// Inserts or updates one normalized DEX row by code.
+pub async fn upsert_dex(
+    database: &crate::KbDatabase,
+    dto: &crate::KbDexDto,
+) -> Result<i64, crate::KbError> {
+    match database.connection() {
+        crate::KbDatabaseConnection::Sqlite(pool) => {
+            let query_result = sqlx::query(
+                r#"
+INSERT INTO kb_dexes (
+    code,
+    name,
+    program_id,
+    router_program_id,
+    is_enabled,
+    created_at,
+    updated_at
+)
+VALUES (?, ?, ?, ?, ?, ?, ?)
+ON CONFLICT(code) DO UPDATE SET
+    name = excluded.name,
+    program_id = excluded.program_id,
+    router_program_id = excluded.router_program_id,
+    is_enabled = excluded.is_enabled,
+    updated_at = excluded.updated_at
+                "#,
+            )
+            .bind(dto.code.clone())
+            .bind(dto.name.clone())
+            .bind(dto.program_id.clone())
+            .bind(dto.router_program_id.clone())
+            .bind(if dto.is_enabled { 1_i64 } else { 0_i64 })
+            .bind(dto.created_at.to_rfc3339())
+            .bind(dto.updated_at.to_rfc3339())
+            .execute(pool)
+            .await;
+            if let Err(error) = query_result {
+                return Err(crate::KbError::Db(format!(
+                    "cannot upsert kb_dexes on sqlite: {}",
+                    error
+                )));
+            }
+            let id_result = sqlx::query_scalar::<sqlx::Sqlite, i64>(
+                r#"
+SELECT id
+FROM kb_dexes
+WHERE code = ?
+LIMIT 1
+                "#,
+            )
+            .bind(dto.code.clone())
+            .fetch_one(pool)
+            .await;
+            match id_result {
+                Ok(id) => Ok(id),
+                Err(error) => Err(crate::KbError::Db(format!(
+                    "cannot fetch kb_dexes id for code '{}' on sqlite: {}",
+                    dto.code, error
+                ))),
+            }
+        }
+    }
+}
+
+/// Lists normalized DEX rows.
+pub async fn list_dexes(
+    database: &crate::KbDatabase,
+) -> Result<std::vec::Vec<crate::KbDexDto>, crate::KbError> {
+    match database.connection() {
+        crate::KbDatabaseConnection::Sqlite(pool) => {
+            let query_result = sqlx::query_as::<sqlx::Sqlite, crate::KbDexEntity>(
+                r#"
+SELECT
+    id,
+    code,
+    name,
+    program_id,
+    router_program_id,
+    is_enabled,
+    created_at,
+    updated_at
+FROM kb_dexes
+ORDER BY code ASC
+                "#,
+            )
+            .fetch_all(pool)
+            .await;
+            let entities = match query_result {
+                Ok(entities) => entities,
+                Err(error) => {
+                    return Err(crate::KbError::Db(format!(
+                        "cannot list kb_dexes on sqlite: {}",
+                        error
+                    )));
+                }
+            };
+            let mut dtos = std::vec::Vec::new();
+            for entity in entities {
+                let dto_result = crate::KbDexDto::try_from(entity);
+                let dto = match dto_result {
+                    Ok(dto) => dto,
+                    Err(error) => return Err(error),
+                };
+                dtos.push(dto);
+            }
+            Ok(dtos)
+        }
+    }
+}