feat: database pragma queries, fix: #132

Author: ElhamAryanpur

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "lua-astra"
-version = "0.38.0"
+version = "0.39.0"
 authors = ["Elham Aryanpur <elhamaryanpur5@gmail.com>"]
 description = "🔥 Blazingly Fast 🔥 runtime environment for Lua"
 documentation = "https://astra.arkforge.net/docs/latest"

Cargo.toml

@@ -3,6 +3,8 @@
 ---@field execute fun(database: Database, sql: string, parameters: table | nil)
 ---@field query_one fun(database: Database, sql: string, parameters: table | nil): table | nil
 ---@field query_all fun(database: Database, sql: string, parameters: table | nil): table | nil
+---@field query_pragma_int fun(database: Database, sql: string): number | nil
+---@field query_pragma_text fun(database: Database, sql: string): string | nil
 ---@field close fun(database: Database)
 
 ---Opens a new SQL connection using the provided URL and returns a table representing the connection.

astra/lua/database.lua

@@ -3,6 +3,8 @@ global interface Database is userdata
     execute: function(database: Database, sql: string, parameters?: table)
     query_one: function(database: Database, sql: string, parameters?: table): table | nil
     query_all: function(database: Database, sql: string, parameters?: table): table | nil
+    query_pragma_int: function(database: Database, sql: string): number | nil
+    query_pragma_text: function(database: Database, sql: string): string | nil
     close: function(database: Database)
 end
 

astra/teal/database.tl

@@ -1,4 +1,4 @@
-use mlua::{LuaSerdeExt, UserData};
+use mlua::{ExternalError, LuaSerdeExt, UserData};
 use sqlx::{Pool, Postgres, Row, Sqlite, migrate::MigrateDatabase};
 use std::sync::LazyLock;
 use tokio::sync::Mutex;
@@ -238,6 +238,45 @@ impl UserData for Database {
             },
         );
 
+        macro_rules! query_pragma {
+            ($type:ty, $lua:ident, $sql:ident, $db:ident) => {
+                match &$db {
+                    DatabaseType::Sqlite(pool) => {
+                        match sqlx::query_scalar::<_, $type>(&$sql)
+                            .fetch_optional(pool)
+                            .await
+                        {
+                            Ok(row) => $lua.to_value(&row),
+                            Err(e) => Err(e.into_lua_err()),
+                        }
+                    }
+                    DatabaseType::Postgres(pool) => {
+                        match sqlx::query_scalar::<_, $type>(&$sql)
+                            .fetch_optional(pool)
+                            .await
+                        {
+                            Ok(row) => $lua.to_value(&row),
+                            Err(e) => Err(e.into_lua_err()),
+                        }
+                    }
+                }
+            };
+        }
+
+        methods.add_async_method("query_pragma_int", |lua, this, sql: String| async move {
+            match &this.db {
+                Some(db) => query_pragma!(i32, lua, sql, db),
+                None => Err(mlua::Error::runtime("The connection is closed")),
+            }
+        });
+
+        methods.add_async_method("query_pragma_text", |lua, this, sql: String| async move {
+            match &this.db {
+                Some(db) => query_pragma!(String, lua, sql, db),
+                None => Err(mlua::Error::runtime("The connection is closed")),
+            }
+        });
+
         methods.add_async_method(
             "query_one",
             |lua, this, (sql, parameters): (String, Option<mlua::Table>)| async move {