Properly exit when receiving SIGINT/SIGTERM/SIGHUP (#377)

fables-tales · Nov 27, 2022 · 3e39251 · 3e39251
1 parent 0db5183
commit 3e39251
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -14,6 +14,7 @@ edition = "2018"
 
 [dependencies]
 clap = { version = "3.2.16", features = ["derive"] }
+ctrlc = { version = "3.2", features = ["termination"] }
 dirs = "3.0.2"
 filetime = "0.2.14"
 glob = "0.3"

diff --git a/src/main.rs b/src/main.rs
@@ -340,6 +340,12 @@ fn puts_stdout(input: &String) {
 }
 
 fn main() {
+    ctrlc::set_handler(move || {
+        eprintln!("`rubyfmt` process was terminated. Exiting...");
+        exit(1);
+    })
+    .expect("Error setting Ctrl-C handler");
+
     updates::begin_checking_for_updates();
 
     let opts = get_command_line_options();