##############################################################################
#
# Copyright (c) 2002-2006 Nexedi SARL and Contributors. All Rights Reserved.
#                         Jean-Paul Smets-Solanes <jp@nexedi.com>
#                         Kevin Deldycke          <kevin@nexedi.com>
#
# WARNING: This program as such is intended to be used by professional
# programmers who take the whole responsability of assessing all potential
# consequences resulting from its eventual inadequacies and bugs
# End users who are looking for a ready-to-use solution with commercial
# garantees and support are strongly adviced to contract a Free Software
# Service Company
#
# This program is Free Software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
#
##############################################################################

from Globals import InitializeClass
from AccessControl import ClassSecurityInfo

from Products.CMFMailIn.MailMessage import MailMessage as CMFMailInMessage
from Products.ERP5Type import Permissions, PropertySheet, Constraint, Interface
from Products.ERP5Type.XMLObject import XMLObject
from Products.CMFCore.WorkflowCore import WorkflowMethod

from Products.ERP5.Document.Event import Event
import smtplib

from zLOG import LOG

# API of base64 has changed between python v2.3 and v2.4
import base64
global supported_encoding
supported_encoding = {}
try:
  # python v2.4 API
  supported_encoding = { 'base64': base64.b64decode
                       , 'base32': base64.b32decode
                       , 'base16': base64.b16decode
                       }
except AttributeError:
  # python v2.3 API
  supported_encoding = { 'base64': base64.decodestring
                       }


class MailMessage(XMLObject, Event, CMFMailInMessage):
  """
    MailMessage subclasses Event objects to implement Email Events.
  """

  meta_type = 'ERP5 Mail Message'
  portal_type = 'Mail Message'
  add_permission = Permissions.AddPortalContent
  isPortalContent = 1
  isRADContent = 1

  # Declarative security
  security = ClassSecurityInfo()
  security.declareObjectProtected(Permissions.AccessContentsInformation)

  # Declarative properties
  property_sheets = ( PropertySheet.Base
                    , PropertySheet.XMLObject
                    , PropertySheet.DublinCore
                    , PropertySheet.Task
                    , PropertySheet.Arrow
                    , PropertySheet.MailMessage
                    )

  def __init__(self, *args, **kw):
    kw = self.cleanIncomingMessage(**kw)
    XMLObject.__init__(self, *args, **kw)

  def _edit(self, *args, **kw):
    #LOG('MailMessage._edit', 0, str(kw))
    kw = self.cleanIncomingMessage(**kw)
    XMLObject._edit(self, *args, **kw)

  def cleanIncomingMessage(self, **kw):
    """
      Clean up the the message data that came from the portal_mailin tool.
    """
    # Delete attachments
    attachments = kw.get('attachments', {})
    if kw.has_key('attachments'):
      del kw['attachments']
    self.attachments = attachments
    # Decode MIME base64/32/16 data
    if kw.has_key('header') and kw['header'].has_key('content-transfer-encoding'):
      content_encoding = kw['header']['content-transfer-encoding']
      if content_encoding in supported_encoding.keys():
        method = supported_encoding[content_encoding]
        kw['body'] = method(kw['body'])
        del kw['header']['content-transfer-encoding']
    return kw

  def getBodyEncoding(self):
    """
      Extract the charset encoding from mail header.
    """
    charset = None
    header = self.getHeader()
    if header != None and header.has_key('content-type'):
      content_type = header['content-type'].replace('\n', ' ')
      content_type_info = content_type.split(';')
      for ct_info in content_type_info:
        info = ct_info.strip().lower()
        if info.startswith('charset='):
          charset = info[len('charset='):]
          # Some charset statements are quoted
          if charset.startswith('"') or charset.startswith("'"): charset = charset[1:]
          if charset.endswith(  '"') or charset.endswith(  "'"): charset = charset[:-1]
          break
    return charset

  def getEncodedBody(self, output_charset="utf-8"):
    """
      Return the entire body message encoded in the given charset.
    """
    body_charset = self.getBodyEncoding()
    if body_charset == None or body_charset.lower() == output_charset.lower():
      return self.getBody()
    else:
      unicode_body = unicode(self.getBody(), body_charset)
      return unicode_body.encode(output_charset)

  def getHeader(self):
    """
      Get the header dict of the message.
    """
    header = self.header
    if header == None or type(header) == type({}):
      return header
    elif type(header) == type(''):
      # Must do an 'eval' because the header is a dict stored as a text (see ERP5/PropertySheet/MailMessage.py)i
      return eval(header)
    else:
      raise 'TypeError', "Type of 'header' property can't be determined."

  def send(self, from_url=None, to_url=None, msg=None, subject=None):
    """
      Sends a reply to this mail message.
    """
    # We assume by default that we are replying to the sender
    if from_url == None:
      from_url = self.getUrlString()
    if to_url == None:
      to_url = self.getSender()
    if msg is not None and subject is not None:
      header = "From: %s\n" % from_url
      header += "To: %s\n\n" % to_url
      header += "Subject: %s\n" % subject
      header += "\n"
      msg = header + msg
      self.MailHost.send( msg )

  def getReplyBody(self):
    """
      This is used in order to respond to a mail,
      this put a '> ' before each line of the body
    """
    reply_body = ''
    if type(self.body) is type('a'):
      reply_body = '> ' + self.body.replace('\n','\n> ')
    return reply_body

  def getReplySubject(self):
    """
      This is used in order to respond to a mail,
      this put a 'Re: ' before the orignal subject
    """
    reply_subject = self.getTitle()
    if reply_subject.find('Re: ')!=0:
      reply_subject = 'Re: ' + reply_subject
    return reply_subject
