from twisted.mail import smtp
from zope.interface import implements
from twisted.internet import reactor, defer

class FileMessage(object):
    implements(smtp.IMessage)

    def __init__(self, fileObj):
        self.fileObj = fileObj
        self.msg = []
        self.envelope_recipients = []

    def lineReceived(self, line):
        self.fileObj.write(line + '\n')
        self.msg.append(line)

    def eomReceived(self):
        self.fileObj.close()
        print "\n".join(self.msg)
        print self.envelope_recipients
        return defer.succeed(None)

    def connectionLost(self):
        self.fileObj.close()
        os.remove(self.fileObj.name)

    def add_envelope_recipient(self, user):
        self.envelope_recipients.append(user)


class MUMessageDelivery(object):
    implements(smtp.IMessageDelivery)

    def __init__(self):
        import time
        fileName = 'smtplog.' + str(int(time.time()))
        self.fm = FileMessage(file(fileName, 'w'))

    def receivedHeader(self, helo, origin, recipients):
        return ''

    def validateFrom(self, helo, origin):
        return origin

    def validateTo(self, user):

        self.fm.add_envelope_recipient(user.dest)

        def serve_file_message():
            return self.fm

        return serve_file_message


class MUMessageDeliveryFactory(object):
    implements(smtp.IMessageDeliveryFactory)
    def getMessageDelivery(self):
        return MUMessageDelivery()

class MU_ESMTP(smtp.ESMTP):

    def _cbToValidate(self, to, user=None, code=250, msg='Recipient address accepted'):
        if user is None:
            user = to
        if len(self._to) == 0:
            self._to.append((user, to))
        self.sendCode(code, msg)

class MU_ESMTPFactory(smtp.SMTPFactory):

        protocol = MU_ESMTP

        def buildProtocol(self, addr):
            proto = smtp.SMTPFactory.buildProtocol(self, addr)
            proto.deliveryFactory = MUMessageDeliveryFactory()
            return proto

reactor.listenTCP(8123, MU_ESMTPFactory())
reactor.run()