command.rs 9.33 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;
Seun Lanlege's avatar
Seun Lanlege committed
18
use service::{IdentifyVariant, self};
19
use sc_cli::{SubstrateCli, RuntimeVersion, Role};
Gavin Wood's avatar
Gavin Wood committed
20
use crate::cli::{Cli, Subcommand};
21

22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
#[derive(thiserror::Error, Debug)]
pub enum Error {
	#[error(transparent)]
	PolkadotService(#[from] service::Error),

	#[error(transparent)]
	SubstrateCli(#[from] sc_cli::Error),

	#[error(transparent)]
	SubstrateService(#[from] sc_service::Error),

	#[error("Other: {0}")]
	Other(String),
}

impl std::convert::From<String> for Error {
	fn from(s: String) -> Self {
		Self::Other(s)
	}
}

type Result<T> = std::result::Result<T, Error>;

Gavin Wood's avatar
Gavin Wood committed
45
46
47
48
49
50
51
fn get_exec_name() -> Option<String> {
	std::env::current_exe()
		.ok()
		.and_then(|pb| pb.file_name().map(|s| s.to_os_string()))
		.and_then(|s| s.into_string().ok())
}

52
impl SubstrateCli for Cli {
53
	fn impl_name() -> String { "Parity Polkadot".into() }
54

55
	fn impl_version() -> String { env!("SUBSTRATE_CLI_IMPL_VERSION").into() }
56

57
	fn description() -> String { env!("CARGO_PKG_DESCRIPTION").into() }
58

59
	fn author() -> String { env!("CARGO_PKG_AUTHORS").into() }
60

61
	fn support_url() -> String { "https://github.com/paritytech/polkadot/issues/new".into() }
62
63
64

	fn copyright_start_year() -> i32 { 2017 }

65
	fn executable_name() -> String { "polkadot".into() }
66
67

	fn load_spec(&self, id: &str) -> std::result::Result<Box<dyn sc_service::ChainSpec>, String> {
Gavin Wood's avatar
Gavin Wood committed
68
69
		let id = if id == "" {
			let n = get_exec_name().unwrap_or_default();
70
			["polkadot", "kusama", "westend", "rococo"].iter()
Gavin Wood's avatar
Gavin Wood committed
71
72
73
74
				.cloned()
				.find(|&chain| n.starts_with(chain))
				.unwrap_or("polkadot")
		} else { id };
75
		Ok(match id {
Wei Tang's avatar
Wei Tang committed
76
77
78
79
80
81
			"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()?),
Gavin Wood's avatar
Gavin Wood committed
82
			"polkadot" => Box::new(service::chain_spec::polkadot_config()?),
83
			"westend" => Box::new(service::chain_spec::westend_config()?),
Gavin Wood's avatar
Gavin Wood committed
84
			"kusama" => Box::new(service::chain_spec::kusama_config()?),
Wei Tang's avatar
Wei Tang committed
85
86
87
			"westend-dev" => Box::new(service::chain_spec::westend_development_config()?),
			"westend-local" => Box::new(service::chain_spec::westend_local_testnet_config()?),
			"westend-staging" => Box::new(service::chain_spec::westend_staging_testnet_config()?),
88
89
90
			"rococo-staging" => Box::new(service::chain_spec::rococo_staging_testnet_config()?),
			"rococo-local" => Box::new(service::chain_spec::rococo_local_testnet_config()?),
			"rococo" => Box::new(service::chain_spec::rococo_config()?),
91
92
93
94
95
96
97
98
99
			path => {
				let path = std::path::PathBuf::from(path);

				let starts_with = |prefix: &str| {
					path.file_name().map(|f| f.to_str().map(|s| s.starts_with(&prefix))).flatten().unwrap_or(false)
				};

				// When `force_*` is given or the file name starts with the name of one of the known chains,
				// we use the chain spec for the specific chain.
100
101
102
				if self.run.force_rococo || starts_with("rococo") {
					Box::new(service::RococoChainSpec::from_json_file(path)?)
				} else if self.run.force_kusama || starts_with("kusama") {
103
104
105
106
107
108
					Box::new(service::KusamaChainSpec::from_json_file(path)?)
				} else if self.run.force_westend || starts_with("westend") {
					Box::new(service::WestendChainSpec::from_json_file(path)?)
				} else {
					Box::new(service::PolkadotChainSpec::from_json_file(path)?)
				}
109
110
111
			},
		})
	}
112
113
114
115
116
117

	fn native_runtime_version(spec: &Box<dyn service::ChainSpec>) -> &'static RuntimeVersion {
		if spec.is_kusama() {
			&service::kusama_runtime::VERSION
		} else if spec.is_westend() {
			&service::westend_runtime::VERSION
118
119
		} else if spec.is_rococo() {
			&service::rococo_runtime::VERSION
120
121
122
123
		} else {
			&service::polkadot_runtime::VERSION
		}
	}
124
}
Gavin Wood's avatar
Gavin Wood committed
125

126
127
128
129
130
131
132
133
134
135
136
137
138
139
fn set_default_ss58_version(spec: &Box<dyn service::ChainSpec>) {
	use sp_core::crypto::Ss58AddressFormat;

	let ss58_version = if spec.is_kusama() {
		Ss58AddressFormat::KusamaAccount
	} else if spec.is_westend() {
		Ss58AddressFormat::SubstrateAccount
	} else {
		Ss58AddressFormat::PolkadotAccount
	};

	sp_core::crypto::set_default_ss58_version(ss58_version);
}

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

	match &cli.subcommand {
Gavin Wood's avatar
Gavin Wood committed
145
		None => {
146
147
			let runner = cli.create_runner(&cli.run.base)?;
			let chain_spec = &runner.config().chain_spec;
148
149
150

			set_default_ss58_version(chain_spec);

151
152
153
154
155
			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
156

157
			if chain_spec.is_kusama() {
Gavin Wood's avatar
Gavin Wood committed
158
159
160
161
162
				info!("----------------------------");
				info!("This chain is not in any way");
				info!("      endorsed by the       ");
				info!("     KUSAMA FOUNDATION      ");
				info!("----------------------------");
163
			}
Gavin Wood's avatar
Gavin Wood committed
164

165
166

			Ok(runner.run_node_until_exit(move |config| async move {
167
168
				let role = config.role.clone();

169
170
171
				let task_manager = match role {
					Role::Light => service::build_light(config).map(|(task_manager, _)| task_manager)
					.map_err(|e| sc_service::Error::Other(e.to_string()) ),
172
173
					_ => service::build_full(
						config,
174
						service::IsCollator::No,
175
						grandpa_pause,
176
177
178
179
180
181
					).map(|full| full.task_manager)
					.map_err(|e| sc_service::Error::Other(e.to_string()) )
				};
				task_manager
			}).map_err(|e| -> sc_cli::Error { e.into() })?)

Gavin Wood's avatar
Gavin Wood committed
182
		},
183
184
		Some(Subcommand::BuildSpec(cmd)) => {
			let runner = cli.create_runner(cmd)?;
185
			Ok(runner.sync_run(|config| cmd.run(config.chain_spec, config.network))?)
186
187
188
189
190
191
192
193
		},
		Some(Subcommand::CheckBlock(cmd)) => {
			let runner = cli.create_runner(cmd)?;
			let chain_spec = &runner.config().chain_spec;

			set_default_ss58_version(chain_spec);

			runner.async_run(|mut config| {
194
195
				let (client, _, import_queue, task_manager) = service::new_chain_ops(&mut config)
					.map_err(|e| sc_service::Error::Other(e.to_string()))?;
196
197
198
199
200
				Ok((cmd.run(client, import_queue), task_manager))
			})
		},
		Some(Subcommand::ExportBlocks(cmd)) => {
			let runner = cli.create_runner(cmd)?;
201
			let chain_spec = &runner.config().chain_spec;
202
203

			set_default_ss58_version(chain_spec);
Gavin Wood's avatar
Gavin Wood committed
204

205
			runner.async_run(|mut config| {
206
207
				let (client, _, _, task_manager) = service::new_chain_ops(&mut config)
					.map_err(|e| sc_service::Error::Other(e.to_string()))?;
208
209
210
211
212
213
214
215
216
217
				Ok((cmd.run(client, config.database), task_manager))
			})
		},
		Some(Subcommand::ExportState(cmd)) => {
			let runner = cli.create_runner(cmd)?;
			let chain_spec = &runner.config().chain_spec;

			set_default_ss58_version(chain_spec);

			runner.async_run(|mut config| {
218
219
				let (client, _, _, task_manager) = service::new_chain_ops(&mut config)
					.map_err(|e| sc_service::Error::Other(e.to_string()))?;
220
221
222
223
224
225
226
227
228
229
				Ok((cmd.run(client, config.chain_spec), task_manager))
			})
		},
		Some(Subcommand::ImportBlocks(cmd)) => {
			let runner = cli.create_runner(cmd)?;
			let chain_spec = &runner.config().chain_spec;

			set_default_ss58_version(chain_spec);

			runner.async_run(|mut config| {
230
231
				let (client, _, import_queue, task_manager) = service::new_chain_ops(&mut config)
					.map_err(|e| sc_service::Error::Other(e.to_string()))?;
232
233
234
235
236
				Ok((cmd.run(client, import_queue), task_manager))
			})
		},
		Some(Subcommand::PurgeChain(cmd)) => {
			let runner = cli.create_runner(cmd)?;
237
238
			Ok(runner.sync_run(|config| cmd.run(config.database))
				.map_err(|e| sc_service::Error::Other(e.to_string()))?)
239
240
241
242
243
244
245
246
		},
		Some(Subcommand::Revert(cmd)) => {
			let runner = cli.create_runner(cmd)?;
			let chain_spec = &runner.config().chain_spec;

			set_default_ss58_version(chain_spec);

			runner.async_run(|mut config| {
247
248
				let (client, backend, _, task_manager) = service::new_chain_ops(&mut config)
					.map_err(|e| sc_service::Error::Other(e.to_string()))?;
249
250
				Ok((cmd.run(client, backend), task_manager))
			})
Gavin Wood's avatar
Gavin Wood committed
251
		},
252
		Some(Subcommand::ValidationWorker(cmd)) => {
253
			let _ = sc_cli::init_logger("", sc_tracing::TracingReceiver::Log, None, false);
Gavin Wood's avatar
Gavin Wood committed
254

255
			if cfg!(feature = "browser") || cfg!(target_os = "android") {
256
				Err(sc_cli::Error::Input("Cannot run validation worker in browser".into()))
Gavin Wood's avatar
Gavin Wood committed
257
			} else {
258
259
				#[cfg(not(any(target_os = "android", feature = "browser")))]
				polkadot_parachain::wasm_executor::run_worker(&cmd.mem_id)?;
Gavin Wood's avatar
Gavin Wood committed
260
261
262
				Ok(())
			}
		},
263
		Some(Subcommand::Benchmark(cmd)) => {
264
265
			let runner = cli.create_runner(cmd)?;
			let chain_spec = &runner.config().chain_spec;
266
267

			set_default_ss58_version(chain_spec);
268

269
270
271
			runner.sync_run(|config| {
				cmd.run::<service::kusama_runtime::Block, service::KusamaExecutor>(config)
			})
272
		},
Bastian Köcher's avatar
Bastian Köcher committed
273
		Some(Subcommand::Key(cmd)) => cmd.run(),
274
275
	}?;
	Ok(())
Gavin Wood's avatar
Gavin Wood committed
276
}