feat(core): ask for process restarts after updating

packages/core/src/commands/command.ts

@@ -1,11 +1,14 @@
 import { networks } from "@arkecosystem/crypto";
 import Command, { flags } from "@oclif/command";
+import cli from "cli-ux";
 import envPaths from "env-paths";
 import { existsSync, readdirSync } from "fs";
 import Listr from "listr";
 import { join, resolve } from "path";
 import prompts from "prompts";
 import { configManager } from "../helpers/config";
+import { confirm } from "../helpers/prompts";
+import { processManager } from "../process-manager";
 import { CommandFlags, Options } from "../types";
 
 // tslint:disable-next-line:no-var-requires
@@ -281,6 +284,22 @@ export abstract class BaseCommand extends Command {
         return this.getNetworks().map(network => ({ title: network, value: network }));
     }
 
+    protected async restartProcess(processName: string) {
+        if (processManager.exists(processName)) {
+            await confirm(`Would you like to restart the ${processName} process?`, () => {
+                try {
+                    cli.action.start(`Restarting ${processName}`);
+
+                    processManager.restart(processName);
+                } catch (error) {
+                    this.error(error.message);
+                } finally {
+                    cli.action.stop();
+                }
+            });
+        }
+    }
+
     private getEnvPaths(flags: CommandFlags): envPaths.Paths {
         return envPaths(flags.token, { suffix: "core" });
     }

packages/core/src/commands/config/cli.ts

@@ -56,11 +56,19 @@ $ ark config:cli --channel=mine
 
             await installFromChannel(this.config.name, newChannel);
 
+            this.warn(`${pkg} has been installed.`);
+
             cli.action.stop();
 
-            this.warn(`${pkg} has been installed. Please restart your relay and forger.`);
+            const { flags } = await this.parseWithNetwork(CommandLineInterfaceCommand);
+
+            await this.restartProcess(`${flags.token}-core`);
+            await this.restartProcess(`${flags.token}-relay`);
+            await this.restartProcess(`${flags.token}-forger`);
         } catch (err) {
             this.error(err.message);
+        } finally {
+            cli.action.stop();
         }
     }
 }

packages/core/src/commands/update.ts

@@ -1,7 +1,7 @@
 import Chalk from "chalk";
 import cli from "cli-ux";
 import { removeSync } from "fs-extra";
-import { requestConfirmation } from "../helpers/prompts";
+import { confirm } from "../helpers/prompts";
 import { checkForUpdates, installFromChannel } from "../helpers/update";
 import { BaseCommand } from "./command";
 
@@ -27,7 +27,7 @@ export class UpdateCommand extends BaseCommand {
                 )}.`,
             );
 
-            await requestConfirmation("Would you like to update?", async () => {
+            await confirm("Would you like to update?", async () => {
                 try {
                     cli.action.start(`Updating from ${currentVersion} to ${newVersion}`);
 
@@ -38,17 +38,16 @@ export class UpdateCommand extends BaseCommand {
                     removeSync(state.cache);
 
                     this.warn(`Version ${newVersion} has been installed.`);
-                    this.warn(
-                        'Respectively run "ark relay:restart", "ark forger:restart" or "ark core:restart" to restart your processes.',
-                    );
 
-                    // @TODO ask the user if he wants to restart the core
-                    // @TODO ask the user if he wants to restart the relay
-                    // @TODO ask the user if he wants to restart the forger
+                    const { flags } = await this.parseWithNetwork(UpdateCommand);
 
-                    process.exit();
+                    await this.restartProcess(`${flags.token}-core`);
+                    await this.restartProcess(`${flags.token}-relay`);
+                    await this.restartProcess(`${flags.token}-forger`);
                 } catch (err) {
                     this.error(err.message);
+                } finally {
+                    cli.action.stop();
                 }
             });
         } catch (err) {

packages/core/src/helpers/prompts.ts

@@ -1,6 +1,6 @@
 import prompts from "prompts";
 
-export async function requestConfirmation(message: string, callback: any): Promise<void> {
+export async function confirm(message: string, callback: any): Promise<void> {
     const { confirm } = await prompts([
         {
             type: "confirm",

packages/core/src/helpers/update.ts

@@ -106,6 +106,8 @@ export async function checkForUpdates({ config, error, warn }): Promise<any> {
         }
     } catch (err) {
         error(err.message);
+    } finally {
+        cli.action.stop();
     }
 
     return state;