1234567891011121314151617use sc_chain_spec::{ChainSpecExtension, ChainSpecGroup};18use sc_service::ChainType;19use sp_core::{sr25519, Pair, Public};20use sp_runtime::traits::{IdentifyAccount, Verify};21use std::collections::BTreeMap;2223use serde::{Deserialize, Serialize};24use serde_json::map::Map;2526use up_common::types::opaque::*;2728#[cfg(feature = "unique-runtime")]29pub use unique_runtime as default_runtime;3031#[cfg(all(not(feature = "unique-runtime"), feature = "quartz-runtime"))]32pub use quartz_runtime as default_runtime;3334#[cfg(all(not(feature = "unique-runtime"), not(feature = "quartz-runtime")))]35pub use opal_runtime as default_runtime;363738#[cfg(feature = "unique-runtime")]39pub type UniqueChainSpec = sc_service::GenericChainSpec<unique_runtime::GenesisConfig, Extensions>;404142#[cfg(feature = "quartz-runtime")]43pub type QuartzChainSpec = sc_service::GenericChainSpec<quartz_runtime::GenesisConfig, Extensions>;444546pub type OpalChainSpec = sc_service::GenericChainSpec<opal_runtime::GenesisConfig, Extensions>;4748#[cfg(feature = "unique-runtime")]49pub type DefaultChainSpec = UniqueChainSpec;5051#[cfg(all(not(feature = "unique-runtime"), feature = "quartz-runtime"))]52pub type DefaultChainSpec = QuartzChainSpec;5354#[cfg(all(not(feature = "unique-runtime"), not(feature = "quartz-runtime")))]55pub type DefaultChainSpec = OpalChainSpec;5657#[cfg(not(feature = "unique-runtime"))]5859const PARA_ID: u32 = 2095;6061#[cfg(feature = "unique-runtime")]6263const PARA_ID: u32 = 2037;6465pub trait RuntimeIdentification {66 fn runtime_id(&self) -> RuntimeId;67}6869impl RuntimeIdentification for Box<dyn sc_service::ChainSpec> {70 fn runtime_id(&self) -> RuntimeId {71 #[cfg(feature = "unique-runtime")]72 if self.id().starts_with("unique") || self.id().starts_with("unq") {73 return RuntimeId::Unique;74 }7576 #[cfg(feature = "quartz-runtime")]77 if self.id().starts_with("quartz")78 || self.id().starts_with("qtz")79 || self.id().starts_with("sapphire")80 {81 return RuntimeId::Quartz;82 }8384 if self.id().starts_with("opal") || self.id() == "dev" || self.id() == "local_testnet" {85 return RuntimeId::Opal;86 }8788 RuntimeId::Unknown(self.id().into())89 }90}9192pub enum ServiceId {93 Prod,94 Dev,95}9697pub trait ServiceIdentification {98 fn service_id(&self) -> ServiceId;99}100101impl ServiceIdentification for Box<dyn sc_service::ChainSpec> {102 fn service_id(&self) -> ServiceId {103 if self.id().ends_with("dev") {104 ServiceId::Dev105 } else {106 ServiceId::Prod107 }108 }109}110111112pub fn get_from_seed<TPublic: Public>(seed: &str) -> <TPublic::Pair as Pair>::Public {113 TPublic::Pair::from_string(&format!("//{}", seed), None)114 .expect("static values are valid; qed")115 .public()116}117118119#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, ChainSpecGroup, ChainSpecExtension)]120#[serde(deny_unknown_fields)]121pub struct Extensions {122 123 pub relay_chain: String,124 125 pub para_id: u32,126}127128impl Extensions {129 130 pub fn try_get(chain_spec: &dyn sc_service::ChainSpec) -> Option<&Self> {131 sc_chain_spec::get_extension(chain_spec.extensions())132 }133}134135type AccountPublic = <Signature as Verify>::Signer;136137138pub fn get_account_id_from_seed<TPublic: Public>(seed: &str) -> AccountId139where140 AccountPublic: From<<TPublic::Pair as Pair>::Public>,141{142 AccountPublic::from(get_from_seed::<TPublic>(seed)).into_account()143}144145#[cfg(not(feature = "unique-runtime"))]146macro_rules! testnet_genesis {147 (148 $runtime:path,149 $root_key:expr,150 $initial_invulnerables:expr,151 $endowed_accounts:expr,152 $id:expr153 ) => {{154 use $runtime::*;155156 GenesisConfig {157 system: SystemConfig {158 code: WASM_BINARY159 .expect("WASM binary was not build, please build it!")160 .to_vec(),161 },162 balances: BalancesConfig {163 balances: $endowed_accounts164 .iter()165 .cloned()166 167 .map(|k| (k, 1 << 100))168 .collect(),169 },170 common: Default::default(),171 nonfungible: Default::default(),172 treasury: Default::default(),173 tokens: TokensConfig { balances: vec![] },174 sudo: SudoConfig {175 key: Some($root_key),176 },177 vesting: VestingConfig { vesting: vec![] },178 parachain_info: ParachainInfoConfig {179 parachain_id: $id.into(),180 },181 parachain_system: Default::default(),182 collator_selection: CollatorSelectionConfig {183 invulnerables: $initial_invulnerables184 .iter()185 .cloned()186 .map(|(acc, _)| acc)187 .collect(),188 },189 session: SessionConfig {190 keys: $initial_invulnerables191 .into_iter()192 .map(|(acc, aura)| {193 (194 acc.clone(), 195 acc, 196 SessionKeys { aura }, 197 )198 })199 .collect(),200 },201 aura: Default::default(),202 aura_ext: Default::default(),203 evm: EVMConfig {204 accounts: BTreeMap::new(),205 },206 ethereum: EthereumConfig {},207 polkadot_xcm: Default::default(),208 transaction_payment: Default::default(),209 }210 }};211}212213#[cfg(feature = "unique-runtime")]214macro_rules! testnet_genesis {215 (216 $runtime:path,217 $root_key:expr,218 $initial_invulnerables:expr,219 $endowed_accounts:expr,220 $id:expr221 ) => {{222 use $runtime::*;223224 GenesisConfig {225 system: SystemConfig {226 code: WASM_BINARY227 .expect("WASM binary was not build, please build it!")228 .to_vec(),229 },230 common: Default::default(),231 nonfungible: Default::default(),232 balances: BalancesConfig {233 balances: $endowed_accounts234 .iter()235 .cloned()236 237 .map(|k| (k, 1 << 100))238 .collect(),239 },240 treasury: Default::default(),241 tokens: TokensConfig { balances: vec![] },242 sudo: SudoConfig {243 key: Some($root_key),244 },245 vesting: VestingConfig { vesting: vec![] },246 parachain_info: ParachainInfoConfig {247 parachain_id: $id.into(),248 },249 parachain_system: Default::default(),250 aura: AuraConfig {251 authorities: $initial_invulnerables252 .into_iter()253 .map(|(_, aura)| aura)254 .collect(),255 },256 aura_ext: Default::default(),257 evm: EVMConfig {258 accounts: BTreeMap::new(),259 },260 ethereum: EthereumConfig {},261 polkadot_xcm: Default::default(),262 transaction_payment: Default::default(),263 }264 }};265}266267pub fn development_config() -> DefaultChainSpec {268 let mut properties = Map::new();269 properties.insert("tokenSymbol".into(), default_runtime::TOKEN_SYMBOL.into());270 properties.insert("tokenDecimals".into(), 18.into());271 properties.insert(272 "ss58Format".into(),273 default_runtime::SS58Prefix::get().into(),274 );275276 DefaultChainSpec::from_genesis(277 278 format!(279 "{}{}",280 default_runtime::RUNTIME_NAME.to_uppercase(),281 if cfg!(feature = "unique-runtime") {282 ""283 } else {284 " by UNIQUE"285 }286 )287 .as_str(),288 289 format!("{}_dev", default_runtime::RUNTIME_NAME).as_str(),290 ChainType::Local,291 move || {292 testnet_genesis!(293 default_runtime,294 295 get_account_id_from_seed::<sr25519::Public>("Alice"),296 vec![297 (298 get_account_id_from_seed::<sr25519::Public>("Alice"),299 get_from_seed::<AuraId>("Alice"),300 ),301 (302 get_account_id_from_seed::<sr25519::Public>("Bob"),303 get_from_seed::<AuraId>("Bob"),304 ),305 ],306 307 vec![308 get_account_id_from_seed::<sr25519::Public>("Alice"),309 get_account_id_from_seed::<sr25519::Public>("Bob"),310 get_account_id_from_seed::<sr25519::Public>("Charlie"),311 get_account_id_from_seed::<sr25519::Public>("Dave"),312 get_account_id_from_seed::<sr25519::Public>("Eve"),313 get_account_id_from_seed::<sr25519::Public>("Ferdie"),314 get_account_id_from_seed::<sr25519::Public>("Alice//stash"),315 get_account_id_from_seed::<sr25519::Public>("Bob//stash"),316 get_account_id_from_seed::<sr25519::Public>("Charlie//stash"),317 get_account_id_from_seed::<sr25519::Public>("Dave//stash"),318 get_account_id_from_seed::<sr25519::Public>("Eve//stash"),319 get_account_id_from_seed::<sr25519::Public>("Ferdie//stash"),320 ],321 PARA_ID322 )323 },324 325 vec![],326 327 None,328 329 None,330 None,331 332 Some(properties),333 334 Extensions {335 relay_chain: "rococo-dev".into(),336 para_id: PARA_ID,337 },338 )339}340341pub fn local_testnet_config() -> DefaultChainSpec {342 let mut properties = Map::new();343 properties.insert("tokenSymbol".into(), default_runtime::TOKEN_SYMBOL.into());344 properties.insert("tokenDecimals".into(), 18.into());345 properties.insert(346 "ss58Format".into(),347 default_runtime::SS58Prefix::get().into(),348 );349350 DefaultChainSpec::from_genesis(351 352 format!(353 "{}{}",354 default_runtime::RUNTIME_NAME.to_uppercase(),355 if cfg!(feature = "unique-runtime") {356 ""357 } else {358 " by UNIQUE"359 }360 )361 .as_str(),362 363 format!("{}_local", default_runtime::RUNTIME_NAME).as_str(),364 ChainType::Local,365 move || {366 testnet_genesis!(367 default_runtime,368 369 get_account_id_from_seed::<sr25519::Public>("Alice"),370 vec![371 (372 get_account_id_from_seed::<sr25519::Public>("Alice"),373 get_from_seed::<AuraId>("Alice"),374 ),375 (376 get_account_id_from_seed::<sr25519::Public>("Bob"),377 get_from_seed::<AuraId>("Bob"),378 ),379 ],380 381 vec![382 get_account_id_from_seed::<sr25519::Public>("Alice"),383 get_account_id_from_seed::<sr25519::Public>("Bob"),384 get_account_id_from_seed::<sr25519::Public>("Charlie"),385 get_account_id_from_seed::<sr25519::Public>("Dave"),386 get_account_id_from_seed::<sr25519::Public>("Eve"),387 get_account_id_from_seed::<sr25519::Public>("Ferdie"),388 get_account_id_from_seed::<sr25519::Public>("Alice//stash"),389 get_account_id_from_seed::<sr25519::Public>("Bob//stash"),390 get_account_id_from_seed::<sr25519::Public>("Charlie//stash"),391 get_account_id_from_seed::<sr25519::Public>("Dave//stash"),392 get_account_id_from_seed::<sr25519::Public>("Eve//stash"),393 get_account_id_from_seed::<sr25519::Public>("Ferdie//stash"),394 ],395 PARA_ID396 )397 },398 399 vec![],400 401 None,402 403 None,404 None,405 406 Some(properties),407 408 Extensions {409 relay_chain: "westend-local".into(),410 para_id: PARA_ID,411 },412 )413}