Newer
Older
// This file is part of Substrate.
// Copyright (C) 2017-2020 Parity Technologies (UK) Ltd.
// SPDX-License-Identifier: GPL-3.0-or-later WITH Classpath-exception-2.0
// This program 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.
// This program 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 this program. If not, see <https://www.gnu.org/licenses/>.
use crate::{chain_spec, service, Cli, Subcommand};
Cecile Tonglet
committed
use node_executor::Executor;
use node_runtime::{Block, RuntimeApi};
Ashley
committed
use sc_cli::{Result, SubstrateCli, RuntimeVersion, Role, ChainSpec};
use sc_service::PartialComponents;
use crate::service::new_partial;
Cecile Tonglet
committed
impl SubstrateCli for Cli {
fn impl_name() -> String {
"Substrate Node".into()
Cecile Tonglet
committed
}
fn impl_version() -> String {
env!("SUBSTRATE_CLI_IMPL_VERSION").into()
Cecile Tonglet
committed
}
fn description() -> String {
env!("CARGO_PKG_DESCRIPTION").into()
Cecile Tonglet
committed
}
fn author() -> String {
env!("CARGO_PKG_AUTHORS").into()
Cecile Tonglet
committed
}
fn support_url() -> String {
"https://github.com/paritytech/substrate/issues/new".into()
Cecile Tonglet
committed
}
fn copyright_start_year() -> i32 {
2017
}
fn load_spec(&self, id: &str) -> std::result::Result<Box<dyn sc_service::ChainSpec>, String> {
Ok(match id {
"dev" => Box::new(chain_spec::development_config()),
"local" => Box::new(chain_spec::local_testnet_config()),
"" | "fir" | "flaming-fir" => Box::new(chain_spec::flaming_fir_config()?),
"staging" => Box::new(chain_spec::staging_testnet_config()),
path => Box::new(chain_spec::ChainSpec::from_json_file(
std::path::PathBuf::from(path),
)?),
})
}
Ashley
committed
fn native_runtime_version(_: &Box<dyn ChainSpec>) -> &'static RuntimeVersion {
&node_runtime::VERSION
}
Cecile Tonglet
committed
}
/// Parse command line arguments into service configuration.
Cecile Tonglet
committed
pub fn run() -> Result<()> {
let cli = Cli::from_args();
Cecile Tonglet
committed
match &cli.subcommand {
Cecile Tonglet
committed
None => {
Cecile Tonglet
committed
let runner = cli.create_runner(&cli.run)?;
Ashley
committed
runner.run_node_until_exit(|config| match config.role {
Role::Light => service::new_light(config),
_ => service::new_full(config),
})
Cecile Tonglet
committed
}
Some(Subcommand::Inspect(cmd)) => {
Cecile Tonglet
committed
let runner = cli.create_runner(cmd)?;
Cecile Tonglet
committed
runner.sync_run(|config| cmd.run::<Block, RuntimeApi, Executor>(config))
}
Some(Subcommand::Benchmark(cmd)) => {
if cfg!(feature = "runtime-benchmarks") {
let runner = cli.create_runner(cmd)?;
runner.sync_run(|config| cmd.run::<Block, Executor>(config))
} else {
println!("Benchmarking wasn't enabled when building the node. \
You can enable it with `--features runtime-benchmarks`.");
Ok(())
}
Cecile Tonglet
committed
}
Some(Subcommand::Key(cmd)) => cmd.run(),
Some(Subcommand::Sign(cmd)) => cmd.run(),
Some(Subcommand::Verify(cmd)) => cmd.run(),
Some(Subcommand::Vanity(cmd)) => cmd.run(),
Cecile Tonglet
committed
Some(Subcommand::Base(subcommand)) => {
let runner = cli.create_runner(subcommand)?;
Ashley
committed
runner.run_subcommand(subcommand, |config| {
let PartialComponents { client, backend, task_manager, import_queue, ..}
= new_partial(&config)?;
Ok((client, backend, import_queue, task_manager))
Ashley
committed
})