refactor: server

app/src/main/java/clientserver/App.java

@@ -10,28 +10,48 @@ public class App {
     }
 
     public static void main(String[] args) {
-        if (args.length == 0) {
-            System.out.println(
-                "Please use --args='client' or --args='server'."
-            );
+        String mode = null;
+
+        // Parse arguments
+        for (String arg : args) {
+            switch (arg.toLowerCase()) {
+                case "--client":
+                    mode = "client";
+                    break;
+                case "--server":
+                    mode = "server";
+                    break;
+                default:
+                    System.out.println("Unknown argument: " + arg);
+                    printUsage();
+                    return;
+            }
+        }
+
+        // Check if mode is specified
+        if (mode == null) {
+            System.out.println("No mode specified.");
+            printUsage();
             return;
         }
 
-        String mode = args[0];
-
-        switch (mode.toLowerCase()) {
+        // Execute based on mode
+        switch (mode) {
             case "client":
-                // Run the client
                 System.out.println("Starting client...");
                 Client.main(new String[] {});
                 break;
             case "server":
-                // Run the server
                 System.out.println("Starting server...");
                 Server.run();
                 break;
-            default:
-                System.out.println("Unknown mode: " + mode);
         }
     }
+
+    private static void printUsage() {
+        System.out.println("Usage: ./gradlew run --args='option'");
+        System.out.println("Options:");
+        System.out.println("  --client         Run in client mode");
+        System.out.println("  --server         Run in server mode");
+    }
 }