From 58f59a3335af870bab31879b950e0b7e2516e4d0 Mon Sep 17 00:00:00 2001
From: Vincent Rodley <vindognz@gmail.com>
Date: Mon, 4 Aug 2025 11:08:02 +1200
Subject: [PATCH] added some better error msgs. maybe fixed sockets staying
 open.

---
 main.py | 35 +++++++++++++++++++++--------------
 1 file changed, 21 insertions(+), 14 deletions(-)

diff --git a/main.py b/main.py
index 0d55b30..5455949 100644
--- a/main.py
+++ b/main.py
@@ -103,7 +103,7 @@ def play_game(player1_get_move, player2_get_move):
         try:
             tile = board[col].index("O")
         except ValueError:
-            continue  # column full, skip turn (could add retry logic)
+            continue
 
         board[col][tile] = player
 
@@ -127,19 +127,24 @@ def play_local_pvp():
 
 def play_lan_server():
     HOST, PORT = "0.0.0.0", 65432
-    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
-        s.bind((HOST, PORT))
-        s.listen()
-        print("Waiting for player 2...")
-        conn, addr = s.accept()
-        with conn:
-            print(f"Connected by {addr}")
-            play_game(
-                lambda p, b: send_and_return_local_move(p, b, conn),
-                lambda p, b: socket_receive_move(conn)
-            )
+    try:
+        with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
+            s.bind((HOST, PORT))
+            s.listen()
+            print("Waiting for player 2...")
+            conn, addr = s.accept()
+            with conn:
+                print(f"Connected by {addr}")
+                play_game(
+                    lambda p, b: send_and_return_local_move(p, b, conn),
+                    lambda p, b: socket_receive_move(conn)
+                )
+    except:
+        print("Somebody broke something. Try again.")
+        input("Press ENTER to return to the menu.")
+    finally:
+        s.close()
         
-
 def play_lan_client():
     while True:
         HOST, PORT = input("Enter server IP: "), 65432
@@ -152,8 +157,10 @@ def play_lan_client():
                     lambda p, b: send_and_return_local_move(p, b, s)
                 )
                 break
-        except:
+        except ConnectionRefusedError:
             print("No game found on that IP. Try again.")
+        except ConnectionResetError or ValueError:
+            print("The game was closed by host (I think).")
 
 def send_and_return_local_move(player, board, sock):
     col = local_move_provider(player, board)