node: Mode system

.github/workflows/test-neo-cli.expect

@@ -14,6 +14,10 @@ expect {
     timeout { exit 1 }
 }
 
+sleep 5
+
+spawn dotnet out/neo-cli.dll
+
 #
 # Test 'create wallet'
 #

Neo.ConsoleService/ConsoleServiceBase.cs

@@ -291,7 +291,14 @@ public virtual void OnStart(string[] args)
 
         public virtual void OnStop()
         {
-            _shutdownAcknowledged.Signal();
+            try
+            {
+                _shutdownAcknowledged.Signal();
+            }
+            catch
+            {
+                // ignore
+            }
         }
 
         public string ReadUserInput(string prompt, bool password = false)

neo-cli/CLI/Helper.cs

@@ -8,6 +8,8 @@
 // Redistribution and use in source and binary forms with or without
 // modifications are permitted.
 
+using System.IO;
+
 namespace Neo.CLI
 {
     internal static class Helper

neo-cli/CLI/MainService.Mode.cs

@@ -0,0 +1,209 @@
+// Copyright (C) 2016-2022 The Neo Project.
+//
+// The neo-cli is free software distributed under the MIT software
+// license, see the accompanying file LICENSE in the main directory of
+// the project or http://www.opensource.org/licenses/mit-license.php
+// for more details.
+//
+// Redistribution and use in source and binary forms with or without
+// modifications are permitted.
+
+using System;
+using System.IO;
+using System.Linq;
+using System.Threading.Tasks;
+using Akka.Util.Internal;
+using Neo.ConsoleService;
+namespace Neo.CLI;
+
+partial class MainService
+{
+    /// <summary>
+    /// Process "mode list" command.
+    /// </summary>
+    [ConsoleCommand("mode list", Category = "Mode Commands")]
+    private void OnListModes()
+    {
+        try
+        {
+            Directory.GetDirectories(ModePath).ForEach(p => ConsoleHelper.Info(p));
+        }
+        catch (IOException)
+        {
+        }
+    }
+
+    /// <summary>
+    /// Process "mode save" command
+    /// <param name="modeName">Mode name</param>
+    /// </summary>
+    [ConsoleCommand("mode save", Category = "Mode Commands")]
+    private void OnSaveMode(string modeName)
+    {
+        // if no mode name assigned
+        if (modeName is null)
+        {
+            ConsoleHelper.Error("No mode name assigned.");
+            return;
+        }
+        modeName = modeName.ToLower();
+        try
+        {
+            var targetMode = new DirectoryInfo($"{ModePath}/{modeName}");
+            // Create the mode if it does not exist
+            if (!targetMode.Exists) Directory.CreateDirectory(targetMode.FullName);
+            // Get the config files of the node
+            MoveModeConfig(modeName.ToLower());
+            var plugins = new DirectoryInfo(PluginPath);
+            // Cache directories before we start copying
+            var dirs = plugins.GetDirectories();
+            // Create an empty .PLUGINS file
+            File.Create($"{ModePath}/{modeName}/.PLUGINS").Close();
+            // Save the Plugin files
+            foreach (var plugin in dirs)
+            {
+                foreach (var file in plugin.GetFiles().Where(p => p.Extension == ".json"))
+                {
+                    file.CopyTo($"{ModePath}/{modeName}/{plugin.Name}.json", true);
+                }
+                AddPluginToMode(plugin.Name, modeName);
+            }
+        }
+        catch (Exception e)
+        {
+            Console.WriteLine(e);
+            throw;
+        }
+    }
+
+    /// <summary>
+    /// Process "mode delete" command
+    /// <param name="modeName">Mode name</param>
+    /// </summary>
+    [ConsoleCommand("mode delete", Category = "Mode Commands")]
+    private void OnDeleteMode(string modeName)
+    {
+        try
+        {
+            var dir = new DirectoryInfo($"{ModePath}/{modeName.ToLower()}");
+            if (!dir.Exists)
+                return;
+            Directory.Delete(dir.FullName, true);
+            ConsoleHelper.Info("Mode ", modeName, " was deleted.");
+        }
+        catch (Exception e)
+        {
+            Console.WriteLine(e);
+            throw;
+        }
+    }
+
+    /// <summary>
+    /// Load the target mode, plugin should be according to the mode,
+    /// if the mode contains the plugin, install the plugin, otherwise delete the plugin
+    /// </summary>
+    /// <param name="mode"> name of the mode</param>
+    /// <exception cref="DirectoryNotFoundException"> if the mode is not found</exception>
+    private async Task LoadMode(string mode)
+    {
+        try
+        {
+            mode = mode.ToLower();
+            var dir = new DirectoryInfo($"{ModePath}/{mode}");
+            if (!dir.Exists)
+                throw new DirectoryNotFoundException($"Mode not found: {dir.FullName}");
+            // Process the plugin
+            var modePlugins = File.ReadAllLines($"{ModePath}/{_currentMode}/.PLUGINS");
+            // loop modePlugins
+            foreach (var pluginName in modePlugins)
+            {
+                // if the plugin is not installed, install it
+                if (!Directory.Exists($"{PluginPath}/{pluginName}/"))
+                {
+                    await InstallPluginAsync(pluginName, overWrite: true, saveConfig: false);
+                    _needRestart = true;
+                }
+                // if the mode has the plugin config, load the config from the mode
+                if (File.Exists($"{ModePath}/{mode}/{pluginName}.json"))
+                    File.Copy($"{ModePath}/{mode}/{pluginName}.json",
+                    $"{PluginPath}/{pluginName}/config.json", true);
+            }
+            // get the system config file from the mode
+            MoveModeConfig(mode, false);
+
+            // Get existing plugins and delete them if they are not in the mode
+            new DirectoryInfo($"{PluginPath}/").GetDirectories().ForEach(p =>
+            {
+                if (modePlugins.Any(k => string.Compare(Path.GetFileNameWithoutExtension(k), p.Name, StringComparison.OrdinalIgnoreCase) == 0)
+                    || !File.Exists($"{PluginPath}/{p.Name}/config.json")) return;
+                try
+                {
+                    ConsoleHelper.Info("Removing plugin ", p.Name);
+                    Directory.Delete($"{PluginPath}/{p.Name}", true);
+                    _needRestart = true;
+                }
+                catch
+                {
+                    // ignored
+                }
+            });
+        }
+        catch (Exception e)
+        {
+            Console.WriteLine(e);
+            throw;
+        }
+    }
+
+    /// save config.json and config.fs.json to the mode directory
+    /// <param name="mode"> name of the mode</param>
+    /// <param name="toMode"></param>
+    /// <exception cref="DirectoryNotFoundException"> if the mode is not found</exception>
+    private static void MoveModeConfig(string mode, bool toMode = true)
+    {
+        var modeDir = new DirectoryInfo($"{ModePath}/{mode.ToLower()}");
+        var configDir = new DirectoryInfo($"{StrExeFilePath}");
+        if (!modeDir.Exists)
+            throw new DirectoryNotFoundException($"Mode not found: {modeDir.FullName}");
+        try
+        {
+            if (toMode)
+            {
+                File.Copy(configDir.FullName + "/config.json",
+                    modeDir.FullName + "/config.json", true);
+                File.Copy(configDir.FullName + "/config.fs.json",
+                    modeDir.FullName + "/config.fs.json", true);
+            }
+            else
+            {
+                File.Copy(modeDir.FullName + "/config.json",
+                    configDir.FullName + "/config.json", true);
+                File.Copy(modeDir.FullName + "/config.fs.json",
+                    configDir.FullName + "/config.fs.json", true);
+            }
+        }
+        catch (Exception e)
+        {
+            Console.WriteLine(e);
+            throw;
+        }
+    }
+
+    // Add plugin to .PLUGINS file
+    private static void AddPluginToMode(string pluginName, string modeName)
+    {
+        var plugins = File.ReadAllLines($"{ModePath}/{modeName}/.PLUGINS");
+        if (plugins.Contains(pluginName)) return;
+        var newPlugins = plugins.Append(pluginName).ToArray();
+        File.WriteAllLines($"{ModePath}/{modeName}/.PLUGINS", newPlugins);
+    }
+
+    // Remove plugin from .PLUGINS file
+    private static void RemovePluginFromMode(string pluginName, string modeName)
+    {
+        var plugins = File.ReadAllLines($"{ModePath}/{modeName}/.PLUGINS");
+        // if (plugins.All(p => !string.Equals(p, pluginName, StringComparison.CurrentCultureIgnoreCase))) return;
+        var newPlugins = plugins.Where(p => !string.Equals(p, pluginName, StringComparison.CurrentCultureIgnoreCase)).ToArray();
+        File.WriteAllLines($"{ModePath}/{modeName}/.PLUGINS", newPlugins);
+    }
+}

neo-cli/CLI/MainService.Plugins.cs

@@ -1,5 +1,5 @@
 // Copyright (C) 2016-2022 The Neo Project.
-// The neo-cli is free software distributed under the MIT software 
+// The neo-cli is free software distributed under the MIT software
 // license, see the accompanying file LICENSE in the main directory of
 // the project or http://www.opensource.org/licenses/mit-license.php
 // for more details.
@@ -34,7 +34,7 @@ private async Task OnInstallCommandAsync(string pluginName)
         {
             if (PluginExists(pluginName))
             {
-                ConsoleHelper.Warning($"Plugin already exist.");
+                ConsoleHelper.Warning("Plugin already exist.");
                 return;
             }
 
@@ -119,9 +119,11 @@ private async Task<MemoryStream> DownloadPluginAsync(string pluginName)
         /// Install plugin from stream
         /// </summary>
         /// <param name="pluginName">name of the plugin</param>
+        /// <param name="installed">installed dependency</param>
         /// <param name="overWrite">Install by force for `update`</param>
+        /// <param name="saveConfig">Need to save the config file to the mode</param>
         private async Task InstallPluginAsync(string pluginName, HashSet<string> installed = null,
-            bool overWrite = false)
+            bool overWrite = false, bool saveConfig = true)
         {
             installed ??= new HashSet<string>();
             if (!installed.Add(pluginName)) return;
@@ -140,7 +142,40 @@ private async Task InstallPluginAsync(string pluginName, HashSet<string> install
                 await using Stream es = entry.Open();
                 await InstallDependenciesAsync(es, installed);
             }
-            zip.ExtractToDirectory("./", true);
+            if (!Directory.Exists($"{StrExeFilePath}Plugins"))
+            {
+                Directory.CreateDirectory($"{StrExeFilePath}Plugins");
+            }
+            zip.ExtractToDirectory($"{StrExeFilePath}", true);
+            Console.WriteLine();
+
+            if (!saveConfig) return;
+            // Save the config.json to current mode
+            try
+            {
+                var pluginActualName = GetPluginActualName(pluginName);
+                if (File.Exists($"{ModePath}/{_currentMode}/{pluginActualName}.json"))
+                {
+                    if (File.Exists($"{PluginPath}/{pluginActualName}/config.json"))
+                        // plugin contains config.json && mode contains plugin.json
+                        // replace the config.json with plugin.json from mode
+                        File.Copy($"{ModePath}/{_currentMode}/{pluginActualName}.json", $"{PluginPath}/{pluginActualName}/config.json", true);
+                    else
+                        // plugin doesn't contain config.json && mode contains plugin.json
+                        // delete the plugin.json from mode
+                        File.Delete($"{ModePath}/{_currentMode}/{pluginActualName}.json");
+                }
+                else if (File.Exists($"{PluginPath}/{pluginActualName}/config.json"))
+                    // plugin contains config.json && mode doesn't contain plugin.json
+                    // copy the config.json to mode
+                    File.Copy($"{PluginPath}/{pluginActualName}/config.json", $"{ModePath}/{_currentMode}/{pluginActualName}.json", false);
+                AddPluginToMode(pluginActualName, _currentMode);
+            }
+            catch (Exception e)
+            {
+                // ignored
+                Console.WriteLine(e.Message);
+            }
         }
 
         /// <summary>
@@ -173,7 +208,8 @@ private async Task InstallDependenciesAsync(Stream config, HashSet<string> insta
         /// <returns></returns>
         private static bool PluginExists(string pluginName)
         {
-            return Plugin.Plugins.Any(p => p.Name.Equals(pluginName, StringComparison.InvariantCultureIgnoreCase));
+            return Plugin.Plugins.Any(p => p.Name.Equals(pluginName, StringComparison.InvariantCultureIgnoreCase)) ||
+                   new DirectoryInfo("Plugins").GetDirectories().Any(p => p.Name.Equals(pluginName, StringComparison.InvariantCultureIgnoreCase));
         }
 
         /// <summary>
@@ -188,12 +224,12 @@ private void OnUnInstallCommand(string pluginName)
                 ConsoleHelper.Warning("Plugin not found");
                 return;
             }
-
+            pluginName = GetPluginActualName(pluginName);
             foreach (var p in Plugin.Plugins)
             {
                 try
                 {
-                    using var reader = File.OpenRead($"./Plugins/{p.Name}/config.json");
+                    using var reader = File.OpenRead($"{PluginPath}/{p.Name}/config.json");
                     if (new ConfigurationBuilder()
                         .AddJsonStream(reader)
                         .Build()
@@ -214,7 +250,11 @@ private void OnUnInstallCommand(string pluginName)
             }
             try
             {
-                Directory.Delete($"Plugins/{pluginName}", true);
+                Directory.Delete($"{PluginPath}/{pluginName}", true);
+                var config = $"{ModePath}/{_currentMode}/{pluginName}.json";
+                if (File.Exists(config))
+                    File.Delete(config);
+                RemovePluginFromMode(pluginName, _currentMode);
             }
             catch (IOException) { }
             ConsoleHelper.Info("Uninstall successful, please restart neo-cli.");
@@ -240,5 +280,17 @@ private void OnPluginsCommand()
                 ConsoleHelper.Warning("No loaded plugins");
             }
         }
+
+        private static string GetPluginActualName(string pluginName)
+        {
+            var pluginActualName = "";
+            foreach (var plugin in new DirectoryInfo($"{PluginPath}").GetDirectories())
+            {
+                if (!string.Equals(plugin.Name, pluginName, StringComparison.CurrentCultureIgnoreCase)) continue;
+                pluginActualName = plugin.Name;
+                break;
+            }
+            return pluginActualName;
+        }
     }
 }