supports optional namespace filtering

Author: Skybound1

src/main.rs

@@ -251,16 +251,26 @@ struct Args {
     only: Vec<String>,
     #[arg(long)]
     exclude: Vec<String>,
+    #[arg(short = 'n', long)]
+    namespace: Vec<String>,
     output_dir: String,
 }
 
 #[tokio::main]
 async fn main() -> anyhow::Result<()> {
     let args = Args::parse();
 
+    if args.namespace.iter().any(|n| n.is_empty()) {
+        anyhow::bail!("--namespace must not be empty");
+    }
+
     let client = Client::try_default().await?;
     let discovery = Discovery::new(client.clone()).run().await?;
-    let namespaces = get_namespaces(&client, &discovery).await?;
+    let namespaces = if args.namespace.is_empty() {
+        get_namespaces(&client, &discovery).await?
+    } else {
+        args.namespace.clone()
+    };
 
     let metadata = Metadata {
         kube_version: get_version(&client).await?,
@@ -310,6 +320,20 @@ async fn main() -> anyhow::Result<()> {
                 }
             }
 
+            if items.is_empty()
+                && ar.plural == "namespaces"
+                && ar.group.is_empty()
+                && !args.namespace.is_empty()
+            {
+                let api: Api<DynamicObject> = Api::all_with(client.clone(), &ar);
+                for ns in &args.namespace {
+                    match api.get(ns).await {
+                        Ok(obj) => items.push(obj),
+                        Err(e) => println!("\tFailed to get namespace {}: {}", ns, e),
+                    }
+                }
+            }
+
             if items.is_empty() {
                 continue;
             }