src/hkml_send.py

#!/usr/bin/env python3
# SPDX-License-Identifier: GPL-2.0

import datetime
import os

import _hkml
import _hkml_cli
import hkml_tag
import hkml_write

def draft_or_sent_mail(draft_file, msgid):
    with open(draft_file, 'r') as f:
        draft_content = f.read()
    paragraphs = draft_content.split('\n\n')
    if len(paragraphs) == 0:
        header_lines = []
    else:
        header_lines = paragraphs[0].split('\n')
    has_date = False
    has_msgid = False
    has_from = False
    for line in header_lines:
        if line.startswith('Date: '):
            has_date = True
        if line.startswith('Message-ID: '):
            has_msgid = True
        if line.startswith('From: '):
            has_from = True

    fake_header = ['From hkml_draft Thu Jan  1 00:00:00 1970']
    if has_date is False:
        date = datetime.datetime.now().astimezone()
        fake_header.append('Date: %s' % date.strftime(
            '%a, %d %b %Y %H:%M:%S %z'))
    if has_msgid is False:
        if msgid is None:
            msgid = '%s' % datetime.datetime.now().strftime(
                    'hkml_draft-%Y-%m-%d-%H-%M-%S')
        fake_header.append('Message-ID: %s' % msgid)
    if has_from is False:
        fake_header.append('From: ')
    draft_mbox_str = '\n'.join(fake_header + [draft_content])
    return _hkml.Mail(mbox=draft_mbox_str)

def handle_user_edit_mistakes(tmp_path):
    with open(tmp_path, 'r') as f:
        written_mail = f.read()
    pars = written_mail.split('\n\n')
    header = pars[0]
    header_lines = []
    for line in header.split('\n'):
        if line in [
                'To: /* write recipients here and REMOVE this comment */',
                'Cc: /* write cc recipients here and REMOVE this comment */']:
            continue
        header_lines.append(line)
    header = '\n'.join(header_lines)

    # Seems silly, but we have to re-join the split body, then turn them
    # into individual lines again. This preserves all empty lines.
    body = '\n\n'.join(pars[1:]).split('\n')
    body_lines = []
    idx = 0
    while idx < len(body):
        for notice in hkml_write.coloring_notices:
            if body[idx:idx + len(notice)] == notice:
                idx += len(notice)
                break

        # A user might delete the newline on top of the signature, so just check
        # for the contents of the comment block.
        if len(body) - idx >= hkml_write.SIGNATURE_WARNING_LEN and \
                body[idx:idx + hkml_write.SIGNATURE_WARNING_LEN] == \
                hkml_write.SIGNATURE_WARNING[1:]:

            # If the warning's newline was not touched, remove it as well
            if idx > 0 and body[idx-1] == '':
                body_lines.pop()
            idx += hkml_write.SIGNATURE_WARNING_LEN
            continue

        line = body[idx]
        if line != '/* write your message here (keep the above blank line) */':
            body_lines.append(line)
        idx += 1
    body = '\n'.join(body_lines)

    written_mail = '\n\n'.join([header] + [body])
    with open(tmp_path, 'w') as f:
        f.write(written_mail)

def send_mail(mboxfile, get_confirm, erase_mbox, orig_draft_subject=None):
    handle_user_edit_mistakes(mboxfile)
    if get_confirm:
        with open(mboxfile, 'r') as f:
            print(f.read())
        print('Above is what you wrote.')
    user_input, selection, err = _hkml_cli.ask_selection(
            desc = ''.join([
                'Maybe I could help you managing the tags for this mail.\n\n',
                'I can tag it as sent or drafts, ',
                'depending on your answers to following questions, ',
                'and results of mail posting.\n',
                'I can remove drafts-tagged mails that having same subject.\n',
                'I can sync tags if you already set "hkml sync".\n\n',
                'May I do those?'
                ]),
            selections_txt=[
                'yes', 'no', 'I don\'t know, ask again after sending mail',
                'I opened the editor by mistake, just quit'],
            allow_cancel=False, allow_error=False)
    if selection == 0:
        confirm_tagging = False
        no_tagging = False
    elif selection == 1:
        no_tagging = True
    elif selection == 2:
        confirm_tagging = True
        no_tagging = False
    elif selection == 3:
        if erase_mbox:
            os.remove(mboxfile)
        return

    sent = False
    msgid = None
    for line in _hkml.cmd_lines_output(['git', 'send-email', mboxfile,
                                        # for getting message-id
                                        '--confirm', 'always']):
        fields = line.split()
        if len(fields) == 2 and fields[0].lower() == 'message-id:':
            msgid = fields[1]
        if fields == ['Result:', '250'] or fields == ['Result:' , 'OK']:
            sent = True
    if no_tagging is False:
        hkml_tag.handle_may_sent_mail(
                draft_or_sent_mail(mboxfile, msgid), sent, orig_draft_subject,
                do_confirm=confirm_tagging)
    if erase_mbox:
        os.remove(mboxfile)

def main(args):
    send_mail(args.mbox_file, get_confirm=True, erase_mbox=False)

def set_argparser(parser=None):
    parser.description = 'send a mail'
    parser.add_argument('mbox_file', metavar='<mboxfile>',
            help='Mbox format file of the mail to send.')