Merge pull request #18 from nihaltp/feat/color

Add ColorCommand to change text and background colors using color

Author: anshumanjadiya1102

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

@@ -52,6 +52,7 @@ private static void registerCommands(Map<String, Command> commands) {
         commands.put("exit", new ExitCommand());
         commands.put("ver", new VersionCommand());
         commands.put("title", new TitleCommand());
+        commands.put("color", new ColorCommand());
         commands.put("hostname", new HostnameCommand());
         commands.put("whoami", new WhoamiCommand());
         commands.put("touch", new TouchCommand());

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

@@ -0,0 +1,54 @@
+package com.mycmd.commands;
+
+import com.mycmd.Command;
+import com.mycmd.ShellContext;
+
+public class ColorCommand implements Command {
+    @Override
+    public void execute(String[] args, ShellContext context) {
+        if (args.length == 1) {
+            String color = args[0];
+
+            if (color.length() != 2) {
+                System.out.println("Usage: color <background><text>");
+                return;
+            }
+
+            String background = String.valueOf(color.charAt(0));
+            String text = String.valueOf(color.charAt(1));
+
+            if (background.equals(text)) {
+                return;
+            }
+
+            int bgIndex, fgIndex;
+            try {
+                bgIndex = Integer.parseInt(background,16);
+                fgIndex = Integer.parseInt(text,16);
+            } catch (NumberFormatException e) {
+                System.out.println("Invalid color code. must use two hexadecimal digits.");
+                System.out.println("Example: color 0A");
+                return;
+            }
+
+            String[] ansiForeground = {"30","34","32","36","31","35","33","37","90","94","92","96","91","95","93","97"};
+            String[] ansiBackground = {"40","44","42","46","41","45","43","47","100","104","102","106","101","105","103","107"};
+
+            if (bgIndex >= ansiBackground.length || fgIndex >= ansiForeground.length) {
+                System.out.println("Invalid color code. must use two hexadecimal digits.");
+                System.out.println("Example: color 0A");
+                return;
+            }
+
+            String bg = "\033[" + ansiBackground[bgIndex] + "m";
+            String fg = "\033[" + ansiForeground[fgIndex] + "m";
+
+            System.out.println(bg + fg);
+        }
+
+        else {
+            // set default color
+            System.out.println("\033[0m");
+        }
+    }
+}