Merge pull request #42 from DeterminateSystems/list-label

Enable listing flakes by label

src/cli/cmd/list.rs

@@ -92,6 +92,8 @@ pub(super) struct Release {
 enum Subcommands {
     /// Lists all currently public flakes on FlakeHub.
     Flakes,
+    /// Lists all public flakes with the provided label.
+    Label { label: String },
     /// Lists all currently public organizations on FlakeHub.
     Orgs,
     /// List all releases for a specific flake on FlakeHub.
@@ -148,6 +150,42 @@ impl CommandExecute for ListSubcommand {
                     Err(e) => return Err(e.into()),
                 }
             }
+            Label { label } => {
+                if string_has_whitespace(&label) {
+                    return Err(FhError::LabelParse(String::from("whitespace not allowed")).into());
+                }
+
+                let label = label.to_lowercase();
+
+                match client.flakes_by_label(&label).await {
+                    Ok(flakes) => {
+                        if flakes.is_empty() {
+                            eprintln!("No results");
+                        } else if self.json {
+                            print_json(&flakes)?;
+                        } else {
+                            let mut table = Table::new();
+                            table.set_format(*TABLE_FORMAT);
+
+                            table.set_titles(row!["Flake", "FlakeHub URL"]);
+
+                            for flake in flakes {
+                                table.add_row(Row::new(vec![
+                                    Cell::new(&flake.name()).with_style(Attr::Bold),
+                                    Cell::new(&flake.url()).with_style(Attr::Dim),
+                                ]));
+                            }
+
+                            if std::io::stdout().is_terminal() {
+                                table.printstd();
+                            } else {
+                                table.to_csv(std::io::stdout())?;
+                            }
+                        }
+                    }
+                    Err(e) => return Err(e.into()),
+                }
+            }
             Orgs => {
                 let pb = ProgressBar::new_spinner();
                 pb.set_style(ProgressStyle::default_spinner());
@@ -286,3 +324,7 @@ impl CommandExecute for ListSubcommand {
         Ok(ExitCode::SUCCESS)
     }
 }
+
+fn string_has_whitespace(s: &str) -> bool {
+    s.chars().any(char::is_whitespace)
+}

src/cli/cmd/mod.rs

@@ -10,8 +10,7 @@ use reqwest::Client as HttpClient;
 use serde::Serialize;
 
 use self::{
-    list::Flake,
-    list::{Org, Release, Version},
+    list::{Flake, Org, Release, Version},
     search::SearchResult,
 };
 
@@ -62,6 +61,9 @@ pub(super) enum FhError {
     #[error("json parsing error: {0}")]
     Json(#[from] serde_json::Error),
 
+    #[error("label parsing error: {0}")]
+    LabelParse(String),
+
     #[error("the flake has no inputs")]
     NoInputs,
 
@@ -128,6 +130,27 @@ impl FlakeHubClient {
         Ok(flakes)
     }
 
+    async fn flakes_by_label(&self, label: &str) -> Result<Vec<Flake>, FhError> {
+        let mut url = self.api_addr.clone();
+        {
+            let mut segs = url
+                .path_segments_mut()
+                .expect("flakehub url cannot be base (this should never happen)");
+
+            segs.push("label").push(label);
+        }
+
+        let flakes = self
+            .client
+            .get(&url.to_string())
+            .send()
+            .await?
+            .json::<Vec<Flake>>()
+            .await?;
+
+        Ok(flakes)
+    }
+
     async fn releases(&self, org: &str, project: &str) -> Result<Vec<Release>, FhError> {
         let mut url = self.api_addr.clone();
         {