2022-04-14 16:46:11 +00:00
|
|
|
import uuid
|
|
|
|
from abc import ABC, abstractmethod
|
|
|
|
from io import BytesIO
|
|
|
|
from mailbox import Message
|
|
|
|
from typing import Optional
|
|
|
|
|
|
|
|
from app import s3
|
|
|
|
from app.config import (
|
|
|
|
ALERT_COMPLAINT_REPLY_PHASE,
|
|
|
|
ALERT_COMPLAINT_TO_USER,
|
|
|
|
ALERT_COMPLAINT_FORWARD_PHASE,
|
|
|
|
)
|
|
|
|
from app.email import headers
|
|
|
|
from app.email_utils import (
|
|
|
|
get_header_unicode,
|
|
|
|
parse_full_address,
|
|
|
|
save_email_for_debugging,
|
|
|
|
to_bytes,
|
|
|
|
render,
|
|
|
|
send_email_with_rate_control,
|
|
|
|
)
|
|
|
|
from app.log import LOG
|
|
|
|
from app.models import (
|
|
|
|
User,
|
|
|
|
Alias,
|
|
|
|
DeletedAlias,
|
|
|
|
DomainDeletedAlias,
|
|
|
|
Contact,
|
2022-04-25 12:40:42 +00:00
|
|
|
ProviderComplaint,
|
2022-04-14 16:46:11 +00:00
|
|
|
Phase,
|
2022-04-25 12:40:42 +00:00
|
|
|
ProviderComplaintState,
|
2022-04-14 16:46:11 +00:00
|
|
|
RefusedEmail,
|
|
|
|
)
|
|
|
|
|
|
|
|
|
2022-04-25 12:40:42 +00:00
|
|
|
class ProviderComplaintOrigin(ABC):
|
2022-04-14 16:46:11 +00:00
|
|
|
@classmethod
|
|
|
|
@abstractmethod
|
|
|
|
def get_original_message(cls, message: Message) -> Optional[Message]:
|
|
|
|
pass
|
|
|
|
|
|
|
|
@classmethod
|
|
|
|
@abstractmethod
|
|
|
|
def name(cls):
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
2022-04-28 13:23:52 +00:00
|
|
|
class ProviderComplaintYahoo(ProviderComplaintOrigin):
|
2022-04-14 16:46:11 +00:00
|
|
|
@classmethod
|
|
|
|
def get_original_message(cls, message: Message) -> Optional[Message]:
|
2022-04-22 12:49:03 +00:00
|
|
|
# 1st part is the container
|
|
|
|
# 2nd has empty body
|
|
|
|
# 6th is the original message
|
|
|
|
current_part = 0
|
2022-04-14 16:46:11 +00:00
|
|
|
for part in message.walk():
|
2022-04-22 12:49:03 +00:00
|
|
|
current_part += 1
|
|
|
|
if current_part == 6:
|
2022-04-14 16:46:11 +00:00
|
|
|
return part
|
|
|
|
return None
|
|
|
|
|
|
|
|
@classmethod
|
|
|
|
def name(cls):
|
|
|
|
return "yahoo"
|
|
|
|
|
|
|
|
|
2022-04-28 13:23:52 +00:00
|
|
|
class ProviderComplaintHotmail(ProviderComplaintOrigin):
|
2022-04-14 16:46:11 +00:00
|
|
|
@classmethod
|
|
|
|
def get_original_message(cls, message: Message) -> Optional[Message]:
|
2022-04-22 12:49:03 +00:00
|
|
|
# 1st part is the container
|
|
|
|
# 2nd has empty body
|
|
|
|
# 3rd is the original message
|
|
|
|
current_part = 0
|
2022-04-14 16:46:11 +00:00
|
|
|
for part in message.walk():
|
2022-04-22 12:49:03 +00:00
|
|
|
current_part += 1
|
|
|
|
if current_part == 3:
|
2022-04-14 16:46:11 +00:00
|
|
|
return part
|
|
|
|
return None
|
|
|
|
|
|
|
|
@classmethod
|
|
|
|
def name(cls):
|
|
|
|
return "hotmail"
|
|
|
|
|
|
|
|
|
|
|
|
def handle_hotmail_complaint(message: Message) -> bool:
|
2022-04-28 13:23:52 +00:00
|
|
|
return handle_complaint(message, ProviderComplaintHotmail())
|
2022-04-14 16:46:11 +00:00
|
|
|
|
|
|
|
|
|
|
|
def handle_yahoo_complaint(message: Message) -> bool:
|
2022-04-28 13:23:52 +00:00
|
|
|
return handle_complaint(message, ProviderComplaintYahoo())
|
2022-04-14 16:46:11 +00:00
|
|
|
|
|
|
|
|
|
|
|
def find_alias_with_address(address: str) -> Optional[Alias]:
|
|
|
|
return (
|
|
|
|
Alias.get_by(email=address)
|
|
|
|
or DeletedAlias.get_by(email=address)
|
|
|
|
or DomainDeletedAlias.get_by(email=address)
|
|
|
|
)
|
|
|
|
|
|
|
|
|
2022-04-25 12:40:42 +00:00
|
|
|
def handle_complaint(message: Message, origin: ProviderComplaintOrigin) -> bool:
|
2022-04-14 16:46:11 +00:00
|
|
|
original_message = origin.get_original_message(message)
|
|
|
|
|
|
|
|
try:
|
|
|
|
_, to_address = parse_full_address(
|
|
|
|
get_header_unicode(original_message[headers.TO])
|
|
|
|
)
|
|
|
|
_, from_address = parse_full_address(
|
|
|
|
get_header_unicode(original_message[headers.FROM])
|
|
|
|
)
|
|
|
|
except ValueError:
|
|
|
|
saved_file = save_email_for_debugging(message, "FromParseFailed")
|
2022-04-22 12:49:03 +00:00
|
|
|
LOG.w(f"Cannot parse from header. Saved to {saved_file or 'nowhere'}")
|
|
|
|
return False
|
2022-04-14 16:46:11 +00:00
|
|
|
|
|
|
|
user = User.get_by(email=to_address)
|
|
|
|
if user:
|
2022-04-25 12:40:42 +00:00
|
|
|
LOG.d(f"Handle provider {origin.name()} complaint for {user}")
|
2022-04-28 13:24:45 +00:00
|
|
|
report_complaint_to_user_in_transactional_phase(user, origin)
|
2022-04-14 16:46:11 +00:00
|
|
|
return True
|
|
|
|
|
|
|
|
alias = find_alias_with_address(from_address)
|
|
|
|
# the email is during a reply phase, from=alias and to=destination
|
|
|
|
if alias:
|
|
|
|
LOG.i(
|
2022-04-22 12:49:03 +00:00
|
|
|
f"Complaint from {origin.name} during reply phase {alias} -> {to_address}, {user}"
|
2022-04-14 16:46:11 +00:00
|
|
|
)
|
|
|
|
report_complaint_to_user_in_reply_phase(alias, to_address, origin)
|
2022-04-25 12:40:42 +00:00
|
|
|
store_provider_complaint(alias, message)
|
2022-04-14 16:46:11 +00:00
|
|
|
return True
|
|
|
|
|
|
|
|
contact = Contact.get_by(reply_email=from_address)
|
|
|
|
if contact:
|
|
|
|
alias = contact.alias
|
|
|
|
else:
|
|
|
|
alias = find_alias_with_address(to_address)
|
|
|
|
|
|
|
|
if not alias:
|
|
|
|
LOG.e(
|
|
|
|
f"Cannot find alias from address {to_address} or contact with reply {from_address}"
|
|
|
|
)
|
|
|
|
return False
|
|
|
|
|
|
|
|
report_complaint_to_user_in_forward_phase(alias, origin)
|
|
|
|
return True
|
|
|
|
|
|
|
|
|
|
|
|
def report_complaint_to_user_in_reply_phase(
|
2022-04-25 12:40:42 +00:00
|
|
|
alias: Alias, to_address: str, origin: ProviderComplaintOrigin
|
2022-04-14 16:46:11 +00:00
|
|
|
):
|
|
|
|
capitalized_name = origin.name().capitalize()
|
|
|
|
send_email_with_rate_control(
|
|
|
|
alias.user,
|
|
|
|
f"{ALERT_COMPLAINT_REPLY_PHASE}_{origin.name()}",
|
|
|
|
alias.user.email,
|
|
|
|
f"Abuse report from {capitalized_name}",
|
|
|
|
render(
|
2022-04-25 12:40:42 +00:00
|
|
|
"transactional/provider-complaint-reply-phase.txt.jinja2",
|
2022-04-14 16:46:11 +00:00
|
|
|
user=alias.user,
|
|
|
|
alias=alias,
|
|
|
|
destination=to_address,
|
|
|
|
provider=capitalized_name,
|
|
|
|
),
|
|
|
|
max_nb_alert=1,
|
|
|
|
nb_day=7,
|
|
|
|
)
|
|
|
|
|
|
|
|
|
2022-04-29 13:50:52 +00:00
|
|
|
def report_complaint_to_user_in_transactional_phase(
|
|
|
|
user: User, origin: ProviderComplaintOrigin
|
|
|
|
):
|
2022-04-14 16:46:11 +00:00
|
|
|
capitalized_name = origin.name().capitalize()
|
|
|
|
send_email_with_rate_control(
|
|
|
|
user,
|
|
|
|
f"{ALERT_COMPLAINT_TO_USER}_{origin.name()}",
|
|
|
|
user.email,
|
|
|
|
f"Abuse report from {capitalized_name}",
|
|
|
|
render(
|
2022-04-25 12:40:42 +00:00
|
|
|
"transactional/provider-complaint-to-user.txt.jinja2",
|
2022-04-14 16:46:11 +00:00
|
|
|
user=user,
|
|
|
|
provider=capitalized_name,
|
|
|
|
),
|
|
|
|
render(
|
2022-04-25 12:40:42 +00:00
|
|
|
"transactional/provider-complaint-to-user.html",
|
2022-04-14 16:46:11 +00:00
|
|
|
user=user,
|
|
|
|
provider=capitalized_name,
|
|
|
|
),
|
|
|
|
max_nb_alert=1,
|
|
|
|
nb_day=7,
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
def report_complaint_to_user_in_forward_phase(
|
2022-04-25 12:40:42 +00:00
|
|
|
alias: Alias, origin: ProviderComplaintOrigin
|
2022-04-14 16:46:11 +00:00
|
|
|
):
|
|
|
|
capitalized_name = origin.name().capitalize()
|
|
|
|
user = alias.user
|
|
|
|
send_email_with_rate_control(
|
|
|
|
user,
|
|
|
|
f"{ALERT_COMPLAINT_FORWARD_PHASE}_{origin.name()}",
|
|
|
|
user.email,
|
|
|
|
f"Abuse report from {capitalized_name}",
|
|
|
|
render(
|
2022-04-25 12:40:42 +00:00
|
|
|
"transactional/provider-complaint-forward-phase.txt.jinja2",
|
2022-04-14 16:46:11 +00:00
|
|
|
user=user,
|
|
|
|
provider=capitalized_name,
|
|
|
|
),
|
|
|
|
render(
|
2022-04-25 12:40:42 +00:00
|
|
|
"transactional/provider-complaint-forward-phase.html",
|
2022-04-14 16:46:11 +00:00
|
|
|
user=user,
|
|
|
|
provider=capitalized_name,
|
|
|
|
),
|
|
|
|
max_nb_alert=1,
|
|
|
|
nb_day=7,
|
|
|
|
)
|
|
|
|
|
|
|
|
|
2022-04-25 12:40:42 +00:00
|
|
|
def store_provider_complaint(alias, message):
|
2022-04-14 16:46:11 +00:00
|
|
|
email_name = f"reply-{uuid.uuid4().hex}.eml"
|
2022-04-25 12:40:42 +00:00
|
|
|
full_report_path = f"provider_complaint/{email_name}"
|
2022-04-14 16:46:11 +00:00
|
|
|
s3.upload_email_from_bytesio(
|
|
|
|
full_report_path, BytesIO(to_bytes(message)), email_name
|
|
|
|
)
|
|
|
|
refused_email = RefusedEmail.create(
|
|
|
|
full_report_path=full_report_path,
|
|
|
|
user_id=alias.user_id,
|
|
|
|
path=email_name,
|
|
|
|
commit=True,
|
|
|
|
)
|
2022-04-25 12:40:42 +00:00
|
|
|
ProviderComplaint.create(
|
2022-04-14 16:46:11 +00:00
|
|
|
user_id=alias.user_id,
|
2022-04-25 12:40:42 +00:00
|
|
|
state=ProviderComplaintState.new.value,
|
2022-04-14 16:46:11 +00:00
|
|
|
phase=Phase.reply.value,
|
|
|
|
refused_email_id=refused_email.id,
|
|
|
|
commit=True,
|
|
|
|
)
|