feat: handle more exit signal

Author: eatradish

Cargo.lock

@@ -12,7 +12,6 @@ authors = ["eatradish <sakiiily@aosc.io>"]
 # Cli
 clap = { version = "4.5.19", features = ["cargo", "wrap_help", "color", "derive", "env", "string"] }
 anyhow = "1.0.89"
-ctrlc = "3.5"
 dialoguer = "0.12.0"
 tabled = { version = "0.20", features = ["ansi"] }
 tokio = "1.46.0"
@@ -52,6 +51,7 @@ once_cell = "1.21"
 itertools = "0.14.0"
 clap-i18n-richformatter = "0.1.0"
 debversion = "0.5"
+signal-hook = "0.3.18"
 
 # oma crates
 oma-utils = { path = "./oma-utils", features = ["dbus", "human-bytes"] }

Cargo.toml

@@ -2,6 +2,7 @@ use std::env::{self, args};
 use std::ffi::CString;
 use std::fs::{create_dir_all, read_dir, remove_file};
 use std::io::{self, IsTerminal, stderr, stdin};
+use std::os::raw::c_int;
 use std::path::{Path, PathBuf};
 
 use std::process::{Command, exit};
@@ -37,6 +38,7 @@ use oma_utils::dbus::{create_dbus_connection, get_another_oma_status};
 use oma_utils::{OsRelease, is_termux};
 use reqwest::Client;
 use rustix::stdio::stdout;
+use signal_hook::iterator::Signals;
 use subcommand::utils::{LockError, is_terminal};
 use tokio::runtime::Runtime;
 use tracing::{debug, error, info, warn};
@@ -55,6 +57,9 @@ use crate::config::Config;
 use crate::error::Chain;
 use crate::install_progress::osc94_progress;
 use crate::subcommand::*;
+use signal_hook::consts::*;
+
+const SIGNALS: &[c_int] = &[SIGTERM, SIGQUIT, SIGINT, SIGTSTP, SIGWINCH, SIGHUP, SIGCONT];
 
 static NOT_DISPLAY_ABORT: AtomicBool = AtomicBool::new(false);
 static LOCKED: AtomicBool = AtomicBool::new(false);
@@ -172,7 +177,14 @@ fn main() {
         .completer("oma")
         .complete();
 
-    ctrlc::set_handler(single_handler).expect("oma could not initialize SIGINT handler.");
+    thread::spawn(|| {
+        let Ok(mut sigs) = Signals::new(SIGNALS) else {
+            panic!("Failed to set signal handler.")
+        };
+        for _ in &mut sigs {
+            single_handler();
+        }
+    });
 
     // 要适配额外的插件子命令，所以这里要保留 matches
     let (matches, oma) = parse_args();