// Copyright 2017-2019 Parity Technologies (UK) Ltd.
// This file is part of Substrate.

// Substrate is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// Substrate is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with Substrate.  If not, see <http://www.gnu.org/licenses/>.

use cli::{NoCustom, CoreParams};

use std::{fs, env, path::Path};

use structopt::{StructOpt, clap::Shell};

fn main() {
	build_shell_completion();
}

/// Build shell completion scripts for all known shells
/// Full list in https://github.com/kbknapp/clap-rs/blob/e9d0562a1dc5dfe731ed7c767e6cee0af08f0cf9/src/app/parser.rs#L123
fn build_shell_completion() {
	for shell in &[Shell::Bash, Shell::Fish, Shell::Zsh, Shell::Elvish, Shell::PowerShell] {
		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();

	CoreParams::<NoCustom, NoCustom>::clap().gen_completions("substrate-node", *shell, &path);
}