Refactor: 2nd Update ShutdownCommand.java

Author: SauravBhowmick

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

@@ -44,8 +44,8 @@ public void execute(String[] args, ShellContext context) throws IOException {
         System.out.println("Shutdown cancelled.");
         return;
       }
-      }
     }
+  }
 
     try {
       StringBuilder cmdBuilder = new StringBuilder("shutdown");
@@ -63,15 +63,23 @@ public void execute(String[] args, ShellContext context) throws IOException {
       }
 
       Process process = pb.start();
-      try (java.util.Scanner scanner = new java.util.Scanner(System.in)) {
-        String confirmation = scanner.nextLine().trim().toLowerCase();
+      try (BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream()));
+          BufferedReader errorReader = new BufferedReader(new InputStreamReader(process.getErrorStream()))) {
+        String line;
+        while ((line = reader.readLine()) != null) {
+          System.out.println(line);
+        }
 
-        if (!confirmation.equals("yes")) {
-          System.out.println("Shutdown cancelled.");
-          return;
+        while ((line = errorReader.readLine()) != null) {
+          System.err.println(line);
         }
       }
 
+      if (!process.waitFor(30, TimeUnit.SECONDS)) {
+        process.destroyForcibly();
+        System.out.println("Command timed out.");
+      }
+
     } catch (Exception e) {
       System.out.println("Error executing shutdown: " + e.getMessage());
       System.out.println("Administrator privileges may be required.");