From 144d30504b0838949f2955684275c93ab3a038b7 Mon Sep 17 00:00:00 2001
From: kysshsy <kysshsy@gmail.com>
Date: Wed, 18 Dec 2024 00:17:12 +0800
Subject: [PATCH 1/3] feat: add a duckdb_extensions() shows duckdb extension
 details

---
 src/api/duckdb.rs | 70 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 70 insertions(+)

diff --git a/src/api/duckdb.rs b/src/api/duckdb.rs
index 6f220816..1ae90cae 100644
--- a/src/api/duckdb.rs
+++ b/src/api/duckdb.rs
@@ -11,6 +11,18 @@ type DuckdbSettingsRow = (
     Option<String>,
 );
 
+type DuckdbExtensionsRow = (
+    Option<String>,
+    Option<bool>,
+    Option<bool>,
+    Option<String>,
+    Option<String>,
+    Option<String>,
+    Option<String>,
+    Option<String>,
+    Option<String>,
+);
+
 #[pg_extern]
 pub fn duckdb_execute(query: &str) {
     connection::execute(query, []).unwrap_or_else(|err| panic!("error executing query: {err:?}"));
@@ -52,3 +64,61 @@ fn duckdb_settings_impl() -> Result<Vec<DuckdbSettingsRow>> {
         .map(|row| row.unwrap())
         .collect::<Vec<DuckdbSettingsRow>>())
 }
+
+#[allow(clippy::type_complexity)]
+#[pg_extern]
+pub fn duckdb_extensions() -> iter::TableIterator<
+    'static,
+    (
+        name!(extension_name, Option<String>),
+        name!(loaded, Option<bool>),
+        name!(installed, Option<bool>),
+        name!(install_path, Option<String>),
+        name!(description, Option<String>),
+        name!(aliases, Option<String>),
+        name!(extension_version, Option<String>),
+        name!(install_mode, Option<String>),
+        name!(installed_from, Option<String>),
+    ),
+> {
+    let rows = duckdb_extensions_impl().unwrap_or_else(|e| {
+        panic!("{}", e);
+    });
+    iter::TableIterator::new(rows)
+}
+
+#[inline]
+fn duckdb_extensions_impl() -> Result<Vec<DuckdbExtensionsRow>> {
+    let conn = unsafe { &*connection::get_global_connection().get() };
+    let mut stmt = conn.prepare(
+        "SELECT
+            extension_name,
+            loaded,
+            installed,
+            install_path,
+            description,
+            aliases::varchar,
+            extension_version,
+            install_mode,
+            installed_from
+        FROM
+            duckdb_extensions();",
+    )?;
+
+    Ok(stmt
+        .query_map([], |row| {
+            Ok((
+                row.get::<_, Option<String>>(0)?,
+                row.get::<_, Option<bool>>(1)?,
+                row.get::<_, Option<bool>>(2)?,
+                row.get::<_, Option<String>>(3)?,
+                row.get::<_, Option<String>>(4)?,
+                row.get::<_, Option<String>>(5)?,
+                row.get::<_, Option<String>>(6)?,
+                row.get::<_, Option<String>>(7)?,
+                row.get::<_, Option<String>>(8)?,
+            ))
+        })?
+        .map(|row| row.unwrap())
+        .collect::<Vec<DuckdbExtensionsRow>>())
+}

From af11c74ec1b3b1892fd398520df764b67abf943a Mon Sep 17 00:00:00 2001
From: kysshsy <kysshsy@gmail.com>
Date: Wed, 18 Dec 2024 23:22:39 +0800
Subject: [PATCH 2/3] test: add duckdb_extensions() test

---
 tests/tests/settings.rs | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/tests/tests/settings.rs b/tests/tests/settings.rs
index 6bb8c197..24c1d374 100644
--- a/tests/tests/settings.rs
+++ b/tests/tests/settings.rs
@@ -15,3 +15,15 @@ async fn test_duckdb_settings(mut conn: PgConnection) -> Result<()> {
 
     Ok(())
 }
+
+#[rstest]
+async fn test_duckdb_extensions(mut conn: PgConnection) -> Result<()> {
+    let azure_extension: (Option<String>,) = "SELECT extension_name
+                                                FROM duckdb_extensions()
+                                                WHERE extension_name = 'azure'
+                                                AND installed = true"
+        .fetch_one(&mut conn);
+    assert!(azure_extension.0.is_some());
+
+    Ok(())
+}

From 4e669b4eb4bc78ac8c04b4308ad55b29d56050b6 Mon Sep 17 00:00:00 2001
From: kysshsy <kysshsy@gmail.com>
Date: Thu, 19 Dec 2024 23:12:49 +0800
Subject: [PATCH 3/3] test: fix test azure is installed during execution

---
 tests/tests/settings.rs | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/tests/tests/settings.rs b/tests/tests/settings.rs
index 24c1d374..d87d121a 100644
--- a/tests/tests/settings.rs
+++ b/tests/tests/settings.rs
@@ -18,11 +18,9 @@ async fn test_duckdb_settings(mut conn: PgConnection) -> Result<()> {
 
 #[rstest]
 async fn test_duckdb_extensions(mut conn: PgConnection) -> Result<()> {
-    let azure_extension: (Option<String>,) = "SELECT extension_name
-                                                FROM duckdb_extensions()
-                                                WHERE extension_name = 'azure'
-                                                AND installed = true"
-        .fetch_one(&mut conn);
+    let azure_extension: (Option<String>,) =
+        "SELECT extension_name FROM duckdb_extensions() WHERE extension_name = 'azure'"
+            .fetch_one(&mut conn);
     assert!(azure_extension.0.is_some());
 
     Ok(())