Reimplemented Windows Vault logic

devexlead · Jan 30, 2025 · 667e31c · 667e31c
1 parent f76744c
commit 667e31c
Show file tree

Hide file tree

Showing 6 changed files with 50 additions and 31 deletions.
diff --git a/src/DevEx.Console/DevEx.Console.csproj b/src/DevEx.Console/DevEx.Console.csproj
@@ -8,7 +8,7 @@
 		<PackAsTool>true</PackAsTool>
 		<!-- The command users will run -->
 		<ToolCommandName>dxc</ToolCommandName>
-		<UserSecretsId>5a2cdd97-0170-4aaa-9816-a59401e8de2b</UserSecretsId>
+		<UserSecretsId>6b2e2731-a735-438e-bf6f-e749e0ebcd02</UserSecretsId>
 	</PropertyGroup>
 
 	<ItemGroup>

diff --git a/src/DevEx.Console/Properties/launchSettings.json b/src/DevEx.Console/Properties/launchSettings.json
@@ -2,7 +2,7 @@
   "profiles": {
     "DevEx.Console": {
       "commandName": "Project",
-      "commandLineArgs": "vault delete --key \"a\""
+      "commandLineArgs": "vault delete --key ab"
     }
   }
 }
diff --git a/src/DevEx.Modules/DevEx.Modules.Vault/Windows/Handlers/WindowsVaultCreateCommandHandler.cs b/src/DevEx.Modules/DevEx.Modules.Vault/Windows/Handlers/WindowsVaultCreateCommandHandler.cs
@@ -1,21 +1,28 @@
 using DevEx.Core;
+using DevEx.Core.Storage;
 
 namespace DevEx.Modules.Vault.Windows.Handlers
 {
     public class WindowsVaultCreateCommandHandler : ICommandHandler
     {
         public void Execute(Dictionary<string, string> options)
         {
-            Console.WriteLine("Executing Vault Create Command:");
-            PrintOptions(options);
-        }
+            var key = options["key"];
+            var value = options["value"];
 
-        private void PrintOptions(Dictionary<string, string> options)
-        {
-            foreach (var option in options)
+            if (string.IsNullOrWhiteSpace(key) || string.IsNullOrWhiteSpace(value))
             {
-                Console.WriteLine($"{option.Key}: {option.Value}");
+                Console.WriteLine("Both --key and --value are required for create.");
+                return;
             }
+
+            var userStorage = UserStorageManager.GetUserStorage();
+            value = WindowsVaultHelper.Encrypt(value);
+            userStorage.Vault.Add(key, value);
+            UserStorageManager.SaveUserStorage(userStorage);
+
+            Console.WriteLine($"Created item: Key={key}, New Value={value}");
         }
+
     }
 }
diff --git a/src/DevEx.Modules/DevEx.Modules.Vault/Windows/Handlers/WindowsVaultDeleteCommandHandler.cs b/src/DevEx.Modules/DevEx.Modules.Vault/Windows/Handlers/WindowsVaultDeleteCommandHandler.cs
@@ -1,4 +1,5 @@
 using DevEx.Core;
+using DevEx.Core.Storage;
 
 namespace DevEx.Modules.Vault.Windows.Handlers
 {
@@ -7,16 +8,20 @@ public class WindowsVaultDeleteCommandHandler : ICommandHandler
     {
         public void Execute(Dictionary<string, string> options)
         {
-            Console.WriteLine("Executing Vault Delete Command:");
-            PrintOptions(options);
-        }
+            var key = options["key"];
 
-        private void PrintOptions(Dictionary<string, string> options)
-        {
-            foreach (var option in options)
+            if (string.IsNullOrWhiteSpace(key))
             {
-                Console.WriteLine($"{option.Key}: {option.Value}");
+                Console.WriteLine("--key is required for delete.");
+                return;
             }
+
+            var userStorage = UserStorageManager.GetUserStorage();
+            userStorage.Vault.Remove(key);
+            UserStorageManager.SaveUserStorage(userStorage);
+            Console.WriteLine($"Deleted item with Key={options["key"]}");
         }
+
+
     }
 }
diff --git a/src/DevEx.Modules/DevEx.Modules.Vault/Windows/Handlers/WindowsVaultReadCommandHandler.cs b/src/DevEx.Modules/DevEx.Modules.Vault/Windows/Handlers/WindowsVaultReadCommandHandler.cs
@@ -1,21 +1,23 @@
 using DevEx.Core;
+using DevEx.Core.Storage;
 
 namespace DevEx.Modules.Vault.Windows.Handlers
 {
     public class WindowsVaultReadCommandHandler : ICommandHandler
     {
         public void Execute(Dictionary<string, string> options)
         {
-            Console.WriteLine("Executing Vault Read Command:");
-            PrintOptions(options);
-        }
-
-        private void PrintOptions(Dictionary<string, string> options)
-        {
-            foreach (var option in options)
+            var key = options["key"];
+            if (string.IsNullOrWhiteSpace(key))
             {
-                Console.WriteLine($"{option.Key}: {option.Value}");
+                Console.WriteLine("--key is required for read.");
+                return;
             }
+
+            var userStorage = UserStorageManager.GetUserStorage();
+            var value = userStorage.Vault.FirstOrDefault(v => v.Key.Equals(key)).Value;
+            value = WindowsVaultHelper.Decrypt(value);
+            Console.WriteLine($"Fetched item with Key={options["key"]} and Value={value}");
         }
     }
 }
diff --git a/src/DevEx.Modules/DevEx.Modules.Vault/Windows/Handlers/WindowsVaultUpdateCommandHandler.cs b/src/DevEx.Modules/DevEx.Modules.Vault/Windows/Handlers/WindowsVaultUpdateCommandHandler.cs
@@ -1,21 +1,26 @@
 using DevEx.Core;
+using DevEx.Core.Storage;
 
 namespace DevEx.Modules.Vault.Windows.Handlers
 {
     public class WindowsVaultUpdateCommandHandler : ICommandHandler
     {
         public void Execute(Dictionary<string, string> options)
         {
-            Console.WriteLine("Executing Vault Update Command:");
-            PrintOptions(options);
-        }
+            var key = options["key"];
+            var value = options["value"];
 
-        private void PrintOptions(Dictionary<string, string> options)
-        {
-            foreach (var option in options)
+            if (string.IsNullOrWhiteSpace(key) || string.IsNullOrWhiteSpace(value))
             {
-                Console.WriteLine($"{option.Key}: {option.Value}");
+                Console.WriteLine("Both --key and --value are required for modify.");
+                return;
             }
+
+            var userStorage = UserStorageManager.GetUserStorage();
+            userStorage.Vault[key] = WindowsVaultHelper.Encrypt(value);
+            UserStorageManager.SaveUserStorage(userStorage);
+
+            Console.WriteLine($"Modified item: Key={key}, New Value={value}");
         }
     }
 }