Update main.py

main.py

@@ -1,9 +1,9 @@
 from fastapi import FastAPI
-from fastapi.middleware.cors import CORSMiddleware
 from starlette.responses import JSONResponse
-import subprocess
+from openvpn_api import VPN
 
 app = FastAPI()
+vpn = None
 
 # Add CORS middleware
 app.add_middleware(
@@ -16,19 +16,25 @@ app.add_middleware(
 
 @app.post("/start")
 def start_vpn():
+    global vpn
     try:
-        subprocess.run(["openvpn", "--config", "/path/to/config.conf"], check=True)
-        return JSONResponse(content={"status": "started"}, status_code=200)
-    except:
-        return JSONResponse(content={"error": "Failed to start VPN"}, status_code=500)
+        vpn = VPN('localhost', 7505)  # Ensure your OpenVPN server's management interface is set up on this IP and port.
+        vpn.connect()
+        # You can add more controls or checks here
+        return JSONResponse(content={"status": "connected"}, status_code=200)
+    except Exception as e:
+        return JSONResponse(content={"error": f"Failed to connect to VPN: {str(e)}"}, status_code=500)
 
 @app.post("/stop")
 def stop_vpn():
+    global vpn
     try:
-        subprocess.run(["pkill", "openvpn"], check=True)
-        return JSONResponse(content={"status": "stopped"}, status_code=200)
-    except:
-        return JSONResponse(content={"error": "Failed to stop VPN"}, status_code=500)
+        if vpn:
+            vpn.disconnect()
+            vpn = None
+        return JSONResponse(content={"status": "disconnected"}, status_code=200)
+    except Exception as e:
+        return JSONResponse(content={"error": f"Failed to disconnect from VPN: {str(e)}"}, status_code=500)
 
 if __name__ == "__main__":
     import uvicorn