indentation changes

[edward.git] / edward-bot

#! /usr/bin/env python3

"""*********************************************************************
* Edward is free software: you can redistribute it and/or modify       *
* it under the terms of the GNU Affero Public License as published by  *
* the Free Software Foundation, either version 3 of the License, or    *
* (at your option) any later version.                                  *
*                                                                      *
* Edward is distributed in the hope that it will be useful,            *
* but WITHOUT ANY WARRANTY; without even the implied warranty of       *
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the        *
* GNU Affero Public License for more details.                          *
*                                                                      *
* You should have received a copy of the GNU Affero Public License     *
* along with Edward.  If not, see <http://www.gnu.org/licenses/>.      *
*                                                                      *
* Copyright (C) 2014-2015 Andrew Engelbrecht                (AGPLv3+)  *
* Copyright (C) 2014      Josh Drake                        (AGPLv3+)  *
* Copyright (C) 2014      Lisa Marie Maginnis               (AGPLv3+)  *
*                                                                      *
* Special thanks to Josh Drake for writing the original edward bot! :) *
*                                                                      *
************************************************************************

Code used from:

* http://agpl.fsf.org/emailselfdefense.fsf.org/edward/CURRENT/edward.tar.gz

"""

import sys
import email.parser
import gpgme
import re
import io
import time

def main ():

    txt = sys.stdin.read()

    msg = email.parser.Parser().parsestr(txt)

    print("From: " + msg['From'])
    print("Subject: " + msg['Subject'])
    print()

    msg_walk(msg)


def msg_walk (msg):

    for part in msg.walk():

        if part.get_content_type() == 'multipart':
            continue

        filename  = part.get_filename()
        conttype  = part.get_content_type()
        descrip_p = part.get_params(header='content-description')

        charset   = part.get_content_charset()
        payload_b = part.get_payload(decode=True)

        if payload_b == None:
            continue

        if descrip_p == None:
            decript = ""
        else:
            descript = descrip_p[0][0]

        if charset == None:
            charset = 'utf-8'

        payload = payload_b.decode(charset)

        if conttype == "application/pgp-encrypted":
            if descript == 'PGP/MIME version identification':
                if payload.strip() != "Version: 1":
                    print("*** Warning... unknown pgp/mime version: " \
                            + payload.strip())
            continue

        if (filename == "encrypted.asc") or (conttype == "pgp/mime"):
            dec_msg = decrypt_payload(payload)
            print_decrypted(dec_msg)

        elif conttype == "text/plain":
            print(payload)

        else:
            print(payload)


def print_decrypted (message):

    if message == None:
        return

    for chunk in message:
        msg = email.parser.Parser().parsestr(chunk[0])
        sigs = chunk[1]

        msg_walk(msg)
        for sig in sigs:
            print_sig(sig)

def print_sig (sig):

    fprint = sig.fpr
    fprint_short = re.search("[0-9A-Fa-f]{32}([0-9A-Fa-f]{8})", fprint).groups()[0]

    timestamp = time.localtime(sig.timestamp)
    date = time.strftime("%a %d %b %Y %I:%M:%S %p %Z", timestamp)

    g = gpgme.Context()
    key = g.get_key(fprint)

    # right now i'm just choosing the first user id, even if that id isn't
    # signed by the user yet another is. if a user id is printed, it should
    # at least be one that is signed, and/or correspond to the From:
    # field's email address and full name.

    name = key.uids[0].name
    e_addr = key.uids[0].email
    comment = key.uids[0].comment

    # this section needs some work. signature summary, validity, status,
    # and wrong_key_usage all complicate the picture. their enum/#define
    # values overlap, which makes things more complicated.

    validity = sig.validity
    if validity == gpgme.VALIDITY_ULTIMATE \
            or validity == gpgme.VALIDITY_FULL:
        status = "Good Signature "
    elif validity == gpgme.VALIDITY_MARGINAL:
        status = "Marginal Signature "
    else:
        status = "BAD Signature "


    print("Signature Made " + date + " using key ID " + fprint_short)
    print(status + "from " + name + " (" + comment + ") <" \
            + e_addr + ">")


def decrypt_payload (payload):

    blocks = split_message(payload)
    message = decrypt_blocks(blocks)

    return message


def split_message (text):

    pgp_matches = re.search( \
            '(-----BEGIN PGP MESSAGE-----' \
            '.*' \
            '-----END PGP MESSAGE-----)', \
            text, \
            re.DOTALL)

    if pgp_matches == None:
        return None
    else:
        return pgp_matches.groups()


def decrypt_blocks (blocks):

    if blocks == None:
        return None

    message = []
    for block in blocks:
        plain, sigs = decrypt_block(block)

        message = message + [(plain, sigs)]

    return message


def decrypt_block (block):

    block_b = io.BytesIO(block.encode('ASCII'))
    plain_b = io.BytesIO()

    g = gpgme.Context()
    sigs = g.decrypt_verify(block_b, plain_b)

    plain = plain_b.getvalue().decode('ASCII')
    return (plain, sigs)


main()