From 1d1fe7b0b68719263ea31504e74c2b0f5ad2e053 Mon Sep 17 00:00:00 2001
From: Zhai Can <bczhc0@126.com>
Date: Fri, 10 Nov 2023 20:13:03 +0800
Subject: add command to generate Rust-part CLI completions

---
 rust-src/src/cmd_completions.rs | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)
 create mode 100644 rust-src/src/cmd_completions.rs

(limited to 'rust-src/src/cmd_completions.rs')

diff --git a/rust-src/src/cmd_completions.rs b/rust-src/src/cmd_completions.rs
new file mode 100644
index 00000000..51859696
--- /dev/null
+++ b/rust-src/src/cmd_completions.rs
@@ -0,0 +1,24 @@
+use crate::transform_c_args;
+use clap::{Command, CommandFactory, Parser};
+use clap_complete::{generate, Generator, Shell};
+use std::ffi::{c_char, c_int};
+use std::io;
+
+/// Generate shell completions
+#[derive(clap::Parser, Debug)]
+pub struct Cli {
+    shell: Shell,
+}
+
+fn print_completions<G: Generator>(gen: G, cmd: &mut Command) {
+    generate(gen, cmd, cmd.get_name().to_string(), &mut io::stdout());
+}
+
+#[no_mangle]
+#[allow(clippy::not_unsafe_ptr_arg_deref)]
+pub extern "C" fn cmd_completions(argc: c_int, argv: *const *const c_char) -> c_int {
+    transform_c_args!(argv, argc, argv);
+    let cli = Cli::parse_from(argv);
+    print_completions(cli.shell, &mut super::Cli::command());
+    0
+}
-- 
cgit v1.2.3