feat(cli): add version check

Eason · Eason · commit 4ebf67037ad0 · 2023-04-27T18:15:10.000+08:00
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -6,6 +6,8 @@ edition = "2021"
 repository = "https://github.com/axonweb3/axon"
 
 [dependencies]
+clap = { version = "4.2", features = ["cargo"] }
+
 # byzantine = { path = "./byzantine" }
 core-api = { path = "./core/api" }
 core-cli = { path = "./core/cli" }
diff --git a/common/config-parser/src/types.rs b/common/config-parser/src/types.rs
@@ -61,13 +61,6 @@ impl Config {
         path_state
     }
 
-    pub fn data_path_for_crosschain(&self) -> PathBuf {
-        let mut path_state = self.data_path.clone();
-        path_state.push("rocksdb");
-        path_state.push("crosschain");
-        path_state
-    }
-
     pub fn data_path_for_txs_wal(&self) -> PathBuf {
         let mut path_state = self.data_path.clone();
         path_state.push("txs_wal");
@@ -79,6 +72,12 @@ impl Config {
         path_state.push("consensus_wal");
         path_state
     }
+
+    pub fn data_path_for_version(&self) -> PathBuf {
+        let mut path_state = self.data_path.clone();
+        path_state.push("axon.ver");
+        path_state
+    }
 }
 
 #[derive(Clone, Debug, Deserialize)]
diff --git a/core/cli/Cargo.toml b/core/cli/Cargo.toml
@@ -5,7 +5,8 @@ edition = "2021"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
-clap = { version = "4.0", features = ["cargo"] }
+clap = "4.2"
+semver = "1.0"
 serde = { version = "1.0", features = ["derive"] }
 serde_json = "1.0"
 
diff --git a/core/cli/src/lib.rs b/core/cli/src/lib.rs
@@ -1,19 +1,24 @@
+use std::fs::File;
+use std::io::{Read, Seek, SeekFrom, Write};
 use std::path::Path;
 
-use clap::{crate_version, Arg, ArgMatches, Command};
+use clap::builder::{IntoResettable, Str};
+use clap::{Arg, ArgMatches, Command};
+use semver::Version;
 
 use common_config_parser::{parse_file, types::Config};
 use core_run::Axon;
 use protocol::types::RichBlock;
 
 pub struct AxonCli {
+    version: Version,
     matches: ArgMatches,
 }
 
 impl AxonCli {
-    pub fn init() -> Self {
+    pub fn init(ver: impl IntoResettable<Str>) -> Self {
         let matches = Command::new("axon")
-            .version(crate_version!())
+            .version(ver)
             .arg(
                 Arg::new("config_path")
                     .short('c')
@@ -30,10 +35,12 @@ impl AxonCli {
                     .required(true)
                     .num_args(1),
             )
-            .subcommand(Command::new("run").about("Run axon process"))
-            .get_matches();
+            .subcommand(Command::new("run").about("Run axon process"));
 
-        AxonCli { matches }
+        AxonCli {
+            version: Version::parse(matches.get_version().unwrap()).unwrap(),
+            matches: matches.get_matches(),
+        }
     }
 
     pub fn start(&self) {
@@ -50,10 +57,50 @@ impl AxonCli {
         )
         .unwrap();
 
+        self.check_version(&config);
+
         register_log(&config);
 
         Axon::new(config, genesis).run().unwrap();
     }
+
+    fn check_version(&self, config: &Config) {
+        if !config.data_path.exists() {
+            std::fs::create_dir_all(&config.data_path).unwrap();
+        }
+
+        let f_path = config.data_path_for_version();
+        let mut f = File::options()
+            .create(true)
+            .read(true)
+            .write(true)
+            .open(f_path)
+            .unwrap();
+
+        let mut ver_str = String::new();
+        f.read_to_string(&mut ver_str).unwrap();
+
+        if ver_str.is_empty() {
+            return f.write_all(self.version.to_string().as_bytes()).unwrap();
+        }
+
+        let prev_version = Version::parse(&ver_str).unwrap();
+        if prev_version < latest_compatible_version() {
+            println!(
+                "The previous version {:?} is not compatible with the current version {:?}",
+                prev_version, self.version
+            );
+            std::process::exit(0);
+        }
+
+        f.seek(SeekFrom::Start(0)).unwrap();
+        f.write_all(self.version.to_string().as_bytes()).unwrap();
+        f.sync_all().unwrap();
+    }
+}
+
+fn latest_compatible_version() -> Version {
+    Version::parse("0.1.0-alpha.9").unwrap()
 }
 
 fn register_log(config: &Config) {
diff --git a/src/lib.rs b/src/lib.rs
@@ -8,5 +8,5 @@ use core_executor::FEE_ALLOCATOR;
 
 pub fn run(fee_allocator: impl FeeAllocate + 'static) {
     FEE_ALLOCATOR.swap(Arc::new(Box::new(fee_allocator)));
-    AxonCli::init().start();
+    AxonCli::init(clap::crate_version!()).start();
 }
diff --git a/src/main.rs b/src/main.rs
@@ -1,5 +1,5 @@
 use core_cli::AxonCli;
 
 fn main() {
-    AxonCli::init().start();
+    AxonCli::init(clap::crate_version!()).start();
 }

Original file line number	Diff line number	Diff line change
`@@ -8,5 +8,5 @@ use core_executor::FEE_ALLOCATOR;`
`8`	`8`
`9`	`9`	`pub fn run(fee_allocator: impl FeeAllocate + 'static) {`
`10`	`10`	`FEE_ALLOCATOR.swap(Arc::new(Box::new(fee_allocator)));`
`11`		`- AxonCli::init().start();`
	`11`	`+ AxonCli::init(clap::crate_version!()).start();`
`12`	`12`	`}`
Original file line number	Diff line number	Diff line change
`@@ -1,5 +1,5 @@`
`1`	`1`	`use core_cli::AxonCli;`
`2`	`2`
`3`	`3`	`fn main() {`
`4`		`- AxonCli::init().start();`
	`4`	`+ AxonCli::init(clap::crate_version!()).start();`
`5`	`5`	`}`