feat: add all_collections (#3)

pgstacrs.pyi

@@ -66,5 +66,12 @@ class Client:
             id: The collection id
         """
 
+    async def all_collections(self) -> None:
+        """Returns all collections.
+
+        Returns:
+            All collections in the database
+        """
+
 class PgstacError:
     """An exception returned from pgstac"""

src/lib.rs

@@ -27,6 +27,17 @@ macro_rules! pgstac {
         })
     };
 
+    (json $client:expr,$py:expr,$function:expr) => {
+        let function = $function.to_string();
+        $client.run($py, |pool: PgstacPool| async move {
+            let query = format!("SELECT * FROM pgstac.{}()", function);
+            let connection = pool.get().await?;
+            let row = connection.query_one(&query, &[]).await?;
+            let value: Value = row.try_get(function.as_str())?;
+            Ok(Json(value))
+        })
+    };
+
     (option $client:expr,$py:expr,$function:expr,$params:expr) => {
         let function = $function.to_string();
         $client.run($py, |pool: PgstacPool| async move {
@@ -139,6 +150,12 @@ impl Client {
             void self,py,"delete_collection",[&id]
         }
     }
+
+    fn all_collections<'a>(&self, py: Python<'a>) -> PyResult<Bound<'a, PyAny>> {
+        pgstac! {
+            json self,py,"all_collections"
+        }
+    }
 }
 
 impl Client {

tests/test_collection.py

@@ -46,3 +46,14 @@ async def test_delete_collection(client: Client, collection: dict[str, Any]) ->
     await client.create_collection(collection)
     await client.delete_collection("simple-collection")
     assert await client.get_collection("simple-id") is None
+
+
+async def test_all_collections(client: Client, collection: dict[str, Any]) -> None:
+    assert len(await client.all_collections()) == 0
+    await client.create_collection(collection)
+    assert len(await client.all_collections()) == 1
+    collection["id"] = "just-as-simple-collection"
+    await client.create_collection(collection)
+    assert len(await client.all_collections()) == 2
+    await client.delete_collection("simple-collection")
+    assert len(await client.all_collections()) == 1