#!/usr/bin/python
# -*- coding: utf-8 -*-

# Inject one or more mbox of email, or a single email into the system.
#
# ./inject-mbox.py mbox [mbox ...]
# ./inject-mbox.py < single_email
#
# Note that if you inject the same message or message threads twice,
# then they are processed twice.  patchq does not deduplicate.
#
# This script will inject anything that looks similar enough to an
# email, even non-patches, cover letters, etc.  The ‘threader.py’
# script filters out non-patches.

import email
import mailbox
import pika
import sys
import config

connection = pika.BlockingConnection(pika.ConnectionParameters(
    host = config.mq_server))
channel = connection.channel()

processed = 0

def inject(m):
    global processed

    print("Injecting %s" % m['Subject'])

    channel.basic_publish(exchange = 'patchq_input',
                          routing_key = '',
                          body = m.as_string())
    processed = processed+1

# Read from mboxes passed on the command line, or read a single
# email from stdin.
if len(sys.argv) > 1:
    for arg in sys.argv[1:]:
        mbox = mailbox.mbox(arg)
        for m in mbox:
            inject(m)
else:
    m = email.message_from_file(sys.stdin)
    inject(m)

print ("Processed %d email(s)." % processed)