node: add command line completion

node/Cargo.toml

@@ -111,7 +111,17 @@ testnet-runtime.workspace = true
 #try-runtime-core = { git = "https://github.com/paritytech/try-runtime-cli", tag = "v0.7.0", optional = true }
 
 [build-dependencies]
-polkadot-sdk = { workspace = true, features = [ "substrate-build-script-utils" ] }
+clap.workspace = true
+
+clap_complete = "4.5"
+
+polkadot-sdk = { workspace = true, features = [
+	"substrate-build-script-utils",
+	"frame-benchmarking-cli",
+	"sc-cli",
+	"sc-storage-monitor",
+	"staging-node-inspect",
+] }
 
 [features]
 default = [ "std" ]

node/build.rs

@@ -1,7 +1,48 @@
-use polkadot_sdk::substrate_build_script_utils::{generate_cargo_keys, rerun_if_git_head_changed};
-
 fn main() {
-	generate_cargo_keys();
+	cli::main();
+}
+
+mod cli {
+	include!("src/cli.rs");
+
+	use clap::{CommandFactory, ValueEnum};
+	use clap_complete::{generate_to, Shell};
+	use polkadot_sdk::substrate_build_script_utils::{
+		generate_cargo_keys, rerun_if_git_head_changed,
+	};
+	use std::{env, fs, path::Path};
+
+	pub fn main() {
+		build_shell_completion();
+		generate_cargo_keys();
+
+		rerun_if_git_head_changed();
+	}
+
+	/// Build shell completion scripts for all known shells.
+	fn build_shell_completion() {
+		for shell in Shell::value_variants() {
+			build_completion(shell);
+		}
+	}
+
+	/// Build the shell auto-completion for a given Shell.
+	fn build_completion(shell: &Shell) {
+		let outdir = match env::var_os("OUT_DIR") {
+			None => return,
+			Some(dir) => dir,
+		};
+		let path = Path::new(&outdir)
+			.parent()
+			.unwrap()
+			.parent()
+			.unwrap()
+			.parent()
+			.unwrap()
+			.join("completion-scripts");
+
+		fs::create_dir(&path).ok();
 
-	rerun_if_git_head_changed();
+		let _ = generate_to(*shell, &mut Cli::command(), "timechain-node", &path);
+	}
 }