Update src/dotnet_mytool/Program.cs

guibranco · Nov 18, 2024 · 1828819 · 1828819
1 parent ec7033f
commit 1828819
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/src/dotnet_mytool/Program.cs b/src/dotnet_mytool/Program.cs
@@ -14,6 +14,24 @@ static int Main(string[] args)
                 new Argument<string>("variable", "The environment variable in the format VAR_NAME=value"),
                 new Option<string>("--target", () => "User", "The target scope: User or Machine")
             };
+            setEnvCommand.Handler = CommandHandler.Create<string, string>((variable, target) =>
+            {
+                var parts = variable.Split('=');
+                if (parts.Length != 2)
+
+        static void SetEnvironmentVariable(string variable, string value, string target)
+        {
+            EnvironmentVariableTarget envTarget = target.Equals("Machine", StringComparison.OrdinalIgnoreCase)
+                ? EnvironmentVariableTarget.Machine
+                : EnvironmentVariableTarget.User;
+
+            Environment.SetEnvironmentVariable(variable, value, envTarget);
+            Console.WriteLine($"Environment variable {variable} set to {value} for {target}.");
+        }
+                {
+                    Console.WriteLine("Invalid format. Use VAR_NAME=value.");
+                    return;
+                }
 
             rootCommand.AddCommand(setEnvCommand);