From fad9f50ff80e6bfe01f4fdbd54f415d9d77440bf Mon Sep 17 00:00:00 2001
From: cdez <cdez4200@gmail.com>
Date: Thu, 13 Mar 2025 17:15:15 -0700
Subject: [PATCH] Fix the client to work regardless of the IP/Port combination,
 default to localhost and 8181 if nothing is provided.

---
 chatserver/client.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/chatserver/client.py b/chatserver/client.py
index 35bb3ff..9ad3c98 100644
--- a/chatserver/client.py
+++ b/chatserver/client.py
@@ -2,11 +2,19 @@ import socket
 import sys
 import select
 
-SERVER_HOST = 'localhost'
-SERVER_PORT = 8181
 BUFFER_SIZE = 1024
+DEFAULT_HOST = 'localhost'
+DEFAULT_PORT = 8181
 
 def main():
+    if len(sys.argv) != 3:
+        SERVER_HOST = input(f"Enter server host (default: {DEFAULT_HOST}): ") or DEFAULT_HOST
+        SERVER_PORT = input(f"Enter server port (default: {DEFAULT_PORT}): ") or DEFAULT_PORT
+        SERVER_PORT = int(SERVER_PORT)
+    else:
+        SERVER_HOST = sys.argv[1]
+        SERVER_PORT = int(sys.argv[2])
+
     # create a socket object
     client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
     try: