component push: add --public/private flags

Author: CLEMENTINATOR

crates/cli/src/commands/components/push.rs

@@ -12,6 +12,7 @@ pub struct Options {
     ///
     /// Defaults to the user "self" org
     pub organization: Option<String>,
+
     #[arg(long, conflicts_with = "private")]
     pub public: bool,
 
@@ -118,11 +119,18 @@ pub async fn run(opts: Options) -> anyhow::Result<()> {
                 return Ok(());
             }
 
-            let public_or_private = Select::new(
-                "Would you like to make this component public or private?",
-                vec!["private", "public"],
-            )
-            .prompt()?;
+            let public = match (opts.public, opts.private) {
+                (true, false) => true,
+                (false, true) => false,
+                _ => {
+                    Select::new(
+                        "Would you like to make this component public or private?",
+                        vec!["private", "public"],
+                    )
+                    .prompt()?
+                        == "public"
+                }
+            };
 
             let avatar_url = if let Some(path) = &manifest.component.icon_path {
                 tracing::info!("Uploading Icon... {:?}", manifest.component.icon_path);
@@ -156,7 +164,7 @@ pub async fn run(opts: Options) -> anyhow::Result<()> {
                                 .map(|url| url.to_string()),
                         )
                         .avatar_url(avatar_url)
-                        .public(public_or_private == "public"),
+                        .public(public),
                 )
                 .send()
                 .await
@@ -178,6 +186,49 @@ pub async fn run(opts: Options) -> anyhow::Result<()> {
         Err(err) => anyhow::bail!("Error contacting API: {}", err.into_message()),
     };
 
+    match (opts.public, opts.private) {
+        (true, false) | (false, true) => {
+            let public = opts.public;
+
+            if let Some(public_state) = component.is_public {
+                if public == public_state {
+                    tracing::info!(
+                        "Component is already {}",
+                        if public { "public" } else { "private" }
+                    );
+                } else {
+                    tracing::info!(
+                        "Updating component visibility to {}...",
+                        if public { "public" } else { "private" }
+                    );
+
+                    if !public {
+                        tracing::info!("Only unused components can be made private. If this component is already in use, it will remain public.");
+                    } else {
+                        let confirm = Confirm::new(
+                            "Your component will become publicly visible in the registry. Are you sure?",
+                        )
+                        .with_default(true)
+                        .prompt()?;
+
+                        if !confirm {
+                            return Ok(());
+                        }
+                    }
+                    client
+                        .update_component_by_slug()
+                        .org_slug(&organization.slug)
+                        .component_slug(&component_slug)
+                        .body(api_types::ComponentUpdateParams::builder().public(public))
+                        .send()
+                        .await
+                        .api_context("Could not update component visibility")?;
+                }
+            }
+        }
+        _ => {}
+    }
+
     // Check if version already exists
     if component.versions.contains_key(&manifest.component.version) {
         anyhow::bail!(