# -*- coding: utf-8 -*-

# 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 3 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, see <http://www.gnu.org/licenses/>.

"""

kallithea.lib.diffs

~~~~~~~~~~~~~~~~~~~

Set of diffing helpers, previously part of vcs

This file was forked by the Kallithea project in July 2014.

Original author and date, and relevant copyright and licensing information is below:

:created_on: Dec 4, 2011

:author: marcink

:copyright: (c) 2013 RhodeCode GmbH, and others.

:license: GPLv3, see LICENSE.md for more details.

"""

import difflib

import logging

import re

from tg.i18n import ugettext as _

from kallithea.lib import helpers as h

from kallithea.lib.utils2 import safe_str

from kallithea.lib.vcs.backends.base import EmptyChangeset

from kallithea.lib.vcs.exceptions import VCSError

from kallithea.lib.vcs.nodes import FileNode, SubModuleNode

log = logging.getLogger(__name__)

def _safe_id(idstring):

    r"""Make a string safe for including in an id attribute.

    The HTML spec says that id attributes 'must begin with

    a letter ([A-Za-z]) and may be followed by any number

    of letters, digits ([0-9]), hyphens ("-"), underscores

    ("_"), colons (":"), and periods (".")'. These regexps

    are slightly over-zealous, in that they remove colons

    and periods unnecessarily.

    Whitespace is transformed into underscores, and then

    anything which is not a hyphen or a character that

    matches \w (alphanumerics and underscore) is removed.

    """

    # Transform all whitespace to underscore

    idstring = re.sub(r'\s', "_", idstring)

    # Remove everything that is not a hyphen or a member of \w

    idstring = re.sub(r'(?!-)\W', "", idstring).lower()

    return idstring

def as_html(table_class='code-difftable', line_class='line',

            old_lineno_class='lineno old', new_lineno_class='lineno new',

            no_lineno_class='lineno',

            code_class='code', enable_comments=False, parsed_lines=None):

    """

    Return given diff as html table with customized css classes

    """

    def _link_to_if(condition, label, url):

        """

        Generates a link if condition is meet or just the label if not.

        """

        if condition:

            return '''<a href="%(url)s" data-pseudo-content="%(label)s"></a>''' % {

                'url': url,

                'label': label

            }

        else:

            return label

    _html_empty = True

    _html = []

    _html.append('''<table class="%(table_class)s">\n''' % {

        'table_class': table_class

    })

            _html_empty = False

                _html.append('''<tr class="%(lc)s %(action)s">\n''' % {

                    'lc': line_class,

                    'action': change['action']

                })

                anchor_old_id = ''

                anchor_new_id = ''

                anchor_old = "%(filename)s_o%(oldline_no)s" % {

                    'oldline_no': change['old_lineno']

                }

                }

                cond_old = (change['old_lineno'] != '...' and

                            change['old_lineno'])

                cond_new = (change['new_lineno'] != '...' and

                            change['new_lineno'])

                no_lineno = (change['old_lineno'] == '...' and

                             change['new_lineno'] == '...')

                if cond_old:

                    anchor_old_id = 'id="%s"' % anchor_old

                if cond_new:

                    anchor_new_id = 'id="%s"' % anchor_new

                ###########################################################

                # OLD LINE NUMBER

                ###########################################################

                _html.append('''\t<td %(a_id)s class="%(olc)s" %(colspan)s>''' % {

                    'a_id': anchor_old_id,

                    'olc': no_lineno_class if no_lineno else old_lineno_class,

                    'colspan': 'colspan="2"' if no_lineno else ''

                })

                _html.append('''%(link)s''' % {

                    'link': _link_to_if(not no_lineno, change['old_lineno'],

                                        '#%s' % anchor_old)

                })

                _html.append('''</td>\n''')

                ###########################################################

                # NEW LINE NUMBER

                ###########################################################

                if not no_lineno:

                    _html.append('''\t<td %(a_id)s class="%(nlc)s">''' % {

                        'a_id': anchor_new_id,

                        'nlc': new_lineno_class

                    })

                    _html.append('''%(link)s''' % {

                        'link': _link_to_if(True, change['new_lineno'],

                                            '#%s' % anchor_new)

                    })

                    _html.append('''</td>\n''')

                ###########################################################

                # CODE

                ###########################################################

                comments = '' if enable_comments else 'no-comment'

                _html.append('''\t<td class="%(cc)s %(inc)s">''' % {

                    'cc': code_class,

                    'inc': comments

                })

                _html.append('''\n\t\t<div class="add-bubble"><div>&nbsp;</div></div><pre>%(code)s</pre>\n''' % {

                    'code': change['line']

                })

                _html.append('''\t</td>''')

                _html.append('''\n</tr>\n''')

    _html.append('''</table>''')

    if _html_empty:

        return None

    return ''.join(_html)

def wrap_to_table(html):

    """Given a string with html, return it wrapped in a table, similar to what

    DiffProcessor returns."""

    return '''\

              <table class="code-difftable">

                <tr class="line no-comment">

                <td class="lineno new"></td>

                <td class="code no-comment"><pre>%s</pre></td>

                </tr>

              </table>''' % html

def wrapped_diff(filenode_old, filenode_new, diff_limit=None,

                ignore_whitespace=True, line_context=3,

                enable_comments=False):

    """

    Returns a file diff wrapped into a table.

    Checks for diff_limit and presents a message if the diff is too big.

    """

    if filenode_old is None:

        filenode_old = FileNode(filenode_new.path, '', EmptyChangeset())

    op = None

    a_path = filenode_old.path # default, might be overriden by actual rename in diff

    if filenode_old.is_binary or filenode_new.is_binary:

        html_diff = wrap_to_table(_('Binary file'))

        stats = (0, 0)

    elif diff_limit != -1 and (

            diff_limit is None or

            (filenode_old.size < diff_limit and filenode_new.size < diff_limit)):

        raw_diff = get_gitdiff(filenode_old, filenode_new,

                                ignore_whitespace=ignore_whitespace,

                                context=line_context)

        diff_processor = DiffProcessor(raw_diff)