mirror of
https://github.com/pezkuwichain/pezkuwi-subxt.git
synced 2026-04-26 14:37:57 +00:00
PG Herveou
076529ea55
* Update runtime.rs * Fix * Revert "Update runtime.rs" This reverts commit 808f026a835b1f2707f640799454a50fd3a8be7b. * nit mono-repo fixes
74 lines
2.5 KiB
Rust
74 lines
2.5 KiB
Rust
// This file is part of Substrate.
|
|
|
|
// Copyright (C) Parity Technologies (UK) Ltd.
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
use std::io::Write;
|
|
|
|
/// Get the latest migration version.
|
|
///
|
|
/// Find the highest version number from the available migration files.
|
|
/// Each migration file should follow the naming convention `vXX.rs`, where `XX` is the version
|
|
/// number.
|
|
fn get_latest_version() -> u16 {
|
|
std::fs::read_dir("src/migration")
|
|
.expect("Folder `src/migration` not found.")
|
|
.filter_map(|entry| {
|
|
let file_name = entry.as_ref().ok()?.file_name();
|
|
let file_name = file_name.to_str()?;
|
|
if file_name.starts_with('v') && file_name.ends_with(".rs") {
|
|
let version = &file_name[1..&file_name.len() - 3];
|
|
let version = version.parse::<u16>().ok()?;
|
|
|
|
// Ensure that the version matches the one defined in the file.
|
|
let path = entry.unwrap().path();
|
|
let file_content = std::fs::read_to_string(&path).ok()?;
|
|
assert!(
|
|
file_content.contains(&format!("const VERSION: u16 = {}", version)),
|
|
"Invalid MigrationStep::VERSION in {:?}",
|
|
path
|
|
);
|
|
|
|
return Some(version)
|
|
}
|
|
None
|
|
})
|
|
.max()
|
|
.expect("Failed to find any files matching the 'src/migration/vxx.rs' pattern.")
|
|
}
|
|
|
|
/// Generates a module that exposes the latest migration version, and the benchmark migrations type.
|
|
fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|
let out_dir = std::env::var("OUT_DIR")?;
|
|
let path = std::path::Path::new(&out_dir).join("migration_codegen.rs");
|
|
let mut f = std::fs::File::create(path)?;
|
|
let version = get_latest_version();
|
|
write!(
|
|
f,
|
|
"
|
|
pub mod codegen {{
|
|
use crate::NoopMigration;
|
|
/// The latest migration version, pulled from the latest migration file.
|
|
pub const LATEST_MIGRATION_VERSION: u16 = {version};
|
|
/// The Migration Steps used for benchmarking the migration framework.
|
|
pub type BenchMigrations = (NoopMigration<{}>, NoopMigration<{version}>);
|
|
}}",
|
|
version - 1,
|
|
)?;
|
|
|
|
println!("cargo:rerun-if-changed=src/migration");
|
|
Ok(())
|
|
}
|