Update app.py

app.py

@@ -393,7 +393,7 @@ async def read_process_output(process, websocket):
         #     }
         #     await websocket.send(json.dumps(event))
 
-async def handle_user_input(websocket,key, process, connected):
+async def handle_user_input(websocket,key, process, connected,process_ids):
     while True:
         try:
             await asyncio.sleep(0.1)
@@ -437,7 +437,8 @@ async def handle_user_input(websocket,key, process, connected):
                                             stderr=asyncio.subprocess.PIPE,
                                             # text=True,
                                         )
-                                
+                                        id=process.pid
+                                        process_ids.append(id)
                                         async def send_message(message):
                                             # print('sending msg')
                                             # await websocket.send(f'data: {message}')
@@ -446,7 +447,7 @@ async def handle_user_input(websocket,key, process, connected):
                                         
                                 
                                         await asyncio.gather(
-                                          handle_user_input(websocket,key, process, connected),
+                                          handle_user_input(websocket,key, process, connected,process_ids),
                                           read_process_output(process, websocket)
                                         )
                                 
@@ -464,7 +465,8 @@ async def handle_user_input(websocket,key, process, connected):
                                                
                                             }
                                             await websocket.send(json.dumps(event))
-                                            
+
+                                        process_ids.remove(id)
                                 
                                     except Exception as e:
                                         await error(websocket, str(e))
@@ -480,15 +482,43 @@ async def handle_user_input(websocket,key, process, connected):
                         
                         await wirte_file(websocket, key,event["project_name"], event["path"], event["content"], connected)
                     elif event["command"]["type"]=="curl":
+                        method = event.get("method", "GET")
+                        url = event["url"]
+                        body = event.get("data", "")
+                        headers = event.get("headers", {})
+        
+                        if method.upper() == "GET":
+                            response = requests.get(url, headers=headers)
+                        elif method.upper() == "POST":
+                            response = requests.post(url, data=body, headers=headers)
+                        else:
+                            response = {"error": "Unsupported method"}
+                        # Parse HTML content
+                        soup = BeautifulSoup(response.text, "html.parser")
+                        # Find and modify CSS links
+                        for link in soup.find_all("link", rel="stylesheet", href=True):
+                            if link['href'].startswith('/'):
+                                css_url = url + link['href']
+                                css_response = requests.get(css_url)
+                                css_content = css_response.text
+                                style_tag = soup.new_tag("style")
+                                style_tag.string = css_content
+                                link.replace_with(style_tag)
                         
-                        response = requests.get(event['url'])
-                        event = {
-                        "type": "web-data",
-                        
-                        "data": response.text,
-                        
-                    }
-                        await websocket.send(json.dumps(event))
+                        # Find and modify JS links
+                        for script in soup.find_all("script", src=True):
+                            if script['src'].startswith('/'):
+                                js_url = url + script['src']
+                                js_response = requests.get(js_url)
+                                js_content = js_response.text
+                                script_tag = soup.new_tag("script")
+                                script_tag.string = js_content
+                                script.replace_with(script_tag)
+                        response_data = {
+                            "type": "web-data",
+                            "data": str(soup),
+                        }
+                        await websocket.send(json.dumps(response_data))
                     elif event["command"]["type"]=="create":
                         if event["item"]=="folder":
                             await create_folder(websocket, key,event["project_name"], event["path"],event["name"],event['root'],event['targetElementData'],event["rpath"], connected)
@@ -739,7 +769,7 @@ async def error(websocket, message):
     await websocket.send(json.dumps(event))
 
 
-async def exe(websocket,connected,key):
+async def exe(websocket,connected,key,process_ids):
     """
     Receive and process moves from a player.
 
@@ -778,6 +808,8 @@ async def exe(websocket,connected,key):
                                 stderr=asyncio.subprocess.PIPE,
                                 # text=True,
                             )
+                            id=process.pid
+                            process_ids.append(id)
                     
                             async def send_message(message):
                                 # print('sending msg')
@@ -787,7 +819,7 @@ async def exe(websocket,connected,key):
                             
                     
                             await asyncio.gather(
-                              handle_user_input(websocket,key, process, connected),
+                              handle_user_input(websocket,key, process, connected,process_ids),
                               read_process_output(process, websocket)
                             )
                     
@@ -805,6 +837,7 @@ async def exe(websocket,connected,key):
                                                
                                             }
                                             await websocket.send(json.dumps(event))
+                            process_ids.remove(id)
                     
                         except Exception as e:
                             await error(websocket, str(e))
@@ -1050,6 +1083,7 @@ async def start(websocket,events):
 
     user = User()
     connected = {websocket}
+    process_ids = []
 
     join_key = secrets.token_urlsafe(12)
     JOIN[join_key] = user, connected
@@ -1065,9 +1099,13 @@ async def start(websocket,events):
         }
         await websocket.send(json.dumps(event))
       
-        await exe(websocket,connected,join_key)
+        await exe(websocket,connected,join_key,process_ids)
     finally:
         del JOIN[join_key]
+        project_path = os.path.join(os.getcwd(), 'projects', JOIN[join_key])
+        shutil.rmtree(project_path)
+        for id in process_ids:
+            os.killpg(os.getpgid(id), signal.SIGTERM)