command.rs 6.72 KB
Newer Older
Gavin Wood's avatar
Gavin Wood committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
// Copyright 2017-2020 Parity Technologies (UK) Ltd.
// This file is part of Polkadot.

// Polkadot 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.

// Polkadot 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 Polkadot.  If not, see <http://www.gnu.org/licenses/>.

use log::info;
Gavin Wood's avatar
Gavin Wood committed
18
use sp_runtime::traits::BlakeTwo256;
Gavin Wood's avatar
Gavin Wood committed
19
20
use service::{IsKusama, Block, self, RuntimeApiCollection, TFullClient};
use sp_api::ConstructRuntimeApi;
Cecile Tonglet's avatar
Cecile Tonglet committed
21
use sc_cli::{substrate_cli, SubstrateCli, Result};
Gavin Wood's avatar
Gavin Wood committed
22
use sc_executor::NativeExecutionDispatch;
23
use crate::cli::{Cli, Subcommand};
Cecile Tonglet's avatar
Cecile Tonglet committed
24

Cecile Tonglet's avatar
Cecile Tonglet committed
25
#[substrate_cli(
Cecile Tonglet's avatar
Cecile Tonglet committed
26
27
28
29
30
	impl_name = "parity-polkadot",
	support_url = "https://github.com/paritytech/polkadot/issues/new",
	copyright_start_year = 2017,
	executable_name = "polkadot",
)]
Cecile Tonglet's avatar
Cecile Tonglet committed
31
32
impl SubstrateCli for Cli {
	fn load_spec(&self, id: &str) -> std::result::Result<Box<dyn sc_service::ChainSpec>, String> {
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
		Ok(match id {
			"polkadot-dev" | "dev" => Box::new(service::chain_spec::polkadot_development_config()),
			"polkadot-local" => Box::new(service::chain_spec::polkadot_local_testnet_config()),
			"polkadot-staging" => Box::new(service::chain_spec::polkadot_staging_testnet_config()),
			"kusama-dev" => Box::new(service::chain_spec::kusama_development_config()),
			"kusama-local" => Box::new(service::chain_spec::kusama_local_testnet_config()),
			"kusama-staging" => Box::new(service::chain_spec::kusama_staging_testnet_config()),
			"westend" => Box::new(service::chain_spec::westend_config()?),
			"kusama" | "" => Box::new(service::chain_spec::kusama_config()?),
			path if self.run.force_kusama => {
				Box::new(service::KusamaChainSpec::from_json_file(std::path::PathBuf::from(path))?)
			},
			path => Box::new(service::PolkadotChainSpec::from_json_file(std::path::PathBuf::from(path))?),
		})
	}
Cecile Tonglet's avatar
Cecile Tonglet committed
48
}
Gavin Wood's avatar
Gavin Wood committed
49
50

/// Parses polkadot specific CLI arguments and run the service.
51
pub fn run() -> Result<()> {
52
	let cli = Cli::from_args();
Gavin Wood's avatar
Gavin Wood committed
53

54
	match &cli.subcommand {
Gavin Wood's avatar
Gavin Wood committed
55
		None => {
Cecile Tonglet's avatar
Cecile Tonglet committed
56
			let runtime = cli.create_runner(&cli.run.base)?;
Cecile Tonglet's avatar
Cecile Tonglet committed
57
58
			let config = runtime.config();
			let is_kusama = config.chain_spec.is_kusama();
59
60
61
62
63
64
			let authority_discovery_enabled = cli.run.authority_discovery_enabled;
			let grandpa_pause = if cli.run.grandpa_pause.is_empty() {
				None
			} else {
				Some((cli.run.grandpa_pause[0], cli.run.grandpa_pause[1]))
			};
Gavin Wood's avatar
Gavin Wood committed
65
66

			if is_kusama {
67
				info!("⛓  Native runtime: {}", service::KusamaExecutor::native_version().runtime_version);
Gavin Wood's avatar
Gavin Wood committed
68
69
70
71
72
73
				info!("----------------------------");
				info!("This chain is not in any way");
				info!("      endorsed by the       ");
				info!("     KUSAMA FOUNDATION      ");
				info!("----------------------------");

Cecile Tonglet's avatar
Cecile Tonglet committed
74
				run_node::<
Gavin Wood's avatar
Gavin Wood committed
75
76
77
					service::kusama_runtime::RuntimeApi,
					service::KusamaExecutor,
					service::kusama_runtime::UncheckedExtrinsic,
78
				>(runtime, authority_discovery_enabled, grandpa_pause)
Gavin Wood's avatar
Gavin Wood committed
79
			} else {
80
				info!("⛓  Native runtime: {}", service::PolkadotExecutor::native_version().runtime_version);
Gavin Wood's avatar
Gavin Wood committed
81

Cecile Tonglet's avatar
Cecile Tonglet committed
82
				run_node::<
Gavin Wood's avatar
Gavin Wood committed
83
84
85
					service::polkadot_runtime::RuntimeApi,
					service::PolkadotExecutor,
					service::polkadot_runtime::UncheckedExtrinsic,
86
				>(runtime, authority_discovery_enabled, grandpa_pause)
Gavin Wood's avatar
Gavin Wood committed
87
88
			}
		},
Cecile Tonglet's avatar
Cecile Tonglet committed
89
		Some(Subcommand::Base(subcommand)) => {
Cecile Tonglet's avatar
Cecile Tonglet committed
90
			let runtime = cli.create_runner(subcommand)?;
Cecile Tonglet's avatar
Cecile Tonglet committed
91
			let is_kusama = runtime.config().chain_spec.is_kusama();
Gavin Wood's avatar
Gavin Wood committed
92
93

			if is_kusama {
94
				runtime.run_subcommand(subcommand, |config|
Cecile Tonglet's avatar
Cecile Tonglet committed
95
96
97
98
99
100
					service::new_chain_ops::<
						service::kusama_runtime::RuntimeApi,
						service::KusamaExecutor,
						service::kusama_runtime::UncheckedExtrinsic,
					>(config)
				)
Gavin Wood's avatar
Gavin Wood committed
101
			} else {
102
				runtime.run_subcommand(subcommand, |config|
Cecile Tonglet's avatar
Cecile Tonglet committed
103
104
105
106
107
108
					service::new_chain_ops::<
						service::polkadot_runtime::RuntimeApi,
						service::PolkadotExecutor,
						service::polkadot_runtime::UncheckedExtrinsic,
					>(config)
				)
Gavin Wood's avatar
Gavin Wood committed
109
110
			}
		},
Cecile Tonglet's avatar
Cecile Tonglet committed
111
		Some(Subcommand::ValidationWorker(cmd)) => {
Gavin Wood's avatar
Gavin Wood committed
112
113
114
			sc_cli::init_logger("");

			if cfg!(feature = "browser") {
115
				Err(sc_cli::Error::Input("Cannot run validation worker in browser".into()))
Gavin Wood's avatar
Gavin Wood committed
116
117
			} else {
				#[cfg(not(feature = "browser"))]
Cecile Tonglet's avatar
Cecile Tonglet committed
118
				service::run_validation_worker(&cmd.mem_id)?;
Gavin Wood's avatar
Gavin Wood committed
119
120
121
				Ok(())
			}
		},
122
		Some(Subcommand::Benchmark(cmd)) => {
Cecile Tonglet's avatar
Cecile Tonglet committed
123
			let runtime = cli.create_runner(cmd)?;
Cecile Tonglet's avatar
Cecile Tonglet committed
124
125
			let is_kusama = runtime.config().chain_spec.is_kusama();

126
			if is_kusama {
Cecile Tonglet's avatar
Cecile Tonglet committed
127
128
129
				runtime.sync_run(|config| {
					cmd.run::<service::kusama_runtime::Block, service::KusamaExecutor>(config)
				})
130
			} else {
Cecile Tonglet's avatar
Cecile Tonglet committed
131
132
133
				runtime.sync_run(|config| {
					cmd.run::<service::polkadot_runtime::Block, service::PolkadotExecutor>(config)
				})
134
135
			}
		},
Gavin Wood's avatar
Gavin Wood committed
136
137
138
	}
}

Cecile Tonglet's avatar
Cecile Tonglet committed
139
fn run_node<R, D, E>(
Cecile Tonglet's avatar
Cecile Tonglet committed
140
	runtime: sc_cli::Runner<Cli>,
Gavin Wood's avatar
Gavin Wood committed
141
	authority_discovery_enabled: bool,
142
	grandpa_pause: Option<(u32, u32)>,
143
) -> sc_cli::Result<()>
Gavin Wood's avatar
Gavin Wood committed
144
145
146
147
148
149
150
151
152
153
154
where
	R: ConstructRuntimeApi<Block, service::TFullClient<Block, R, D>>
		+ Send + Sync + 'static,
	<R as ConstructRuntimeApi<Block, service::TFullClient<Block, R, D>>>::RuntimeApi:
		RuntimeApiCollection<E, StateBackend = sc_client_api::StateBackendFor<service::TFullBackend<Block>, Block>>,
	<R as ConstructRuntimeApi<Block, service::TLightClient<Block, R, D>>>::RuntimeApi:
		RuntimeApiCollection<E, StateBackend = sc_client_api::StateBackendFor<service::TLightBackend<Block>, Block>>,
	E: service::Codec + Send + Sync + 'static,
	D: service::NativeExecutionDispatch + 'static,
	// Rust bug: https://github.com/rust-lang/rust/issues/24159
	<<R as ConstructRuntimeApi<Block, TFullClient<Block, R, D>>>::RuntimeApi as sp_api::ApiExt<Block>>::StateBackend:
Gavin Wood's avatar
Gavin Wood committed
155
		sp_api::StateBackend<BlakeTwo256>,
Gavin Wood's avatar
Gavin Wood committed
156
157
158
159
160
161
	// Rust bug: https://github.com/rust-lang/rust/issues/43580
	R: ConstructRuntimeApi<
		Block,
		TLightClient<R, D>
	>,
{
Cecile Tonglet's avatar
Cecile Tonglet committed
162
163
164
165
166
167
168
169
170
171
172
	runtime.run_node(
		|config| service::new_light::<R, D, E>(config),
		|config| service::new_full::<R, D, E>(
			config,
			None,
			None,
			authority_discovery_enabled,
			6000,
			grandpa_pause,
		).map(|(s, _)| s),
	)
Gavin Wood's avatar
Gavin Wood committed
173
174
175
176
177
}

// We can't simply use `service::TLightClient` due to a
// Rust bug: https://github.com/rust-lang/rust/issues/43580
type TLightClient<Runtime, Dispatch> = sc_client::Client<
Gavin Wood's avatar
Gavin Wood committed
178
	sc_client::light::backend::Backend<sc_client_db::light::LightStorage<Block>, BlakeTwo256>,
Gavin Wood's avatar
Gavin Wood committed
179
	sc_client::light::call_executor::GenesisCallExecutor<
Gavin Wood's avatar
Gavin Wood committed
180
		sc_client::light::backend::Backend<sc_client_db::light::LightStorage<Block>, BlakeTwo256>,
Gavin Wood's avatar
Gavin Wood committed
181
		sc_client::LocalCallExecutor<
Gavin Wood's avatar
Gavin Wood committed
182
			sc_client::light::backend::Backend<sc_client_db::light::LightStorage<Block>, BlakeTwo256>,
Gavin Wood's avatar
Gavin Wood committed
183
184
185
186
187
188
			sc_executor::NativeExecutor<Dispatch>
		>
	>,
	Block,
	Runtime
>;