bugfix: fixed migration. Added migration saving to be more clear (#434)

near · Jan 20, 2025 · a781f2e · a781f2e
1 parent c9758d7
commit a781f2e
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 4 deletions.
diff --git a/src/config/migrations.rs b/src/config/migrations.rs
@@ -49,7 +49,7 @@ impl From<ConfigV2> for ConfigV3 {
                 .network_connection
                 .into_iter()
                 .map(|(network_name, mut network_config)| {
-                    if network_name == "testnet" && network_config.faucet_url.is_none() {
+                    if network_name == "testnet" && network_config.fastnear_url.is_none() {
                         network_config.fastnear_url =
                             Some("https://test.api.fastnear.com/".parse().unwrap());
                     }
@@ -112,7 +112,7 @@ impl From<NetworkConfigV1> for NetworkConfigV2 {
     }
 }
 
-#[derive(serde::Serialize, serde::Deserialize)]
+#[derive(serde::Serialize, serde::Deserialize, Debug)]
 #[serde(tag = "version")]
 pub enum ConfigVersion {
     #[serde(rename = "1")]
@@ -123,3 +123,13 @@ pub enum ConfigVersion {
     #[serde(rename = "3")]
     V3(ConfigV3),
 }
+
+impl ConfigVersion {
+    pub fn is_latest_version(&self) -> bool {
+        // Used match instead of matches! to compile fail if new version is added
+        match self {
+            ConfigVersion::V3(_) => true,
+            ConfigVersion::V2(_) | ConfigVersion::V1(_) => false,
+        }
+    }
+}
diff --git a/src/config/mod.rs b/src/config/mod.rs
@@ -52,7 +52,7 @@ impl Default for Config {
                 near_social_db_contract_account_id: Some("v1.social08.testnet".parse().unwrap()),
                 faucet_url: Some("https://helper.nearprotocol.com/account".parse().unwrap()),
                 meta_transaction_relayer_url: None,
-                fastnear_url: None,
+                fastnear_url: Some("https://test.api.fastnear.com/".parse().unwrap()),
                 staking_pools_factory_account_id: Some("pool.f863973.m0".parse().unwrap()),
                 coingecko_url: None,
             },
@@ -99,7 +99,14 @@ impl Config {
                 }
             })?;
 
-            Ok(config_version.into())
+            let is_latest_version = config_version.is_latest_version();
+            let config: Config = config_version.into();
+
+            if !is_latest_version {
+                Self::write_config_toml(config.clone())?;
+            }
+
+            Ok(config)
         } else {
             Ok(crate::config::Config::default())
         }
@@ -190,9 +197,11 @@ impl From<migrations::ConfigVersion> for Config {
         loop {
             config_version = match config_version {
                 migrations::ConfigVersion::V1(config_v1) => {
+                    eprintln!("Migrating config.toml from V1 to V2...");
                     migrations::ConfigVersion::V2(config_v1.into())
                 }
                 migrations::ConfigVersion::V2(config_v2) => {
+                    eprintln!("Migrating config.toml from V2 to V3...");
                     migrations::ConfigVersion::V3(config_v2.into())
                 }
                 migrations::ConfigVersion::V3(config_v3) => {