import re
import time
from paste import wsgilib
from paste.response import header_value, remove_header
from wareweb.funcs import *
from commentary import store
from paste.deploy import CONFIG
from commentary import dumbpath, bodysplit
from lxml.html import fromstring, tostring

def filter_app(app):
    def replacement_app(environ, start_response):
        # @@: I have to rewrite content-length
        path_info = environ.get('PATH_INFO', '')
        script_name = environ.get('SCRIPT_NAME', '')
        status, headers, body = wsgilib.intercept_output(
            environ, app)
        if int(status.split(None, 1)[0]) != 200:
            start_response(status, headers)
            return [body]
        content_type = header_value(headers, 'content-type')
        if (not content_type
            or not content_type.startswith('text/html')):
            start_response(status, headers)
            return [body]
        body = filter_body(environ, script_name, path_info, body)
        remove_header(headers, 'content-length')
        # Fix up the encoding:
        remove_header(headers, 'content-type')
        headers.append(('content-type', 'text/html; charset=utf8'))
        
        headers.append(('content-length', str(len(body))))
        start_response(status, headers)
        return [body]
    return replacement_app

def filter_body(environ, script_name, path_info, body):
    base = environ['commentary.base_href']
    start_re = CONFIG.get('body_start_re')
    if start_re:
        start_re = re.compile(start_re)
    end_re = CONFIG.get('body_end_re')
    if end_re:
        end_re = re.compile(end_re)
    # Clearly I should be reading a header for this or something...
    if CONFIG.get('input_encoding'):
        body = body.decode(CONFIG.get('input_encoding'))
    (head, body_tag, header, 
     body, footer, end_body_tag) = bodysplit.split_body(
        body, start_re, end_re)
    vars = {'base': base,
            'mochi_base': CONFIG.get('mochikit_uri', base+'/mochilib/MochiKit/'),
            'static_base': CONFIG.get('static_uri', base),
            'scriptaculous_base': CONFIG.get('scriptaculous_uri', base+'/scriptaculous'),
            'script_name': script_name,
            'path_info': path_info,
            'path_info_html': html_quote(path_info),
            'commentary_id': int(time.time()),
            'comment_body_html': '',
            'page_html': '',
            'action_name': 'add_comment',
            }
    content = filter_content(environ, base, path_info, body, vars)
    parts = [head,
             filter_head % vars,
             body_tag,
             filter_page_head % vars,
             header,
             filter_content_head % vars,
             content,
             filter_content_foot % vars,
             footer,
             filter_foot % vars,
             end_body_tag]
    return ''.join(map(_encode, parts))

def _encode(s):
    if isinstance(s, unicode):
        return s.encode('utf8')
    return s

def filter_content(environ, base, path_info, body, vars):
    body = '<span id="start"></span>' + body
    doc = fromstring(body)
    doc_comment = store.DocumentComments(path_info, CONFIG['storage'])
    found_nodes = {}
    for position_id, comments in doc_comment.iter_comments():
        if position_id.startswith('__'): #special
            continue
        found_nodes[position_id] = dumbpath.find_path(
            doc, position_id)
    for position_id, comments in doc_comment.iter_comments():
        if position_id.startswith('__'): #special
            continue
        node = found_nodes[position_id]
        node.attrib['class'] = node.attrib.get('class', '')+' commentary-commented-element'
        html = ['<div class="commentary-inline" comment-meta="1">']
        for comment in comments:
            html.append('<div class="commentary-comment" '
                        'comment-meta="1">%s</div>' % comment.html(**vars))
        html.append('</div>')
        html = fromstring(''.join(html))
        node.getparent().insert(node.getparent().index(node)+1, html)
    body = tostring(doc)
    if doc_comment.comments('__page__'):
        for comment in doc_comment.comments('__page__'):
            html = comment.html(**vars)
            if isinstance(html, unicode):
                html = html.encode('utf-8')
            body += ('<div id="commentary-existing-comments" '
                     'class="commentary-comment">%s</div>' % html)
    return body

############################################################
## Templates
############################################################

filter_head = '''
<script type="text/javascript">
  commentary_base = '%(base)s';
  commentary_id = %(commentary_id)s;
  commentary_this_page = '%(path_info)s';
  commentary_edit_uri = "%(base)s/edit_comment";
  commentary_add_uri = "%(base)s/add_comment";
</script>
<script type="text/javascript" src="%(mochi_base)s/MochiKit.js"></script>
<script type="text/javascript" src="%(static_base)s/common.js"></script>
<script type="text/javascript" src="%(static_base)s/dumbpath.js"></script>
<script src="%(scriptaculous_base)s/prototype.js" type="text/javascript"></script>
 <script src="%(scriptaculous_base)s/effects.js" type="text/javascript"></script>
 <script src="%(scriptaculous_base)s/dragdrop.js" type="text/javascript"></script>
 <script src="%(scriptaculous_base)s/controls.js" type="text/javascript"></script>

<link rel="stylesheet" type="text/css" href="%(base)s/stylesheet.css">
'''

filter_page_head = '''
<div comment-meta="1" align="right" id="commentary-nav-div">
<span id="commentary-nav">
comments:
<a
href="#"
 onclick="return hideComments(this)"
 title="Hide all comments" class="notlast"
 id="commentary-hide-comments">hide</a><a
href="#" onclick="return foldComments(this)"
 title="Make all comments compact" class="notlast"
 id="commentary-fold-comments">fold</a><a
href="#" onclick="return showComments(this)"
 title="Show all comments" class="notlast clicked"
 id="commentary-show-comments">show</a><a
href="#" onclick="return helpButton(this)"
 title="Help on Commentary (double-click to comment)"><img
 src="%(base)s/media/help-icon-small.png"
 alt="help" border="0" /></a></span>
</div>
'''

filter_content_head = '''
<div comment-meta="0">
'''

filter_content_foot = '</div>'

comment_form_template = '''

<form action="%(base)s/%(action_name)s" method="POST" id="commentary-form"
 onsubmit="return submitCommentForm(this)">
 <input type="hidden" name="page" value="%(page_html)s" />
 [[ID]]
 <input type="text" name="username" style="width: 49%%" /><input
  type="text" name="email" style="width: 49%%"
  title="Your email will not be displayed publicly" />
 <br />
 <textarea name="comment" style="width: 100%%" rows="6"
  cols="60">%(comment_body_html)s</textarea>
 <br />
 <span style="float: right"><a style="font-size: x-small"
  title="Quick help for markup (opens in new window)"
  href="http://docutils.sourceforge.net/docs/user/rst/quickref.html"
  target="_blank">markup help</a>
  <a
  href="#" onclick="return helpButton(this)"
  title="help on Commentary"><img
  src="%(base)s/media/help-icon-small.png" alt="help"
  border="0" /></a></span>
 <button type="submit">Save</button>
 <button onclick="return submitFormInPlace(this)">...&amp; edit</button>
 <button onclick="return cancelSubmitForm(this)">Cancel</button>
</form>
'''

comment_form_position_id = '<input type="hidden" name="position_id" />'

filter_foot = '''
<div style="display: none">
%s
</div>

<p align="right" comment-meta="1"><i>Filtered for Commentary</i></p>
''' % comment_form_template.replace('[[ID]]', comment_form_position_id)