command.rs 5.89 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
21
22
23
use service::{IsKusama, Block, self, RuntimeApiCollection, TFullClient};
use sp_api::ConstructRuntimeApi;
use sc_executor::NativeExecutionDispatch;
use crate::chain_spec::load_spec;
use crate::cli::{Cli, Subcommand};
24
use sc_cli::VersionInfo;
Gavin Wood's avatar
Gavin Wood committed
25
26

/// Parses polkadot specific CLI arguments and run the service.
27
pub fn run(version: VersionInfo) -> sc_cli::Result<()> {
Gavin Wood's avatar
Gavin Wood committed
28
29
	let opt = sc_cli::from_args::<Cli>(&version);

30
	let mut config = service::Configuration::from_version(&version);
Gavin Wood's avatar
Gavin Wood committed
31
	config.impl_name = "parity-polkadot";
Gavin Wood's avatar
Gavin Wood committed
32
	let force_kusama = opt.run.force_kusama;
Gavin Wood's avatar
Gavin Wood committed
33
34
35

	match opt.subcommand {
		None => {
Gavin Wood's avatar
Gavin Wood committed
36
37
38
39
40
41
			opt.run.base.init(&version)?;
			opt.run.base.update_config(
				&mut config,
				|id| load_spec(id, force_kusama),
				&version
			)?;
Gavin Wood's avatar
Gavin Wood committed
42

Gavin Wood's avatar
Gavin Wood committed
43
			let is_kusama = config.expect_chain_spec().is_kusama();
Gavin Wood's avatar
Gavin Wood committed
44
45
46
47
48
49

			info!("{}", version.name);
			info!("  version {}", config.full_version());
			info!("  by {}, 2017-2020", version.author);
			info!("Chain specification: {}", config.expect_chain_spec().name());
			info!("Node name: {}", config.name);
50
			info!("Roles: {}", config.display_role());
Gavin Wood's avatar
Gavin Wood committed
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75

			if is_kusama {
				info!("Native runtime: {}", service::KusamaExecutor::native_version().runtime_version);
				info!("----------------------------");
				info!("This chain is not in any way");
				info!("      endorsed by the       ");
				info!("     KUSAMA FOUNDATION      ");
				info!("----------------------------");

				run_service_until_exit::<
					service::kusama_runtime::RuntimeApi,
					service::KusamaExecutor,
					service::kusama_runtime::UncheckedExtrinsic,
				>(config, opt.authority_discovery_enabled)
			} else {
				info!("Native runtime: {}", service::PolkadotExecutor::native_version().runtime_version);

				run_service_until_exit::<
					service::polkadot_runtime::RuntimeApi,
					service::PolkadotExecutor,
					service::polkadot_runtime::UncheckedExtrinsic,
				>(config, opt.authority_discovery_enabled)
			}
		},
		Some(Subcommand::Base(cmd)) => {
76
			cmd.init(&version)?;
Gavin Wood's avatar
Gavin Wood committed
77
78
79
80
81
			cmd.update_config(
				&mut config,
				|id| load_spec(id, force_kusama),
				&version
			)?;
Gavin Wood's avatar
Gavin Wood committed
82

Gavin Wood's avatar
Gavin Wood committed
83
			let is_kusama = config.expect_chain_spec().is_kusama();
Gavin Wood's avatar
Gavin Wood committed
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102

			if is_kusama {
				cmd.run(config, service::new_chain_ops::<
					service::kusama_runtime::RuntimeApi,
					service::KusamaExecutor,
					service::kusama_runtime::UncheckedExtrinsic,
				>)
			} else {
				cmd.run(config, service::new_chain_ops::<
					service::polkadot_runtime::RuntimeApi,
					service::PolkadotExecutor,
					service::polkadot_runtime::UncheckedExtrinsic,
				>)
			}
		},
		Some(Subcommand::ValidationWorker(args)) => {
			sc_cli::init_logger("");

			if cfg!(feature = "browser") {
103
				Err(sc_cli::Error::Input("Cannot run validation worker in browser".into()))
Gavin Wood's avatar
Gavin Wood committed
104
105
106
107
108
109
			} else {
				#[cfg(not(feature = "browser"))]
				service::run_validation_worker(&args.mem_id)?;
				Ok(())
			}
		},
110
111
		Some(Subcommand::Benchmark(cmd)) => {
			cmd.init(&version)?;
Gavin Wood's avatar
Gavin Wood committed
112
113
			cmd.update_config(&mut config, |id| load_spec(id, force_kusama), &version)?;
			let is_kusama = config.expect_chain_spec().is_kusama();
114
			if is_kusama {
Gavin Wood's avatar
Gavin Wood committed
115
				cmd.run::<service::kusama_runtime::Block, service::KusamaExecutor>(config)
116
			} else {
Gavin Wood's avatar
Gavin Wood committed
117
				cmd.run::<service::polkadot_runtime::Block, service::PolkadotExecutor>(config)
118
119
			}
		},
Gavin Wood's avatar
Gavin Wood committed
120
121
122
123
124
125
	}
}

fn run_service_until_exit<R, D, E>(
	config: service::Configuration,
	authority_discovery_enabled: bool,
126
) -> sc_cli::Result<()>
Gavin Wood's avatar
Gavin Wood committed
127
128
129
130
131
132
133
134
135
136
137
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
138
		sp_api::StateBackend<BlakeTwo256>,
Gavin Wood's avatar
Gavin Wood committed
139
140
141
142
143
144
145
146
147
148
	// Rust bug: https://github.com/rust-lang/rust/issues/43580
	R: ConstructRuntimeApi<
		Block,
		TLightClient<R, D>
	>,
{
	match config.roles {
		service::Roles::LIGHT =>
			sc_cli::run_service_until_exit(
				config,
149
				|config| service::new_light::<R, D, E>(config),
Gavin Wood's avatar
Gavin Wood committed
150
151
152
153
			),
		_ =>
			sc_cli::run_service_until_exit(
				config,
154
155
				|config| service::new_full::<R, D, E>(config, None, None, authority_discovery_enabled, 6000)
					.map(|(s, _)| s),
Gavin Wood's avatar
Gavin Wood committed
156
157
158
159
160
161
162
			),
	}
}

// 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
163
	sc_client::light::backend::Backend<sc_client_db::light::LightStorage<Block>, BlakeTwo256>,
Gavin Wood's avatar
Gavin Wood committed
164
	sc_client::light::call_executor::GenesisCallExecutor<
Gavin Wood's avatar
Gavin Wood committed
165
		sc_client::light::backend::Backend<sc_client_db::light::LightStorage<Block>, BlakeTwo256>,
Gavin Wood's avatar
Gavin Wood committed
166
		sc_client::LocalCallExecutor<
Gavin Wood's avatar
Gavin Wood committed
167
			sc_client::light::backend::Backend<sc_client_db::light::LightStorage<Block>, BlakeTwo256>,
Gavin Wood's avatar
Gavin Wood committed
168
169
170
171
172
173
			sc_executor::NativeExecutor<Dispatch>
		>
	>,
	Block,
	Runtime
>;