Update app.py

app.py

@@ -16,7 +16,24 @@ JOIN = {}
 
 WATCH = {}
 
+async def create_file_structure(websocket, data, base_path='.'):
+    if data['type'] == 'folder':
+        folder_path = os.path.join(base_path, data['name'])
+        os.makedirs(folder_path, exist_ok=True)
+        for child in data['children']:
+            await create_file_structure(child, base_path=folder_path)
+    elif data['type'] == 'file':
+        file_path = os.path.join(base_path, data['name'])
+        with open(file_path, 'w', encoding='utf-8') as file:
+            file.write(data['content'])
 
+    event = {
+            "type": "msg",
+            "message": "project created",
+        }
+    await websocket.send(json.dumps(event))
+
+            
 async def error(websocket, message):
     """
     Send an error message.
@@ -85,7 +102,7 @@ async def play(websocket, game, player, connected):
             websockets.broadcast(connected, json.dumps(event))
 
 
-async def start(websocket):
+async def start(websocket,events):
     """
     Handle a connection from the first player: start a new game.
 
@@ -110,8 +127,14 @@ async def start(websocket):
             "watch": watch_key,
         }
         await websocket.send(json.dumps(event))
+        
+        js = json.loads(events)
+        assert js["type"] == "init"
+        base_path = os.path.join(os.getcwd(), 'projects', js["project_name"])
+        data=js["file_structure"]
         # Receive and process moves from the first player.
-        await play(websocket, game, PLAYER1, connected)
+        # await play(websocket, game, PLAYER1, connected)
+        await create_file_structure(websocket,data, base_path=base_path)
     finally:
         del JOIN[join_key]
         del WATCH[watch_key]
@@ -181,7 +204,7 @@ async def handler(websocket):
         await watch(websocket, event["watch"])
     else:
         # First player starts a new game.
-        await start(websocket)
+        await start(websocket, event)
 
 
 async def main():