paul@49 | 1 | #!/usr/bin/env python |
paul@49 | 2 | |
paul@49 | 3 | from email import message_from_file |
paul@64 | 4 | from email.mime.message import MIMEMessage |
paul@49 | 5 | from email.mime.multipart import MIMEMultipart |
paul@49 | 6 | from email.mime.text import MIMEText |
paul@60 | 7 | from smtplib import LMTP, SMTP |
paul@49 | 8 | from imiptools.content import handle_itip_part |
paul@49 | 9 | import sys |
paul@49 | 10 | |
paul@49 | 11 | MESSAGE_SENDER = "resources+agent@example.com" |
paul@49 | 12 | |
paul@49 | 13 | MESSAGE_SUBJECT = "Calendar system message" |
paul@49 | 14 | |
paul@49 | 15 | MESSAGE_TEXT = """\ |
paul@49 | 16 | This is a response to a calendar message sent by your calendar program. |
paul@49 | 17 | """ |
paul@49 | 18 | |
paul@49 | 19 | # Postfix exit codes. |
paul@49 | 20 | |
paul@49 | 21 | EX_TEMPFAIL = 75 |
paul@49 | 22 | |
paul@49 | 23 | # Permitted iTIP content types. |
paul@49 | 24 | |
paul@49 | 25 | itip_content_types = [ |
paul@49 | 26 | "text/calendar", # from RFC 6047 |
paul@49 | 27 | "text/x-vcalendar", "application/ics", # other possibilities |
paul@49 | 28 | ] |
paul@49 | 29 | |
paul@49 | 30 | # Sending of outgoing messages. |
paul@49 | 31 | |
paul@60 | 32 | def sendmail(sender, recipients, data, lmtp_socket=None): |
paul@60 | 33 | if lmtp_socket: |
paul@60 | 34 | smtp = LMTP(lmtp_socket) |
paul@60 | 35 | else: |
paul@60 | 36 | smtp = SMTP("localhost") |
paul@49 | 37 | smtp.sendmail(sender, recipients, data) |
paul@49 | 38 | smtp.quit() |
paul@49 | 39 | |
paul@49 | 40 | # Processing of incoming messages. |
paul@49 | 41 | |
paul@49 | 42 | def get_all_values(msg, key): |
paul@49 | 43 | l = [] |
paul@49 | 44 | for v in msg.get_all(key) or []: |
paul@49 | 45 | l += [s.strip() for s in v.split(",")] |
paul@49 | 46 | return l |
paul@49 | 47 | |
paul@49 | 48 | class Processor: |
paul@49 | 49 | |
paul@49 | 50 | "The processing framework." |
paul@49 | 51 | |
paul@49 | 52 | def __init__(self, handlers, sender=None, subject=None, body_text=None): |
paul@49 | 53 | self.handlers = handlers |
paul@49 | 54 | self.sender = sender or MESSAGE_SENDER |
paul@49 | 55 | self.subject = subject or MESSAGE_SUBJECT |
paul@49 | 56 | self.body_text = body_text or MESSAGE_TEXT |
paul@60 | 57 | self.lmtp_socket = None |
paul@49 | 58 | |
paul@49 | 59 | def process(self, f, original_recipients, recipients): |
paul@49 | 60 | |
paul@49 | 61 | """ |
paul@49 | 62 | Process content from the stream 'f' accompanied by the given |
paul@49 | 63 | 'original_recipients' and 'recipients'. |
paul@49 | 64 | """ |
paul@49 | 65 | |
paul@49 | 66 | msg = message_from_file(f) |
paul@49 | 67 | senders = msg.get_all("Reply-To") or msg.get_all("From") |
paul@49 | 68 | original_recipients = original_recipients or get_all_values(msg, "To") |
paul@49 | 69 | |
paul@49 | 70 | # Handle messages with iTIP parts. |
paul@49 | 71 | |
paul@60 | 72 | all_responses = [] |
paul@60 | 73 | handled = False |
paul@49 | 74 | |
paul@49 | 75 | for part in msg.walk(): |
paul@49 | 76 | if part.get_content_type() in itip_content_types and \ |
paul@49 | 77 | part.get_param("method"): |
paul@49 | 78 | |
paul@60 | 79 | all_responses += handle_itip_part(part, original_recipients, self.handlers) |
paul@60 | 80 | handled = True |
paul@49 | 81 | |
paul@59 | 82 | # Pack any returned parts into a single message. |
paul@49 | 83 | |
paul@60 | 84 | if all_responses: |
paul@60 | 85 | outgoing_parts = [] |
paul@60 | 86 | forwarded_parts = [] |
paul@60 | 87 | |
paul@60 | 88 | for outgoing, part in all_responses: |
paul@60 | 89 | if outgoing: |
paul@60 | 90 | outgoing_parts.append(part) |
paul@60 | 91 | else: |
paul@60 | 92 | forwarded_parts.append(part) |
paul@60 | 93 | |
paul@60 | 94 | # Reply using any outgoing parts in a new message. |
paul@60 | 95 | |
paul@60 | 96 | if outgoing_parts: |
paul@60 | 97 | message = self.make_message(outgoing_parts, senders) |
paul@49 | 98 | |
paul@60 | 99 | if "-d" in sys.argv: |
paul@60 | 100 | print message |
paul@60 | 101 | else: |
paul@60 | 102 | sendmail(self.sender, senders, message.as_string()) |
paul@60 | 103 | |
paul@60 | 104 | # Forward messages to their recipients using the existing message. |
paul@60 | 105 | |
paul@60 | 106 | if forwarded_parts: |
paul@64 | 107 | message = self.wrap_message(msg, forwarded_parts) |
paul@60 | 108 | |
paul@60 | 109 | if "-d" in sys.argv: |
paul@64 | 110 | print message |
paul@60 | 111 | elif self.lmtp_socket: |
paul@64 | 112 | sendmail(self.sender, original_recipients, message.as_string(), self.lmtp_socket) |
paul@60 | 113 | |
paul@60 | 114 | # Unhandled messages are delivered as they are. |
paul@60 | 115 | |
paul@60 | 116 | if not handled: |
paul@49 | 117 | if "-d" in sys.argv: |
paul@60 | 118 | print msg |
paul@60 | 119 | elif self.lmtp_socket: |
paul@60 | 120 | sendmail(self.sender, original_recipients, msg.as_string(), self.lmtp_socket) |
paul@59 | 121 | |
paul@49 | 122 | def make_message(self, parts, recipients): |
paul@49 | 123 | |
paul@49 | 124 | "Make a message from the given 'parts' for the given 'recipients'." |
paul@49 | 125 | |
paul@49 | 126 | message = MIMEMultipart("alternative", _subparts=parts) |
paul@49 | 127 | message.preamble = self.body_text |
paul@49 | 128 | payload = message.get_payload() |
paul@49 | 129 | payload.insert(0, MIMEText(self.body_text)) |
paul@49 | 130 | |
paul@49 | 131 | message["From"] = self.sender |
paul@49 | 132 | for recipient in recipients: |
paul@49 | 133 | message["To"] = recipient |
paul@49 | 134 | message["Subject"] = self.subject |
paul@49 | 135 | |
paul@49 | 136 | return message |
paul@49 | 137 | |
paul@64 | 138 | def wrap_message(self, msg, parts): |
paul@64 | 139 | |
paul@64 | 140 | "Wrap 'msg' and provide the given 'parts' as the primary content." |
paul@64 | 141 | |
paul@64 | 142 | message = MIMEMultipart("mixed", _subparts=parts) |
paul@64 | 143 | message.preamble = self.body_text |
paul@64 | 144 | message.get_payload().append(MIMEMessage(msg)) |
paul@64 | 145 | |
paul@64 | 146 | message["From"] = msg["From"] |
paul@64 | 147 | message["To"] = msg["To"] |
paul@64 | 148 | message["Subject"] = msg["Subject"] |
paul@64 | 149 | |
paul@64 | 150 | return message |
paul@64 | 151 | |
paul@49 | 152 | def process_args(self, args, stream): |
paul@49 | 153 | |
paul@49 | 154 | """ |
paul@49 | 155 | Interpret the given program arguments 'args' and process input from the |
paul@49 | 156 | given 'stream'. |
paul@49 | 157 | """ |
paul@49 | 158 | |
paul@49 | 159 | # Obtain the different kinds of recipients plus sender address. |
paul@49 | 160 | |
paul@49 | 161 | original_recipients = [] |
paul@49 | 162 | recipients = [] |
paul@49 | 163 | senders = [] |
paul@60 | 164 | lmtp = [] |
paul@49 | 165 | |
paul@49 | 166 | l = [] |
paul@49 | 167 | |
paul@49 | 168 | for arg in args: |
paul@49 | 169 | |
paul@49 | 170 | # Switch to collecting recipients. |
paul@49 | 171 | |
paul@49 | 172 | if arg == "-o": |
paul@49 | 173 | l = original_recipients |
paul@49 | 174 | elif arg == "-r": |
paul@49 | 175 | l = recipients |
paul@49 | 176 | |
paul@49 | 177 | # Switch to collecting senders. |
paul@49 | 178 | |
paul@49 | 179 | elif arg == "-s": |
paul@49 | 180 | l = senders |
paul@49 | 181 | |
paul@60 | 182 | # Switch to getting the LMTP socket. |
paul@60 | 183 | |
paul@60 | 184 | elif arg == "-l": |
paul@60 | 185 | l = lmtp |
paul@60 | 186 | |
paul@49 | 187 | # Ignore debugging options. |
paul@49 | 188 | |
paul@49 | 189 | elif arg == "-d": |
paul@49 | 190 | pass |
paul@49 | 191 | else: |
paul@49 | 192 | l.append(arg) |
paul@49 | 193 | |
paul@49 | 194 | self.sender = senders and senders[0] or self.sender |
paul@60 | 195 | self.lmtp_socket = lmtp and lmtp[0] or None |
paul@49 | 196 | self.process(stream, original_recipients, recipients) |
paul@49 | 197 | |
paul@49 | 198 | def __call__(self): |
paul@49 | 199 | |
paul@49 | 200 | """ |
paul@49 | 201 | Obtain arguments from the command line to initialise the processor |
paul@49 | 202 | before invoking it. |
paul@49 | 203 | """ |
paul@49 | 204 | |
paul@49 | 205 | args = sys.argv[1:] |
paul@49 | 206 | |
paul@49 | 207 | if "-d" in args: |
paul@49 | 208 | self.process_args(args, sys.stdin) |
paul@49 | 209 | else: |
paul@49 | 210 | try: |
paul@49 | 211 | self.process_args(args, sys.stdin) |
paul@49 | 212 | except SystemExit, value: |
paul@49 | 213 | sys.exit(value) |
paul@49 | 214 | except Exception, exc: |
paul@60 | 215 | if "-v" in args: |
paul@60 | 216 | raise |
paul@49 | 217 | type, value, tb = sys.exc_info() |
paul@49 | 218 | print >>sys.stderr, "Exception %s at %d" % (exc, tb.tb_lineno) |
paul@49 | 219 | sys.exit(EX_TEMPFAIL) |
paul@49 | 220 | sys.exit(0) |
paul@49 | 221 | |
paul@49 | 222 | # vim: tabstop=4 expandtab shiftwidth=4 |