[MERGE] forward port of branch 8.0 up to 2b192be
[odoo/odoo.git] / openerp / tools / mail.py
index 431fb56..e3e8f80 100644 (file)
@@ -2,7 +2,7 @@
 ##############################################################################
 #
 #    OpenERP, Open Source Business Applications
-#    Copyright (C) 2012-2013 OpenERP S.A. (<http://openerp.com>).
+#    Copyright (C) 2012-TODAY OpenERP S.A. (<http://openerp.com>).
 #
 #    This program is free software: you can redistribute it and/or modify
 #    it under the terms of the GNU Affero General Public License as
@@ -29,6 +29,7 @@ import re
 import socket
 import threading
 import time
+from email.utils import getaddresses
 
 import openerp
 from openerp.loglevels import ustr
@@ -44,10 +45,15 @@ tags_to_kill = ["script", "head", "meta", "title", "link", "style", "frame", "if
 tags_to_remove = ['html', 'body', 'font']
 
 # allow new semantic HTML5 tags
-allowed_tags = clean.defs.tags | frozenset('article section header footer hgroup nav aside figure'.split())
-safe_attrs = clean.defs.safe_attrs | frozenset(['style'])
+allowed_tags = clean.defs.tags | frozenset('article section header footer hgroup nav aside figure main'.split() + [etree.Comment])
+safe_attrs = clean.defs.safe_attrs | frozenset(
+    ['style',
+     'data-oe-model', 'data-oe-id', 'data-oe-field', 'data-oe-type', 'data-oe-expression', 'data-oe-translate', 'data-oe-nodeid',
+     'data-publish', 'data-id', 'data-res_id', 'data-member_id', 'data-view-id'
+     ])
 
-def html_sanitize(src, silent=True):
+
+def html_sanitize(src, silent=True, strict=False):
     if not src:
         return src
     src = ustr(src, errors='replace')
@@ -57,6 +63,9 @@ def html_sanitize(src, silent=True):
     # html encode email tags
     part = re.compile(r"(<(([^a<>]|a[^<>\s])[^<>]*)@[^<>]+>)", re.IGNORECASE | re.DOTALL)
     src = part.sub(lambda m: cgi.escape(m.group(1)), src)
+    # html encode mako tags <% ... %> to decode them later and keep them alive, otherwise they are stripped by the cleaner
+    src = src.replace('<%', cgi.escape('<%'))
+    src = src.replace('%>', cgi.escape('%>'))
 
     kwargs = {
         'page_structure': True,
@@ -64,6 +73,8 @@ def html_sanitize(src, silent=True):
         'forms': True,              # remove form tags
         'remove_unknown_tags': False,
         'allow_tags': allowed_tags,
+        'comments': False,
+        'processing_instructions': False
     }
     if etree.LXML_VERSION >= (2, 3, 1):
         # kill_tags attribute has been added in version 2.3.1
@@ -74,20 +85,33 @@ def html_sanitize(src, silent=True):
     else:
         kwargs['remove_tags'] = tags_to_kill + tags_to_remove
 
-    if etree.LXML_VERSION >= (3, 1, 0):
-        kwargs.update({
-            'safe_attrs_only': True,
-            'safe_attrs': safe_attrs,
-        })
+    if strict:
+        if etree.LXML_VERSION >= (3, 1, 0):
+            # lxml < 3.1.0 does not allow to specify safe_attrs. We keep all attributes in order to keep "style"
+            kwargs.update({
+                'safe_attrs_only': True,
+                'safe_attrs': safe_attrs,
+            })
     else:
-        # lxml < 3.1.0 does not allow to specify safe_attrs. We keep all attributes in order to keep "style"
-        kwargs['safe_attrs_only'] = False
+        kwargs['safe_attrs_only'] = False    # keep oe-data attributes + style
+        kwargs['frames'] = False,            # do not remove frames (embbed video in CMS blogs)
 
     try:
         # some corner cases make the parser crash (such as <SCRIPT/XSS SRC=\"http://ha.ckers.org/xss.js\"></SCRIPT> in test_mail)
         cleaner = clean.Cleaner(**kwargs)
         cleaned = cleaner.clean_html(src)
-    except etree.ParserError:
+        # MAKO compatibility: $, { and } inside quotes are escaped, preventing correct mako execution
+        cleaned = cleaned.replace('%24', '$')
+        cleaned = cleaned.replace('%7B', '{')
+        cleaned = cleaned.replace('%7D', '}')
+        cleaned = cleaned.replace('%20', ' ')
+        cleaned = cleaned.replace('%5B', '[')
+        cleaned = cleaned.replace('%5D', ']')
+        cleaned = cleaned.replace('&lt;%', '<%')
+        cleaned = cleaned.replace('%&gt;', '%>')
+    except etree.ParserError, e:
+        if 'empty' in str(e):
+            return ""
         if not silent:
             raise
         logger.warning('ParserError obtained when sanitizing %r', src, exc_info=True)
@@ -97,6 +121,11 @@ def html_sanitize(src, silent=True):
             raise
         logger.warning('unknown error obtained when sanitizing %r', src, exc_info=True)
         cleaned = '<p>Unknown error when sanitizing</p>'
+
+    # this is ugly, but lxml/etree tostring want to put everything in a 'div' that breaks the editor -> remove that
+    if cleaned.startswith('<div>') and cleaned.endswith('</div>'):
+        cleaned = cleaned[5:-6]
+
     return cleaned
 
 
@@ -104,7 +133,8 @@ def html_sanitize(src, silent=True):
 # HTML Cleaner
 #----------------------------------------------------------
 
-def html_email_clean(html, remove=False, shorten=False, max_length=300):
+def html_email_clean(html, remove=False, shorten=False, max_length=300, expand_options=None,
+                     protect_sections=False):
     """ html_email_clean: clean the html by doing the following steps:
 
      - try to strip email quotes, by removing blockquotes or having some client-
@@ -129,6 +159,32 @@ def html_email_clean(html, remove=False, shorten=False, max_length=300):
                             be flagged as to remove
     :param int max_length: if shortening, maximum number of characters before
                            shortening
+    :param dict expand_options: options for the read more link when shortening
+                                the content.The used keys are the following:
+
+                                 - oe_expand_container_tag: class applied to the
+                                   container of the whole read more link
+                                 - oe_expand_container_class: class applied to the
+                                   link container (default: oe_mail_expand)
+                                 - oe_expand_container_content: content of the
+                                   container (default: ...)
+                                 - oe_expand_separator_node: optional separator, like
+                                   adding ... <br /><br /> <a ...>read more</a> (default: void)
+                                 - oe_expand_a_href: href of the read more link itself
+                                   (default: #)
+                                 - oe_expand_a_class: class applied to the <a> containing
+                                   the link itself (default: oe_mail_expand)
+                                 - oe_expand_a_content: content of the <a> (default: read more)
+
+                                The formatted read more link is the following:
+                                <cont_tag class="oe_expand_container_class">
+                                    oe_expand_container_content
+                                    if expand_options.get('oe_expand_separator_node'):
+                                        <oe_expand_separator_node/>
+                                    <a href="oe_expand_a_href" class="oe_expand_a_class">
+                                        oe_expand_a_content
+                                    </a>
+                                </span>
     """
     def _replace_matching_regex(regex, source, replace=''):
         """ Replace all matching expressions in source by replace """
@@ -209,17 +265,41 @@ def html_email_clean(html, remove=False, shorten=False, max_length=300):
         node.text = innertext
 
         # create <span> ... <a href="#">read more</a></span> node
-        read_more_node = _create_node('span', ' ... ', None, {'class': 'oe_mail_expand'})
-        read_more_link_node = _create_node('a', 'read more', None, {'href': '#', 'class': 'oe_mail_expand'})
+        read_more_node = _create_node(
+            expand_options.get('oe_expand_container_tag', 'span'),
+            expand_options.get('oe_expand_container_content', ' ... '),
+            None,
+            {'class': expand_options.get('oe_expand_container_class', 'oe_mail_expand')}
+        )
+        if expand_options.get('oe_expand_separator_node'):
+            read_more_separator_node = _create_node(
+                expand_options.get('oe_expand_separator_node'),
+                '',
+                None,
+                {}
+            )
+            read_more_node.append(read_more_separator_node)
+        read_more_link_node = _create_node(
+            'a',
+            expand_options.get('oe_expand_a_content', 'read more'),
+            None,
+            {
+                'href': expand_options.get('oe_expand_a_href', '#'),
+                'class': expand_options.get('oe_expand_a_class', 'oe_mail_expand'),
+            }
+        )
         read_more_node.append(read_more_link_node)
         # create outertext node
-        overtext_node = _create_node('span', outertext[truncate_idx:])
+        overtext_node = _create_node('span', outertext)
         # tag node
         overtext_node.set('in_overlength', '1')
         # add newly created nodes in dom
         node.append(read_more_node)
         node.append(overtext_node)
 
+    if expand_options is None:
+        expand_options = {}
+
     if not html or not isinstance(html, basestring):
         return html
     html = ustr(html)
@@ -262,9 +342,15 @@ def html_email_clean(html, remove=False, shorten=False, max_length=300):
     # signature_begin = False  # try dynamic signature recognition
     quote_begin = False
     overlength = False
+    overlength_section_id = None
+    overlength_section_count = 0
     cur_char_nbr = 0
     for node in root.iter():
-        # node_text = re.sub('\s{2,}', ' ', node.text and node.text.strip(' \t\r\n') or '')  # do not take into account multiple spaces that are displayed as max 1 space in html
+        # comments do not need processing
+        # note: bug in node.get(value, default) for HtmlComments, default never returned
+        if node.tag == etree.Comment:
+            continue
+        # do not take into account multiple spaces that are displayed as max 1 space in html
         node_text = ' '.join((node.text and node.text.strip(' \t\r\n') or '').split())
 
         # root: try to tag the client used to write the html
@@ -273,14 +359,22 @@ def html_email_clean(html, remove=False, shorten=False, max_length=300):
         if 'SkyDrivePlaceholder' in node.get('class', '') or 'SkyDrivePlaceholder' in node.get('id', ''):
             root.set('hotmail', '1')
 
+        # protect sections by tagging section limits and blocks contained inside sections, using an increasing id to re-find them later
+        if node.tag == 'section':
+            overlength_section_count += 1
+            node.set('section_closure', str(overlength_section_count))
+        if node.getparent() is not None and (node.getparent().get('section_closure') or node.getparent().get('section_inner')):
+            node.set('section_inner', str(overlength_section_count))
+
         # state of the parsing: flag quotes and tails to remove
         if quote_begin:
             node.set('in_quote', '1')
             node.set('tail_remove', '1')
-        # state of the parsing: flag when being in over-length content
+        # state of the parsing: flag when being in over-length content, depending on section content if defined (only when having protect_sections)
         if overlength:
-            node.set('in_overlength', '1')
-            node.set('tail_remove', '1')
+            if not overlength_section_id or int(node.get('section_inner', overlength_section_count + 1)) > overlength_section_count:
+                node.set('in_overlength', '1')
+                node.set('tail_remove', '1')
 
         # find quote in msoffice / hotmail / blockquote / text quote and signatures
         if root.get('msoffice') and node.tag == 'div' and 'border-top:solid' in node.get('style', ''):
@@ -292,16 +386,33 @@ def html_email_clean(html, remove=False, shorten=False, max_length=300):
             node.set('in_quote', '1')
             node.set('tail_remove', '1')
         if node.tag == 'blockquote' or node.get('text_quote') or node.get('text_signature'):
+            # here no quote_begin because we want to be able to remove some quoted
+            # text without removing all the remaining context
+            node.set('in_quote', '1')
+        if node.getparent() is not None and node.getparent().get('in_quote'):
+            # inside a block of removed text but not in quote_begin (see above)
             node.set('in_quote', '1')
 
         # shorten:
-        # 1/ truncate the text at the next available space
-        # 2/ create a 'read more' node, next to current node
-        # 3/ add the truncated text in a new node, next to 'read more' node
+        # if protect section:
+        #   1/ find the first parent not being inside a section
+        #   2/ add the read more link
+        # else:
+        #   1/ truncate the text at the next available space
+        #   2/ create a 'read more' node, next to current node
+        #   3/ add the truncated text in a new node, next to 'read more' node
+        node_text = (node.text or '').strip().strip('\n').strip()
         if shorten and not overlength and cur_char_nbr + len(node_text) > max_length:
             node_to_truncate = node
-            while node_to_truncate.get('in_quote') and node_to_truncate.getparent() is not None:
-                node_to_truncate = node_to_truncate.getparent()
+            while node_to_truncate.getparent() is not None:
+                if node_to_truncate.get('in_quote'):
+                    node_to_truncate = node_to_truncate.getparent()
+                elif protect_sections and (node_to_truncate.getparent().get('section_inner') or node_to_truncate.getparent().get('section_closure')):
+                    node_to_truncate = node_to_truncate.getparent()
+                    overlength_section_id = node_to_truncate.get('section_closure')
+                else:
+                    break
+
             overlength = True
             node_to_truncate.set('truncate', '1')
             if node_to_truncate == node:
@@ -337,7 +448,7 @@ def html_email_clean(html, remove=False, shorten=False, max_length=300):
         if remove:
             node.getparent().remove(node)
         else:
-            if not 'oe_mail_expand' in node.get('class', ''):  # trick: read more link should be displayed even if it's in overlength
+            if not expand_options.get('oe_expand_a_class', 'oe_mail_expand') in node.get('class', ''):  # trick: read more link should be displayed even if it's in overlength
                 node_class = node.get('class', '') + ' oe_mail_cleaned'
                 node.set('class', node_class)
 
@@ -397,6 +508,9 @@ def html2plaintext(html, body_id=None, encoding='utf-8'):
     html = re.sub('<br\s*/?>', '\n', html)
     html = re.sub('<.*?>', ' ', html)
     html = html.replace(' ' * 2, ' ')
+    html = html.replace('&gt;', '>')
+    html = html.replace('&lt;', '<')
+    html = html.replace('&amp;', '&')
 
     # strip all lines
     html = '\n'.join([x.strip() for x in html.splitlines()])
@@ -465,7 +579,7 @@ def append_content_to_html(html, content, plaintext=True, preserve=False, contai
     elif plaintext:
         content = '\n%s\n' % plaintext2html(content, container_tag)
     else:
-        content = re.sub(r'(?i)(</?html.*>|</?body.*>|<!\W*DOCTYPE.*>)', '', content)
+        content = re.sub(r'(?i)(</?(?:html|body|head|!\s*DOCTYPE)[^>]*>)', '', content)
         content = u'\n%s\n' % ustr(content)
     # Force all tags to lowercase
     html = re.sub(r'(</?)\W*(\w+)([ >])',
@@ -493,12 +607,8 @@ command_re = re.compile("^Set-([a-z]+) *: *(.+)$", re.I + re.UNICODE)
 # Updated in 7.0 to match the model name as well
 # Typical form of references is <timestamp-openerp-record_id-model_name@domain>
 # group(1) = the record ID ; group(2) = the model (if any) ; group(3) = the domain
-reference_re = re.compile("<.*-open(?:object|erp)-(\\d+)(?:-([\w.]+))?.*@(.*)>", re.UNICODE)
+reference_re = re.compile("<.*-open(?:object|erp)-(\\d+)(?:-([\w.]+))?[^>]*@([^>]*)>", re.UNICODE)
 
-# Bounce regex
-# Typical form of bounce is bounce-128-crm.lead-34@domain
-# group(1) = the mail ID; group(2) = the model (if any); group(3) = the record ID
-bounce_re = re.compile("[\w]+-(\d+)-?([\w.]+)?-?(\d+)?", re.UNICODE)
 
 def generate_tracking_message_id(res_id):
     """Returns a string that can be used in the Message-ID RFC822 header field
@@ -530,7 +640,7 @@ def email_send(email_from, email_to, subject, body, email_cc=None, email_bcc=Non
     if not cr:
         db_name = getattr(threading.currentThread(), 'dbname', None)
         if db_name:
-            local_cr = cr = openerp.registry(db_name).db.cursor()
+            local_cr = cr = openerp.registry(db_name).cursor()
         else:
             raise Exception("No database cursor found, please pass one explicitly")
 
@@ -557,4 +667,9 @@ def email_split(text):
     """ Return a list of the email addresses found in ``text`` """
     if not text:
         return []
-    return re.findall(r'([^ ,<@]+@[^> ,]+)', text)
+    return [addr[1] for addr in getaddresses([text])
+                # getaddresses() returns '' when email parsing fails, and
+                # sometimes returns emails without at least '@'. The '@'
+                # is strictly required in RFC2822's `addr-spec`.
+                if addr[1]
+                if '@' in addr[1]]