##############################################################################
#
# Copyright (c) 2002 Nexedi SARL and Contributors. All Rights Reserved.
#                    Jean-Paul Smets-Solanes <jp@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 AccessControl import ClassSecurityInfo
from Products.ERP5Type import Permissions, PropertySheet, Constraint, Interface
from Products.ERP5Type.Base import Base
from Products.ERP5.Document.Coordinate import Coordinate
from cStringIO import StringIO
from MimeWriter import MimeWriter
from base64 import encode
from mimetools import choose_boundary
from mimetypes import guess_type

class UrlMixIn:

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

  no_host_protocol_list = ['mailto', 'news', ]
  default_protocol_dict = { 'Email' : 'mailto:',
                          }

  security.declareProtected(Permissions.AccessContentsInformation,
                            'asURL')
  def asURL(self):
    """
    Returns a text representation of the Url
    """
    protocol = self.getUrlProtocol()
    if not protocol:
      # A quick fix for all objects which did not
      # define protocol such as email addresses
      ptype = self.getPortalType()
      if UrlMixIn.default_protocol_dict.has_key():
        protocol = UrlMixIn.default_protocol_dict[ptype]
      else:
        protocol = 'http'
    url_string = self.getUrlString()
    if protocol in UrlMixIn.no_host_protocol_list or url_string.startswith('//'):
      return '%s:%s' % (protocol, url_string)
    return '%s://%s' % (protocol, url_string)

  security.declareProtected(Permissions.ModifyPortalContent, 'fromText')
  def fromURL(self, url):
    """
    Analyses a URL and splits it into two parts.
    """
    protocol, url_string = url.split(':')
    if url_string.startswith('//'): url_string = url_string[2:]
    self._setUrlProtocol(protocol)
    self.setUrlString(url_string)

class Url(Coordinate, Base, UrlMixIn):
  """
  A Url is allows to represent in a standard way coordinates
  such as web sites, emails, ftp sites, etc.
  """

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

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

  # Default Properties
  property_sheets = (   PropertySheet.Base
                      , PropertySheet.SimpleItem
                      , PropertySheet.Url
                      )

  security.declareProtected(Permissions.AccessContentsInformation,
                            'asText')
  def asText(self):
    """
    Returns a text representation of the Url
    """
    return self.asURL()

  security.declareProtected(Permissions.ModifyPortalContent, 'fromText')
  def fromText(self, text):
    """
    set the Url from its text representation
    """
    self.fromURL(text)

  security.declareProtected(Permissions.AccessContentsInformation,
                            'standardTextFormat')
  def standardTextFormat(self):
    """
    Returns the standard text formats for urls
    """
    return ("http://www.erp5.org", "mailto:info@erp5.org")

  security.declareProtected(Permissions.UseMailhostServices, 'send')
  def send(self, from_url=None, to_url=None, msg=None, subject=None,  attachment_list=None):
    """
    This method was previously named 'SendMail' and is used to send email
    attachment_list is a list of dictionnary wich has keys :
     - name : name of the attachment,
     - content: data of the attachment
     - mime_type: mime-type corresponding to the attachment     
    """
    # get the mailhost object
    try:
      mailhost=self.getPortalObject().MailHost
    except:
      raise AttributeError, "Cannot find a Mail Host object"
    else:
      if from_url is None:
        from_url = self.getUrlString(None)
      if to_url is None:
        to_url = self.getUrlString(None)
      if from_url is None or to_url is None:
        raise AttributeError, "No mail defined"

      # Create multi-part MIME message.
      message = StringIO()
      writer = MimeWriter(message)
      writer.addheader('From', from_url)
      writer.addheader('To', to_url)
      writer.addheader('Subject', subject)
      writer.addheader('MimeVersion', '1.0')
      # Don't forget to flush the headers for Communicator
      writer.flushheaders()
      # Generate a unique section boundary:
      outer_boundary = choose_boundary()

      # Start the main message body. Write a brief message
      # for non-MIME-capable readers:
      dummy_file=writer.startmultipartbody("mixed",outer_boundary)
      dummy_file.write("If you can read this, your mailreader\n")
      dummy_file.write("can not handle multi-part messages!\n")

      submsg = writer.nextpart()
      submsg.addheader("Content-Transfer-Encoding", "7bit")
      FirstPartFile=submsg.startbody("text/plain", [("charset","UTF8")])
      FirstPartFile.write(msg)

      if attachment_list!=None:
        for attachment in attachment_list:
          if attachment.has_key('name'):
            attachment_name = attachment['name']
          else:
            attachment_name = ''
          # try to guess the mime type
          if not attachment.has_key('mime_type'):
            type, encoding = guess_type( attachment_name )
            if type != None:
              attachment['mime_type'] = type
            else:
              attachment['mime_type'] = 'application/octet-stream'
          # attach it
          submsg = writer.nextpart()
          if attachment['mime_type'] == 'text/plain':
            attachment_file = StringIO(attachment['content'] )
            submsg.addheader("Content-Transfer-Encoding", "7bit")
            submsg.addheader("Content-Disposition", "attachment;\nfilename="+attachment_name)
            submsg.flushheaders()

            f = submsg.startbody(attachment['mime_type'] , [("name", attachment_name)])
            f.write(attachment_file.getvalue())
          else:
            #  encode non-plaintext attachment in base64
            attachment_file = StringIO(attachment['content'] )
            submsg.addheader("Content-Transfer-Encoding", "base64")
            submsg.flushheaders()

            f = submsg.startbody(attachment['mime_type'] , [("name", attachment_name)])
            encode(attachment_file, f)
      # close the writer
      writer.lastpart()
      # send mail to user
      mailhost.send(message.getvalue(), to_url, from_url)
      return None