# -*- coding: utf-8 -*-

# Copyright (C) Stéphane Glondu + ???
# Licence : ???

"""
Envoi de mails textes encodés et bien formatés (encodages spécifiés là il faut).
Autres outils relatifs aux mails.

format_sender et send_email adaptés depuis /usr/scripts/impression/crans_backend.py.
"""


import re

import gestion.mail as mail_module
# TODO deprecated

format_sender = mail_module.format_sender


def send_email(sender, recipient, subject, body, server='localhost', cc=None, debug=False, actual_sender=None):
    """
    Send an email.

    All arguments should be Unicode strings (plain ASCII works as well).
    Only the real name part of sender and recipient addresses may contain
    non-ASCII characters. The email will be properly MIME encoded.

    The charset of the email will be the first one out of US-ASCII or UTF-8
    that can represent all the characters occurring in the email.

    Argument server is ignored.

    The resulting mail will be sent to debug if not False.
    Otherwise, it will be sent to recipient and cc.
    """
    from email.MIMEText import MIMEText
    from email.Header import Header

    # Header class is smart enough to try US-ASCII, then the charset we
    # provide, then fall back to UTF-8.
    header_charset = 'UTF-8'

    # We must choose the body charset manually
    for body_charset in 'US-ASCII', 'UTF-8':
        try:
            body.encode(body_charset)
        except UnicodeError:
            pass
        else:
            break

    sender = format_sender(sender)
    recipient = format_sender(recipient)
    if cc: cc = format_sender(cc)

    # Create the message ('plain' stands for Content-Type: text/plain)
    msg = MIMEText(body.encode(body_charset), 'plain', body_charset)
    msg['From'] = sender
    msg['To'] = recipient
    msg['Subject'] = Header(unicode(subject), header_charset)
    if cc: msg['Cc'] = cc

    if debug:
        actual_recipient = [debug]
    else:
        actual_recipient = [recipient]
        if cc: actual_recipient.append(cc)

    if actual_sender:
        actual_sender = format_sender(actual_sender)
    else:
        actual_sender = sender

    # Send the message
    with mail_module.ServerConnection() as smtp:
        smtp.sendmail(actual_sender, actual_recipient, msg.as_string())


def parse_mail_template(fichier):
    """
    Lit fichier et renvoie le couple sujet, corps en unicode.
    Les trois premières lignes de fichier doivent être :
        Encoding: <encodage du fichier>
        Subject: <sujet du mail>
        <ligne vide>
    Le reste forme le corps du mail.
    L'argument fichier peut être un nom de fichier, ou directement une
    instance de file.
    """
    if not isinstance(fichier, file):
        fichier = file(fichier)

    encoding = fichier.readline()
    matched = re.search(r'^Encoding:\s*([^\r\n]*)', encoding)
    if matched:
        encoding = matched.group(1)
    else:
        raise SyntaxError("Encoding manquant dans template")

    subject = fichier.readline()
    matched = re.search(r'^Subject:\s*([^\r\n]*)', subject)
    if matched:
        subject = matched.group(1).decode(encoding)
    else:
        raise SyntaxError("Subject manquant dans template")

    if fichier.readline() not in ['\r\n', '\n']:
        raise SyntaxError("le fichier n'a pas le bon format")

    body = fichier.read().decode(encoding)
    return subject, body