mirror of
https://github.com/mskatoni/ni-mail.git
synced 2026-08-29 12:07:24 +08:00
106 lines
4.0 KiB
Python
106 lines
4.0 KiB
Python
from flask import Blueprint, jsonify, request
|
|
from app.models import Account
|
|
from app.services import graph, imap, smtp, verification
|
|
import os
|
|
|
|
bp = Blueprint("emails", __name__, url_prefix="/api/emails")
|
|
|
|
@bp.get("/")
|
|
def get_emails():
|
|
email = request.args.get("email")
|
|
if not email:
|
|
return jsonify({"error": "email query param required"}), 400
|
|
|
|
acct = Account.query.filter_by(email=email).first()
|
|
if not acct:
|
|
return jsonify({"error": "account not found"}), 404
|
|
|
|
proxy_url = os.getenv("PROXY_URL")
|
|
messages = []
|
|
|
|
try:
|
|
if acct.method == "graph":
|
|
token_res = graph.get_access_token_graph_result(
|
|
client_id=acct.client_id,
|
|
refresh_token=acct.refresh_token,
|
|
proxy_url=proxy_url
|
|
)
|
|
if isinstance(token_res, dict) and "access_token" in token_res:
|
|
token = token_res["access_token"]
|
|
messages = graph.fetch_messages(token, top=15, proxy_url=proxy_url)
|
|
elif isinstance(token_res, dict) and "error" in token_res:
|
|
return jsonify({"error": token_res["error"]}), 400
|
|
else:
|
|
messages = imap.fetch_inbox_messages(
|
|
email=acct.email,
|
|
password=acct.password or acct.refresh_token,
|
|
limit=15,
|
|
proxy_url=proxy_url
|
|
)
|
|
except Exception as e:
|
|
return jsonify({"error": f"Failed to fetch emails: {str(e)}"}), 500
|
|
|
|
results = []
|
|
for msg in messages:
|
|
body = msg.get("body", {}).get("content", "") or msg.get("body_text", "") or ""
|
|
code, link = verification.extract_code_and_link(body)
|
|
results.append({
|
|
"id": msg.get("id"),
|
|
"subject": msg.get("subject"),
|
|
"from": msg.get("from", {}).get("emailAddress", {}).get("address", "") if isinstance(msg.get("from"), dict) else str(msg.get("from", "")),
|
|
"received_at": msg.get("receivedDateTime"),
|
|
"otp_code": code,
|
|
"link": link,
|
|
"body_preview": body[:300]
|
|
})
|
|
|
|
return jsonify(results)
|
|
|
|
@bp.post("/send")
|
|
def send_email():
|
|
data = request.json or {}
|
|
email = data.get("email")
|
|
to_email = data.get("to_email")
|
|
subject = data.get("subject", "No Subject")
|
|
body = data.get("body", "")
|
|
|
|
if not email or not to_email:
|
|
return jsonify({"error": "email and to_email required"}), 400
|
|
|
|
acct = Account.query.filter_by(email=email).first()
|
|
if not acct:
|
|
return jsonify({"error": "account not found"}), 404
|
|
|
|
proxy_url = os.getenv("PROXY_URL")
|
|
|
|
try:
|
|
if acct.method == "graph":
|
|
token_res = graph.get_access_token_graph_result(
|
|
client_id=acct.client_id,
|
|
refresh_token=acct.refresh_token,
|
|
proxy_url=proxy_url
|
|
)
|
|
if isinstance(token_res, dict) and "access_token" in token_res:
|
|
token = token_res["access_token"]
|
|
success = graph.send_mail_graph(token, to_email, subject, body, proxy_url=proxy_url)
|
|
if success:
|
|
return jsonify({"success": True, "message": "Email sent via Graph API"})
|
|
return jsonify({"error": "Graph API send mail failed"}), 500
|
|
return jsonify({"error": "Token refresh failed"}), 400
|
|
else:
|
|
success = smtp.send_mail_smtp(
|
|
host=os.getenv("SMTP_HOST", "smtp.office365.com"),
|
|
port=int(os.getenv("SMTP_PORT", "587")),
|
|
username=acct.email,
|
|
password=acct.password or acct.refresh_token,
|
|
to_email=to_email,
|
|
subject=subject,
|
|
body_text=body,
|
|
use_tls=True
|
|
)
|
|
if success:
|
|
return jsonify({"success": True, "message": "Email sent via SMTP"})
|
|
return jsonify({"error": "SMTP send mail failed"}), 500
|
|
except Exception as e:
|
|
return jsonify({"error": f"Send mail failed: {str(e)}"}), 500
|