Update main.py

main.py

@@ -1,15 +1,283 @@
-from fastapi import FastAPI
+import json
+from fastapi import FastAPI, Request
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import JSONResponse
+
+from greenapi import (
+    extract_text_message,
+    get_device_key_by_instance_id,
+    send_text_message,
+    send_url_button,
+)
+
+from engine.conversation_engine import process_message
+from messages import log_message
+
+from handoff.sales import create_sales_handoff
+from handoff.support import create_support_handoff
 
 app = FastAPI()
 
-@app.get("/")
-async def root():
-    return {"version": "ONLY_TEST_2026_03_16_A"}
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=False,
+    allow_methods=["POST", "OPTIONS"],
+    allow_headers=["*"],
+)
+
 
 @app.get("/health")
 async def health():
-    return {"ok": True, "version": "ONLY_TEST_2026_03_16_A"}
+    print(">>> /health called")
+    return {"ok": True, "service": "up"}
+
+
+@app.get("/")
+async def root():
+    return {
+        "ok": True,
+        "service": "ADK Bot API",
+        "message": "Server is running"
+    }
+
 
 @app.get("/receive-test")
 async def receive_test():
-    return {"ok": True, "message": "receive endpoint reachable", "version": "ONLY_TEST_2026_03_16_A"}
+    print(">>> /receive-test called")
+    return {"ok": True, "message": "receive endpoint reachable"}
+
+
+@app.post("/receive")
+async def webhook_receiver(req: Request):
+    print("========== /receive HIT ==========")
+
+    try:
+        raw_body = await req.body()
+        print(">>> RAW BODY BYTES:")
+        print(raw_body)
+    except Exception as e:
+        print(">>> FAILED TO READ RAW BODY:", str(e))
+        raw_body = b""
+
+    try:
+        headers = dict(req.headers)
+        print(">>> HEADERS:")
+        print(json.dumps(headers, ensure_ascii=False, indent=2))
+    except Exception as e:
+        print(">>> FAILED TO READ HEADERS:", str(e))
+
+    body = None
+
+    try:
+        if raw_body:
+            body = json.loads(raw_body.decode("utf-8"))
+        else:
+            body = {}
+    except Exception as e:
+        print(">>> FAILED TO PARSE JSON:", str(e))
+        return JSONResponse(
+            status_code=200,
+            content={
+                "ok": True,
+                "debug": "request reached /receive but body is not valid JSON"
+            }
+        )
+
+    print("===== RECEIVED JSON =====")
+    print(json.dumps(body, ensure_ascii=False, indent=2))
+    print("=========================")
+
+    # مؤقتًا نطبع typeWebhook حتى لو مختلفة
+    webhook_type = body.get("typeWebhook")
+    print(">>> typeWebhook:", webhook_type)
+
+    if webhook_type != "incomingMessageReceived":
+        print(">>> ignored non-incoming webhook")
+        return {
+            "ok": True,
+            "ignored": True,
+            "typeWebhook": webhook_type
+        }
+
+    instance_data = body.get("instanceData") or {}
+    sender_data = body.get("senderData") or {}
+    msg_data = body.get("messageData") or {}
+
+    # مبدئيًا هنطبع كل الاحتمالات
+    print(">>> instanceData:", instance_data)
+    print(">>> senderData:", sender_data)
+    print(">>> messageData:", msg_data)
+
+    # مهم: بعض payloads فيها idInstance وبعضها wid فقط
+    instance_id = str(
+        instance_data.get("idInstance")
+        or instance_data.get("wid")
+        or ""
+    )
+
+    chat_id = sender_data.get("chatId")
+    text_message = extract_text_message(msg_data)
+
+    print(">>> resolved instance_id:", instance_id)
+    print(">>> resolved chat_id:", chat_id)
+    print(">>> resolved text_message:", text_message)
+
+    if not instance_id or not chat_id or not text_message:
+        print(">>> skipped because missing required fields")
+        return {
+            "ok": True,
+            "skipped": True,
+            "resolved_instance_id": instance_id,
+            "resolved_chat_id": chat_id,
+            "resolved_text_message": text_message,
+        }
+
+    device_key = get_device_key_by_instance_id(instance_id)
+    print(">>> resolved device_key:", device_key)
+
+    if not device_key:
+        print(">>> Unknown instance_id:", instance_id)
+        return {
+            "ok": True,
+            "skipped": True,
+            "reason": "unknown_instance_id",
+            "instance_id": instance_id
+        }
+
+    customer_phone = chat_id.replace("@c.us", "")
+    bot_number = device_key
+
+    try:
+        log_message(customer_phone, bot_number, "in", text_message)
+    except Exception as e:
+        print(">>> failed to log incoming message:", str(e))
+
+    try:
+        result = process_message(
+            bot_number=bot_number,
+            customer_phone=customer_phone,
+            text=text_message
+        )
+        print(">>> engine result:")
+        print(json.dumps(result, ensure_ascii=False, indent=2))
+    except Exception as e:
+        print(">>> engine error:", str(e))
+
+        fallback_reply = "تمام، حصلت مشكلة بسيطة. ممكن تعيد إرسال رسالتك مرة تانية؟"
+        try:
+            status, txt = send_text_message(device_key, chat_id, fallback_reply)
+            print(">>> fallback send status:", status)
+            print(">>> fallback send body:", txt)
+            log_message(customer_phone, bot_number, "out", fallback_reply)
+        except Exception as send_err:
+            print(">>> failed to send fallback:", str(send_err))
+
+        return JSONResponse(
+            status_code=200,
+            content={"ok": True, "error": str(e), "fallback_sent": True}
+        )
+
+    reply = (result.get("reply") or "").strip()
+    action = result.get("action")
+
+    if action and action.get("type") == "url_button":
+        try:
+            status, txt = send_url_button(
+                device_key=device_key,
+                chat_id=chat_id,
+                header=action.get("header", "مرحباً بك"),
+                body=action.get("body", reply),
+                footer=action.get("footer", ""),
+                url=action.get("url"),
+                button_text=action.get("button_text", "فتح الرابط")
+            )
+            print(">>> sent url button status:", status)
+            print(">>> sent url button body:", txt)
+
+            try:
+                log_message(customer_phone, bot_number, "out", action.get("body", reply))
+            except Exception as e:
+                print(">>> failed to log outgoing url button body:", str(e))
+
+        except Exception as e:
+            print(">>> failed to send url button:", str(e))
+            fallback_reply = action.get("body", reply) or "تمام، افتح الرابط من فضلك."
+            try:
+                status, txt = send_text_message(device_key, chat_id, fallback_reply)
+                print(">>> url fallback send status:", status)
+                print(">>> url fallback send body:", txt)
+                log_message(customer_phone, bot_number, "out", fallback_reply)
+            except Exception as send_err:
+                print(">>> failed to send url fallback:", str(send_err))
+
+    else:
+        try:
+            status, txt = send_text_message(device_key, chat_id, reply)
+            print(">>> sent text status:", status)
+            print(">>> sent text body:", txt)
+
+            try:
+                log_message(customer_phone, bot_number, "out", reply)
+            except Exception as e:
+                print(">>> failed to log outgoing text:", str(e))
+
+        except Exception as e:
+            print(">>> failed to send text:", str(e))
+
+    if action and action.get("type") == "handoff":
+        department = action.get("department")
+        summary = action.get("summary", "")
+        metadata = result.get("flow_data", {})
+
+        try:
+            if department == "sales":
+                create_sales_handoff(
+                    customer_phone=customer_phone,
+                    bot_number=bot_number,
+                    summary=summary,
+                    metadata=metadata
+                )
+            elif department == "support":
+                create_support_handoff(
+                    customer_phone=customer_phone,
+                    bot_number=bot_number,
+                    summary=summary,
+                    metadata=metadata
+                )
+            print(">>> handoff created:", department)
+        except Exception as e:
+            print(">>> failed to create handoff:", str(e))
+
+    return {
+        "ok": True,
+        "device_key": device_key,
+        "customer_phone": customer_phone,
+        "state": result.get("next_state"),
+        "has_action": bool(action),
+    }
+
+
+@app.post("/complaints/callback")
+async def complaints_callback(req: Request):
+    print("========== /complaints/callback HIT ==========")
+
+    try:
+        payload = await req.json()
+    except Exception:
+        return JSONResponse(status_code=400, content={"ok": False, "error": "invalid json"})
+
+    print("===== COMPLAINT CALLBACK =====")
+    print(json.dumps(payload, ensure_ascii=False, indent=2))
+    print("==============================")
+
+    uid = payload.get("uid", "")
+    complaint_number = payload.get("complaint_number", "")
+    status = payload.get("status", "")
+
+    return {
+        "ok": True,
+        "uid": uid,
+        "complaint_number": complaint_number,
+        "status": status
+    }