service reconfiguration on khost update

src/kaspad.rs

@@ -511,7 +511,7 @@ pub fn purge_data_folder(config: &Config) -> Result<()> {
     )
 }
 
-pub fn check_for_updates(ctx: &Context) -> Result<()> {
+pub fn check_for_updates(ctx: &Context) -> Result<bool> {
     let mut updates = Vec::new();
     for origin in unique_origins(ctx) {
         let path = folder(&origin);
@@ -532,9 +532,10 @@ pub fn check_for_updates(ctx: &Context) -> Result<()> {
             .interact()?
     {
         update(ctx)?;
+        Ok(true)
+    } else {
+        Ok(false)
     }
-
-    Ok(())
 }
 
 pub fn find_config_by_network<'a>(

src/khost.rs

@@ -2,6 +2,8 @@ use crate::imports::*;
 
 pub const VERSION: &str = env!("CARGO_PKG_VERSION");
 
+const CONFIGURATION_REBUILD_REQUIRED: bool = true;
+
 pub fn binary() -> Result<PathBuf> {
     Ok(std::env::current_exe()?)
 }
@@ -38,6 +40,31 @@ pub fn update() -> Result<()> {
     Ok(())
 }
 
+pub fn reconfigure_if_needed(ctx: &mut Context, _services_updated: bool) -> Result<()> {
+    let version_file = data_folder().join("version");
+    let rebuild = if !version_file.exists() {
+        fs::write(version_file, VERSION)?;
+        CONFIGURATION_REBUILD_REQUIRED
+    } else {
+        let current_version = fs::read_to_string(&version_file)?;
+        if current_version != VERSION {
+            fs::write(version_file, VERSION)?;
+            CONFIGURATION_REBUILD_REQUIRED
+        } else {
+            false
+        }
+    };
+
+    if rebuild {
+        log::warning("Service configuration rebuild is required...")?;
+        kaspad::reconfigure(ctx, true)?;
+        resolver::reconfigure(ctx, true)?;
+        nginx::reconfigure(ctx)?;
+    }
+
+    Ok(())
+}
+
 pub fn surrender() {
     let _ = duct::cmd!("khost").run();
     std::process::exit(0);

src/main.rs

@@ -71,14 +71,20 @@ fn main() {
 
     init_user_interaction();
 
-    if first_run {
+    let services_updated = if first_run {
         if let Err(err) = actions::Bootstrap::select(&mut ctx) {
             log::error(err).ok();
             log::info("You can attempt another full install from 'Advanced' menu").ok();
         }
+        false
     } else {
-        kaspad::check_for_updates(&ctx).ok();
-        resolver::check_for_updates(&ctx).ok();
+        let kaspad_update = kaspad::check_for_updates(&ctx).unwrap_or_default();
+        let resolver_update = resolver::check_for_updates(&ctx).unwrap_or_default();
+        kaspad_update || resolver_update
+    };
+
+    if let Err(err) = khost::reconfigure_if_needed(&mut ctx, services_updated) {
+        log::error(err).ok();
     }
 
     actions::Main::run(&mut ctx).ok();

src/resolver.rs

@@ -286,10 +286,10 @@ pub fn status(config: &Config) -> Result<String> {
     systemd::status(config)
 }
 
-pub fn check_for_updates(ctx: &Context) -> Result<()> {
+pub fn check_for_updates(ctx: &Context) -> Result<bool> {
     let config = &ctx.config.resolver;
     if !config.enabled() {
-        return Ok(());
+        return Ok(false);
     }
     let mut updates = Vec::new();
     let origin = &config.origin;
@@ -310,9 +310,10 @@ pub fn check_for_updates(ctx: &Context) -> Result<()> {
             .interact()?
     {
         update(ctx)?;
+        Ok(true)
+    } else {
+        Ok(false)
     }
-
-    Ok(())
 }
 
 pub fn reconfigure(ctx: &mut Context, _force: bool) -> Result<()> {
@@ -328,7 +329,13 @@ pub fn reconfigure(ctx: &mut Context, _force: bool) -> Result<()> {
         && systemd::is_enabled(config.service_name())?
         && systemd::is_active(config.service_name())?
     {
-        restart(ctx)?;
+        step("Configuring 'kaspa-resolver'", || {
+            systemd::stop(config)?;
+            create_systemd_unit(ctx, &ctx.config.resolver)?;
+            systemd::daemon_reload()?;
+            systemd::start(config)?;
+            Ok(())
+        })?;
     } else {
         step("Configuring 'kaspa-resolver'", || {
             if config.enabled() {
@@ -345,7 +352,6 @@ pub fn reconfigure(ctx: &mut Context, _force: bool) -> Result<()> {
             Ok(())
         })?;
     }
-    // restart(ctx)?;
     Ok(())
 }