File size: 3,356 Bytes
3827a07 25fb0b9 3827a07 e59981a 3827a07 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 |
from fastapi import FastAPI, WebSocket, HTTPException
from fastapi.staticfiles import StaticFiles
from pydantic import BaseModel
from fastapi.responses import FileResponse
import uuid, random, asyncio
app = FastAPI()
# Serve the index.html at root
@app.get("/")
def read_index():
return FileResponse("index.html")
# --- Data Models ---
class CreateLobby(BaseModel):
max_players: int
starting_amount: int
passkey: str
class JoinLobby(BaseModel):
lobby_id: str
player_name: str
passkey: str
# --- In-memory state ---
lobbies = {} # lobby_id -> Lobby instance
class Player:
def __init__(self, name, amount):
self.name = name
self.amount = amount
self.cards = []
self.active = True
class Lobby:
def __init__(self, max_players, start_amt, passkey):
self.id = str(uuid.uuid4())[:8]
self.passkey = passkey
self.max_players = max_players
self.start_amt = start_amt
self.players = [] # list[Player]
self.state = "waiting" # waiting | dealing | betting | showdown
self.deck = []
self.pot = 0
self.community = [] # flop/turn/river cards
def build_deck(self):
suits = ['♠','♥','♦','♣']
ranks = [str(r) for r in range(2,11)] + list("JQKA")
self.deck = [r + s for s in suits for r in ranks]
random.shuffle(self.deck)
print(f"[Lobby {self.id}] deck shuffled")
def deal(self):
self.build_deck()
# deal hole cards
for p in self.players:
p.cards = [self.deck.pop(), self.deck.pop()]
print(f"[Lobby {self.id}] Dealt to {p.name}: {p.cards}")
# --- Endpoints ---
@app.post("/create_lobby")
async def create_lobby(data: CreateLobby):
lobby = Lobby(data.max_players, data.starting_amount, data.passkey)
lobbies[lobby.id] = lobby
print(f"Lobby {lobby.id} created, max_players={lobby.max_players}")
return {"lobby_id": lobby.id}
@app.post("/join_lobby")
async def join_lobby(data: JoinLobby):
lobby = lobbies.get(data.lobby_id)
if not lobby or lobby.passkey != data.passkey:
raise HTTPException(404, "Lobby not found or wrong passkey")
if len(lobby.players) >= lobby.max_players:
raise HTTPException(400, "Lobby is full")
p = Player(data.player_name, lobby.start_amt)
lobby.players.append(p)
print(f"{p.name} joined Lobby {lobby.id}")
# Auto-start when full
if len(lobby.players) == lobby.max_players:
lobby.state = "dealing"
lobby.deal()
return {"status": "joined"}
@app.get("/lobby/{lobby_id}/state")
async def lobby_state(lobby_id: str):
lobby = lobbies.get(lobby_id)
if not lobby:
raise HTTPException(404, "Lobby not found")
# Simplified: only reveal each player’s name, chips, cards if dealing
return {
"state": lobby.state,
"players": [
{"name": p.name, "amount": p.amount, "cards": p.cards if lobby.state!="waiting" else []}
for p in lobby.players
],
"community": lobby.community,
"pot": lobby.pot
}
# (You would add endpoints for posting “bet”, “fold”, advancing rounds, showdown, etc.)
if __name__ == "__main__":
import uvicorn
uvicorn.run("app:app", host="0.0.0.0", port=7860, log_level="info") |