Update app.py

app.py

@@ -95,7 +95,7 @@ async def handle_action(request: Request):
         return {"success": True, "challenge": game.get("challenge", None)}
 
     if action == 'steal':
-        # Check if the player has permission to steal
+        # Existing steal code…
         if game["permissions"].get(player, {}).get("steal", True):
             game["challenge"] = {
                 "action": 'steal',
@@ -108,21 +108,15 @@ async def handle_action(request: Request):
             return {"success": True, "message": f"Steal initiated by {player} targeting {target}. Awaiting response from {target}."}
         else:
             return {"success": False, "message": "You can only steal once per turn."}
-    
+
     if action == 'coup':
-        # Find the player object
+        # Existing coup code…
         player_data = next((p for p in game["players"] if p["name"] == player), None)
-    
         if not player_data:
             return {"success": False, "message": "Player not found."}
-    
         if player_data["coins"] < 7:
             return {"success": False, "message": "You don't have enough coins to coup another player."}
-    
-        # Deduct 7 coins for the coup
         player_data["coins"] -= 7
-    
-        # Initialize the coup challenge
         game["challenge"] = {
             "action": 'coup',
             "challenger": player,
@@ -130,16 +124,14 @@ async def handle_action(request: Request):
             "challengeType": 'coup',
             "status": 'choose'
         }
-    
         return {"success": True, "message": f"Coup initiated by {player} targeting {target}."}
 
     if action == 'challengeResponse':
+        # Existing challengeResponse code…
         if not game.get("challenge"):
             return {"success": False, "message": "No challenge pending."}
-
         challenger = next(p for p in game["players"] if p["name"] == game["challenge"]["challenger"])
         target_player = next(p for p in game["players"] if p["name"] == game["challenge"]["target"])
-
         if response == 'accept':
             coins_to_steal = min(target_player["coins"], 2)
             target_player["coins"] -= coins_to_steal
@@ -181,6 +173,52 @@ async def handle_action(request: Request):
         if not player_found:
             raise HTTPException(status_code=404, detail="Player not found in the game.")
 
+    # --- NEW: Foreign Aid ---
+    if action == 'foreignAid':
+        # Ensure it is the player's turn and they haven't already performed a gain action.
+        if game["turn"] != player:
+            return {"success": False, "message": "Not your turn."}
+        if not game["permissions"].get(player, {}).get("gain", True):
+            return {"success": False, "message": "You can only perform a gain action once per turn."}
+        # Create a pending foreign aid challenge.
+        game["challenge"] = {
+            "action": "foreignAid",
+            "challenger": player,
+            "challengeType": "foreignAid",
+            "responses": {},
+            "status": "pending"
+        }
+        game["permissions"][player]["gain"] = False
+        return {"success": True, "message": f"Foreign Aid initiated by {player}. Waiting for opponents to respond."}
+
+    if action == 'foreignAidResponse':
+        # This branch handles responses from opponents for the foreign aid action.
+        if not game.get("challenge") or game["challenge"].get("challengeType") != "foreignAid":
+            return {"success": False, "message": "No foreign aid action pending."}
+        # Only opponents may respond.
+        if player == game["challenge"]["challenger"]:
+            return {"success": False, "message": "Challenger cannot respond to their own action."}
+        # Record the response.
+        game["challenge"]["responses"][player] = response
+        # If any opponent chooses to block, immediately cancel the foreign aid.
+        if response == "block":
+            game["challenge"]["status"] = "blocked"
+            challenger = game["challenge"]["challenger"]
+            game["challenge"] = None
+            return {"success": True, "message": f"Foreign Aid blocked by {player}."}
+        else:
+            # Check if all opponents have responded (all players except the challenger).
+            total_opponents = len(game["players"]) - 1
+            if len(game["challenge"]["responses"]) == total_opponents:
+                # No opponent blocked; grant the acting player +2 coins.
+                challenger_player = next(p for p in game["players"] if p["name"] == game["challenge"]["challenger"])
+                challenger_player["coins"] += 2
+                game["challenge"] = None
+                return {"success": True, "message": f"Foreign Aid accepted. {challenger_player['name']} gains 2 coins."}
+            else:
+                return {"success": True, "message": f"{player} allowed Foreign Aid. Awaiting other responses."}
+
+    # Default response.
     return {"success": True, "message": f"Action '{action}' processed for player {player}."}
 
 @app.post("/api/joinGame")