Change module command (neo-project#187)

* change command * Change comment * Add nuget * Update StatesDumper.cs
chenzhitong · Oct 13, 2020 · b0ff063 · b0ff063
1 parent 9d1b7f3
commit b0ff063
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 33 deletions.
diff --git a/src/StatesDumper/StatesDumper.cs b/src/StatesDumper/StatesDumper.cs
@@ -1,3 +1,4 @@
+using Neo.ConsoleService;
 using Neo.IO;
 using Neo.IO.Caching;
 using Neo.IO.Json;
@@ -35,40 +36,15 @@ private static void Dump<TKey, TValue>(IEnumerable<(TKey Key, TValue Value)> sta
             Console.WriteLine($"States ({array.Count}) have been dumped into file {path}");
         }
 
-        protected override bool OnMessage(object message)
+        /// <summary>
+        /// Process "dump storage" command
+        /// </summary>
+        [ConsoleCommand("dump storage", Category = "Storage", Description = "You can specify the key or use null to get the corresponding information from the storage")]
+        private void OnDumpStorage(UInt160 key = null)
         {
-            if (!(message is string[] args)) return false;
-            if (args.Length == 0) return false;
-            return (args[0].ToLower()) switch
-            {
-                "help" => OnHelp(args),
-                "dump" => OnDump(args),
-                _ => false,
-            };
-        }
-
-        private bool OnDump(string[] args)
-        {
-            if (args.Length < 2) return false;
-            switch (args[1].ToLower())
-            {
-                case "storage":
-                    Dump(args.Length >= 3
-                        ? Blockchain.Singleton.View.Storages.Find(UInt160.Parse(args[2]).ToArray())
-                        : Blockchain.Singleton.View.Storages.Find());
-                    return true;
-                default:
-                    return false;
-            }
-        }
-
-        private bool OnHelp(string[] args)
-        {
-            if (args.Length < 2) return false;
-            if (!string.Equals(args[1], Name, StringComparison.OrdinalIgnoreCase))
-                return false;
-            Console.Write($"{Name} Commands:\n" + "\tdump storage <key>\n");
-            return true;
+            Dump(key != null
+                ? Blockchain.Singleton.View.Storages.Find(key.ToArray())
+                : Blockchain.Singleton.View.Storages.Find());
         }
 
         public void OnPersist(StoreView snapshot, IReadOnlyList<Blockchain.ApplicationExecuted> applicationExecutedList)

diff --git a/src/StatesDumper/StatesDumper.csproj b/src/StatesDumper/StatesDumper.csproj
@@ -15,6 +15,7 @@
 
   <ItemGroup>
     <PackageReference Include="Neo" Version="3.0.0-CI00863" />
+    <PackageReference Include="Neo.ConsoleService" Version="1.0.0" />
   </ItemGroup>
 
 </Project>