Update app/gm_crate.py

app/gm_crate.py

@@ -1,40 +1,54 @@
 import os
-import email.message
-import aiosmtplib
+import base64
+from email.message import EmailMessage
+from google.oauth2.credentials import Credentials
+from googleapiclient.discovery import build
+from google.auth.transport.requests import Request
 
 class GmailLogic:
     def __init__(self):
-        self.mail_user = os.getenv("GMAIL_USER")
-        self.mail_pass = os.getenv("GMAIL_PASS")
+        # Bạn cần các biến này trong Secrets của Hugging Face
+        self.client_id = os.getenv("GMAIL_CLIENT_ID")
+        self.client_secret = os.getenv("GMAIL_CLIENT_SECRET")
+        self.refresh_token = os.getenv("GMAIL_REFRESH_TOKEN")
+        self.user_email = os.getenv("GMAIL_USER")
 
-    async def send_order_email(self, to_email: str, subject: str, customer_name: str, order_id: str, status: str):
-        if not self.mail_user or not self.mail_pass:
-            return {"ok": False, "error": "ENV GMAIL_USER/PASS missing"}
+    def get_service(self):
+        creds = Credentials(
+            None,
+            refresh_token=self.refresh_token,
+            token_uri="https://oauth2.googleapis.com/token",
+            client_id=self.client_id,
+            client_secret=self.client_secret,
+        )
+        # Tự động làm mới token nếu hết hạn
+        creds.refresh(Request())
+        return build('gmail', 'v1', credentials=creds)
 
-        msg = email.message.EmailMessage()
-        msg["From"] = self.mail_user
-        msg["To"] = to_email
-        msg["Subject"] = subject
+    async def send_order_email(self, to_email: str, subject: str, customer_name: str, order_id: str, status: str):
+        try:
+            service = self.get_service()
+            
+            # Tạo email chuẩn
+            message = EmailMessage()
+            message.set_content(f"Chào {customer_name}, đơn hàng {order_id} của bạn: {status}")
+            
+            # Nội dung HTML
+            html_content = f"<h2>Thông báo đơn hàng</h2><p>Chào <b>{customer_name}</b>, đơn hàng <b>#{order_id}</b> của bạn đã <b>{status}</b>.</p>"
+            message.add_alternative(html_content, subtype='html')
 
-        content = f"""
-        <div style="font-family: sans-serif; padding: 20px;">
-            <h2>Thông báo đơn hàng</h2>
-            <p>Chào <b>{customer_name}</b>, đơn hàng <b>#{order_id}</b>: <b>{status}</b>.</p>
-        </div>
-        """
-        msg.add_alternative(content, subtype="html")
+            message['To'] = to_email
+            message['From'] = self.user_email
+            message['Subject'] = subject
 
-        try:
-            # Thay đổi Port sang 587 và dùng STARTTLS
-            await aiosmtplib.send(
-                msg,
-                hostname="smtp.gmail.com",
-                port=587,
-                start_tls=True, # Dùng STARTTLS thay vì TLS trực tiếp
-                username=self.mail_user,
-                password=self.mail_pass,
-                timeout=15 # Thêm timeout để không bị treo server
-            )
-            return {"ok": True, "message": f"Sent to {to_email}"}
+            # Mã hóa base64 đúng chuẩn Gmail API
+            raw_message = base64.urlsafe_b64encode(message.as_bytes()).decode()
+            
+            create_message = {'raw': raw_message}
+            
+            # Gửi qua HTTP POST (Cổng 443)
+            service.users().messages().send(userId="me", body=create_message).execute()
+            
+            return {"ok": True, "message": "Gửi qua Gmail API thành công!"}
         except Exception as e:
-            return {"ok": False, "error": f"Lỗi kết nối SMTP: {str(e)}"}
+            return {"ok": False, "error": f"Gmail API Error: {str(e)}"}