remove duplicate code, handle control c

Cargo.toml

@@ -8,6 +8,7 @@ edition = "2021"
 [dependencies]
 anyhow = "1.0.76"
 clap = { version = "4.4.11", features = ["derive"] }
+ctrlc = "3.4.2"
 futures = "0.3.30"
 k8s-openapi = { version = "0.20.0", features = ["latest"] }
 kube = { version = "0.87.2", features = ["runtime", "derive"] }

src/kubeclient.rs

@@ -1,5 +1,4 @@
 use anyhow::{bail, Context, Result};
-use clap::Parser;
 use kube::{
     api::{Patch, PatchParams},
     core::{DynamicObject, GroupVersionKind},
@@ -8,19 +7,6 @@ use kube::{
 };
 use log::{info, trace, warn};
 
-#[derive(Parser, Debug)]
-#[clap(author, version, about, long_about = None)]
-struct Args {
-    #[clap(long, default_value = "kapplier")]
-    user_agent: String,
-    #[clap(long, default_value = ".")]
-    directory: String,
-    #[clap(long, default_values = [".git"])]
-    ignore_directories: Vec<String>,
-    #[clap(long, default_values = ["yml", "yaml"])]
-    supported_extensions: Vec<String>,
-}
-
 // https://github.com/kube-rs/kube/blob/main/examples/kubectl.rs#L249
 fn multidoc_deserialize(data: &str) -> Result<Vec<serde_yaml::Value>> {
     use serde::Deserialize;

src/main.rs

@@ -1,4 +1,4 @@
-use std::{path::Path, time::Duration};
+use std::{path::Path, process, time::Duration};
 
 use anyhow::Result;
 use clap::Parser;
@@ -34,6 +34,13 @@ async fn main() -> Result<()> {
     let args = Args::parse();
     let full_run_args = args.clone();
 
+    // handle control c
+    ctrlc::set_handler(move || {
+        info!("received Ctrl+C! Exiting...");
+        // exit immediately
+        process::exit(0);
+    })?;
+
     // full run thread
     tokio::spawn(async move {
         let args = full_run_args;