Update main.py

main.py

@@ -5,39 +5,45 @@ import requests
 from fastapi import FastAPI, Request, HTTPException
 from starlette.concurrency import run_in_threadpool
 from playwright.sync_api import sync_playwright
-from typing import Optional
 
-# ------------ إعداد المتغيرات عبر البيئة ------------
+# ------------ إعدادات ثابتة/بيئة ------------
 ODOO_BASE = os.getenv("ODOO_BASE", "https://odoo.binrushd.care")
 ODOO_API  = f"{ODOO_BASE}/api/crm.lead"
+AUTH_URL = f"{ODOO_BASE}/api/auth/token"
+SESSION_AUTH_URL = f"{ODOO_BASE}/web/session/authenticate"
 
 CF_ID       = os.getenv("CF_ACCESS_CLIENT_ID", "0491b36d7dcabce5b04f1a53f347bb4e.access")
 CF_SECRET   = os.getenv("CF_ACCESS_CLIENT_SECRET", "22152cb41b62393e159daaff7dce433006c3744c5850e6adc15fa3544bb5eb09")
-ACCESS_TOKEN= os.getenv("ODOO_ACCESS_TOKEN", "access_token_242cc8a47d7a5a1f1102228867c1ad4724919f09")   # من /api/auth/token
-SESSION_ID  = os.getenv("ODOO_SESSION_ID", "51a026396c3db05dc39fd4a0b7827e11924148eb")     # من /web/session/authenticate (اختياري)
+
+LOGIN     = os.getenv("ODOO_LOGIN", "binrushd.automation@gmail.com")
+PASSWORD  = os.getenv("ODOO_PASSWORD", "BR2025")
+DB_NAME   = os.getenv("ODOO_DB", "BR_EMR_16.0_202401")
 
 USER_AGENT = os.getenv("USER_AGENT",
                        "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 "
                        "(KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36")
 
-# ضبط وقت الانتظار عند طلب Odoo
 REQUEST_TIMEOUT = int(os.getenv("REQUEST_TIMEOUT", "15"))
 
 app = FastAPI(title="HF Proxy -> Odoo (with Cloudflare cookie bootstrap)")
 
-# ---- دالة مزودة: تشغّل Playwright متزامنًا وترجع Cookie header ----
+# ---------- Utilities ----------
+
+def _mask(s: str, keep: int = 6) -> str:
+    """أظهر أول/آخر أحرف قليلة للّوج فقط"""
+    if not s: return ""
+    if len(s) <= keep*2: return s
+    return f"{s[:keep]}…{s[-keep:]}"
+
 def _get_cloudflare_cookie_header_sync(base_url: str, wait_ms: int = 2000) -> str:
     """
-    يفتح الصفحة الرئيسية بالدومين، يترك الـ JS يتم تنفيذه، ويجمع الكوكيز في صيغة Cookie header.
-    يعاد string مثل: "cf_clearance=xxx; __cf_bm=yyy"
+    يفتح الصفحة الرئيسية بالدومين، ينفذ JS (Cloudflare challenge) ويجمع الكوكيز كـ Cookie header.
     """
     with sync_playwright() as p:
         browser = p.chromium.launch(headless=True)
         context = browser.new_context(user_agent=USER_AGENT, locale="en-US")
         page = context.new_page()
-        # افتح الصفحة الرئيسية (ستنفّذ أي JS challenge مثل Cloudflare)
         page.goto(base_url, timeout=30000)
-        # انتظر ms بسيطة لتكميل الـ JS (يمكن رفعها لو لازم)
         page.wait_for_timeout(wait_ms)
         cookies = context.cookies()
         browser.close()
@@ -45,81 +51,134 @@ def _get_cloudflare_cookie_header_sync(base_url: str, wait_ms: int = 2000) -> st
         return ""
     return "; ".join(f"{c['name']}={c['value']}" for c in cookies)
 
-# ---- endpoint رئيسي: يستقبل أي JSON payload ثم يرسله لـ Odoo بعد الحصول على الكوكيز ----
+def _get_token_and_session() -> tuple[str, str]:
+    """
+    يستخرج access_token + session_id في كل مرة:
+      1) GET /api/auth/token مع CF-Access + (login/password/db) كهيدرز
+      2) لو session_id غير موجود: POST /web/session/authenticate (JSON-RPC)
+    """
+    s = requests.Session()
+    s.headers.update({
+        "Accept": "*/*",
+        "Content-Type": "application/json",
+        "Accept-Encoding": "gzip, deflate",  # بدون br
+        "Connection": "keep-alive",
+        "CF-Access-Client-Id": CF_ID,
+        "CF-Access-Client-Secret": CF_SECRET,
+    })
+
+    # (1) token
+    auth_headers = {
+        "CF-Access-Client-Id": CF_ID,
+        "CF-Access-Client-Secret": CF_SECRET,
+        "login": LOGIN,
+        "password": PASSWORD,
+        "db": DB_NAME,
+    }
+    r = s.get(AUTH_URL, headers=auth_headers, timeout=(5, 30))
+    r.raise_for_status()
+
+    # Cloudflare غالبًا يرجّع JSON؛ requests يفك gzip تلقائيًا
+    data = r.json()
+    token = data.get("access_token") or (data.get("data") or {}).get("access_token")
+    if not token:
+        raise RuntimeError(f"No access_token in response: {data}")
+
+    # (2) session_id
+    session_id = s.cookies.get("session_id") or r.cookies.get("session_id")
+    if not session_id:
+        payload = {
+            "jsonrpc": "2.0",
+            "params": {"db": DB_NAME, "login": LOGIN, "password": PASSWORD}
+        }
+        r2 = s.post(SESSION_AUTH_URL, json=payload, timeout=(5, 30))
+        r2.raise_for_status()
+        session_id = s.cookies.get("session_id") or r2.cookies.get("session_id")
+        if not session_id:
+            raise RuntimeError("Could not obtain session_id cookie.")
+
+    # اطبع للّوج (مخفّض)
+    print(f"[AUTH] token={_mask(token)}  session_id={_mask(session_id)}")
+    return token, session_id
+
+# ---------- Endpoints ----------
+
+@app.get("/health")
+def health():
+    return {"ok": True, "note": "ready to bootstrap cookies and forward requests"}
+
+@app.get("/ping")
+def ping():
+    return {"pong": True}
+
 @app.post("/forward")
 async def forward_lead(request: Request):
     """
-    استقبل أي JSON body من العميل، شغّل Playwright للحصول على كوكيز Cloudflare،
-    ثم أرسل نفس الـ body الى Odoo مع الهيدرز المطلوبة.
+    - يستقبل JSON payload
+    - يستخرج access_token + session_id في نفس اللحظة ويطبعهما في اللوج
+    - يأخذ كوكيز Cloudflare بـ Playwright
+    - يرسل الطلب إلى Odoo مع كل الهيدرز/الكوكيز المطلوبة
     """
+    # 0) payload
     try:
         payload = await request.json()
     except Exception:
-        # لو الجسم مش JSON نقرأه كسطر خام
         body_text = await request.body()
         try:
             payload = json.loads(body_text.decode("utf-8") or "{}")
         except Exception:
             payload = {}
+    # Debug (اختياري)
+    print(f"[FORWARD] incoming payload keys={list(payload.keys())}")
+
+    # 1) token + session_id
+    try:
+        access_token, session_id = await run_in_threadpool(_get_token_and_session)
+    except Exception as e:
+        raise HTTPException(status_code=502, detail=f"Auth failure: {str(e)}")
 
-    # 1) اجلب كوكيز Cloudflare (تشغيل Playwright في threadpool حتى لا يعرقل event loop)
+    # 2) Cloudflare cookies via Playwright
     try:
         cookie_header = await run_in_threadpool(_get_cloudflare_cookie_header_sync, ODOO_BASE, 2000)
     except Exception as e:
-        # لو Playwright فشل، نستمر بدون كوكيز (ربما CF لا يطلب JS). أرجع خطأ لكن لا تقطع التنفيذ نهائيًا.
         cookie_header = ""
-        app.logger = getattr(app, "logger", None)
-        # لا نرمي استثناء لأن أحيان كثيرة ممكن ينجح بدون JS
-        # ولكن نُسجل السبب لو حابب تفحص الـ logs
-        print("Playwright error (will continue without cookies):", str(e))
+        print("Playwright error (continue without cookies):", str(e))
 
-    # 2) جهّز الهيدرز (نفس التي كنت تستخدمها محليًا)
+    # 3) headers
     headers = {
         "Accept": "application/json",
         "Content-Type": "application/json",
         "User-Agent": USER_AGENT,
         "Accept-Encoding": "gzip, deflate",
         "Connection": "keep-alive",
-        # Cloudflare Access service token headers (لو مستخدمة)
         "CF-Access-Client-Id": CF_ID,
         "CF-Access-Client-Secret": CF_SECRET,
-        # Odoo access token لو متوفر
+        "access-token": access_token,
     }
-    if ACCESS_TOKEN:
-        headers["access-token"] = ACCESS_TOKEN
-
-    # أضف كوكيز الصفحة التي حصلنا عليها (إن وُجدت)
     cookie_list = []
     if cookie_header:
         cookie_list.append(cookie_header)
-    if SESSION_ID:
-        cookie_list.append(f"session_id={SESSION_ID}")
+    if session_id:
+        cookie_list.append(f"session_id={session_id}")
     if cookie_list:
         headers["Cookie"] = "; ".join(cookie_list)
 
-    # 3) نفّذ طلب POST إلى Odoo
+    # 4) POST -> Odoo
     try:
-        resp = requests.post(ODOO_API, headers=headers, json=payload, timeout=REQUEST_TIMEOUT, allow_redirects=False)
+        resp = requests.post(ODOO_API, headers=headers, json=payload,
+                             timeout=REQUEST_TIMEOUT, allow_redirects=False)
     except requests.RequestException as e:
-        # لو فشل الاتصال خارجيًا، نعيد 502 مع رسالة الخطأ
         raise HTTPException(status_code=502, detail=f"Error connecting to Odoo: {str(e)}")
 
-    # 4) حاول نرجع الرد كما هو (JSON أو نص)
     status = resp.status_code
     ctype = resp.headers.get("Content-Type", "")
-    text = resp.text
+    # اطبع ملخص
+    print(f"[FORWARD] Odoo status={status} content-type={ctype}")
 
-    # إذا كان JSON نعيده مفككًا، وإلا نعيد النص الخام
     if "application/json" in ctype:
         try:
             return resp.json()
         except Exception:
-            return {"status": status, "body": text}
+            return {"status": status, "body": resp.text}
     else:
-        return {"status": status, "body": text}
-
-@app.get("/health")
-def health():
-    return {"ok": True, "note": "ready to bootstrap cookies and forward requests"}
-
-# ---------- نهاية الملف ----------
+        return {"status": status, "body": resp.text}