Skip to content
Snippets Groups Projects
command.rs 2.34 KiB
Newer Older
// This file is part of Substrate.

// Copyright (C) 2017-2020 Parity Technologies (UK) Ltd.
// SPDX-License-Identifier: Apache-2.0

// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// 	http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

use crate::chain_spec;
use crate::cli::Cli;
Ricardo Rius's avatar
Ricardo Rius committed
use crate::service;
Dan Forbes's avatar
Dan Forbes committed
use sc_cli::{SubstrateCli, RuntimeVersion, Role, ChainSpec};
Dan Forbes's avatar
Dan Forbes committed
use sc_service::PartialComponents;
use crate::service::new_partial;
Ricardo Rius's avatar
Ricardo Rius committed
impl SubstrateCli for Cli {
Dan Forbes's avatar
Dan Forbes committed
	fn impl_name() -> String {
		"Substrate Node".into()
Ricardo Rius's avatar
Ricardo Rius committed
	}

Dan Forbes's avatar
Dan Forbes committed
	fn impl_version() -> String {
		env!("SUBSTRATE_CLI_IMPL_VERSION").into()
Ricardo Rius's avatar
Ricardo Rius committed
	}
Dan Forbes's avatar
Dan Forbes committed
	fn description() -> String {
		env!("CARGO_PKG_DESCRIPTION").into()
Ricardo Rius's avatar
Ricardo Rius committed
	}

Dan Forbes's avatar
Dan Forbes committed
	fn author() -> String {
		env!("CARGO_PKG_AUTHORS").into()
Ricardo Rius's avatar
Ricardo Rius committed
	}

Dan Forbes's avatar
Dan Forbes committed
	fn support_url() -> String {
		"support.anonymous.an".into()
Ricardo Rius's avatar
Ricardo Rius committed
	}
Ricardo Rius's avatar
Ricardo Rius committed
	fn copyright_start_year() -> i32 {
		2017
	}

	fn load_spec(&self, id: &str) -> Result<Box<dyn sc_service::ChainSpec>, String> {
		Ok(match id {
Dan Forbes's avatar
Dan Forbes committed
			"dev" => Box::new(chain_spec::development_config()?),
			"" | "local" => Box::new(chain_spec::local_testnet_config()?),
Ricardo Rius's avatar
Ricardo Rius committed
			path => Box::new(chain_spec::ChainSpec::from_json_file(
				std::path::PathBuf::from(path),
			)?),
		})
	}
Dan Forbes's avatar
Dan Forbes committed

	fn native_runtime_version(_: &Box<dyn ChainSpec>) -> &'static RuntimeVersion {
		&node_template_runtime::VERSION
	}
Ricardo Rius's avatar
Ricardo Rius committed
}

/// Parse and run command line arguments
pub fn run() -> sc_cli::Result<()> {
	let cli = Cli::from_args();

Dan Forbes's avatar
Dan Forbes committed
	match cli.subcommand {
		Some(ref subcommand) => {
Ricardo Rius's avatar
Ricardo Rius committed
			let runner = cli.create_runner(subcommand)?;
Dan Forbes's avatar
Dan Forbes committed
			runner.run_subcommand(subcommand, |config| {
Dan Forbes's avatar
Dan Forbes committed
				let PartialComponents { client, backend, task_manager, import_queue, .. }
					= new_partial(&config)?;
Dan Forbes's avatar
Dan Forbes committed
				Ok((client, backend, import_queue, task_manager))
			})
Ricardo Rius's avatar
Ricardo Rius committed
		}
		None => {
Ricardo Rius's avatar
Ricardo Rius committed
			let runner = cli.create_runner(&cli.run)?;
Dan Forbes's avatar
Dan Forbes committed
			runner.run_node_until_exit(|config| match config.role {
				Role::Light => service::new_light(config),
				_ => service::new_full(config),
			})
Ricardo Rius's avatar
Ricardo Rius committed
		}