Update main.py

main.py

@@ -10,6 +10,7 @@ import json
 import base64
 import hashlib
 import hmac
+import time
 
 app = FastAPI(title="Cié Cié Backend API")
 
@@ -38,6 +39,7 @@ RETURN_URL = "https://ciecietaipei.github.io/booking.html"
 
 supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY) if SUPABASE_URL else None
 
+# --- 資料結構定義 ---
 class OrderPayload(BaseModel):
     service_type: str
     name: str
@@ -55,6 +57,12 @@ class ConfirmPayload(BaseModel):
     order_id: str
     amount: int
 
+# 🌟 新增：用來接收補付款請求的資料結構
+class RepayPayload(BaseModel):
+    order_id: str
+
+# --- API 端點定義 ---
+
 @app.get("/")
 def read_root():
     return {"status": "online", "message": "Cié Cié FastAPI is running."}
@@ -86,7 +94,6 @@ async def submit_booking(payload: OrderPayload):
                 "products": [{"name": "餐飲訂金與預付金", "quantity": 1, "price": final_deposit}]
             }],
             "redirectUrls": {
-                # 這裡最關鍵：帶上 action=payment_confirm 讓前端知道要處理結帳確認
                 "confirmUrl": f"{RETURN_URL}?action=payment_confirm&amount={final_deposit}", 
                 "cancelUrl": f"{RETURN_URL}?action=payment_cancel"
             }
@@ -139,7 +146,7 @@ async def submit_booking(payload: OrderPayload):
         return { "status": "success", "message": "訂位已成功建立！" }
     except Exception as e: raise HTTPException(status_code=500, detail=f"寫入資料庫失敗: {str(e)}")
 
-# 🌟 新增：確認收錢的端點 (Confirm API) 🌟
+# 確認收錢的端點 (Confirm API)
 @app.post("/api/linepay/confirm")
 async def confirm_payment(payload: ConfirmPayload):
     uri = f"/v3/payments/{payload.transaction_id}/confirm"
@@ -172,6 +179,66 @@ async def confirm_payment(payload: ConfirmPayload):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
+# 🌟 新增：處理重新產生付款連結的 API 🌟
+@app.post("/api/linepay/repay")
+async def repay_payment(payload: RepayPayload):
+    if not supabase: raise HTTPException(status_code=500, detail="資料庫未連線")
+    
+    try:
+        # 1. 找回原本的訂單
+        res = supabase.table("bookings").select("*").ilike("remarks", f"%{payload.order_id}%").execute()
+        if not res.data: raise HTTPException(status_code=404, detail="找不到該筆訂單")
+        
+        booking = res.data[0]
+        # 防呆：如果客人其實已經付過錢了，阻止他重複刷卡
+        if "已付" in booking.get("status", "") or "確認" in booking.get("status", ""):
+            raise HTTPException(status_code=400, detail="此訂單已完成付款或確認，無需重新結帳")
+
+        # 2. 獲取要收取的金額 (PoC 階段先以防護訂金的 1000 元作為補收預設值)
+        # 備註：在最終版的 orders 表格中，我們會直接從資料庫精準讀取 deposit_amount
+        amount = 1000 
+        
+        # 3. 重新向 LINE Pay 申請新的交易連結
+        # 技巧：在原本的 order_id 後面加上時間戳記，繞過 LINE Pay 「同訂單號不可重複請求」的限制
+        new_order_id = f"{payload.order_id}-R{int(time.time())}"
+        
+        request_body = {
+            "amount": amount,
+            "currency": "TWD",
+            "orderId": new_order_id, 
+            "packages": [{
+                "id": "pkg_repay", "amount": amount, "name": "Cié Cié Taipei 補繳訂金",
+                "products": [{"name": "餐飲訂金", "quantity": 1, "price": amount}]
+            }],
+            "redirectUrls": {
+                # 付款成功後，一樣帶回前端進行確認
+                "confirmUrl": f"{RETURN_URL}?action=payment_confirm&amount={amount}&orderId={payload.order_id}", 
+                "cancelUrl": f"{RETURN_URL}?action=payment_cancel"
+            }
+        }
+        
+        uri = "/v3/payments/request"
+        nonce = str(uuid.uuid4())
+        body_str = json.dumps(request_body)
+        message = LINE_PAY_CHANNEL_SECRET + uri + body_str + nonce
+        signature = base64.b64encode(hmac.new(LINE_PAY_CHANNEL_SECRET.encode(), message.encode(), hashlib.sha256).digest()).decode()
+
+        headers = {
+            "Content-Type": "application/json", "X-LINE-ChannelId": LINE_PAY_CHANNEL_ID,
+            "X-LINE-Authorization-Nonce": nonce, "X-LINE-Authorization": signature
+        }
+        
+        line_pay_res = requests.post(f"{LINE_PAY_BASE_URL}{uri}", headers=headers, data=body_str)
+        res_data = line_pay_res.json()
+        
+        if res_data.get("returnCode") == "0000":
+            return {"payment_url": res_data["info"]["paymentUrl"]["web"]}
+        else:
+            raise HTTPException(status_code=500, detail=f"LINE Pay 錯誤: {res_data.get('returnMessage')}")
+            
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
 def notify_boss(name, tel, date, time, pax, amount):
     if not LINE_ACCESS_TOKEN or not BOSS_LINE_ID: return
     msg = f"🔔 【新訂單通知】\n姓��：{name}\n電話：{tel}\n時間：{date} {time}\n人數：{pax}位"