Merge pull request #74 from Kaveeshakavindi/feat/Kaveesha/setCommand

Feat/kaveesha/set command

Author: anshumanjadiya1102

src/main/java/com/mycmd/App.java

@@ -92,6 +92,7 @@ private static void registerCommands(Map<String, Command> commands) {
         commands.put("whoami", new WhoamiCommand());
         commands.put("touch", new TouchCommand());
         commands.put("time", new TimeCommand());
+        commands.put("tasklist", new TasklistCommand());
         commands.put("tree", new TreeCommand());
         commands.put("date", new DateCommand());
         commands.put("history", new HistoryCommand());
@@ -104,6 +105,8 @@ private static void registerCommands(Map<String, Command> commands) {
         commands.put("alias", new AliasCommand());
         commands.put("unalias", new UnaliasCommand());
         commands.put("rename", new RenameCommand());
-        commands.put("tasklist", new TasklistCommand());
-    }
+        commands.put("set", new SetCommand());
+          }
 }
+
+        

src/main/java/com/mycmd/ShellContext.java

@@ -130,4 +130,22 @@ public File resolvePath(String path) {
             return new File(currentDir, path);
         }
     }
+
+    // environmental variable map
+    // author: Kaveesha Fernando
+    // date: 2024-06-10
+    private final Map<String, String> envVars = new HashMap<>();
+
+    public void setEnvVar(String key, String value) {
+        envVars.put(key, value);
+    }
+
+    public String getEnvVar(String key) {
+        return envVars.get(key);
+    }
+
+    public Map<String, String> getEnvVars() {
+        return new HashMap<>(envVars);
+    }
+
 }

src/main/java/com/mycmd/commands/SetCommand.java

@@ -0,0 +1,45 @@
+package com.mycmd.commands;
+
+import com.mycmd.*;
+import java.io.IOException;
+import java.util.Map;
+
+public class SetCommand implements Command {
+
+    @Override
+    public void execute(String[] args, ShellContext context) throws IOException {
+        if (args.length == 0) {
+            // Print all environment variables
+            for (Map.Entry<String, String> entry : context.getEnvVars().entrySet()) {
+                System.out.println(entry.getKey() + "=" + entry.getValue());
+            }
+            return;
+        }
+
+        String arg = args[0];
+        if (arg.contains("=")) {
+            String[] parts = arg.split("=", 2);
+            String key = parts[0].trim();
+            String value = parts[1].trim();
+            context.setEnvVar(key, value);
+            System.out.println("Variable set: " + key + "=" + value);
+        } else {
+            String value = context.getEnvVar(arg);
+            if (value != null) {
+                System.out.println(arg + "=" + value);
+            } else {
+                System.out.println("Variable not found: " + arg);
+            }
+        }
+    }
+
+    @Override
+    public String description() {
+        return "Displays, sets, or retrieves shell environment variables.";
+    }
+
+    @Override
+    public String usage() {
+        return "Usage: set [variable[=value]]";
+    }
+}