command.rs 6.97 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;
21
use sc_cli::{SubstrateCli, Result};
Gavin Wood's avatar
Gavin Wood committed
22
23
use sc_executor::NativeExecutionDispatch;
use crate::cli::{Cli, Subcommand};
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

impl SubstrateCli for Cli {
	fn impl_name() -> &'static str { "parity-polkadot" }

	fn impl_version() -> &'static str { env!("SUBSTRATE_CLI_IMPL_VERSION") }

	fn description() -> &'static str { env!("CARGO_PKG_DESCRIPTION") }

	fn author() -> &'static str { env!("CARGO_PKG_AUTHORS") }

	fn support_url() -> &'static str { "https://github.com/paritytech/polkadot/issues/new" }

	fn copyright_start_year() -> i32 { 2017 }

	fn executable_name() -> &'static str { "polkadot" }

	fn load_spec(&self, id: &str) -> std::result::Result<Box<dyn sc_service::ChainSpec>, String> {
		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))?),
		})
	}
}
Gavin Wood's avatar
Gavin Wood committed
57
58

/// Parses polkadot specific CLI arguments and run the service.
59
60
61
62
pub fn run() -> Result<()> {
	let cli = Cli::from_args();

	match &cli.subcommand {
Gavin Wood's avatar
Gavin Wood committed
63
		None => {
64
65
66
67
68
69
70
71
72
			let runtime = cli.create_runner(&cli.run.base)?;
			let config = runtime.config();
			let is_kusama = config.chain_spec.is_kusama();
			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
73
74

			if is_kusama {
75
				info!("⛓  Native runtime: {}", service::KusamaExecutor::native_version().runtime_version);
Gavin Wood's avatar
Gavin Wood committed
76
77
78
79
80
81
				info!("----------------------------");
				info!("This chain is not in any way");
				info!("      endorsed by the       ");
				info!("     KUSAMA FOUNDATION      ");
				info!("----------------------------");

82
				run_node::<
Gavin Wood's avatar
Gavin Wood committed
83
84
85
					service::kusama_runtime::RuntimeApi,
					service::KusamaExecutor,
					service::kusama_runtime::UncheckedExtrinsic,
86
				>(runtime, authority_discovery_enabled, grandpa_pause)
Gavin Wood's avatar
Gavin Wood committed
87
			} else {
88
				info!("⛓  Native runtime: {}", service::PolkadotExecutor::native_version().runtime_version);
Gavin Wood's avatar
Gavin Wood committed
89

90
				run_node::<
Gavin Wood's avatar
Gavin Wood committed
91
92
93
					service::polkadot_runtime::RuntimeApi,
					service::PolkadotExecutor,
					service::polkadot_runtime::UncheckedExtrinsic,
94
				>(runtime, authority_discovery_enabled, grandpa_pause)
Gavin Wood's avatar
Gavin Wood committed
95
96
			}
		},
97
98
99
		Some(Subcommand::Base(subcommand)) => {
			let runtime = cli.create_runner(subcommand)?;
			let is_kusama = runtime.config().chain_spec.is_kusama();
Gavin Wood's avatar
Gavin Wood committed
100
101

			if is_kusama {
102
103
104
105
106
107
108
				runtime.run_subcommand(subcommand, |config|
					service::new_chain_ops::<
						service::kusama_runtime::RuntimeApi,
						service::KusamaExecutor,
						service::kusama_runtime::UncheckedExtrinsic,
					>(config)
				)
Gavin Wood's avatar
Gavin Wood committed
109
			} else {
110
111
112
113
114
115
116
				runtime.run_subcommand(subcommand, |config|
					service::new_chain_ops::<
						service::polkadot_runtime::RuntimeApi,
						service::PolkadotExecutor,
						service::polkadot_runtime::UncheckedExtrinsic,
					>(config)
				)
Gavin Wood's avatar
Gavin Wood committed
117
118
			}
		},
119
		Some(Subcommand::ValidationWorker(cmd)) => {
Gavin Wood's avatar
Gavin Wood committed
120
121
122
			sc_cli::init_logger("");

			if cfg!(feature = "browser") {
123
				Err(sc_cli::Error::Input("Cannot run validation worker in browser".into()))
Gavin Wood's avatar
Gavin Wood committed
124
125
			} else {
				#[cfg(not(feature = "browser"))]
126
				service::run_validation_worker(&cmd.mem_id)?;
Gavin Wood's avatar
Gavin Wood committed
127
128
129
				Ok(())
			}
		},
130
		Some(Subcommand::Benchmark(cmd)) => {
131
132
133
			let runtime = cli.create_runner(cmd)?;
			let is_kusama = runtime.config().chain_spec.is_kusama();

134
			if is_kusama {
135
136
137
				runtime.sync_run(|config| {
					cmd.run::<service::kusama_runtime::Block, service::KusamaExecutor>(config)
				})
138
			} else {
139
140
141
				runtime.sync_run(|config| {
					cmd.run::<service::polkadot_runtime::Block, service::PolkadotExecutor>(config)
				})
142
143
			}
		},
Gavin Wood's avatar
Gavin Wood committed
144
145
146
	}
}

147
148
fn run_node<R, D, E>(
	runtime: sc_cli::Runner<Cli>,
Gavin Wood's avatar
Gavin Wood committed
149
	authority_discovery_enabled: bool,
150
	grandpa_pause: Option<(u32, u32)>,
151
) -> sc_cli::Result<()>
Gavin Wood's avatar
Gavin Wood committed
152
153
154
155
156
157
158
159
160
161
162
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
163
		sp_api::StateBackend<BlakeTwo256>,
Gavin Wood's avatar
Gavin Wood committed
164
165
166
167
168
169
	// Rust bug: https://github.com/rust-lang/rust/issues/43580
	R: ConstructRuntimeApi<
		Block,
		TLightClient<R, D>
	>,
{
170
171
172
173
174
175
176
177
178
179
180
	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
181
182
183
184
185
}

// 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
186
	sc_client::light::backend::Backend<sc_client_db::light::LightStorage<Block>, BlakeTwo256>,
Gavin Wood's avatar
Gavin Wood committed
187
	sc_client::light::call_executor::GenesisCallExecutor<
Gavin Wood's avatar
Gavin Wood committed
188
		sc_client::light::backend::Backend<sc_client_db::light::LightStorage<Block>, BlakeTwo256>,
Gavin Wood's avatar
Gavin Wood committed
189
		sc_client::LocalCallExecutor<
Gavin Wood's avatar
Gavin Wood committed
190
			sc_client::light::backend::Backend<sc_client_db::light::LightStorage<Block>, BlakeTwo256>,
Gavin Wood's avatar
Gavin Wood committed
191
192
193
194
195
196
			sc_executor::NativeExecutor<Dispatch>
		>
	>,
	Block,
	Runtime
>;