add list command

Author: imor

cli/src/commands/list.rs

@@ -0,0 +1,40 @@
+use std::collections::HashMap;
+
+use futures::TryStreamExt;
+use sqlx::PgConnection;
+
+pub(crate) async fn list(conn: &mut PgConnection) -> anyhow::Result<()> {
+    let available_extensions = available_extensions(conn).await?;
+
+    for (extension, versions) in available_extensions {
+        println!("{extension}");
+        for version in versions {
+            println!("  {version}");
+        }
+    }
+
+    Ok(())
+}
+
+#[derive(sqlx::FromRow)]
+struct ExtensionRow {
+    name: String,
+    version: String,
+}
+
+async fn available_extensions(
+    conn: &mut PgConnection,
+) -> anyhow::Result<HashMap<String, Vec<String>>> {
+    let mut rows = sqlx::query_as::<_, ExtensionRow>(
+        "select name, version from pgtle.available_extension_versions()",
+    )
+    .fetch(conn);
+
+    let mut available_extensions = HashMap::new();
+    while let Some(row) = rows.try_next().await? {
+        let versions: &mut Vec<String> = available_extensions.entry(row.name).or_default();
+        versions.push(row.version);
+    }
+
+    Ok(available_extensions)
+}

cli/src/commands/mod.rs

@@ -2,3 +2,4 @@ pub mod install;
 pub mod login;
 pub mod publish;
 pub mod uninstall;
+pub mod list;

cli/src/main.rs

@@ -62,6 +62,14 @@ enum Commands {
         #[arg(long)]
         registry_name: Option<String>,
     },
+
+    /// List available packages
+    #[clap(alias = "ls")]
+    List {
+        /// PostgreSQL connection string
+        #[arg(short, long)]
+        connection: String,
+    },
 }
 
 #[derive(Debug, clap::Args)]
@@ -139,5 +147,11 @@ async fn main() -> anyhow::Result<()> {
             println!("Login successful");
             Ok(())
         }
+
+        Commands::List { connection } => {
+            let mut conn = util::get_connection(connection).await?;
+            commands::list::list(&mut conn).await?;
+            Ok(())
+        }
     }
 }