Enhance error handling in bypass_shinigami function and update API response structure

app/main.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, Query
+from fastapi import FastAPI, Query, HTTPException
 from app.utils import bypass_shinigami
 
 app = FastAPI()
@@ -9,5 +9,9 @@ async def root():
 
 @app.get("/bypass")
 async def bypass(url: str = Query(..., description="URL Shinigami yang mau dibypass")):
+    if not url.startswith("http"):
+        raise HTTPException(status_code=400, detail="URL tidak valid")
     result = bypass_shinigami(url)
-    return {"original": url, "bypassed": result}
+    if result["status"] == "error":
+        raise HTTPException(status_code=502, detail=result["message"])
+    return result

app/utils.py

@@ -3,12 +3,23 @@ from selenium import webdriver
 from selenium.webdriver.chrome.options import Options
 from selenium.webdriver.common.by import By
 import time
+import requests
 
 # Force all cache to go to /tmp/.cache
 os.environ["XDG_CACHE_HOME"] = "/tmp/.cache"
 os.environ["XDG_CONFIG_HOME"] = "/tmp/.config"
 
-def bypass_shinigami(url: str) -> str:
+def bypass_shinigami(url: str) -> dict:
+    # Cek status code dulu
+    try:
+        resp = requests.get(url, timeout=10)
+        status_code = resp.status_code
+    except Exception as e:
+        return {"status": "error", "message": f"Gagal akses url: {e}", "code": None, "bypassed_url": None}
+
+    if status_code != 200:
+        return {"status": "error", "message": f"Status code: {status_code}", "code": status_code, "bypassed_url": None}
+
     chrome_options = Options()
     chrome_options.add_argument("--headless")
     chrome_options.add_argument("--disable-gpu")
@@ -23,12 +34,21 @@ def bypass_shinigami(url: str) -> str:
 
     try:
         driver.get(url)
-        time.sleep(5)  # atau pakai wait_for_element kalau mau halus
-
-        # Ambil link pertama dari tombol (misal tombol lanjut/link keluar)
+        time.sleep(5)
+        # Ambil link pertama yang ditemukan
         link = driver.find_element(By.TAG_NAME, "a").get_attribute("href")
-        return link or "Gagal ambil href"
+        return {
+            "status": "success",
+            "message": "OK",
+            "code": status_code,
+            "bypassed_url": link or None
+        }
     except Exception as e:
-        return f"Error: {e}"
+        return {
+            "status": "error",
+            "message": f"Error: {e}",
+            "code": status_code,
+            "bypassed_url": None
+        }
     finally:
         driver.quit()